@sh0dan
As expected, it's not possible... :'(
@siwalters
Here is the OOP version I promised...
... in 3.0 style, but besides it allowed me to skip the copy part, it don't change much.
class SimpleSampleParent : public GenericVideoFilter {
public:
SimpleSampleParent(PClip _child, int _SquareSize) : GenericVideoFilter(_child), SquareSize(_SquareSize) { }
virtual CPVideoFrame _stdcall GetFrame(int n)
//in 3.0 there is a distinction from CPVideoFrame and PVideoFrame
//C stands for const, those cannot be modified
{
PVideoFrame frame = chid->GetFrame(n);
//conversion a CPVideoFrame to a PVideoFrame
//does what is needed to allow modifications
//(I can develop if you want)
Process(frame); //call the filling method
return frame; //implicit cast from PVideoFrame to CPVideoFrame
}
protected:
// define the parameter variable
const int SquareSize;
virtual void _stdcall Process(PVideoFrame frame) = 0;
void _stdcall ByteSquareFill(BYTE * dstp, int pitch, int width, int height, BYTE fillPattern)
//method to fill a rect with a BYTE value
//used in RGB and YV12
{
for(int h = height; h --> 0; dstp += pitch)
for(int w = width; w --> 0; )
*(dstp + w) = fillPattern;
}
BYTE * _stdcall FindSquareLeftCorner(PVideoFrame frame, Plane plane, int BpP, int scaleFactor = 1)
{
Byte * result = frame->GetWritePtr(plane);
result += (frame->GetHeight(plane)>>1 - SquareSize>>1/scaleFactor) * frame->GetPitch(plane);
result += frame->GetRowSize(plane)>>1 - SquareSize>>1/scaleFactor * BpP;
return result;
}
};
class SimpleSampleRGB : public SimpleSampleParent {
//no distinction between RGB24 and RGB32
//I fill alpha with 255 too
//if that is unwanted make a special version for RGB32
public:
SimpleSampleRGB(PClip _child, int _SquareSize) : SimpleSampleParent(_child, _SquareSize) { }
protected:
virtual void _stdcall Process(PVideoFrame frame)
{
int BpP = vi.IsRGB32()? 4 : 3;
ByteSquareFill(FindSquareLeftCorner(frame, NOT_PLANAR, BpP), frame->GetPitch(), BpP*SquareSize, SquareSize, 255));
}
};
class SimpleSampleYUY2 : public SimpleSampleParent {
public:
SimpleSampleYUY2(PClip _child, int _SquareSize) : SimpleSampleInterleaved(_child, _SquareSize) { }
protected:
virtual void _stdcall Process(PVideoFrame frame)
{
BYTE * dstp = FindSquareLeftCorner(frame, NOT_PLANAR, 2);
int pitch = frame->GetPitch(NOT_PLANAR);
for(int h = SquareSize; h --> 0; dstp += pitch)
for(int w = SquareSize>>1; w --> 0; )
*((unsigned int *)dstp + w) = 0x80EB80EB;
}
};
class SimpleSampleYV12 : SimpleSampleParent {
public:
SimpleSampleYV12(PClip _child, int _SquareSize) : SimpleSampleParent(_child, _SquareSize) { }
protected:
virtual void _stdcall Process(PVideoFrame frame)
{
static const BYTE fillPatterns[] = { 235, 128, 128 };
static const int scaleFactor[] = { 1, 2, 2 };
for(int i = 3; i --> 0; )
Plane p = VideoFrame::IndexToPlane(i);
int s = scaleFactor[i];
ByteSquareFill(FindSquareLeftCorner(frame, p, 1, s), frame->GetPitch(), SquareSize/s, SquareSize/s, fillPatterns[i]);
}
}
};
AVSValue __cdecl Create_SimpleSample(const vector<AVSValue>& args) {
PClip clip = args[0]; //implicit conversions
int sq = args[1];
switch(clip->GetVideoInfo().GetColorSpace().id)
{
case I_RGB24:
case I_RGB32:
return new SimpleSampleRGB(clip, sq);
case I_YUY2:
return new SimpleSampleYUY2(clip, sq);
case I_YV12:
return new SimpleSampleYV12(clip, sq);
}
}