1 files changed, 100 insertions, 0 deletions
diff --git a/decoder.c b/decoder.c
new file mode 100644
index 0000000..eaa3160
--- /dev/null
+++ b/decoder.c
@@ -0,0 +1,100 @@
+/*
+ * OSD Picture in Picture plugin for the Video Disk Recorder
+ *
+ * See the README file for copyright information and how to reach the author.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "decoder.h"
+#include "config.h"
+
+
+int cDecoder::Open()
+{
+	m_Codec = avcodec_find_decoder(CODEC_ID_MPEG2VIDEO);
+	if (!m_Codec)
+	{
+		printf("codec not found\n");
+		return -1;
+	}
+	m_Context = avcodec_alloc_context();
+	if (avcodec_open(m_Context, m_Codec) < 0)
+	{
+		printf("could not open codec\n");
+		return -1;
+	}
+	m_PicDecoded = avcodec_alloc_frame();
+	m_PicResample = avcodec_alloc_frame();
+	m_BufferResample = new unsigned char[(400 * 300 * 3) / 2]; // size for YUV 420
+	m_PicConvert = avcodec_alloc_frame();
+	m_BufferConvert = new unsigned char[400 * 300 * 4]; // size for RGBA32
+
+	return 0;
+}
+
+int cDecoder::Close()
+{
+	delete[] m_BufferConvert;
+	free(m_PicConvert);
+	delete[] m_BufferResample;
+	free(m_PicResample);
+	avcodec_close(m_Context);
+	free(m_Context);
+	free(m_PicDecoded);
+
+	return 0;
+}
+
+int cDecoder::Decode(unsigned char * data, int length)
+{
+	int gotPicture, len;
+
+	len = avcodec_decode_video(m_Context, m_PicDecoded, &gotPicture, data, length);
+	if (len < 0)
+	{
+		printf("Error while decoding frame\n");
+		return -1;
+	}
+	if (!gotPicture)
+	{
+		return -1;
+	}
+
+	return 0;
+}
+
+int cDecoder::Resample(int width, int height)
+{
+	ImgReSampleContext * contextResample;
+
+	m_Width = width;
+	m_Height = height;
+	contextResample = img_resample_full_init(m_Width, m_Height,
+			m_Context->width, m_Context->height,
+			OsdPipSetup.CropTop, OsdPipSetup.CropBottom,
+			OsdPipSetup.CropLeft, OsdPipSetup.CropRight);
+	if (!contextResample) {
+		printf("Error initializing resample context.\n");
+		return -1;
+	}
+	avpicture_fill((AVPicture *) m_PicResample, m_BufferResample,
+			PIX_FMT_YUV420P, m_Width, m_Height);
+	img_resample(contextResample, (AVPicture *) m_PicResample, (AVPicture *) m_PicDecoded);
+	img_resample_close(contextResample);
+
+	return 0;
+}
+
+int cDecoder::ConvertToRGB()
+{
+	avpicture_fill((AVPicture *) m_PicConvert, m_BufferConvert,
+			PIX_FMT_RGBA32, m_Width, m_Height);
+	img_convert((AVPicture *) m_PicConvert, PIX_FMT_RGBA32,
+				(AVPicture *) m_PicResample, PIX_FMT_YUV420P,
+				m_Width, m_Height);
+
+	return 0;
+}
+