/*
 * PerformanceCounterBrowser.h 
 * Copyright (c) 2011 Antillia.com TOSHIYUKI ARAI. ALL RIGHTS RESERVED. 
 */


// SOL9
// 2009/09/30

#pragma once
#include <sol/Object.h>
#include <sol/String.h>

#include <pdh.h>    // DWORD,PDH_FMT_LONG‘¼
#include <pdhmsg.h>    // PDH_MORE_DATA
//#include <stdio.h>

#pragma comment( lib, "pdh.lib" )

namespace SOL {

class PerformanceCounterBrowser :public Object {

private:
    HWND hOwner;
    
private:
    PDH_BROWSE_DLG_CONFIG config;

private:
    TCHAR counterPath[1024];

public:
    PerformanceCounterBrowser()
    :hOwner(NULL)
    {
        memset(counterPath, (TCHAR)0, SizeOf(counterPath));
    }

public:
    Boolean create(HWND hparent, const TCHAR* title, DWORD level=PERF_DETAIL_WIZARD)
    {
        Boolean rc = True;
        memset(counterPath, (TCHAR)0, SizeOf(counterPath));

        memset(&config, 0, sizeof(config));

        hOwner = hparent;
        if (title) {
            config.szDialogBoxCaption = (TCHAR*)title; 
        }
        config.hWndOwner = hOwner;
        config.bIncludeInstanceIndex    = TRUE;//FALSE;
        config.bSingleCounterPerAdd     = TRUE;
        config.bSingleCounterPerDialog  = TRUE;
        config.bLocalCountersOnly       = TRUE;
        config.bWildCardInstances       = TRUE;
        config.bHideDetailBox           = TRUE;

        config.bInitializePath          = FALSE;
        config.bDisableMachineSelection = FALSE;
        config.bIncludeCostlyObjects    = FALSE;
        config.bShowObjectBrowser       = FALSE;

        config.szDataSource             = NULL;
        config.szReturnPathBuffer       = counterPath;
        config.cchReturnPathLength      = SizeOf(counterPath);
        config.pCallBack                = NULL;
        config.dwCallBackArg            = 0;
        config.CallBackStatus           = ERROR_SUCCESS;
        config.dwDefaultDetailLevel     = level;

        return rc;
    }


public:
    DWORD select(__out String& path)
    {
        path = "";
        memset(counterPath, (TCHAR)0, SizeOf(counterPath));

        config.szReturnPathBuffer = counterPath;  
        config.cchReturnPathLength = SizeOf(counterPath);

        PDH_STATUS status = PdhBrowseCounters(&config);
        
        if (status == ERROR_SUCCESS) {
            path = counterPath;
        }
        
        return status;
    }

public:
    bool getCounterPath(__out String& path)
    {
        bool rc = false;
        path = "";
        if (strlen(counterPath)>0) {
            path = counterPath;
            rc = true;
        }
        return rc;
    }

// Setter
public:
    void setIncludeInstanceIndex(BOOL value)
    {
        config.bIncludeInstanceIndex    = value;
    }

public:
    void setSingleCounterPerAdd(BOOL value)
    {
        config.bSingleCounterPerAdd     = value;
    }

public:
    void setSingleCounterPerDialog(BOOL value)
    {
        config.bSingleCounterPerDialog  = value;
    }

public:
    void setLocalCountersOnly(BOOL value)
    {
        config.bLocalCountersOnly       = value;
    }

public:
    void setWildCardInstances(BOOL value)
    {

        config.bWildCardInstances       = value;
    }

public:
    void setHideDetailBox(BOOL value)
    {
        config.bHideDetailBox           = value;
    }

public:
    void setInitializePath(BOOL value)
    {
        config.bInitializePath          = FALSE;
    }

public:
    void setDisableMachineSelection(BOOL value)
    {
        config.bDisableMachineSelection = FALSE;
    }
public:
    void setIncludeCostlyObjects(BOOL value)
    {
        config.bIncludeCostlyObjects    = value;
    }

public:
    void setShowObjectBrowser(BOOL value)
    {
        config.bShowObjectBrowser       = value;
    }

public:
    void setDefaultDetailLevel(DWORD level)
    {
        config.dwDefaultDetailLevel     = level;
    }

// getter
public:
    BOOL getIncludeInstanceIndex()
    {
        return config.bIncludeInstanceIndex;
    }

public:
    BOOL getSingleCounterPerAdd()
    {
        return config.bSingleCounterPerAdd ;
    }

public:
    BOOL getSingleCounterPerDialog()
    {
        return config.bSingleCounterPerDialog;
    }

public:
    BOOL setLocalCountersOnly()
    {
        return config.bLocalCountersOnly;
    }

public:
    BOOL getWildCardInstances()
    {
        return config.bWildCardInstances;
    }

public:
    BOOL getHideDetailBox()
    {
        return config.bHideDetailBox;
    }

public:
    BOOL getInitializePath()
    {
        return config.bInitializePath;
    }

public:
    BOOL getDisableMachineSelection()
    {
        return config.bDisableMachineSelection;
    }

public:
    BOOL getIncludeCostlyObjects()
    {
        return config.bIncludeCostlyObjects;
    }

public:
    BOOL getShowObjectBrowser()
    {
        return config.bShowObjectBrowser;
    }

public:
    BOOL getDefaultDetailLevel()
    {
        return config.dwDefaultDetailLevel;
    }

};

}