#pragma once

#include "stdafx.h"

#include <map>
#include <stdio.h>
#include <stdlib.h>
#include <Windows.h>
#include <stdint.h>
#include <tchar.h>
#include <string>
#include <direct.h>
#include <iostream>

#include <algorithm>

//BOOST
#include <boost/filesystem.hpp>
#include <boost/algorithm/string.hpp>
#include "boost/shared_ptr.hpp"
//#include <boost/thread.hpp>
#include <boost/format.hpp>

//////////////////////////////////////////////////////////////////////////
//Definitions
//////////////////////////////////////////////////////////////////////////
typedef int (*pinit_h264ToBMP)(uint32_t width, uint32_t height, uint32_t pps);
typedef BOOL (*ppost_BMP_RGB)(const uint8_t *h264_data, uint32_t h264_data_size,unsigned char *decodebuf);
typedef void (*pFree_all)();

struct H264Decode
{
	pinit_h264ToBMP		m_DLL_init_h264ToBMP_without_encoder;
	ppost_BMP_RGB		m_DLL_post_BMP_RGB;
	pFree_all			m_Dll_pFree_all;

	HINSTANCE		m_h264Dll;
	unsigned char*      m_tmpBuf;
	
	bool Init(int decodeId)
	{
		char dllname[255];
		sprintf(dllname,"h264ToBMP_%d.dll",decodeId);
		std::string destfilename = boost::str(boost::format("h264ToBMP_%d.dll") % decodeId);
		boost::filesystem::path destPath(destfilename);
		boost::filesystem::path basePath("h264ToBMP.dll");
		if (!boost::filesystem::is_regular_file(destPath))
		{
			//winXP下，如果已存文件再copy一下会出现问题，如果catch住，就可以避免其发生
			try
			{
				boost::filesystem::copy_file(basePath, destPath);
			}catch(boost::filesystem::filesystem_error)
			{
			}
		}

		m_h264Dll = LoadLibrary(destfilename.c_str()); 

		if (nullptr == m_h264Dll)
		{
			return false;
		}
		m_DLL_init_h264ToBMP_without_encoder = (pinit_h264ToBMP) GetProcAddress(m_h264Dll, "init_h264ToBMP_without_encoder");
		m_DLL_post_BMP_RGB = (ppost_BMP_RGB) GetProcAddress(m_h264Dll, "post_BMP_RGB");
		m_Dll_pFree_all = (pFree_all) GetProcAddress(m_h264Dll, "free_all");
		m_tmpBuf = new unsigned char[2500*1500];
		return true;
	}

	int InitDecoder(int width, int height, int pps)
	{
		if (m_h264Dll != nullptr)
		{
			int ret = m_DLL_init_h264ToBMP_without_encoder(width, height, pps);
			return ret;
		}
		else
		{
			return 0;
		}
	}
	
	int	DecodeVideo(const char* data,int msgLen,unsigned char* imgbuf)
	{
		try
		{
			if (m_h264Dll != nullptr)
			{
				static bool firstTime = true;
				int length = msgLen;
				if (true == firstTime)
				{
					firstTime = false;
					uint8_t sps_frame[16] = { 0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x80, 0x1E, 0x89, 0x95, 0x41, 0x68, 0x22, 0xD0, 0x04 };//GAODE NEW

					memcpy(m_tmpBuf, sps_frame, 15);
					memcpy(m_tmpBuf + 15, data, msgLen);
					length = msgLen + 15;
				}
				else
				{
					memcpy(m_tmpBuf, data, msgLen);
				}

				if (m_tmpBuf != nullptr && length > 0 && imgbuf != nullptr)
				{
					// 此处容易在测控状态不好的情况下报错。
					try
					{
							int got_bmp = m_DLL_post_BMP_RGB(m_tmpBuf, length, imgbuf);
							return got_bmp;
					}
					catch (...)
					{
						
					}
				}
				else
				{
					return 0;
				}
			}
			else
			{
				return 0;
			}
		}
		catch (CMemoryException* e)
		{
			e;
			return 0;
		}
		catch (CFileException* e)
		{
			e;
			return 0;
		}
		catch (CException* e)
		{
			e;
			return 0;
		}

		return 1;
	}

	void DeInit()
	{
		delete m_tmpBuf;
		m_tmpBuf = nullptr;

		m_h264Dll = nullptr;

		m_Dll_pFree_all();
	}
};