VS_IT Initial Porting

HomeOfVapourSynthEvolution · Oct 1, 2014 · 97d69c6 · 97d69c6
1 parent 8ce8a31
commit 97d69c6
Show file tree

Hide file tree

Showing 10 changed files with 1,912 additions and 0 deletions.
diff --git a/vs_it.sln b/vs_it.sln
@@ -0,0 +1,22 @@
+
+Microsoft Visual Studio Solution File, Format Version 12.00
+# Visual Studio 2013
+VisualStudioVersion = 12.0.21005.1
+MinimumVisualStudioVersion = 10.0.40219.1
+Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "vs_it", "vs_it\vs_it.vcxproj", "{296F3E49-8C9B-47E2-BF8A-B1EDD6A322CB}"
+EndProject
+Global
+	GlobalSection(SolutionConfigurationPlatforms) = preSolution
+		Debug|Win32 = Debug|Win32
+		Release|Win32 = Release|Win32
+	EndGlobalSection
+	GlobalSection(ProjectConfigurationPlatforms) = postSolution
+		{296F3E49-8C9B-47E2-BF8A-B1EDD6A322CB}.Debug|Win32.ActiveCfg = Debug|Win32
+		{296F3E49-8C9B-47E2-BF8A-B1EDD6A322CB}.Debug|Win32.Build.0 = Debug|Win32
+		{296F3E49-8C9B-47E2-BF8A-B1EDD6A322CB}.Release|Win32.ActiveCfg = Release|Win32
+		{296F3E49-8C9B-47E2-BF8A-B1EDD6A322CB}.Release|Win32.Build.0 = Release|Win32
+	EndGlobalSection
+	GlobalSection(SolutionProperties) = preSolution
+		HideSolutionNode = FALSE
+	EndGlobalSection
+EndGlobal
diff --git a/vs_it/IScriptEnvironment.cpp b/vs_it/IScriptEnvironment.cpp
@@ -0,0 +1,35 @@
+/*
+VS_IT Copyright(C) 2002 thejam79, 2003 minamina, 2014 msg7086
+
+This program is free software; you can redistribute it and / or
+modify it under the terms of the GNU General Public License
+as published by the Free Software Foundation; either version 2
+of the License, or(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110 - 1301, USA.
+*/
+
+#include "IScriptEnvironment.h"
+
+IScriptEnvironment::IScriptEnvironment(VSFrameContext *_frameCtx, VSCore *_core, const VSAPI *_vsapi)
+: frameCtx(_frameCtx), core(_core), vsapi(_vsapi)
+{
+	m_iSumC = m_iSumP = m_iSumN = 0;
+	m_iUsePrev = m_iUseNext = 0;
+}
+
+IScriptEnvironment::~IScriptEnvironment()
+{
+}
+
+VSFrameRef *IScriptEnvironment::NewVideoFrame(const VSVideoInfo *vi)
+{
+	return vsapi->newVideoFrame(vi->format, vi->width, vi->height, nullptr, core);
+}
diff --git a/vs_it/IScriptEnvironment.h b/vs_it/IScriptEnvironment.h
@@ -0,0 +1,83 @@
+/*
+VS_IT Copyright(C) 2002 thejam79, 2003 minamina, 2014 msg7086
+
+This program is free software; you can redistribute it and / or
+modify it under the terms of the GNU General Public License
+as published by the Free Software Foundation; either version 2
+of the License, or(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110 - 1301, USA.
+*/
+
+#pragma once
+#include "vs_it_interface.h"
+
+enum REFTYPE;
+
+class CFrameInfo {
+public:
+	char pos;
+	char match;
+	char matchAcc;
+	char ip;
+	char out;
+	char mflag;
+	int diffP0;
+	int diffP1;
+	int diffS0;
+	int diffS1;
+	long ivC, ivP, ivN, ivM;
+	long ivPC, ivPP, ivPN;
+};
+
+class CTFblockInfo {
+public:
+	int cfi;
+	char level;
+	char itype;
+};
+
+class IScriptEnvironment
+{
+public:
+	int m_iRealFrame;
+	CFrameInfo *m_frameInfo;
+	CTFblockInfo *m_blockInfo;
+	unsigned char *m_edgeMap, *m_motionMap4DI, *m_motionMap4DIMax;
+
+	long m_iSumC, m_iSumP, m_iSumN, m_iSumM;
+	long m_iSumPC, m_iSumPP, m_iSumPN, m_iSumPM;
+	int m_iCurrentFrame;
+	bool m_bRefP;
+	int m_iUsePrev, m_iUseNext;
+	int m_iUseFrame;
+
+	VSFrameContext *frameCtx;
+	VSCore *core;
+	const VSAPI *vsapi;
+	IScriptEnvironment(VSFrameContext *_frameCtx, VSCore *_core, const VSAPI *_vsapi);
+	~IScriptEnvironment();
+	VSFrameRef *IScriptEnvironment::NewVideoFrame(const VSVideoInfo * vi);
+	void BitBlt(BYTE* dstp, int dst_pitch, const BYTE* srcp, int src_pitch, int row_size, int height) {
+		if ((!height) || (!row_size)) return;
+
+		if (height == 1 || (dst_pitch == src_pitch && src_pitch == row_size)) {
+			memcpy(dstp, srcp, row_size*height);
+		}
+		else {
+			for (int y = height; y > 0; --y) {
+				memcpy(dstp, srcp, row_size);
+				dstp += dst_pitch;
+				srcp += src_pitch;
+			}
+		}
+	}
+};
+
diff --git a/vs_it/vs_it.cpp b/vs_it/vs_it.cpp
@@ -0,0 +1,236 @@
+/*
+VS_IT Copyright(C) 2002 thejam79, 2003 minamina, 2014 msg7086
+
+This program is free software; you can redistribute it and / or
+modify it under the terms of the GNU General Public License
+as published by the Free Software Foundation; either version 2
+of the License, or(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110 - 1301, USA.
+*/
+
+#include "vs_it.h"
+
+IT::IT(VSVideoInfo * vi,
+	VSNodeRef * node,
+	int _fps,
+	int _threshold,
+	int _pthreshold,
+	const VSAPI *vsapi) :
+	vi(vi),
+	node(node),
+	m_iFPS(_fps),
+	m_iThreshold(_threshold),
+	m_iPThreshold(_pthreshold)
+{
+	m_iMaxFrames = vi->numFrames;
+	m_iCounter = 0;
+	width = vi->width;
+	height = vi->height;
+	m_bSwap = false;
+	m_iField = false;
+
+	for (int k = 0; k < 32; ++k) {
+		m_pvf[k] = 0;
+		m_ipvfIndex[k] = -1;
+	}
+
+	m_iPThreshold = AdjPara(m_iPThreshold);
+
+	if (m_iFPS == 24) {
+		vi->numFrames = vi->numFrames * 4 / 5;
+		vi->fpsNum *= 4;
+		if (vi->fpsNum % 5 == 0)
+			vi->fpsNum /= 5;
+		else
+			vi->fpsDen *= 5;
+	}
+}
+
+
+void IT::GetFramePre(IScriptEnvironment* env, int n)
+{
+	if (m_iFPS == 24)
+	{
+		int base = n + n / 4;
+		base = (base / 5) * 5;
+		int i;
+		for (i = 0; i < 6; ++i)
+			env->vsapi->requestFrameFilter(base + i, node, env->frameCtx);
+	}
+	else
+		env->vsapi->requestFrameFilter(n, node, env->frameCtx);
+}
+
+PVideoFrame IT::GetFrame(IScriptEnvironment* env, int n)
+{
+	++m_iCounter;
+	env->m_iRealFrame = n;
+	env->m_frameInfo = new CFrameInfo[m_iMaxFrames + 6];
+
+	int i;
+	for (i = 0; i < 8; ++i) {
+		m_PVOut[i] = 0;
+		m_iPVOutIndex[i] = -1;
+	}
+	for (i = 0; i < m_iMaxFrames + 6; ++i) {
+		env->m_frameInfo[i].match = 'U';
+		env->m_frameInfo[i].matchAcc = 'U';
+		env->m_frameInfo[i].pos = 'U';
+		env->m_frameInfo[i].ip = 'U';
+		env->m_frameInfo[i].mflag = 'U';
+		env->m_frameInfo[i].diffP0 = -1;
+		env->m_frameInfo[i].diffP1 = -1;
+	}
+	env->m_blockInfo = new CTFblockInfo[m_iMaxFrames / 5 + 6];
+	for (i = 0; i < m_iMaxFrames / 5 + 1; ++i) {
+		env->m_blockInfo[i].level = 'U';
+		env->m_blockInfo[i].itype = 'U';
+	}
+
+	env->m_edgeMap = new unsigned char[width * height];
+	memset(env->m_edgeMap, width * height, 0);
+
+	env->m_motionMap4DI = new unsigned char[width * height];
+	memset(env->m_motionMap4DI, width * height, 0);
+
+	env->m_motionMap4DIMax = new unsigned char[width * height];
+	memset(env->m_motionMap4DIMax, width * height, 0);
+
+
+	int tfFrame;
+	if (m_iFPS == 24) {
+		tfFrame = n + n / (5 - 1);
+
+		int base = (tfFrame / 5) * 5;
+		int i;
+
+		for (i = 0; i < 5; ++i)
+			GetFrameSub(env, base + i);
+		Decide(env, base);
+
+		bool iflag = true;
+		for (i = 0; i < 5; ++i) {
+			if (FII(base + i).ivC >= m_iPThreshold) {
+				iflag = false;
+			}
+		}
+		if (iflag) {
+			env->m_blockInfo[base / 5].itype = '3';
+		}
+		else {
+			env->m_blockInfo[base / 5].itype = '2';
+		}
+		int no = tfFrame - base;
+		for (i = 0; i < 5; ++i) {
+			char f = FII(base + i).mflag;
+			if (f != 'D' && f != 'd' && f != 'X' && f != 'x' && f != 'y' && f != 'z' && f != 'R') {
+				if (no == 0)
+					break;
+				--no;
+			}
+		}
+		if (m_iFPS != 30)
+			n = clipFrame(i + base);
+	}
+	else {
+		GetFrameSub(env, n);
+	}
+	VSFrameRef * dst = env->NewVideoFrame(vi);
+	MakeOutput(env, dst, n);
+	delete[] env->m_frameInfo;
+	delete[] env->m_blockInfo;
+	delete[] env->m_edgeMap;
+	delete[] env->m_motionMap4DI;
+	delete[] env->m_motionMap4DIMax;
+	return dst;
+}
+
+void IT::GetFrameSub(IScriptEnvironment* env, int n)
+{
+	if (n >= m_iMaxFrames)
+		return;
+	if (env->m_frameInfo[n].ip != 'U') {
+		return;
+	}
+	env->m_iCurrentFrame = n;
+
+	env->m_iUseFrame = 'C';
+	env->m_iSumC = env->m_iSumP = env->m_iSumN = env->m_iSumM = 720 * 480;
+	env->m_bRefP = true;
+
+	ChooseBest(env, n);
+	env->m_frameInfo[n].match = (unsigned char)env->m_iUseFrame;
+	switch (toupper(env->m_iUseFrame)) {
+	case 'C':
+		env->m_iSumM = env->m_iSumC;
+		env->m_iSumPM = env->m_iSumPC;
+		break;
+	case 'P':
+		env->m_iSumM = env->m_iSumP;
+		env->m_iSumPM = env->m_iSumPP;
+		break;
+	case 'N':
+		env->m_iSumM = env->m_iSumN;
+		env->m_iSumPM = env->m_iSumPN;
+		break;
+	}
+
+	env->m_frameInfo[n].ivC = env->m_iSumC;
+	env->m_frameInfo[n].ivP = env->m_iSumP;
+	env->m_frameInfo[n].ivN = env->m_iSumN;
+	env->m_frameInfo[n].ivM = env->m_iSumM;
+	env->m_frameInfo[n].ivPC = env->m_iSumPC;
+	env->m_frameInfo[n].ivPP = env->m_iSumPP;
+	env->m_frameInfo[n].ivPN = env->m_iSumPN;
+	if (env->m_iSumM < m_iPThreshold && env->m_iSumPM < m_iPThreshold * 3) {
+		env->m_frameInfo[n].ip = 'P';
+	}
+	else {
+		env->m_frameInfo[n].ip = 'I';
+	}
+	return;
+}
+
+const VSFrameRef* IT::MakeOutput(IScriptEnvironment* env, VSFrameRef* dst, int n)
+{
+	env->m_iCurrentFrame = n;
+
+	env->m_iSumC = env->m_frameInfo[n].ivC;
+	env->m_iSumP = env->m_frameInfo[n].ivP;
+	env->m_iSumN = env->m_frameInfo[n].ivN;
+	env->m_iSumM = env->m_frameInfo[n].ivM;
+	env->m_iSumPC = env->m_frameInfo[n].ivPC;
+	env->m_iSumPP = env->m_frameInfo[n].ivPP;
+	env->m_iSumPN = env->m_frameInfo[n].ivPN;
+
+	env->m_bRefP = true;
+
+	env->m_iUseFrame = toupper(env->m_frameInfo[n].match);
+
+#ifdef DEBUG_SHOW_INTERLACE
+	//    ShowInterlaceArea(dst, n);
+	//    PrintDebugInfo(dst, n);
+	USE_MMX2
+		return dst;
+#endif // DEBUG_SHOW_INTERLACE
+
+	if (env->m_frameInfo[n].ip == 'P')
+	{
+		CopyCPNField(env, dst, n);
+	}
+	else {
+		if (!DrawPrevFrame(env, dst, n))
+			DeintOneField_YV12(env, dst, n);
+	}
+
+	return dst;
+}
+