#include "stdafx.h"
#include "Image.h"

#include "hDIB.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif


/////////////////////////////////////////////////////////////////////////////
// CCImage load/save bmp

// Function	: CImage::LoadBMP
// Descript	: BMP 이미지를 읽어들인다.
// Return	: BOOL
// Argument	: LPCTSTR lpszFileName
BOOL CImage::LoadBMP(LPCTSTR lpszFileName)
{
	CFile file;
	CFileException fe;
	BITMAPFILEHEADER bfHeader;

	// 읽기 모드로 파일 열기
	if (!file.Open(lpszFileName, CFile::modeRead|CFile::shareDenyWrite, &fe))
		return FALSE;

	// 파일 헤더 읽기
	if (file.Read((LPSTR)&bfHeader, sizeof(bfHeader)) != sizeof(bfHeader))
		return FALSE;

	// BMP 파일임을 나타내는 "BM" 마커가 있는지 확인
	if (bfHeader.bfType != DIB_HEADER_MARKER)
		return FALSE;

	// 파일의 길이를 구함
	DWORD dwBitsSize = file.GetLength() - sizeof(BITMAPFILEHEADER);
	// 메모리 할당
	HGLOBAL hImage = ::GlobalAlloc(GHND, dwBitsSize);
	if (hImage == NULL)
		return FALSE;
	// 메모리 고정
	LPSTR pDIB = (LPSTR) ::GlobalLock(hImage);

	// 파일 읽기
	if (file.ReadHuge(pDIB, dwBitsSize) != dwBitsSize) 
	{
		::GlobalUnlock(hImage);
		::GlobalFree(hImage);
		return FALSE;
	}

	// 메모리 풀어줌
	::GlobalUnlock(hImage);

	return ConvertToImage(hImage);
}

// Function	: CImage::ConvertToImage
// Descript	: BMP 이미지를 CImage로 변환한다.
// Return	: BOOL
// Argument	: HGLOBAL hDIB
BOOL CImage::ConvertToImage(HGLOBAL hDIB)
{
	if (hDIB == NULL)
		return FALSE;

	// 메모리 고정
	BYTE * srcAddress = (BYTE *) ::GlobalLock(hDIB);

	// 그릴 이미지 생성
	long dx = ::DibWidth(srcAddress);
	long dy = ::DibHeight(srcAddress);
	if (Create(dx, dy) == FALSE)
		return FALSE;

	// 그릴 이미지의 메모리 고정
	BYTE * address = (BYTE *) ::GlobalLock(m_hImage);
	long columns = GetRealWidth();

	RGBQUAD color;
	long indexY, index;
	for (long y = 0; y < dy; y++)
	{
		// Y 축상의 시작 위치 계산
		indexY = y * columns;
		for (long x = 0; x < dx; x++)
		{
			// 칠한 이미지의 위치 계산
			index = indexY + x * 3;

			// 색상 가져오기
			::DibGetPixelColor(srcAddress, x, y, &color);

			// 색상 쓰기
			address[index + 2] = color.rgbRed;
			address[index + 1] = color.rgbGreen;
			address[index    ] = color.rgbBlue;
		}
	}

	// 메모리 풀어줌
	::GlobalUnlock(m_hImage);
	::GlobalUnlock(hDIB);
	::GlobalFree(hDIB);

	return TRUE;
}

// Function	: CImage::ConvertToDIB
// Descript	: CImage 형식을 DIB로 변환한다. 24-bit DIB
// Return	: HGLOBAL
// Argument	: void
HGLOBAL CImage::ConvertToDIB()
{
	if (m_hImage == NULL)
		return NULL;

	// 새로운 DIB룰 생성한다. 생성될때 정보도 세팅 된다.
	HDIB hHandle = ::DibCreate(GetWidth(), GetHeight(), GetBitCount());
	if (hHandle == NULL)
		return NULL;

	// 메모리 고정
	void * des = ::GlobalLock(hHandle);
	void * desAddress = ::DibImage(des);

	void * srcAddress = ::GlobalLock(m_hImage);

	// 이미지를 그대로 복사하면 된다.
	memcpy(desAddress, srcAddress, GetHeight() * GetRealWidth());

	::GlobalUnlock(m_hImage);
	::GlobalUnlock(hHandle);

	return hHandle;
}

// Function	: CImage::SaveBMP
// Descript	: BMP 포맷으로 이미지를 저장한다.
// Return	: BOOL
// Argument	: LPCTSTR lpszFileName
BOOL CImage::SaveBMP(LPCTSTR lpszFileName)
{
	CFile file;
	CFileException fe;
	BITMAPFILEHEADER bmFileHeader;

	// 메모리 핸들이 유효한지 확인
	if (m_hImage == NULL)
		return FALSE;

	// 쓰기 모드로 파일 열기
	if (!file.Open(lpszFileName, CFile::modeCreate | CFile::modeWrite | CFile::shareDenyWrite, &fe))
		return FALSE;

	// 메모리 고정
	LPSTR address = (LPSTR) ::GlobalLock(m_hImage);
	if (address == NULL)
		return FALSE;

	// 비트맵 파일 헤더 정보를 설정
	bmFileHeader.bfType = DIB_HEADER_MARKER;		// "BM"
	DWORD dwDIBSize = m_biHeader.biSize;
	dwDIBSize += m_biHeader.biClrUsed * sizeof(RGBQUAD);
	dwDIBSize += m_biHeader.biSizeImage;
	bmFileHeader.bfSize = dwDIBSize + sizeof(BITMAPFILEHEADER);
	bmFileHeader.bfReserved1 = 0;
	bmFileHeader.bfReserved2 = 0;
	bmFileHeader.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER) + m_biHeader.biSize;
	TRY
	{
		// 비트맵 파일 헤더를 파일에 쓰기
		file.Write((LPSTR)&bmFileHeader, sizeof(BITMAPFILEHEADER));
		// 비트맵 정보 헤더를 파일에 쓰기
		file.Write((LPSTR)&m_biHeader, sizeof(BITMAPINFOHEADER));
		// 나머지 데이터를 파일에 쓰기
		file.WriteHuge(address, dwDIBSize);
	}
	CATCH (CFileException, e)
	{
		::GlobalUnlock( m_hImage);
		THROW_LAST();
	}
	END_CATCH

	// 메모리 풀어줌
	::GlobalUnlock( m_hImage);

	return TRUE;
}