OTS/OTSCPP/OTSRptCalculate/Model/BSEImgFileMgr.cpp

#pragma once
#include "stdafx.h"
#include "BSEImgFileMgr.h"

namespace OTSMODEL {

	// CMsrParamFileMrg

	//IMPLEMENT_SERIAL(CBSEImgFileMgr, CObject, 1)

		// constructor
	CBSEImgFileMgr::CBSEImgFileMgr()
	{
		// initialization
		
		 //Init();
	}

	// destructor
	CBSEImgFileMgr::~CBSEImgFileMgr()
	{
		// cleanup
		//Cleanup();
	}

	// CMsrParamFileMrg member functions
	// public

	// serialization
	//void CBSEImgFileMgr::Serialize(CArchive& ar)
	//{
	//	// store?
	//	if (ar.IsStoring())
	//	{
	//		// store
	//		ar << BSE_IMG_FILE_MARK;
	//		ar << BSE_IMG_FILE_VERSION;
	//	}
	//	else
	//	{
	//		// load
	//		int nMark;
	//		ar >> nMark;
	//		if (nMark != BSE_IMG_FILE_MARK)
	//		{
	//			// invalid stage file
	//			LogErrorTrace(__FILE__, __LINE__, _T("Serialize: invalid BSE image file."));
	//			return;
	//		}
	//		CString strFileVersion;
	//		ar >> strFileVersion;
	//	}

	//	// member objects serialization
	//	m_poBSE->Serialize(ar);

	//	// base object serialization 
	//	CObject::Serialize(ar);
	//}

	// Load/Save
	//BOOL CBSEImgFileMgr::Load(CString a_strPathName /*= _T("")*/, BOOL a_bClear /*= TRUE*/)
	//{

	//	AFX_MANAGE_STATE(AfxGetStaticModuleState());

	//	// clear all data if necessary
	//	if (a_bClear)
	//	{
	//		Init();
	//	}

	//	// check file pathname
	//	a_strPathName.Trim();
	//	if (a_strPathName.IsEmpty())
	//	{
	//		// file open dialog
	//		CFileDialog dlg(TRUE, BSE_IMG_FILE_EXT, NULL, OFN_FILEMUSTEXIST, BSE_IMG_FILE_FILTER);
	//		if (dlg.DoModal() != IDOK)
	//		{
	//			return FALSE;
	//		}

	//		// get file pathname
	//		a_strPathName = dlg.GetPathName();
	//	}
	//	// open the particle analysis standard file
	//	CFile hFile;
	//	CFileException ex;
	//	if (!hFile.Open(a_strPathName, CFile::modeRead, &ex))
	//	{
	//		// failed to open the file
	//		TCHAR szCause[255];
	//		ex.GetErrorMessage(szCause, 255);
	//		LogErrorTrace(__FILE__, __LINE__, _T("Load: can't open file %s. error: %s"), a_strPathName, szCause);
	//		return FALSE;
	//	}

	//	// create a loading archive 
	//	CArchive ar(&hFile, CArchive::load);

	//	// file serialization (load)
	//	Serialize(ar);

	//	// close the file
	//	ar.Close();

	//	// file pathname
	//	m_strPathName = a_strPathName;

	//	// ok, return TRUE
	//	return TRUE;
	//}

	//BOOL CBSEImgFileMgr::Save(CString a_strPathName /*= _T("")*/)
	//{

	//	AFX_MANAGE_STATE(AfxGetStaticModuleState());
	//
	//	// check file pathname
	//	a_strPathName.Trim();
	//	if (a_strPathName.IsEmpty())
	//	{
	//		// file save as dialog
	//		CFileDialog dlg(FALSE, BSE_IMG_FILE_EXT, NULL, OFN_OVERWRITEPROMPT, BSE_IMG_FILE_FILTER);
	//		if (dlg.DoModal() != IDOK)
	//		{
	//			return FALSE;
	//		}

	//		// get file pathname
	//		a_strPathName = dlg.GetPathName();
	//	}

	//	// create the file
	//	CFile hFile;
	//	CFileException ex;
	//	if (!hFile.Open(a_strPathName, CFile::modeCreate | CFile::modeWrite, &ex))
	//	{
	//		// failed to open file
	//		TCHAR szCause[255];
	//		ex.GetErrorMessage(szCause, 255);
	//		LogErrorTrace(__FILE__, __LINE__, _T("Save: failed to create file. %s. error: %s"), a_strPathName, szCause);
	//		return FALSE;
	//	}

	//	// create a store archive 
	//	CArchive ar(&hFile, CArchive::store);

	//	// file serialization (store)
	//	Serialize(ar);

	//	// close the file
	//	//hFile.Close();

	//	// file pathname
	//	m_strPathName = a_strPathName;

	//	// ok, return TRUE
	//	return TRUE;
	//}

	BOOL CBSEImgFileMgr::LoadFromBitmap(CString a_strPathName /*= _T("")*/, BOOL a_bClear/* = TRUE*/)
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());

		// prepare file
		CFile file;
		CFileException fe;
		if (a_bClear)
		{
			//Init();
		}
		// check file pathname
		a_strPathName.Trim();
		if (a_strPathName.IsEmpty())
		{
			// file save as dialog
			CFileDialog dlg(TRUE, BMP_IMG_FILE_EXT, NULL, OFN_FILEMUSTEXIST, BMP_IMG_FILE_FILTER);
			if (dlg.DoModal() != IDOK)
			{
				return FALSE;
			}

			// get file pathname
			m_strPathName = dlg.GetPathName();
		}
		else
		{
			m_strPathName = a_strPathName;
		}

		BYTE* pPixel;
		BYTE* pBmInfo;
		BITMAPFILEHEADER* pBmfh;
		pBmfh = new BITMAPFILEHEADER;
		

		if (!file.Open(m_strPathName, CFile::modeRead, &fe))
		{
			LogInfoTrace(__FILE__, __LINE__, _T("CBSEImgFileMgr::LoadFromBitmap: failed to open image file %s."), a_strPathName);
			return FALSE;
		}

		// read file header
		file.Read(pBmfh, sizeof(BITMAPFILEHEADER));
		pBmInfo = new BYTE[pBmfh->bfOffBits - 14];

		file.Read(pBmInfo, pBmfh->bfOffBits - 14);

		BITMAPINFOHEADER bmih;
		memcpy(&bmih, pBmInfo, sizeof(BITMAPINFOHEADER));

		long width = bmih.biWidth;//获取宽度
		int bitCount = bmih.biBitCount;//获取位数
		long  height = bmih.biHeight;//获取高度
		long LineBytes = (width*bitCount + 31) / 32 * 4;//计算每行像素所占的字节数

		pPixel = new BYTE[height*LineBytes];
		file.Read(pPixel, height*LineBytes);
		file.Close();		

		CRect imageRect(0, 0, width, height);
		m_poBSE = CBSEImgPtr(new CBSEImg(imageRect));
		//m_poBSE->SetImageRect(imageRect);
		m_poBSE->InitImageData(width,height);
		BYTE* pImageData = m_poBSE->GetImageDataPointer();

		long nActImageSize = width * height;

		for (int i = 0; i < height; i++)
		{
			memcpy(pImageData + i*width, pPixel+(height-1-i)*width, width);
		}
		
		delete[]pPixel;
		delete[]pBmInfo;
		delete pBmfh;

		return TRUE;
	}

	BOOL CBSEImgFileMgr::SaveIntoBitmap(CString a_strPathName)
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());

		// prepare file
		CFile file;
		CFileException fe;

		// check file pathname
		a_strPathName.Trim();
		if (a_strPathName.IsEmpty())
		{
			// file save as dialog
			CFileDialog dlg(FALSE, BMP_IMG_FILE_EXT, NULL, OFN_OVERWRITEPROMPT, BMP_IMG_FILE_FILTER);
			if (dlg.DoModal() != IDOK)
			{
				return FALSE;
			}

			// get file pathname
			a_strPathName = dlg.GetPathName();
		}
		
		// create the file	
		if (!file.Open(a_strPathName, CFile::modeCreate | CFile::modeWrite, &fe))
		{
			// failed to open file
			TCHAR szCause[255];
			fe.GetErrorMessage(szCause, 255);
			LogErrorTrace(__FILE__, __LINE__, _T("SaveIntoBitmap: failed to create file. %s. error: %s"), a_strPathName, szCause);
			return FALSE;
		}

		LONG nHeight = 0;
		LONG nWidth = 0;
		DWORD bfSize = 0;
		
		nHeight = m_poBSE->GetHeight();
		nWidth = m_poBSE->GetWidth();
						
		BITMAPFILEHEADER* pBmfh;
		BITMAPINFOHEADER* pBmInfo;
		RGBQUAD* pColorTable;
		BYTE* pPixel;

		bfSize = nHeight + nWidth + 1078;

		pBmfh = new BITMAPFILEHEADER;
		pBmfh->bfType = 0x4d42; // BM
		pBmfh->bfSize = bfSize;
		pBmfh->bfReserved1 = 0;
		pBmfh->bfReserved2 = 0;
		pBmfh->bfOffBits = 1078; // 40 + 14 +1024

		pBmInfo = new BITMAPINFOHEADER;
		pBmInfo->biSize = 40;
		pBmInfo->biWidth = nWidth;
		pBmInfo->biHeight = nHeight;
		pBmInfo->biPlanes = 0;
		pBmInfo->biBitCount = 8;
		pBmInfo->biCompression = 0;
		pBmInfo->biSizeImage = nWidth * nHeight;
		pBmInfo->biXPelsPerMeter = 0;
		pBmInfo->biYPelsPerMeter = 0;
		pBmInfo->biClrUsed = 256;
		pBmInfo->biClrImportant = 0;
		
		pColorTable = new RGBQUAD[256];
		for (int i = 0; i<256; i++)
		{
			pColorTable[i].rgbBlue = i;
			pColorTable[i].rgbGreen = i;
			pColorTable[i].rgbRed = i;
			pColorTable[i].rgbReserved = 0;
		}

		try 
		{
			file.Write(pBmfh, sizeof(BITMAPFILEHEADER));
			file.Write(pBmInfo, sizeof(BITMAPINFOHEADER));
			file.Write(pColorTable, sizeof(RGBQUAD) * 256);
			long LineBytes = (nWidth*pBmInfo->biBitCount + 31) / 32 * 4;//计算每行像素所占的字节数
			pPixel = new BYTE[nHeight*LineBytes];
			
			BYTE* pImageData = m_poBSE->GetImageDataPointer();

			for (int i = 0; i < nHeight; i++)
			{
				memcpy(pPixel + (nHeight - 1 - i)*nWidth, pImageData + i*nWidth, nWidth);
			}
			file.Write(pPixel, nHeight*LineBytes);
			delete[]pPixel;
		}
		catch (CException *pe)
		{
			pe->Delete();
			LogErrorTrace(__FILE__, __LINE__, _T("SaveIntoBitmap:can't write into file."));
		}


		
		delete pBmInfo;
		delete pBmfh;
		delete[] pColorTable;

		return TRUE;

	}

	void CBSEImgFileMgr::SetBSEImg(CBSEImgPtr a_poBSE)
	{
		ASSERT(a_poBSE);
		if (!a_poBSE)
		{
			LogErrorTrace(__FILE__, __LINE__, _T("SetBSEImg::invalid BSE image pointer"));
			return;
		}
		m_poBSE = a_poBSE;
		
	}

	//void CBSEImgFileMgr::Serialize(bool isStoring, tinyxml2::XMLDocument * classDoc, tinyxml2::XMLElement * rootNode)
	//{
	//	//xmls::xBool xbModify;
	//	
	//	xmls::xInt xBseImgFileMark;
	//	xmls::xString xBseImgFileVersion; 
	//	//xmls::Collection<CStage> xStagelist;
	//	xmls::Slo slo;
	//	slo.Register("BseImgFileMark", &xBseImgFileMark);
	//	slo.Register("BseImgFileVersion", &xBseImgFileVersion);
	//	//slo.Register("BSE", m_poBSE.get());
	//	//this->Register("WorkingStageId", &xBseImgFileMark);
	//	//this->Register("Stagelist", &xStagelist);
	//	//->Register("StageData", m_pStageData.get());
	//	if (isStoring)
	//	{
	//		xBseImgFileMark = BSE_IMG_FILE_MARK;
	//		xBseImgFileVersion = BSE_IMG_FILE_VERSION;
	//		//xBseImgFileMark = m_nWorkingStageId;

	//		slo.Serialize(true, classDoc, rootNode);
	//	}
	//	else
	//	{
	//		slo.Serialize(false, classDoc, rootNode);
	//	}
	//}
	// protected

	// cleanup 
	//void CBSEImgFileMgr::Cleanup()
	//{
	//	// need to do nothing at the moment
	//}

	//// initialization
	//void CBSEImgFileMgr::Init(int w,int h)
	//{
	//	// initialization
	//	m_poBSE = CBSEImgPtr(new CBSEImg(CRect(0,0,w,h)));
	//}

	// duplication 
	//void CBSEImgFileMgr::Duplicate(const CBSEImgFileMgr& a_oSource)
	//{
	//	// initialization
	//	//Init();

	//	// copy data over
	//	m_poBSE = CBSEImgPtr(new CBSEImg(a_oSource.m_poBSE.get()));
	//}


}