#include "StdAfx.h"
#include "SharedImageData.h"

CSharedImageData::CSharedImageData(void)
{
	m_pSMImageData		= NULL;
	m_nImageCount		= 0;
	m_nImageInfoSize	= sizeof(SSharedImageInfo);
	//m_nMaxImageSize		= 2456 * 2058 * 4;
	m_nMaxImageSize		= 2464 * 2056 * 4;
	m_hAsignedMemory	= NULL;
	m_pSharedMemory		= NULL;
	m_pImageInfo		= NULL;
	m_pImageData		= NULL;
}

CSharedImageData::~CSharedImageData(void)
{
	DeleteMemory();
}

void CSharedImageData::DeleteMemory()
{
	if(m_pSharedMemory)
	{
		UnmapViewOfFile(m_pSharedMemory); 
		m_pSharedMemory = NULL;
	}

	if(m_hAsignedMemory)
	{
		CloseHandle(m_hAsignedMemory);
		m_hAsignedMemory = NULL;
	}

	if (m_pSMImageData)
	{
		for (int i=0; i<m_nImageCount; i++)
		{
			delete m_pSMImageData[i];
		}
		delete [] m_pSMImageData;
		m_pSMImageData = NULL;
	}

	if (m_pImageInfo)
	{
		delete [] m_pImageInfo;
		m_pImageInfo = NULL;
	}

	if (m_pImageData)
	{
		delete [] m_pImageData;
		m_pImageData = NULL;
	}

	m_nImageCount	= 0;

}

BOOL CSharedImageData::CreateMemory(int nImageCount)
{
	DeleteMemory(); 

	m_nImageCount = nImageCount;

	// alloc semapore
	m_pSMImageData = new CSemaphore*[m_nImageCount];
	m_pImageInfo = new SSharedImageInfo*[m_nImageCount];
	m_pImageData = new BYTE*[m_nImageCount];

	for (int i=0; i<m_nImageCount; i++)
	{
		m_pSMImageData[i] = new CSemaphore(1, 1);
		m_pImageInfo[i] = NULL;
		m_pImageData[i] = NULL;
	}

	long nTotalImageInfoSize = m_nImageInfoSize * m_nImageCount;
	long nTotalImageSize = m_nMaxImageSize * m_nImageCount;
	long nTotalAsignedBuffSize = nTotalImageInfoSize + nTotalImageSize;

	m_hAsignedMemory = (HANDLE) CreateFileMapping(
		INVALID_HANDLE_VALUE, 
		NULL, 
		PAGE_READWRITE, 
		0,
		(DWORD)nTotalAsignedBuffSize, 
		_T("SHARED_IMAGE_DATA"));

	if(!m_hAsignedMemory) return FALSE;

	m_pSharedMemory = (BYTE *)MapViewOfFile(m_hAsignedMemory, FILE_MAP_READ | FILE_MAP_WRITE, 0, 0, 0);

	if(!m_pSharedMemory) return FALSE;

	for (int i=0; i<m_nImageCount; i++)
	{
		m_pImageInfo[i]	= (SSharedImageInfo*)( m_pSharedMemory + (m_nImageInfoSize*i) );

		m_pImageData[i]	= (BYTE *)( m_pSharedMemory + nTotalImageInfoSize + (m_nMaxImageSize*i) );
	}

	return TRUE;
}



int CSharedImageData::GetUpdated(int nIndex)
{
	if (nIndex<0 || nIndex>=m_nImageCount) return 0;

	if (m_pImageInfo[nIndex]) 
	{
		return m_pImageInfo[nIndex]->nUpdated;
	}

	return 0;
}

int CSharedImageData::GetSelected(int nIndex)
{
	if (nIndex<0 || nIndex>=m_nImageCount) return 0;

	if (m_pImageInfo[nIndex]) 
	{
		return m_pImageInfo[nIndex]->nSelected;
	}

	return 0;
}


BOOL CSharedImageData::LockImageData(int nIndex)
{
	if (nIndex<0 || nIndex>=m_nImageCount) return FALSE;

	if (m_pSMImageData[nIndex]) 
	{
		m_pSMImageData[nIndex]->Lock();
	}
	else
	{
		return FALSE;
	}

	return TRUE;
}

BOOL CSharedImageData::UnlockImageData(int nIndex)
{
	if (nIndex<0 || nIndex>=m_nImageCount) return FALSE;

	if (m_pImageInfo[nIndex]) 
	{
		m_pImageInfo[nIndex]->nUpdated--;
	}
	else
	{
		return FALSE;
	}

	if (m_pSMImageData[nIndex])
	{
		m_pSMImageData[nIndex]->Unlock();
	}
	else
	{
		return FALSE;
	}

	return TRUE;
}

const BYTE* CSharedImageData::GetImageData(int nIndex)
{
	if (nIndex<0 || nIndex>=m_nImageCount) return NULL;

	return m_pImageData[nIndex];
}

const SSharedImageInfo* CSharedImageData::GetImageInfo(int nIndex)
{
	if (nIndex<0 || nIndex>=m_nImageCount) return NULL;

	return m_pImageInfo[nIndex];
}

BOOL CSharedImageData::SetImageData(int nIndex, SSharedImageInfo *pImageInfo, BYTE* pImageData)
{
	if (pImageInfo==NULL || pImageData==NULL) return FALSE;

	if(m_hAsignedMemory==NULL) return FALSE;

	if (nIndex<0 || nIndex>=m_nImageCount) return FALSE;

	int nSrcSize = pImageInfo->nImageWidthStep * pImageInfo->nImageHeight;

	if (nSrcSize>m_nMaxImageSize) return FALSE;

	if (m_pSMImageData[nIndex]==NULL) return FALSE;

	m_pSMImageData[nIndex]->Lock();	// 함수를 Atomic 하게 동작시키자

	// 이미지 정보 기록
	pImageInfo->nUpdated = 1;
	memcpy(m_pImageInfo[nIndex], pImageInfo, m_nImageInfoSize);

	// 이미지 기록
	memcpy(m_pImageData[nIndex], pImageData, nSrcSize);

	m_pSMImageData[nIndex]->Unlock();// 함수 동기화 종료

	return TRUE;
}