static const VSFrameRef *VS_CC mvrecalculateGetFrame(int n, int activationReason, void **instanceData, void **frameData, VSFrameContext *frameCtx, VSCore *core, const VSAPI *vsapi) { (void)frameData; MVRecalculateData *d = (MVRecalculateData *)*instanceData; if (activationReason == arInitial) { vsapi->requestFrameFilter(n, d->vectors, frameCtx); int nref; int offset = d->analysisData.nDeltaFrame; if (offset > 0) { nref = d->analysisData.isBackward ? n + offset : n - offset; } else { nref = -offset; } if (nref >= 0 && nref < d->vi->numFrames) { if (n < nref) { vsapi->requestFrameFilter(n, d->node, frameCtx); vsapi->requestFrameFilter(nref, d->node, frameCtx); } else { vsapi->requestFrameFilter(nref, d->node, frameCtx); vsapi->requestFrameFilter(n, d->node, frameCtx); } } else { // too close to beginning/end of clip vsapi->requestFrameFilter(n, d->node, frameCtx); } } else if (activationReason == arAllFramesReady) { GroupOfPlanes vectorFields; gopInit(&vectorFields, d->analysisData.nBlkSizeX, d->analysisData.nBlkSizeY, d->analysisData.nLvCount, d->analysisData.nPel, d->analysisData.nMotionFlags, d->analysisData.nCPUFlags, d->analysisData.nOverlapX, d->analysisData.nOverlapY, d->analysisData.nBlkX, d->analysisData.nBlkY, d->analysisData.xRatioUV, d->analysisData.yRatioUV, d->divideExtra, d->vi->format->bitsPerSample); const uint8_t *pSrc[3] = { NULL }; const uint8_t *pRef[3] = { NULL }; int nSrcPitch[3] = { 0 }; int nRefPitch[3] = { 0 }; int nref; int offset = d->analysisData.nDeltaFrame; if (offset > 0) { nref = d->analysisData.isBackward ? n + offset : n - offset; } else { nref = -offset; } const VSFrameRef *src = vsapi->getFrameFilter(n, d->node, frameCtx); const VSMap *srcprops = vsapi->getFramePropsRO(src); int err; int src_top_field = !!vsapi->propGetInt(srcprops, "_Field", 0, &err); if (err && d->fields && !d->tff_exists) { vsapi->setFilterError("Recalculate: _Field property not found in input frame. Therefore, you must pass tff argument.", frameCtx); gopDeinit(&vectorFields); vsapi->freeFrame(src); return NULL; } // if tff was passed, it overrides _Field. if (d->tff_exists) src_top_field = d->tff ^ (n % 2); for (int plane = 0; plane < d->vi->format->numPlanes; plane++) { pSrc[plane] = vsapi->getReadPtr(src, plane); nSrcPitch[plane] = vsapi->getStride(src, plane); } FakeGroupOfPlanes fgop; fgopInit(&fgop, &d->vectors_data); const VSFrameRef *mvn = vsapi->getFrameFilter(n, d->vectors, frameCtx); const VSMap *mvprops = vsapi->getFramePropsRO(mvn); fgopUpdate(&fgop, (const int *)vsapi->propGetData(mvprops, prop_MVTools_vectors, 0, NULL)); vsapi->freeFrame(mvn); int vectors_size = gopGetArraySize(&vectorFields) * sizeof(int); int *vectors = (int *)malloc(vectors_size); if (fgopIsValid(&fgop) && nref >= 0 && nref < d->vi->numFrames) { const VSFrameRef *ref = vsapi->getFrameFilter(nref, d->node, frameCtx); const VSMap *refprops = vsapi->getFramePropsRO(ref); int ref_top_field = !!vsapi->propGetInt(refprops, "_Field", 0, &err); if (err && d->fields && !d->tff_exists) { vsapi->setFilterError("Recalculate: _Field property not found in input frame. Therefore, you must pass tff argument.", frameCtx); gopDeinit(&vectorFields); vsapi->freeFrame(src); vsapi->freeFrame(ref); free(vectors); fgopDeinit(&fgop); return NULL; } // if tff was passed, it overrides _Field. if (d->tff_exists) ref_top_field = d->tff ^ (nref % 2); int fieldShift = 0; if (d->fields && d->analysisData.nPel > 1 && (d->analysisData.nDeltaFrame % 2)) { fieldShift = (src_top_field && !ref_top_field) ? d->analysisData.nPel / 2 : ((ref_top_field && !src_top_field) ? -(d->analysisData.nPel / 2) : 0); // vertical shift of fields for fieldbased video at finest level pel2 } for (int plane = 0; plane < d->vi->format->numPlanes; plane++) { pRef[plane] = vsapi->getReadPtr(ref, plane); nRefPitch[plane] = vsapi->getStride(ref, plane); } MVGroupOfFrames pSrcGOF, pRefGOF; mvgofInit(&pSrcGOF, d->nSuperLevels, d->analysisData.nWidth, d->analysisData.nHeight, d->nSuperPel, d->nSuperHPad, d->nSuperVPad, d->nSuperModeYUV, d->opt, d->analysisData.xRatioUV, d->analysisData.yRatioUV, d->vi->format->bitsPerSample); mvgofInit(&pRefGOF, d->nSuperLevels, d->analysisData.nWidth, d->analysisData.nHeight, d->nSuperPel, d->nSuperHPad, d->nSuperVPad, d->nSuperModeYUV, d->opt, d->analysisData.xRatioUV, d->analysisData.yRatioUV, d->vi->format->bitsPerSample); // cast away the const, because why not. mvgofUpdate(&pSrcGOF, (uint8_t **)pSrc, nSrcPitch); mvgofUpdate(&pRefGOF, (uint8_t **)pRef, nRefPitch); DCTFFTW *DCTc = NULL; if (d->dctmode >= 1 && d->dctmode <= 4) { DCTc = (DCTFFTW *)malloc(sizeof(DCTFFTW)); dctInit(DCTc, d->analysisData.nBlkSizeX, d->analysisData.nBlkSizeY, d->vi->format->bitsPerSample, d->opt); } gopRecalculateMVs(&vectorFields, &fgop, &pSrcGOF, &pRefGOF, d->searchType, d->nSearchParam, d->nLambda, d->pnew, vectors, fieldShift, d->thSAD, DCTc, d->dctmode, d->smooth, d->meander); if (d->divideExtra) { // make extra level with divided sublocks with median (not estimated) motion gopExtraDivide(&vectorFields, vectors); } gopDeinit(&vectorFields); if (DCTc) { dctDeinit(DCTc); free(DCTc); } mvgofDeinit(&pSrcGOF); mvgofDeinit(&pRefGOF); vsapi->freeFrame(ref); } else {// too close to the beginning or end to do anything gopWriteDefaultToArray(&vectorFields, vectors); gopDeinit(&vectorFields); } VSFrameRef *dst = vsapi->copyFrame(src, core); VSMap *dstprops = vsapi->getFramePropsRW(dst); vsapi->propSetData(dstprops, prop_MVTools_MVAnalysisData, (const char *)(d->divideExtra ? &d->analysisDataDivided : &d->analysisData), sizeof(MVAnalysisData), paReplace); vsapi->propSetData(dstprops, prop_MVTools_vectors, (const char *)vectors, vectors_size, paReplace); free(vectors); #if defined(MVTOOLS_X86) // FIXME: Get rid of all mmx shit. mvtools_cpu_emms(); #endif vsapi->freeFrame(src); fgopDeinit(&fgop); return dst; } return 0; }
static const VSFrameRef *VS_CC mvrecalculateGetFrame(int n, int activationReason, void **instanceData, void **frameData, VSFrameContext *frameCtx, VSCore *core, const VSAPI *vsapi) { MVRecalculateData *d = (MVRecalculateData *) * instanceData; if (activationReason == arInitial) { vsapi->requestFrameFilter(n, d->vectors, frameCtx); int offset = ( d->analysisData.isBackward ) ? d->analysisData.nDeltaFrame : -d->analysisData.nDeltaFrame; int nref = n + offset; if (nref >= 0 && (nref < d->vi->numFrames || !d->vi->numFrames)) { if (n < nref) { vsapi->requestFrameFilter(n, d->node, frameCtx); vsapi->requestFrameFilter(nref, d->node, frameCtx); } else { vsapi->requestFrameFilter(nref, d->node, frameCtx); vsapi->requestFrameFilter(n, d->node, frameCtx); } } else { // too close to beginning/end of clip vsapi->requestFrameFilter(n, d->node, frameCtx); } } else if (activationReason == arAllFramesReady) { GroupOfPlanes *vectorFields = new GroupOfPlanes(d->analysisData.nBlkSizeX, d->analysisData.nBlkSizeY, d->analysisData.nLvCount, d->analysisData.nPel, d->analysisData.nMotionFlags, d->analysisData.nCPUFlags, d->analysisData.nOverlapX, d->analysisData.nOverlapY, d->analysisData.nBlkX, d->analysisData.nBlkY, d->analysisData.xRatioUV, d->analysisData.yRatioUV, d->divideExtra, d->supervi->format->bitsPerSample); const uint8_t *pSrc[3]; const uint8_t *pRef[3]; uint8_t *pDst; int nSrcPitch[3]; int nRefPitch[3]; int offset = ( d->analysisData.isBackward ) ? d->analysisData.nDeltaFrame : -d->analysisData.nDeltaFrame; int nref = n + offset; const VSFrameRef *src = vsapi->getFrameFilter(n, d->node, frameCtx); const VSMap *srcprops = vsapi->getFramePropsRO(src); int err; bool srctff = !!vsapi->propGetInt(srcprops, "_Field", 0, &err); if (err && d->fields && !d->tffexists) { vsapi->setFilterError("Recalculate: _Field property not found in input frame. Therefore, you must pass tff argument.", frameCtx); delete vectorFields; vsapi->freeFrame(src); return NULL; } // if tff was passed, it overrides _Field. if (d->tffexists) srctff = d->tff && (n % 2 == 0); //child->GetParity(n); // bool tff; for (int plane = 0; plane < d->supervi->format->numPlanes; plane++) { pSrc[plane] = vsapi->getReadPtr(src, plane); nSrcPitch[plane] = vsapi->getStride(src, plane); } int dst_height = 1; int dst_width = d->headerSize / sizeof(int) + vectorFields->GetArraySize(); //v1.8.1 // In Avisynth the frame was packed BGR32, which has 4 bytes per pixel. // It's GRAY8 here. dst_width *= 4; VSFrameRef *dst = vsapi->newVideoFrame(d->vi->format, dst_width, dst_height, src, core); pDst = vsapi->getWritePtr(dst, 0); // write analysis parameters as a header to frame memcpy(pDst, &d->headerSize, sizeof(int)); if (d->divideExtra) memcpy(pDst+sizeof(int), &d->analysisDataDivided, sizeof(d->analysisData)); else memcpy(pDst+sizeof(int), &d->analysisData, sizeof(d->analysisData)); pDst += d->headerSize; const VSFrameRef *mvn = vsapi->getFrameFilter(n, d->vectors, frameCtx); MVClipBalls balls(d->mvClip, vsapi); balls.Update(mvn); vsapi->freeFrame(mvn); if (balls.IsUsable() && nref >= 0 && (nref < d->vi->numFrames || !d->vi->numFrames)) { const VSFrameRef *ref = vsapi->getFrameFilter(nref, d->node, frameCtx); const VSMap *refprops = vsapi->getFramePropsRO(ref); bool reftff = !!vsapi->propGetInt(refprops, "_Field", 0, &err); if (err && d->fields && !d->tffexists) { vsapi->setFilterError("Recalculate: _Field property not found in input frame. Therefore, you must pass tff argument.", frameCtx); delete vectorFields; vsapi->freeFrame(src); vsapi->freeFrame(ref); vsapi->freeFrame(dst); return NULL; } // if tff was passed, it overrides _Field. if (d->tffexists) reftff = d->tff && (nref % 2 == 0); //child->GetParity(n); // bool tff; int fieldShift = 0; if (d->fields && d->analysisData.nPel > 1 && (d->analysisData.nDeltaFrame % 2)) { fieldShift = (srctff && !reftff) ? d->analysisData.nPel/2 : ( (reftff && !srctff) ? -(d->analysisData.nPel/2) : 0); // vertical shift of fields for fieldbased video at finest level pel2 } for (int plane = 0; plane < d->supervi->format->numPlanes; plane++) { pRef[plane] = vsapi->getReadPtr(ref, plane); nRefPitch[plane] = vsapi->getStride(ref, plane); } MVGroupOfFrames *pSrcGOF = new MVGroupOfFrames(d->nSuperLevels, d->analysisData.nWidth, d->analysisData.nHeight, d->nSuperPel, d->nSuperHPad, d->nSuperVPad, d->nSuperModeYUV, d->isse, d->analysisData.xRatioUV, d->analysisData.yRatioUV, d->supervi->format->bitsPerSample); MVGroupOfFrames *pRefGOF = new MVGroupOfFrames(d->nSuperLevels, d->analysisData.nWidth, d->analysisData.nHeight, d->nSuperPel, d->nSuperHPad, d->nSuperVPad, d->nSuperModeYUV, d->isse, d->analysisData.xRatioUV, d->analysisData.yRatioUV, d->supervi->format->bitsPerSample); // cast away the const, because why not. pSrcGOF->Update(d->nModeYUV, (uint8_t *)pSrc[0], nSrcPitch[0], (uint8_t *)pSrc[1], nSrcPitch[1], (uint8_t *)pSrc[2], nSrcPitch[2]); // v2.0 pRefGOF->Update(d->nModeYUV, (uint8_t *)pRef[0], nRefPitch[0], (uint8_t *)pRef[1], nRefPitch[1], (uint8_t *)pRef[2], nRefPitch[2]); // v2.0 DCTClass *DCTc = NULL; if (d->dctmode != 0) { /* // FIXME: deal with this inline asm shit if (d->isse && (d->blksize == 8) && d->blksizev == 8) DCTc = new DCTINT(d->blksize, d->blksizev, d->dctmode); else */ DCTc = new DCTFFTW(d->blksize, d->blksizev, d->dctmode, d->vi->format->bitsPerSample); } vectorFields->RecalculateMVs(balls, pSrcGOF, pRefGOF, d->searchType, d->nSearchParam, d->nLambda, d->pnew, reinterpret_cast<int*>(pDst), NULL, fieldShift, d->thSAD, DCTc, d->smooth, d->meander); if (d->divideExtra) { // make extra level with divided sublocks with median (not estimated) motion vectorFields->ExtraDivide(reinterpret_cast<int*>(pDst)); } delete vectorFields; if (DCTc) delete DCTc; delete pSrcGOF; delete pRefGOF; vsapi->freeFrame(ref); } else // too close to the beginning or end to do anything { vectorFields->WriteDefaultToArray(reinterpret_cast<int*>(pDst)); delete vectorFields; } // FIXME: Get rid of all mmx shit. mvtools_cpu_emms(); vsapi->freeFrame(src); return dst; } return 0; }