// Compression.cpp: implementation of the CCompression class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "Phone.h"
#include "Compression.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

#pragma comment(lib,"msacm32")

BOOL CALLBACK FindDriverEnum(HACMDRIVERID hadid, DWORD dwInstance,
							 DWORD fdwSupport);
BOOL CALLBACK FindFormatEnum(HACMDRIVERID hadid, LPACMFORMATDETAILS pafd, 
							 DWORD dwInstance, DWORD fdwSupport);
BOOL CALLBACK FindFormatEnumOld(HACMDRIVERID hadid, LPACMFORMATDETAILS pafd, 
							 DWORD dwInstance, DWORD fdwSupport);

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CCompression::CCompression()
{
	pDst1Data=NULL;
	pDst2Data=NULL;
	m_pSrcData=NULL;
}

CCompression::~CCompression()
{
	if(pDst1Data!=NULL)
		free(pDst1Data);
	if(pDst2Data!=NULL)
		free(pDst2Data);
	if(m_pSrcData!=NULL)
		free(m_pSrcData);
	if(pwfDrv!=NULL)
		free(pwfDrv);
	if(pwfPCM!=NULL)
		free(pwfPCM);
}

void CCompression::SetWaveFormat(int channels, long samplerate, int bitspersample)
{
	memset(&m_wfSrc, 0, sizeof(m_wfSrc));
	m_wfSrc.cbSize = 0;
	m_wfSrc.wFormatTag = WAVE_FORMAT_PCM; // pcm
	m_wfSrc.nChannels = channels; // mono
	m_wfSrc.nSamplesPerSec = samplerate; // 11.025 kHz
	m_wfSrc.wBitsPerSample = bitspersample; // 8 bit
	m_wfSrc.nBlockAlign = m_wfSrc.nChannels *m_wfSrc.wBitsPerSample / 8;
	m_wfSrc.nAvgBytesPerSec = m_wfSrc.nSamplesPerSec * m_wfSrc.nBlockAlign;
}

void CCompression::SetSrcSamples(long samples,unsigned char* pSrcData)
{
	int n=(int)(m_wfSrc.wBitsPerSample/8);

	m_dwSrcSamples=samples;
	m_dwSrcBytes=(DWORD)(samples*n);
	m_pSrcData=(BYTE*)malloc(m_dwSrcBytes);
	memcpy(m_pSrcData,pSrcData,m_dwSrcBytes);
}

void CCompression::SelectMethod(int i)
{
	if((i<0) || (i>5)){
		AfxMessageBox("No support for such compression method!");
		return;
	}
	switch(i){
	case 0:{
		wFormatTag = 0x32; // MSN
		break;
		   }
	case 1:{
		wFormatTag = WAVE_FORMAT_DSPGROUP_TRUESPEECH;
		break;}
	case 2:{
		wFormatTag = WAVE_FORMAT_ADPCM;
		break;}
	case 3:{
		wFormatTag = WAVE_FORMAT_GSM610;
		break;}
	case 4:{
		wFormatTag = WAVE_FORMAT_ALAW;
		break;}
	case 5:{
		wFormatTag = WAVE_FORMAT_MULAW;
		break;}
	}
}

HACMDRIVERID CCompression::FindDriver(WORD wFormatTag)
{
	FIND_DRIVER_INFO fdi;
	fdi.hadid = NULL;
	fdi.wFormatTag = wFormatTag;
	MMRESULT mmr = acmDriverEnum(FindDriverEnum, (DWORD)(VOID*)&fdi,0); 
	if (mmr) 
		return NULL;
	return fdi.hadid;
}

BOOL CALLBACK FindDriverEnum(HACMDRIVERID hadid, DWORD dwInstance,
							 DWORD fdwSupport)
{
	FIND_DRIVER_INFO* pdi = (FIND_DRIVER_INFO*) dwInstance;

	// open the driver
	HACMDRIVER had = NULL;
	MMRESULT mmr = acmDriverOpen(&had, hadid, 0);
	if (mmr) {
		// some error
		return FALSE; // stop enumerating
	}
	// enumerate the formats it supports
	DWORD dwSize = 0;
	mmr = acmMetrics((HACMOBJ)had, ACM_METRIC_MAX_SIZE_FORMAT, &dwSize);
	if (dwSize < sizeof(WAVEFORMATEX)) dwSize = sizeof(WAVEFORMATEX); // for MS-PCM
	WAVEFORMATEX* pwf = (WAVEFORMATEX*) malloc(dwSize);
	memset(pwf, 0, dwSize);
	pwf->cbSize = LOWORD(dwSize) - sizeof(WAVEFORMATEX);
	pwf->wFormatTag = pdi->wFormatTag;
	ACMFORMATDETAILS fd;
	memset(&fd, 0, sizeof(fd));
	fd.cbStruct = sizeof(fd);
	fd.pwfx = pwf;
	fd.cbwfx = dwSize;
	fd.dwFormatTag = pdi->wFormatTag;
	mmr = acmFormatEnum(had, &fd, FindFormatEnumOld, (DWORD)(VOID*)pdi, 0);  
	free(pwf);
	acmDriverClose(had, 0);
	if (pdi->hadid || mmr) {
		// found it or some error
		return FALSE; // stop enumerating
	}
	return TRUE; // continue enumeration
}

BOOL CALLBACK FindFormatEnum(HACMDRIVERID hadid, LPACMFORMATDETAILS pafd, DWORD dwInstance, DWORD fdwSupport)
{
	FIND_DRIVER_INFO* pdi = (FIND_DRIVER_INFO*) dwInstance;
	if (pafd->dwFormatTag == (DWORD)pdi->wFormatTag) {
		// found it
		WAVEFORMATEX pwfPCM;
		pwfPCM=*(pafd->pwfx);
		pdi->hadid = hadid;
		TRACE("PCM format: %u bits, %lu samples per second\n", pwfPCM.wBitsPerSample, pwfPCM.nSamplesPerSec);
		if((pwfPCM.nSamplesPerSec==(unsigned long)pdi->lSampleRate) && (pwfPCM.wBitsPerSample==(unsigned short)pdi->nBits)){
			pdi->hadid = hadid;
			return FALSE; // stop enumerating
		}
	}

	return TRUE; // continue enumerating
}

BOOL CALLBACK FindFormatEnumOld(HACMDRIVERID hadid, LPACMFORMATDETAILS pafd, DWORD dwInstance, DWORD fdwSupport)
{
	FIND_DRIVER_INFO* pdi = (FIND_DRIVER_INFO*) dwInstance;
	if (pafd->dwFormatTag == (DWORD)pdi->wFormatTag) {
		// found it
		pdi->hadid = hadid;
		return FALSE; // stop enumerating
	}

	return TRUE; // continue enumerating
}

WAVEFORMATEX* CCompression::GetDriverFormat(HACMDRIVERID hadid, WORD wFormatTag,BOOL bPCM)
{
	// open the driver
	HACMDRIVER had = NULL;
	MMRESULT mmr = acmDriverOpen(&had, hadid, 0);
	if (mmr) {
		return NULL;
	}

	// allocate a structure for the info
	DWORD dwSize = 0;
	mmr =acmMetrics((HACMOBJ)had, ACM_METRIC_MAX_SIZE_FORMAT, &dwSize);
	if (dwSize < sizeof(WAVEFORMATEX)) dwSize = sizeof(WAVEFORMATEX); // for MS-PCM
	WAVEFORMATEX* pwf = (WAVEFORMATEX*) malloc(dwSize);
	memset(pwf, 0, dwSize);
	pwf->cbSize = LOWORD(dwSize) - sizeof(WAVEFORMATEX);
	pwf->wFormatTag = wFormatTag;

	ACMFORMATDETAILS fd;
	memset(&fd, 0, sizeof(fd));
	fd.cbStruct = sizeof(fd);
	fd.pwfx = pwf;
	fd.cbwfx = dwSize;
	fd.dwFormatTag = wFormatTag;

	// set up a struct to control the enumeration
	FIND_DRIVER_INFO fdi;
	fdi.hadid = NULL;
	fdi.wFormatTag = wFormatTag;
	fdi.nBits=(unsigned short)m_wfSrc.wBitsPerSample;
	fdi.lSampleRate=8000;//(unsigned long)m_wfSrc.nSamplesPerSec;
	
	if(!bPCM)
		mmr=acmFormatEnum(had, &fd, FindFormatEnumOld, (DWORD)(VOID*)&fdi,0);
	else
		mmr = acmFormatEnum(had, &fd, FindFormatEnum, (DWORD)(VOID*)&fdi,0);  
	
	acmDriverClose(had, 0);
	if ((fdi.hadid == NULL) || mmr) {
		free(pwf);
		return NULL;
	}
	
	return pwf;
}

BOOL CCompression::Initialize()
{
	hadid = FindDriver(wFormatTag);
	if (hadid == NULL) {
		AfxMessageBox("No driver found\n");
		return FALSE;
	}
	TRACE("Driver found (hadid: %4.4lXH)\n", hadid);
	// show some information about the driver
	ACMDRIVERDETAILS dd;
	dd.cbStruct = sizeof(dd);
	MMRESULT mmr = acmDriverDetails(hadid, &dd, 0);
	TRACE("   Short name: %s\n", dd.szShortName);
	TRACE("   Long name:  %s\n", dd.szLongName);
	TRACE("   Copyright:  %s\n", dd.szCopyright);
	TRACE("   Licensing:  %s\n", dd.szLicensing);
	TRACE("   Features:   %s\n", dd.szFeatures);

	// get the details of the format
	// Note: this is just the first of one or more possible formats for the given tag
	pwfDrv = GetDriverFormat(hadid, wFormatTag,FALSE);
	if (pwfDrv == NULL) {
		AfxMessageBox("Error getting format info\n");
		return FALSE;
	}
	TRACE("Driver format: %u bits, %lu samples per second\n", pwfDrv->wBitsPerSample, pwfDrv->nSamplesPerSec);

	// get a PCM format tag the driver supports
	// Note: we just pick the first supported PCM format which might not really
	// be the best choice.
	pwfPCM = GetDriverFormat(hadid, WAVE_FORMAT_PCM,TRUE);
	if (pwfPCM == NULL) {
		AfxMessageBox("Error getting PCM format info\n");
		return FALSE;
	}
	TRACE("PCM format: %u bits, %lu samples per second\n", pwfPCM->wBitsPerSample, pwfPCM->nSamplesPerSec);
	return TRUE;
}

long CCompression::MyConvert(BOOL bQuery,unsigned char *pDetData, long detLength)
{
	if(bQuery){
		MMRESULT mmr;