Main-Repo/SourceCode/AIReconstruction/AIReconstructionCppCore/VolumeDataPreProcessor.h

#pragma once
#include "Common.h"


class VolumeDataPreProcessor
{
public:
	VolumeDataPreProcessor();
	VolumeDataPreProcessor(int size,VolumeDataPreProcessorInfo dataInfo);
	~VolumeDataPreProcessor();
	cv::Mat GetImage(int index);

private:
	void AddImage(int index, cv::Mat image);
	bool Contains(int index);
	void RemoveTopMostImage();
	std::map<int, cv::Mat> _images;
	std::queue<int> _imageIndex;
	int _queueSize;

	const uint8_t* _dataBuffer;
	int _origWidth;
	int _origHeight;
	int _origImgCount;
	int _dstWidth;
	int _dstHeight;
	int _depthFlag;
	int _bytePerPixel;
	cv::Rect _cropRect;
	int _origImgByteCount;

};