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

#include "stdafx.h"
#include <afxctl.h>

#include "ScrDemo.h"

#include "MainFrm.h"
#include "ChildFrm.h"
#include "ScrDoc.h"
#include "CScrView.h"

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

/////////////////////////////////////////////////////////////////////////////
// CScrDemoApp

BEGIN_MESSAGE_MAP(CScrDemoApp, CWinApp)
   //{{AFX_MSG_MAP(CScrDemoApp)
   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()

/////////////////////////////////////////////////////////////////////////////
// CScrDemoApp construction

CScrDemoApp::CScrDemoApp()
{
   // TODO: add construction code here,
   // Place all significant initialization in InitInstance
   m_pRasterScr = NULL;
}

/////////////////////////////////////////////////////////////////////////////
// The one and only CScrDemoApp object

CScrDemoApp theApp;

/////////////////////////////////////////////////////////////////////////////
// CScrDemoApp initialization

BOOL CScrDemoApp::InitInstance()
{
   AfxEnableControlContainer();

   // Standard initialization
   // If you are not using these features and wish to reduce the size
   //  of your final executable, you should remove from the following
   //  the specific initialization routines you do not need.

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

   // 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("Local AppWizard-Generated Applications"));

   LoadStdProfileSettings();  // Load standard INI file options (including MRU)

   // Register the application's document templates.  Document templates
   //  serve as the connection between documents, frame windows and views.

   CMultiDocTemplate* pDocTemplate;
   pDocTemplate = new CMultiDocTemplate(
      IDR_SCRDEMTYPE,
      RUNTIME_CLASS(CScrDoc),
      RUNTIME_CLASS(CChildFrame), // custom MDI child frame
      RUNTIME_CLASS(CScrView));
   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);
//
//   // Dispatch commands specified on the command line
//   if (!ProcessShellCommand(cmdInfo))
//      return FALSE;

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

   //Instantiate the thumbanil browswer sink class and hold a pointer to it.
   m_pRasterScrSink = new CRasterScrSink;
   //Create the RasterScr object
   HRESULT hr = CoCreateInstance(
      CLSID_LEADRasterScr,
      NULL,
      CLSCTX_ALL,
      IID_ILEADRasterScr,
      (void**)&m_pRasterScr
      );
   if(FAILED(hr) || !m_pRasterScr)
   {
      AfxMessageBox(TEXT("Error creating Screen Capture Object"));
return FALSE;  
   }
   
   
   m_pRasterScr->PutEnableMethodErrors(FALSE);
   m_pRasterScr->PutCaptureHotKey(VK_F11);

   //Establish a connection between source and sink.
   LPUNKNOWN pUnkSink = m_pRasterScrSink->GetIDispatch(FALSE);
   AfxConnectionAdvise(m_pRasterScr, DIID__LTSCREvents, pUnkSink, FALSE, &m_dwScrCookie);
   
   m_pRasterKrnDlg = NULL;
   hr = ::CoCreateInstance(CLSID_LEADRasterDlgKrn, NULL, CLSCTX_ALL,
            IID_ILEADRasterDlgKrn, (void**)&m_pRasterKrnDlg);
   if (FAILED (hr) || (m_pRasterKrnDlg == NULL))
   {
      m_pRasterKrnDlg ->Release ();
      m_pRasterKrnDlg = NULL;
   }
   m_pRasterKrnDlg->InitDlg(INIT_WITH_COLOR);
   
   m_pRasterFileDlg = NULL;
   hr = ::CoCreateInstance(CLSID_LEADRasterDlgFile, NULL, CLSCTX_ALL,
            IID_ILEADRasterDlgFile, (void**)&m_pRasterFileDlg);
   if (FAILED (hr) || (m_pRasterFileDlg == NULL))
   {
      m_pRasterFileDlg->Release ();
      m_pRasterFileDlg = NULL;
   }  

   return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// 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 CScrDemoApp::OnAppAbout()
{
   CAboutDlg aboutDlg;
   aboutDlg.DoModal();
}

/////////////////////////////////////////////////////////////////////////////
// CScrDemoApp commands

int CScrDemoApp::ExitInstance() 
{
   if (m_pRasterScr != NULL)
   {
      //Terminate a connection between Scr source and sink.
      LPUNKNOWN pUnkSink = m_pRasterScrSink->GetIDispatch(FALSE);
      AfxConnectionUnadvise(m_pRasterScr, DIID__LTSCREvents, pUnkSink, FALSE, m_dwScrCookie);
      delete m_pRasterScrSink;
      m_pRasterScr->Release();
   }

  
   if (m_pRasterFileDlg)
      m_pRasterFileDlg->Release ();
    
   m_pRasterKrnDlg->FreeDlg();
   
   if(m_pRasterKrnDlg)
      m_pRasterKrnDlg->Release ();
   
   m_pRasterFileDlg = NULL;

   CoUninitialize();

   return CWinApp::ExitInstance();
}

void CScrDemoApp::OnFileOpen() 
{
   CString m_FileName;
   CMainFrame *m_pMainFrame = (CMainFrame*)AfxGetMainWnd ();

   TCHAR * pszFilter = TEXT("Executable files(*.EXE)|*.exe|Dynamic Link Libraries(*.DLL)|*.dll|All files(*.*)|*.*|");
   CFileDialog COpen(TRUE, NULL, NULL, OFN_HIDEREADONLY|OFN_FILEMUSTEXIST, pszFilter);

   if (COpen.DoModal()==IDOK)
   {
      CMainFrame *pMain = (CMainFrame *) AfxGetMainWnd();
      m_FileName = COpen.GetPathName();
      AddToRecentFileList(m_FileName);
      m_pRasterScr->CaptureEXEDlg ((long)m_pMainFrame->m_hWnd,0, (LPCTSTR)m_FileName, SCR_CAP_ALL, SCR_CAP_EXEDLG_TREEVIEW);
   }
}

void CScrDemoApp::CreateNewDoc()
{
   OnFileNew();
}