static int filter_frame(AVFilterLink *inlink, AVFilterBufferRef *in)
{
AVFilterContext *ctx = inlink->dst;
FilterData *fd = ctx->priv;
TransformData* td = &(fd->td);
AVFilterLink *outlink = inlink->dst->outputs[0];
//const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(inlink->format);
//int hsub0 = desc->log2_chroma_w;
//int vsub0 = desc->log2_chroma_h;
int direct = 0;
AVFilterBufferRef *out;
VSFrame inframe;
VSFrame outframe;
int plane;
if (in->perms & AV_PERM_WRITE) {
direct = 1;
out = in;
} else {
out = ff_get_video_buffer(outlink, AV_PERM_WRITE, outlink->w, outlink->h);
if (!out) {
avfilter_unref_bufferp(&in);
return AVERROR(ENOMEM);
}
avfilter_copy_buffer_ref_props(out, in);
}
for(plane=0; plane < td->fiSrc.planes; plane++){
inframe.data[plane] = in->data[plane];
inframe.linesize[plane] = in->linesize[plane];
}
for(plane=0; plane < td->fiDest.planes; plane++){
outframe.data[plane] = out->data[plane];
outframe.linesize[plane] = out->linesize[plane];
}
transformPrepare(td, &inframe, &outframe);
if (fd->td.fiSrc.pFormat > PF_PACKED) {
transformRGB(td, getNextTransform(td, &fd->trans));
} else {
transformYUV(td, getNextTransform(td, &fd->trans));
}
transformFinish(td);
if (!direct)
avfilter_unref_bufferp(&in);
return ff_filter_frame(outlink, out);
}
static int deshake_filter_video(TCModuleInstance *self,
vframe_list_t *frame)
{
DeshakeData *sd = NULL;
TC_MODULE_SELF_CHECK(self, "filter_video");
TC_MODULE_SELF_CHECK(frame, "filter_video");
sd = self->userdata;
MotionDetect* md = &(sd->md);
TransformData* td = &(sd->td);
LocalMotions localmotions;
Transform motion;
VSFrame vsFrame;
fillFrameFromBuffer(&vsFrame,frame->video_buf, &td->fiSrc);
if(motionDetection(md, &localmotions, &vsFrame)!= VS_OK){
tc_log_error(MOD_NAME, "motion detection failed");
return TC_ERROR;
}
if(writeToFile(md, sd->f, &localmotions) != VS_OK)
motion = simpleMotionsToTransform(td, &localmotions);
vs_vector_del(&localmotions);
transformPrepare(td, &vsFrame, &vsFrame);
Transform t = lowPassTransforms(td, &sd->avg, &motion);
/* tc_log_error(MOD_NAME, "Trans: det: %f %f %f \n\t\t act: %f %f %f %f", */
/* motion.x, motion.y, motion.alpha, */
/* t.x, t.y, t.alpha, t.zoom); */
if (sd->vob->im_v_codec == CODEC_RGB) {
transformRGB(td, t);
} else if (sd->vob->im_v_codec == CODEC_YUV) {
transformYUV(td, t);
} else {
tc_log_error(MOD_NAME, "unsupported Codec: %i\n", sd->vob->im_v_codec);
return TC_ERROR;
}
transformFinish(td);
return TC_OK;
}
void generateFrames(TestData* testdata, int num){
int i;
for(i=0; i<num; i++){
allocateFrame(&testdata->frames[i],&testdata->fi);
}
// first frame noise
fillArrayWithNoise(testdata->frames[0].data[0],
testdata->fi.width*testdata->fi.height, 10);
fillArrayWithNoise(testdata->frames[0].data[1],
testdata->fi.width/2*testdata->fi.height/2, 5);
fillArrayWithNoise(testdata->frames[0].data[2],
testdata->fi.width/2*testdata->fi.height/2, 5);
// add rectangles
int k;
for(k=0; k<NUM_RECTANGLES; k++){
paintRectangle(testdata->frames[0].data[0],&testdata->fi,
randUpTo(testdata->fi.width), randUpTo(testdata->fi.height),
randUpTo((testdata->fi.width>>4)+4),
randUpTo((testdata->fi.height>>4)+4),randPixel());
}
TransformData td;
test_bool(initTransformData(&td, &testdata->fi, &testdata->fi, "generate") == VS_OK);
td.interpolType=Zero;
test_bool(configureTransformData(&td)== VS_OK);
fprintf(stderr, "testframe transforms\n");
for(i=1; i<num; i++){
Transform t = getTestFrameTransform(i);
fprintf(stderr, "%i, %6.4lf %6.4lf %8.5lf %6.4lf %i\n",
i, t.x, t.y, t.alpha, t.zoom, t.extra);
test_bool(transformPrepare(&td,&testdata->frames[i-1],&testdata->frames[i])== VS_OK);
test_bool(transformYUV_float(&td, t)== VS_OK);
test_bool(transformFinish(&td)== VS_OK);
}
cleanupTransformData(&td);
}
