#pragma warning(disable : 4996)

#include <map>
#include <string>
#include <vector>
#include <fstream>
#include <iostream>
#include <iomanip>
#include <ios>
#include <streambuf>
#include <sstream>

#include <windows.h>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <commctrl.h>
#include <time.h>
#include <limits.h>
#include <shlobj.h>
#include <process.h>
#include <fcntl.h>
#include <io.h>
#include <shlwapi.h>
#include <tchar.h>
#include <stdarg.h>

//boost libraries
#include <boost/foreach.hpp>


// ObjectModuleの内容をXmlで生成する場合はこの行を有効にする
//#define OBJECT_MODULE_IS_NOT_BINARY


#ifdef OBJECT_MODULE_IS_NOT_BINARY
#include <boost/archive/xml_oarchive.hpp>
#include <boost/archive/xml_iarchive.hpp>
#else
#include <boost/archive/binary_oarchive.hpp>
#include <boost/archive/binary_iarchive.hpp>
#endif
#include <boost/serialization/string.hpp>
#include <boost/serialization/access.hpp>
#include <boost/serialization/level.hpp>
#include <boost/serialization/vector.hpp>
#include <boost/serialization/map.hpp>
#include <boost/serialization/version.hpp>
#include <boost/serialization/is_abstract.hpp>
#include <boost/serialization/serialization.hpp>
#include <boost/serialization/nvp.hpp>
#include <boost/serialization/export.hpp>

#include <jenga/include/jenga.h>
#include <abdev/ab_common/include/ab_common.h>


void ObjectModule::StaticLink( ObjectModule &objectModule, bool isSll )
{
	const std::vector<int> relationTable = this->GetRelationTable( objectModule.relationalObjectModuleNames );

	long dataSectionBaseOffset = dataTable.GetSize();

	// メタ情報を結合
	meta.StaticLink( objectModule.meta, dataSectionBaseOffset, relationTable );

	if( !isSll )
	{
		// グローバル ネイティブコードを結合
		objectModule.globalNativeCode.ResetDataSectionBaseOffset( dataSectionBaseOffset );
		objectModule.globalNativeCode.ResetRelationalObjectModuleIndex( relationTable );
		globalNativeCode.PutEx( objectModule.globalNativeCode );

		// データテーブルを結合
		objectModule.dataTable.ResetDataSectionBaseOffset( dataSectionBaseOffset );
		dataTable.Add( objectModule.dataTable );
	}

	// 依存関係を解決
	this->Resolve();

	// TODO: basbufがいらなくなったら消す
	extern char *basbuf;
	basbuf = this->source.GetBuffer();
}

void ObjectModule::Resolve()
{
	this->meta.Resolve();

	// グローバルネイティブコードを解決（スケジュールを解決する）
	this->globalNativeCode.Resolve();

	// データテーブルを解決（スケジュールを解決する）
	this->dataTable.Resolve();
}

const std::vector<int> ObjectModule::GetRelationTable( const Jenga::Common::Strings &oldRelationalObjectModuleNames )
{
	// 要素 = 古いインデックス、値 = 新しいインデックス
	std::vector<int> relationTable;

	// リレーションテーブルを構築
	BOOST_FOREACH( const std::string &oldRelationalObjectModuleName, oldRelationalObjectModuleNames )
	{
		bool isMatch = false;
		for( int i=0; i<static_cast<int>(this->relationalObjectModuleNames.size()); i++ )
		{
			if( oldRelationalObjectModuleName == this->relationalObjectModuleNames[i] )
			{
				isMatch = true;
				relationTable.push_back( i );
				break;
			}
		}

		if( !isMatch )
		{
			// エラー
			_ASSERT( false );
		}
	}

	return relationTable;
}

bool ObjectModule::Read( const std::string &filePath )
{
	// XMLとして読み込む
	bool isSuccessful = false;

	try{
#ifdef OBJECT_MODULE_IS_NOT_BINARY
		std::ifstream ifs( filePath.c_str() );
		boost::archive::xml_iarchive ia(ifs);
#else
		std::ifstream ifs( filePath.c_str(), std::ios::in | std::ios::binary );
		boost::archive::binary_iarchive ia(ifs);
#endif

		// ファイルから読込
		ia >> boost::serialization::make_nvp( RootTagName(), *this );

		isSuccessful = true;
	}
	catch( boost::archive::archive_exception e )
	{
		MessageBox( NULL, e.what(), "XMLシリアライズの例外", MB_OK );
	}
	catch(...){
		MessageBox( NULL, "archive_exception以外の不明な例外", "XMLシリアライズの例外", MB_OK );
	}

	if( !isSuccessful )
	{
		return false;
	}

	return true;
}
bool ObjectModule::Write( const std::string &filePath ) const
{
	bool isSuccessful = false;

	try{
#ifdef OBJECT_MODULE_IS_NOT_BINARY
		std::ofstream ofs( filePath.c_str() );
		boost::archive::xml_oarchive oa(ofs);
#else
		std::ofstream ofs( filePath.c_str(), std::ios::out | std::ios::binary );
		boost::archive::binary_oarchive oa(ofs);
#endif

		// ファイルに書き出し
		oa << boost::serialization::make_nvp( RootTagName(), *this );

		isSuccessful = true;
	}
	catch( boost::archive::archive_exception e )
	{
		MessageBox( NULL, e.what(), "XMLシリアライズの例外", MB_OK );
	}
	catch(...){
		MessageBox( NULL, "archive_exception以外の不明な例外", "XMLシリアライズの例外", MB_OK );
	}

	if( !isSuccessful )
	{
		return false;
	}

	return true;
}
bool ObjectModule::ReadString( const std::string &str )
{
	bool isSuccessful = false;

	// 入力アーカイブの作成

	try{
#ifdef OBJECT_MODULE_IS_NOT_BINARY
		std::istringstream iss( str );
		boost::archive::xml_iarchive ia(iss);
#else
		std::istringstream iss( str, std::ios::in | std::ios::binary );
		boost::archive::binary_iarchive ia(iss);
#endif

		// 文字列ストリームから読込
		ia >> boost::serialization::make_nvp( RootTagName(), *this );

		isSuccessful = true;
	}
	catch( boost::archive::archive_exception e )
	{
		MessageBox( NULL, e.what(), "XMLシリアライズの例外", MB_OK );
	}
	catch(...){
		MessageBox( NULL, "archive_exception以外の不明な例外", "XMLシリアライズの例外", MB_OK );
	}

	if( !isSuccessful )
	{
		return false;
	}

	return true;
}
bool ObjectModule::WriteString( std::string &str ) const
{
	// 出力アーカイブの作成

	bool isSuccessful = false;
	try{
#ifdef OBJECT_MODULE_IS_NOT_BINARY
		std::ostringstream oss;
		boost::archive::xml_oarchive oa(oss);
#else
		std::ostringstream oss( "", std::ios::out | std::ios::binary );
		boost::archive::binary_oarchive oa(oss);
#endif

		// 文字列ストリームに書き出し
		oa << boost::serialization::make_nvp( RootTagName(), *this );

		str = oss.str();

		isSuccessful = true;
	}
	catch( boost::archive::archive_exception e )
	{
		MessageBox( NULL, e.what(), "XMLシリアライズの例外", MB_OK );
	}
	catch(...){
		MessageBox( NULL, "archive_exception以外の不明な例外", "XMLシリアライズの例外", MB_OK );
	}

	return isSuccessful;
}