// JPEG2000.cpp : Defines the class behaviors for the application.
//

#include "stdafx.h"
#include "JPEG2000.h"

#include "MainFrm.h"
#include "ChildFrm.h"
#include "JPEG2000Doc.h"
#include "JPEG2000View.h"

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

// some global vars
FILESAVEFORMATBPP gJ2K[] =
{
   {sizeof(FILESAVEFORMATBPP),24, DLG_FF_SAVE_SUB_J2K_STREAM|DLG_FF_SAVE_SUB_J2K_JP2},
   {sizeof(FILESAVEFORMATBPP), 8, DLG_FF_SAVE_SUB_J2K_STREAM|DLG_FF_SAVE_SUB_J2K_JP2},
   {sizeof(FILESAVEFORMATBPP),12, DLG_FF_SAVE_SUB_J2K_STREAM|DLG_FF_SAVE_SUB_J2K_JP2},
   {sizeof(FILESAVEFORMATBPP),16, DLG_FF_SAVE_SUB_J2K_STREAM|DLG_FF_SAVE_SUB_J2K_JP2},
};

FILESAVEFORMATBPP gCMW[] = 
{
   {sizeof(FILESAVEFORMATBPP),24, 0},
   {sizeof(FILESAVEFORMATBPP), 8, 0},
   {sizeof(FILESAVEFORMATBPP),12, 0},
   {sizeof(FILESAVEFORMATBPP),16, 0},
};

FILESAVEFORMATBPP gJPEG[] = 
{
   {sizeof(FILESAVEFORMATBPP),24, DLG_FF_SAVE_SUB_JPEG24_YUV_444 | DLG_FF_SAVE_SUB_JPEG24_YUV_422 |
                                  DLG_FF_SAVE_SUB_JPEG24_YUV_411 | DLG_FF_SAVE_SUB_JPEG24_PROG_444|
                                  DLG_FF_SAVE_SUB_JPEG24_PROG_422| DLG_FF_SAVE_SUB_JPEG24_PROG_411|
                                  DLG_FF_SAVE_SUB_JPEG24_LOSSLESS},
   {sizeof(FILESAVEFORMATBPP), 8, DLG_FF_SAVE_SUB_JPEG8_YUV_400  | DLG_FF_SAVE_SUB_JPEG8_PROG_400 |
                                  DLG_FF_SAVE_SUB_JPEG8_LOSSLESS},
};

FILESAVEFORMATBPP gLEAD[] = 
{
   {sizeof(FILESAVEFORMATBPP),24, DLG_FF_SAVE_SUB_CMP_NONPROGRESSIVE | DLG_FF_SAVE_SUB_CMP_PROGRESSIVE},
   {sizeof(FILESAVEFORMATBPP), 8, 0},
};

FILESAVEFORMAT gaFormats[] =
{
   /*JPEG 2000*/
   {sizeof(FILESAVEFORMAT) ,DLG_FF_SAVE_J2K,  sizeof(gJ2K)/sizeof(FILESAVEFORMATBPP),  gJ2K},
   /*LEAD Wavelet*/
   {sizeof(FILESAVEFORMAT) ,DLG_FF_SAVE_CMW,  sizeof(gCMW)/sizeof(FILESAVEFORMATBPP),  gCMW},
   /*JPEG*/
   {sizeof(FILESAVEFORMAT) ,DLG_FF_SAVE_JPEG, sizeof(gJPEG)/sizeof(FILESAVEFORMATBPP), gJPEG},
   /*LEAD CMP*/
   {sizeof(FILESAVEFORMAT) ,DLG_FF_SAVE_LEAD, sizeof(gLEAD)/sizeof(FILESAVEFORMATBPP), gLEAD},
};
#define SAVEFORMATCOUNT sizeof(gaFormats)/sizeof(FILESAVEFORMAT)

/////////////////////////////////////////////////////////////////////////////
// CJPEG2000App

BEGIN_MESSAGE_MAP(CJPEG2000App, CWinApp)
	//{{AFX_MSG_MAP(CJPEG2000App)
	ON_COMMAND(ID_APP_ABOUT, OnAppAbout)
	ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
	//}}AFX_MSG_MAP
	// Standard file based document commands
	ON_COMMAND(ID_FILE_NEW, CWinApp::OnFileNew)
	ON_COMMAND(ID_FILE_OPEN, CWinApp::OnFileOpen)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CJPEG2000App construction

CJPEG2000App::CJPEG2000App()
{
   m_nPaintScaling = Paint_Normal;
   m_bPaintWhileLoad = FALSE;
   m_bEnableROI = FALSE;
   
   lstrcpy(m_szFileName, TEXT(""));
   lstrcpy(m_szFileName2, TEXT(""));
   lstrcpy(m_szImageDir, TEXT("c:\\\0"));
   
   m_OpenFileName.lStructSize       = sizeof(OPENFILENAME);
   m_OpenFileName.lpstrInitialDir   = NULL;
   m_OpenFileName.lpstrTitle        = TEXT("Save a File");

   m_OpenFileName.lpstrFile         = m_szFileName;
   m_OpenFileName.nMaxFile          = sizeof(m_szFileName)/sizeof(L_TCHAR);
   m_OpenFileName.lpstrFileTitle    = m_FSParm.szFileName;
   m_OpenFileName.nMaxFileTitle     = sizeof(m_FSParm.szFileName)/sizeof(L_TCHAR);

   m_OpenFileName.nFileOffset       = 0;
   m_OpenFileName.Flags             = 0;

   memset(&m_FSParm, 0, sizeof(SAVEDLGPARAMS));
   m_FSParm.uStructSize = sizeof(SAVEDLGPARAMS);
   m_FSParm.pBitmap = NULL;
   m_FSParm.nFormat = FILE_JFIF;
   m_FSParm.uSaveMulti = MULTIPAGE_OPERATION_OVERWRITE;
   m_FSParm.nPasses = -1;
   m_FSParm.bSaveWithStamp = FALSE;
   m_FSParm.nStampBits = 24;
   m_FSParm.nStampWidth = 120;
   m_FSParm.nStampHeight = 120;
   m_FSParm.nQFactor = 2;
   m_FSParm.uDlgFlags = DLG_SAVE_SHOW_FILEOPTIONS_PROGRESSIVE | DLG_SAVE_SHOW_FILEOPTIONS_QFACTOR | DLG_SAVE_SHOW_FILEOPTIONS_J2KOPTIONS | DLG_SAVE_SHOW_FILEOPTIONS_STAMP;
   m_FSParm.pFileFormats = gaFormats;
   m_FSParm.uFileFormatsCount = SAVEFORMATCOUNT;
   
   L_GetDefaultLoadFileOption(&m_lfo, sizeof(LOADFILEOPTION));
}

/////////////////////////////////////////////////////////////////////////////
// The one and only CJPEG2000App object

CJPEG2000App theApp;

/////////////////////////////////////////////////////////////////////////////
// CJPEG2000App initialization

BOOL CJPEG2000App::InitInstance()
{

#ifdef _AFXDLL
	Enable3dControls();			// Call this when using MFC in a shared DLL
#else
	Enable3dControlsStatic();	// Call this when linking to MFC statically
#endif

	// Change the registry key under which our settings are stored.
	// You should modify this string to be something appropriate
	// such as the name of your company or organization.
	SetRegistryKey(_T("LEAD Technologies, Inc."));
	LoadStdProfileSettings();  // Load standard INI file options (including MRU)

   CString cs;
   cs = GetProfileString(TEXT("Settings"), TEXT("ImageDir"), TEXT("c:\\"));
   lstrcpy(m_szImageDir, (LPCTSTR)cs);
   m_bPaintWhileLoad    = GetProfileInt(TEXT("Settings"), TEXT("PaintWhileLoad"), FALSE);
   //m_bEnableROI         = GetProfileInt("Settings", "RegionOfInterest", FALSE);

	// Register the application's document templates.  Document templates
	// serve as the connection between documents, frame windows and views.
	CMultiDocTemplate* pDocTemplate;
	pDocTemplate = new CMultiDocTemplate(
		IDR_JPEG20TYPE,
		RUNTIME_CLASS(CJPEG2000Doc),
		RUNTIME_CLASS(CChildFrame), // custom MDI child frame
		RUNTIME_CLASS(CJPEG2000View));
	AddDocTemplate(pDocTemplate);

	// create main MDI Frame window
	CMainFrame* pMainFrame = new CMainFrame;
	if (!pMainFrame->LoadFrame(IDR_MAINFRAME))
		return FALSE;
	m_pMainWnd = pMainFrame;

	// Parse command line for standard shell commands, DDE, file open
	CCommandLineInfo cmdInfo;
	ParseCommandLine(cmdInfo);

   // The main window has been initialized, so show and update it.
	pMainFrame->ShowWindow(m_nCmdShow);
	pMainFrame->UpdateWindow();

   UNLOCKSUPPORT();
   L_DlgInit(DLG_INIT_COLOR);
   if(L_IsSupportLocked(L_SUPPORT_J2K))
   {
      AfxMessageBox(TEXT("JPEG 2000 Support is locked!"));
      return FALSE;
   }
   
	return TRUE;
}


int CJPEG2000App::ExitInstance() 
{
	// TODO: Add your specialized code here and/or call the base class
   L_DlgFree();
	
	return CWinApp::ExitInstance();
}

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
		// No message handlers
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

// App command to run the dialog
void CJPEG2000App::OnAppAbout()
{
	CAboutDlg aboutDlg;
	aboutDlg.DoModal();
}

/////////////////////////////////////////////////////////////////////////////
// CJPEG2000App commands

L_VOID CJPEG2000App::CleanOpenDlgParam(LPOPENDLGPARAMS pFOParam)
{
   int i=0;

   if(pFOParam!=NULL && pFOParam->pFileData!= NULL )
   {
      for(i=0; i < pFOParam->nNumOfFiles; ++i )
      {
         if(pFOParam->pFileData[i].pBitmap != NULL)
         {
            L_FreeBitmap(pFOParam->pFileData[i].pBitmap);
            GlobalFreePtr(pFOParam->pFileData[i].pBitmap);
         }

         if(pFOParam->pFileData[i].pThumbnail != NULL)
         {
            L_FreeBitmap(pFOParam->pFileData[i].pThumbnail);
            GlobalFreePtr(pFOParam->pFileData[i].pThumbnail);
         }

         if(pFOParam->pFileData[i].pFileInfo != NULL)
         {
            GlobalFreePtr(pFOParam->pFileData[i].pFileInfo);
         }
      }

      GlobalFreePtr(pFOParam->pFileData);

      pFOParam->pFileData    =NULL;
      pFOParam->nNumOfFiles  = 0;
   }
}

void CJPEG2000App::OnFileOpen() 
{
   L_INT nRet;
   HWND hWndChild = NULL;
   OPENFILENAME OpenFileName;
   L_TCHAR szFileTitle[_MAX_PATH]=TEXT("");
   OPENDLGPARAMS FOParm;
   L_TCHAR szFilter[] = TEXT("All Files (*.*)\0*.*\0")
                       TEXT("JPEG 2000 (*.jp2;*.j2k)\0*.jp2;*.j2k\0")
                       TEXT("JPEG (*.jpg)\0*.jpg;*.jpeg\0")
                       TEXT("LEAD Wavelet (*.cmw)\0*.cmw\0")
                       TEXT("LEAD CMP (*.cmp)\0*.cmp\0")
                       TEXT("\0");

   memset(&FOParm, 0, sizeof(OPENDLGPARAMS));
   FOParm.uStructSize = sizeof(OPENDLGPARAMS);
   FOParm.uDlgFlags = DLG_OPEN_SHOW_MULTIPAGE  |
                      DLG_OPEN_SHOW_FILEINFO   |
                      DLG_OPEN_SHOW_PREVIEW    |
                      DLG_OPEN_VIEWTOTALPAGES  |
                      DLG_OPEN_SHOW_PROGRESSIVE;
   FOParm.bPreviewEnabled = TRUE;

   memset(&m_lfo, 0, sizeof(LOADFILEOPTION));
   L_GetDefaultLoadFileOption(&m_lfo, sizeof(LOADFILEOPTION));

   memset(&OpenFileName,0,sizeof(OpenFileName));
   OpenFileName.lStructSize       = sizeof(OPENFILENAME);
   OpenFileName.hwndOwner         = NULL;
   OpenFileName.lpstrFilter       = szFilter;
   OpenFileName.lpstrCustomFilter = NULL;
   OpenFileName.nMaxCustFilter    = 0;
   OpenFileName.nFilterIndex      = 0;

   OpenFileName.lpstrFileTitle    = szFileTitle;
   OpenFileName.nMaxFileTitle     = sizeof(szFileTitle)/sizeof (L_TCHAR);

   OpenFileName.lpstrInitialDir   = NULL;
   OpenFileName.lpstrTitle        = TEXT("Open a File");
   OpenFileName.nFileOffset       = 0;
   OpenFileName.nFileExtension    = 0;
   OpenFileName.lpstrDefExt       = NULL;
   OpenFileName.Flags             = 0;

#if(WINVER >= 0x0400)
   OpenFileName.Flags |= OFN_ENABLESIZING;
#endif /* WINVER >= 0x0400 */

   nRet = L_DlgOpen( m_pMainWnd->m_hWnd,
                     &OpenFileName,
                     &FOParm);

   if (nRet == SUCCESS_DLG_OK)
   {
      L_GetDefaultLoadFileOption(&m_lfo, sizeof(LOADFILEOPTION));
      m_lfo.PageNumber = FOParm.pFileData[0].nPageNumber;
      if(FOParm.pFileData[0].nPasses!=0)
         m_lfo.Passes = FOParm.pFileData[0].nPasses; 

      OpenDocumentFile((LPCTSTR)FOParm.pFileData[0].szFileName);

      CleanOpenDlgParam(&FOParm);
   }
   else
   {
      if(nRet != SUCCESS_DLG_CANCEL)
         theApp.DisplayLEADError(nRet); 
   }
   return;
}

CDocument* CJPEG2000App::OpenDocumentFile(LPCTSTR lpszFileName) 
{
   FILEINFO fi;
   int OldPage;
   
   OldPage = m_lfo.PageNumber;
   m_lfo.PageNumber = 32000;
   fi.uStructSize = sizeof(FILEINFO);
   memset(&fi, 0, sizeof(FILEINFO));
   L_FileInfo((LPTSTR) lpszFileName, &fi, sizeof(FILEINFO), 0, &m_lfo);
   m_lfo.PageNumber = OldPage;
   
   CDocument * pDoc = CWinApp::OpenDocumentFile(lpszFileName);
   return pDoc;   
}

int CJPEG2000App::DisplayLEADError(int nCode)
{
   CString cs;
   if(nCode != SUCCESS)
   {
      cs.Format(TEXT("A LEADTOOLS error has occured.\nCODE = %d"), nCode);
      m_pMainWnd->MessageBox(cs, TEXT("Error"), MB_OK|MB_ICONERROR);
   }
   return(nCode);
}

DWORD CJPEG2000App::SetDisplayMode(void)
{
   DWORD dwFlags;

   dwFlags = L_GetDisplayMode();

   L_SetDisplayMode (DISPLAYMODE_ORDEREDDITHER, DISPLAYMODE_ORDEREDDITHER);
   L_SetDisplayMode (DISPLAYMODE_FIXEDPALETTE, DISPLAYMODE_FIXEDPALETTE);
   L_SetDisplayMode (DISPLAYMODE_SCALETOGRAY, DISPLAYMODE_SCALETOGRAY);
   switch(theApp.m_nPaintScaling)
   {
   case theApp.Paint_Normal:
      L_SetDisplayMode (DISPLAYMODE_RESAMPLE|DISPLAYMODE_BICUBIC, 0);
      break;
   case theApp.Paint_Resample:
      L_SetDisplayMode (DISPLAYMODE_RESAMPLE|DISPLAYMODE_BICUBIC, DISPLAYMODE_RESAMPLE);
      break;
   case theApp.Paint_Bicubic:
      L_SetDisplayMode (DISPLAYMODE_RESAMPLE|DISPLAYMODE_BICUBIC, DISPLAYMODE_BICUBIC);
      break;
   }
   return dwFlags;
}