blob: 30d62a1a6fa7d7bd91a26d57ea083b6c6c297149 [file] [log] [blame]
#ifndef IMAGE_IO_JPEG_JPEG_XMP_DATA_EXTRACTOR_H_ // NOLINT
#define IMAGE_IO_JPEG_JPEG_XMP_DATA_EXTRACTOR_H_ // NOLINT
#include "image_io/base/data_destination.h"
#include "image_io/jpeg/jpeg_info.h"
namespace photos_editing_formats {
namespace image_io {
/// A class that can make use of the data in a JpegInfo instance to extract
/// the xmp data JpegSegments passed to it and forward it to a DataDestination.
class JpegXmpDataExtractor : public DataDestination {
public:
/// @param xmp_info_type The type of xmp data being extracted.
/// @param segment_count The number of segment ranges over which the xmp
/// data is spread.
/// @param data_destination The destination to which the extracted xmp data
/// is to be sent.
JpegXmpDataExtractor(JpegXmpInfo::Type xmp_info_type, size_t segment_count,
DataDestination* data_destination)
: xmp_info_type_(xmp_info_type),
last_segment_index_(segment_count - 1),
data_destination_(data_destination),
has_error_(false) {}
/// Set the current segment index to the given value.
/// @param segment_index The index of the segment currently being processed.
void SetSegmentIndex(size_t segment_index) { segment_index_ = segment_index; }
/// @return True if there was an error in the extraction process.
bool HasError() const { return has_error_; }
void StartTransfer() override;
TransferStatus Transfer(const DataRange& transfer_range,
const DataSegment& data_segment) override;
void FinishTransfer() override;
/// @return The number of bytes written not to this extractor destination, but
/// to the next destination. Returns zero if the next destination is null.
size_t GetBytesTransferred() const override {
return data_destination_ ? data_destination_->GetBytesTransferred() : 0;
}
private:
/// The type of xmp data being extracted.
JpegXmpInfo::Type xmp_info_type_;
/// The xmp data require special processing when the last segment is being
/// transferred. This value is the index of the last segment.
size_t last_segment_index_;
/// The DataDestination that the extracted xmp data is sent to.
DataDestination* data_destination_;
/// The xmp data is spread over one or more segments in the DataSource. This
/// index tracks which one is being transferred.
size_t segment_index_;
/// A true value indicates that an error occurred in the decoding process.
bool has_error_;
};
} // namespace image_io
} // namespace photos_editing_formats
#endif // IMAGE_IO_JPEG_JPEG_XMP_DATA_EXTRACTOR_H_ // NOLINT