#pragma once
|
|
#include "CHImageControls/CHImageData.h"
|
#include "CHThreadPools/WorkThreadPools.h"
|
|
class CFrameThreadData : public CWorkThreadData
|
{
|
public:
|
CFrameThreadData(PVOID pPtr) : CWorkThreadData(pPtr)
|
{
|
m_nIndex = -1;
|
}
|
|
virtual ~CFrameThreadData()
|
{
|
m_nIndex = -1;
|
}
|
|
int SetFrameData(int nFrameIndex, BYTE* pFrameBuffer, int nFrameWidth, int nFrameHeight, int nFrameChannels)
|
{
|
m_nIndex = nFrameIndex;
|
|
if (m_Image.CreateImageHeader(nFrameWidth, nFrameHeight, 8, nFrameChannels)==FALSE)
|
{
|
return 0;
|
}
|
|
m_Image.SetImageDataPtr((char*)pFrameBuffer);
|
|
return 1;
|
}
|
|
int GetIndex() { return m_nIndex; }
|
CCHImageData* GetFrameImage() { return &m_Image; }
|
|
protected:
|
int m_nIndex;
|
CCHImageData m_Image;
|
};
|
|
interface IImageReSampler2Parent
|
{
|
virtual void IIRS2P_ImageReSampleComplete(int nFrameCount) = 0;
|
};
|
|
class AFX_EXT_CLASS CImageReSampler : public CWorkThreadPools
|
{
|
public:
|
CImageReSampler(int nThreadCount=8, IImageReSampler2Parent* pIIRS2P=NULL);
|
virtual ~CImageReSampler(void);
|
void Reset();
|
void SetIIRS2P(IImageReSampler2Parent* pIIRS2P);
|
|
int GetReSampledFrameCount() const { return m_nReSampledFrameCount; }
|
void Start_ReSampling();
|
int SetBufferInfo(int nSrcWidth, int nSrcHeight, int nSrcChannels, int nFrameSize, int nTagWidth, int nTagHeight);
|
int SetFrameBuffer(int nFrameIndex, BYTE* pFrameBuffer, int nFrameWidth, int nFrameHeight, int nFrameChannels);
|
int SaveImage(const CString& strFilename, int nQual);
|
|
int GetWidth() const { return m_TargetImage.GetWidth(); }
|
int GetHeight() const { return m_TargetImage.GetHeight(); }
|
int GetFrameHeight() const { return m_nTargetFrameHeight; }
|
BYTE* GetImageBuffer() { return (BYTE*) m_TargetImage.GetImageBuffer(); }
|
|
protected:
|
virtual void WorkThreadProcess(PVOID pParameter);
|
|
protected:
|
int m_nSourceWidth;
|
int m_nSourceHeight;
|
int m_nSourceChannels;
|
int m_nSourceFrameHeight;
|
|
int m_nTargetFrameHeight;
|
int m_nTargetFrameSize;
|
int m_nTotalFrameCount;
|
int m_nReSampledFrameCount;
|
CCHImageData m_TargetImage;
|
|
IImageReSampler2Parent* m_pIIRS2P;
|
CCriticalSection m_csTargetImage;
|
};
|
