source: trunk/Utilities/StressEvr/StressEvr.cpp @ 174

////////////////////////////////////////////////////////////
// Copyright (C) Roman Ryltsov, 2008-2013
// Created by Roman Ryltsov roman@alax.info

#include "stdafx.h"
#include "resource.h"
#include "StressEvr_i.h"
#include "MainDialog.h"

////////////////////////////////////////////////////////////
// CStressEvrModule

class CStressEvrModule : 
        public CAtlExeModuleT<CStressEvrModule>,
        public CWtlExeModuleT<CStressEvrModule>
{
public:

//DECLARE_LIBID(LIBID_AlaxInfoStressEvr)

//DECLARE_REGISTRY_APPID_RESOURCEID(IDR_MODULE, "{0B0FCF8F-E12E-4E97-B683-06E3F93830E3}")

public:
// CStressEvrModule
        CStressEvrModule() throw()
        {
                AtlTraceSetDefaultSettings();
                _W(CExceptionFilter::Initialize());
                _Z4(atlTraceRefcount, 4, _T("this 0x%p\n"), this);
        }
        ~CStressEvrModule() throw()
        {
                _Z4(atlTraceRefcount, 4, _T("this 0x%p\n"), this);
                CExceptionFilter::Terminate();
        }
        HRESULT PreMessageLoop(INT nShowCommand) throw()
        {
                _ATLTRY
                {
                        TraceModuleVersion();
                        __E(AtlInitCommonControls(ICC_WIN95_CLASSES | ICC_DATE_CLASSES | ICC_COOL_CLASSES));
                        //__C(CWtlExeModule::Initialize());
                        __C(__super::PreMessageLoop(nShowCommand));
                }
                _ATLCATCH(Exception)
                {
                        _C(Exception);
                }
                return S_OK;
        }
        VOID RunMessageLoop() throw()
        {
                Lock();
                {
                        CMainDialog MainDialog;
                        MainDialog.DoModal();
                }
                if(GetLockCount() == 1)
                {
                        m_dwTimeOut = 0L;
                        m_dwPause = 0L;
                }
                Unlock();
        }
        HRESULT PostMessageLoop() throw()
        {
                _ATLTRY
                {
                        _V(__super::PostMessageLoop());
                        //CWtlExeModule::Terminate();
                }
                _ATLCATCH(Exception)
                {
                        _C(Exception);
                }
                return S_OK;
        }
};

CStressEvrModule _AtlModule;

////////////////////////////////////////////////////////////
// Main

extern "C" int WINAPI _tWinMain(HINSTANCE, HINSTANCE, LPTSTR, INT nShowCommand)
{
    return _AtlModule.WinMain(nShowCommand);
}