diff options
Diffstat (limited to 'utils/vmpi/vmpi_job_watch/JobWatchDlg.cpp')
| -rw-r--r-- | utils/vmpi/vmpi_job_watch/JobWatchDlg.cpp | 648 |
1 files changed, 648 insertions, 0 deletions
diff --git a/utils/vmpi/vmpi_job_watch/JobWatchDlg.cpp b/utils/vmpi/vmpi_job_watch/JobWatchDlg.cpp new file mode 100644 index 0000000..4c7909b --- /dev/null +++ b/utils/vmpi/vmpi_job_watch/JobWatchDlg.cpp @@ -0,0 +1,648 @@ +//========= Copyright Valve Corporation, All rights reserved. ============// +// +// Purpose: +// +// $NoKeywords: $ +// +//=============================================================================// +// JobWatchDlg.cpp : implementation file +// + +#include "stdafx.h" +#include "JobWatchDlg.h" +#include "tier1/strtools.h" +#include "consolewnd.h" +#include "vmpi_browser_helpers.h" + + +#ifdef _DEBUG +#define new DEBUG_NEW +#undef THIS_FILE +static char THIS_FILE[] = __FILE__; +#endif + + +#define IMPLEMENT_SORT_NUMBER_FN( FnName, VarName ) \ + static int CALLBACK FnName( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam ) \ + { \ + CWorkerInfo *pInfo1 = (CWorkerInfo*)iItem1; \ + CWorkerInfo *pInfo2 = (CWorkerInfo*)iItem2; \ + return pInfo1->VarName > pInfo2->VarName; \ + } + +static int CALLBACK SortByName( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam ) +{ + CWorkerInfo *pInfo1 = (CWorkerInfo*)iItem1; + CWorkerInfo *pInfo2 = (CWorkerInfo*)iItem2; + + return strcmp( pInfo1->m_ComputerName, pInfo2->m_ComputerName ); +} + +static int CALLBACK SortByCurrentStage( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam ) +{ + CWorkerInfo *pInfo1 = (CWorkerInfo*)iItem1; + CWorkerInfo *pInfo2 = (CWorkerInfo*)iItem2; + + return strcmp( pInfo1->m_CurrentStage, pInfo2->m_CurrentStage ); +} + +IMPLEMENT_SORT_NUMBER_FN( SortByConnected, m_bConnected ) +IMPLEMENT_SORT_NUMBER_FN( SortByWorkUnitsDone, m_nWorkUnitsDone ); +IMPLEMENT_SORT_NUMBER_FN( SortByRunningTime, m_RunningTimeMS ); +IMPLEMENT_SORT_NUMBER_FN( SortByThread0WU, m_ThreadWUs[0] ); +IMPLEMENT_SORT_NUMBER_FN( SortByThread1WU, m_ThreadWUs[1] ); +IMPLEMENT_SORT_NUMBER_FN( SortByThread2WU, m_ThreadWUs[2] ); +IMPLEMENT_SORT_NUMBER_FN( SortByThread3WU, m_ThreadWUs[3] ); + +typedef int (CALLBACK *ServicesSortFn)( LPARAM iItem1, LPARAM iItem2, LPARAM lpParam ); + +struct +{ + char *pText; + int width; + ServicesSortFn sortFn; +} g_ColumnInfos[] = +{ + {"Computer Name", 150, SortByName}, + {"Connected", 70, SortByConnected}, + {"Work Units Done", 100, SortByWorkUnitsDone}, + {"Running Time", 80, SortByRunningTime}, + {"Current Stage", 180, SortByCurrentStage}, + {"Thread 0", 70, SortByThread0WU}, + {"Thread 1", 70, SortByThread1WU}, + {"Thread 2", 70, SortByThread2WU}, + {"Thread 3", 70, SortByThread3WU} +}; + +#define COLUMN_COMPUTER_NAME 0 +#define COLUMN_CONNECTED 1 +#define COLUMN_WORK_UNITS_DONE 2 +#define COLUMN_RUNNING_TIME 3 +#define COLUMN_CURRENT_STAGE 4 +#define COLUMN_THREAD0_WU 5 +#define COLUMN_THREAD1_WU 6 +#define COLUMN_THREAD2_WU 7 +#define COLUMN_THREAD3_WU 8 + +int g_iSortColumn = 0; + + +///////////////////////////////////////////////////////////////////////////// +// CJobWatchDlg dialog + + +CJobWatchDlg::CJobWatchDlg(CWnd* pParent /*=NULL*/) + : CIdleDialog(CJobWatchDlg::IDD, pParent) +{ + m_CurMessageIndex = 0; + m_CurGraphTime = 0; + m_pSQL = NULL; + m_hMySQLDLL = NULL; + m_CurWorkerTextToken = 0; + m_LastQueryTime = 0; + + //{{AFX_DATA_INIT(CJobWatchDlg) + // NOTE: the ClassWizard will add member initialization here + //}}AFX_DATA_INIT +} + + +CJobWatchDlg::~CJobWatchDlg() +{ + if ( m_pSQL ) + { + m_pSQL->Release(); + } + + if ( m_hMySQLDLL ) + { + Sys_UnloadModule( m_hMySQLDLL ); + } +} + + +void CJobWatchDlg::DoDataExchange(CDataExchange* pDX) +{ + CDialog::DoDataExchange(pDX); + //{{AFX_DATA_MAP(CJobWatchDlg) + DDX_Control(pDX, IDC_WORKERS, m_Workers); + DDX_Control(pDX, IDC_TEXTOUTPUT, m_TextOutput); + //}}AFX_DATA_MAP +} + + +BEGIN_MESSAGE_MAP(CJobWatchDlg, CIdleDialog) + //{{AFX_MSG_MAP(CJobWatchDlg) + ON_LBN_SELCHANGE(IDC_WORKERS, OnSelChangeWorkers) + ON_WM_SIZE() + ON_NOTIFY(LVN_ODSTATECHANGED, IDC_WORKERS, OnOdstatechangedWorkers) + ON_NOTIFY(LVN_ITEMCHANGED, IDC_WORKERS, OnItemchangedWorkers) + //}}AFX_MSG_MAP +END_MESSAGE_MAP() + +///////////////////////////////////////////////////////////////////////////// +// CJobWatchDlg message handlers + +const char* FindArg( const char *pArgName, const char *pDefault="" ) +{ + for ( int i=1; i < __argc; i++ ) + { + if ( Q_stricmp( pArgName, __argv[i] ) == 0 ) + { + if ( (i+1) < __argc ) + return __argv[i+1]; + else + return pDefault; + } + } + return NULL; +} + + +bool ReadStringFromFile( FILE *fp, char *pStr, int strSize ) +{ + int i=0; + for ( i; i < strSize-2; i++ ) + { + if ( fread( &pStr[i], 1, 1, fp ) != 1 || + pStr[i] == '\n' ) + { + break; + } + } + + pStr[i] = 0; + return i != 0; +} + + +BOOL CJobWatchDlg::OnInitDialog() +{ + CDialog::OnInitDialog(); + + + m_Workers.SetExtendedStyle( LVS_EX_FULLROWSELECT ); + + // Setup the headers. + for ( int i=0; i < ARRAYSIZE( g_ColumnInfos ); i++ ) + { + m_Workers.InsertColumn( i, g_ColumnInfos[i].pText, LVCFMT_LEFT, g_ColumnInfos[i].width, i ); + } + + + m_GraphControl.SubclassDlgItem( IDC_GRAPH_AREA, this ); + + + CString str; + + // Get all our startup info from the command line. + const char *pJobID = FindArg( "-JobID", NULL ); + const char *pDBName = FindArg( "-dbname", NULL ); + const char *pHostName = FindArg( "-hostname", NULL ); + const char *pUserName = FindArg( "-username", NULL ); + + if ( !pJobID ) + { + str.Format( "Missing a command line parameter (-JobID or -dbname or -hostname or -username)" ); + MessageBox( str, "Error", MB_OK ); + EndDialog( 1 ); + return FALSE; + } + + char hostName[512], dbName[512], userName[512]; + if ( !pDBName || !pHostName || !pUserName ) + { + char errString[512]; + + // If they don't specify the DB info, get it from where + const char *pFilename = "dbinfo_job_search.txt"; + FILE *fp = fopen( pFilename, "rt" ); + if ( !fp ) + { + Q_snprintf( errString, sizeof( errString ), "Can't open '%s' for database info.", pFilename ); + MessageBox( errString, "Error", MB_OK ); + EndDialog( 0 ); + return FALSE; + } + + if ( !ReadStringFromFile( fp, hostName, sizeof( hostName ) ) || + !ReadStringFromFile( fp, dbName, sizeof( dbName ) ) || + !ReadStringFromFile( fp, userName, sizeof( userName ) ) + ) + { + fclose( fp ); + Q_snprintf( errString, sizeof( errString ), "'%s' has invalid format.", pFilename ); + MessageBox( errString, "Error", MB_OK ); + EndDialog( 0 ); + return FALSE; + } + + pDBName = dbName; + pHostName = hostName; + pUserName = userName; + + fclose( fp ); + } + + m_JobID = atoi( pJobID ); + + // Get the mysql interface. + IMySQL *pSQL; + if ( !Sys_LoadInterface( "mysql_wrapper", MYSQL_WRAPPER_VERSION_NAME, &m_hMySQLDLL, (void**)&pSQL ) ) + return false; + + if ( !pSQL->InitMySQL( pDBName, pHostName, pUserName ) ) + { + pSQL->Release(); + str.Format( "Can't init MYSQL db (db = '%s', host = '%s', user = '%s')", pDBName, pHostName, pUserName ); + MessageBox( str, "Error", MB_OK ); + EndDialog( 0 ); + return FALSE; + } + + m_pSQL = CreateMySQLAsync( pSQL ); + if ( !m_pSQL ) + { + pSQL->Release(); + str.Format( "Can't create IMySQLAsync" ); + MessageBox( str, "Error", MB_OK ); + EndDialog( 0 ); + return FALSE; + } + + + memset( m_bQueriesInProgress, 0, sizeof( m_bQueriesInProgress ) ); + + + // (Init the idle processor so we can update text and graphs). + StartIdleProcessing( 100 ); + + // Fill in the command line control. + char cmdLine[2048]; + Q_snprintf( cmdLine, sizeof( cmdLine ), "vmpi_job_watch -JobID %s -hostname %s -dbname %s -username %s", + pJobID, pHostName, pDBName, pUserName ); + SetDlgItemText( IDC_COMMAND_LINE, cmdLine ); + + // Setup anchors. + m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_WORKERS_PANEL ), ANCHOR_LEFT, ANCHOR_TOP, ANCHOR_WIDTH_PERCENT, ANCHOR_HEIGHT_PERCENT ); + m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_WORKERS ), ANCHOR_LEFT, ANCHOR_TOP, ANCHOR_WIDTH_PERCENT, ANCHOR_HEIGHT_PERCENT ); + + m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_TEXT_OUTPUT_PANEL ), ANCHOR_LEFT, ANCHOR_HEIGHT_PERCENT, ANCHOR_WIDTH_PERCENT, ANCHOR_BOTTOM ); + m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_TEXTOUTPUT ), ANCHOR_LEFT, ANCHOR_HEIGHT_PERCENT, ANCHOR_WIDTH_PERCENT, ANCHOR_BOTTOM ); + + m_AnchorMgr.AddAnchor( this, GetDlgItem( IDC_GRAPHS_PANEL ), ANCHOR_WIDTH_PERCENT, ANCHOR_TOP, ANCHOR_RIGHT, ANCHOR_HEIGHT_PERCENT ); + m_AnchorMgr.AddAnchor( this, &m_GraphControl, ANCHOR_WIDTH_PERCENT, ANCHOR_TOP, ANCHOR_RIGHT, ANCHOR_HEIGHT_PERCENT ); + + return TRUE; // return TRUE unless you set the focus to a control + // EXCEPTION: OCX Property Pages should return FALSE +} + + +CWorkerInfo* CJobWatchDlg::FindWorkerByID( unsigned long jobWorkerID ) +{ + int nIndex = -1; + while ( ( nIndex = m_Workers.GetNextItem( nIndex, LVNI_ALL ) ) != -1 ) + { + CWorkerInfo *pInfo = (CWorkerInfo*)m_Workers.GetItemData( nIndex ); + if ( pInfo->m_JobWorkerID == jobWorkerID ) + return pInfo; + } + + return NULL; +} + + +CWorkerInfo* CJobWatchDlg::FindWorkerByMachineName( const char *pMachineName ) +{ + int nIndex = -1; + while ( ( nIndex = m_Workers.GetNextItem( nIndex, LVNI_ALL ) ) != -1 ) + { + CWorkerInfo *pInfo = (CWorkerInfo*)m_Workers.GetItemData( nIndex ); + if ( Q_stricmp( pInfo->m_ComputerName, pMachineName ) == 0 ) + return pInfo; + } + + return NULL; +} + + +void CJobWatchDlg::SetWorkerListItemInt( int nIndex, int iColumn, int value ) +{ + char str[512]; + Q_snprintf( str, sizeof( str ), "%d", value ); + m_Workers.SetItemText( nIndex, iColumn, str ); +} + + +void CJobWatchDlg::UpdateWorkersList() +{ + int nIndex = -1; + while ( ( nIndex = m_Workers.GetNextItem( nIndex, LVNI_ALL ) ) != -1 ) + { + CWorkerInfo *pInfo = (CWorkerInfo*)m_Workers.GetItemData( nIndex ); + + char *pConnectedStr = pInfo->m_bConnected ? "yes" : "no"; + m_Workers.SetItemText( nIndex, COLUMN_CONNECTED, pConnectedStr ); + + SetWorkerListItemInt( nIndex, COLUMN_WORK_UNITS_DONE, pInfo->m_nWorkUnitsDone ); + + char timeStr[1024]; + FormatTimeString( pInfo->m_RunningTimeMS / 1000, timeStr, sizeof( timeStr ) ); + m_Workers.SetItemText( nIndex, COLUMN_RUNNING_TIME, timeStr ); + + // Current stage. + SetWorkerListItemInt( nIndex, COLUMN_WORK_UNITS_DONE, pInfo->m_nWorkUnitsDone ); + + m_Workers.SetItemText( nIndex, COLUMN_CURRENT_STAGE, pInfo->m_CurrentStage ); + + SetWorkerListItemInt( nIndex, COLUMN_THREAD0_WU, pInfo->m_ThreadWUs[0] ); + SetWorkerListItemInt( nIndex, COLUMN_THREAD1_WU, pInfo->m_ThreadWUs[1] ); + SetWorkerListItemInt( nIndex, COLUMN_THREAD2_WU, pInfo->m_ThreadWUs[2] ); + SetWorkerListItemInt( nIndex, COLUMN_THREAD3_WU, pInfo->m_ThreadWUs[3] ); + } + + ResortItems(); +} + + +bool CJobWatchDlg::GetCurJobWorkerID( unsigned long &id ) +{ + POSITION pos = m_Workers.GetFirstSelectedItemPosition(); + if ( !pos ) + return false; + + int index = m_Workers.GetNextSelectedItem( pos ); + CWorkerInfo *pInfo = (CWorkerInfo*)m_Workers.GetItemData( index ); + id = pInfo->m_JobWorkerID; + return true; +} + + +void CJobWatchDlg::OnSelChangeWorkers() +{ + // Clear the text output and invalidate any old queries for text. + int nLen = m_TextOutput.SendMessage( EM_GETLIMITTEXT, 0, 0 ); + m_TextOutput.SendMessage( EM_SETSEL, 0, nLen ); + m_TextOutput.SendMessage( EM_REPLACESEL, FALSE, (LPARAM)"" ); + + m_CurMessageIndex = 0; + m_CurWorkerTextToken++; + m_LastQueryTime = 0; // force a query. + + m_GraphControl.Clear(); + m_CurGraphTime = -1; +} + + +void CJobWatchDlg::ResortItems() +{ + m_Workers.SortItems( g_ColumnInfos[g_iSortColumn].sortFn, (LPARAM)this ); +} + + + +void CJobWatchDlg::OnIdle() +{ + // Issue any queries that we need to. + DWORD curTime = GetTickCount(); + if ( curTime - m_LastQueryTime >= 1000 ) + { + m_LastQueryTime = curTime; + char query[2048]; + + unsigned long jobWorkerID; + bool bJobWorkerIDValid = GetCurJobWorkerID( jobWorkerID ); + + if ( !m_bQueriesInProgress[QUERY_TEXT] && bJobWorkerIDValid ) + { + Q_snprintf( query, sizeof( query ), "select * from text_messages where JobWorkerID=%lu and MessageIndex >= %lu", jobWorkerID, m_CurMessageIndex ); + m_pSQL->Execute( query, (void*)(QUERY_TEXT | (m_CurWorkerTextToken << 16)) ); + m_bQueriesInProgress[QUERY_TEXT] = true; + } + + if ( !m_bQueriesInProgress[QUERY_GRAPH] && bJobWorkerIDValid ) + { + Q_snprintf( query, sizeof( query ), "select * from graph_entry where JobWorkerID=%lu", jobWorkerID ); + m_pSQL->Execute( query, (void*)QUERY_GRAPH ); + m_bQueriesInProgress[QUERY_GRAPH] = true; + } + + if ( !m_bQueriesInProgress[QUERY_WORKER_STATS] ) + { + Q_snprintf( query, sizeof( query ), "select JobWorkerID, WorkerState, NumWorkUnits, " + "RunningTimeMS, CurrentStage, Thread0WU, Thread1WU, Thread2WU, Thread3WU, IsMaster, MachineName " + " from job_worker_start where JobID=%lu", m_JobID ); + + m_pSQL->Execute( query, (void*)QUERY_WORKER_STATS ); + m_bQueriesInProgress[QUERY_WORKER_STATS] = true; + } + } + + + // Pickup query results. + CQueryResults results; + while ( m_pSQL->GetNextResults( results ) && results.m_pResults ) + { + int iQueryID = ((int)results.m_pUserData) & 0xFFFF; + int iExtraData = ((int)results.m_pUserData) >> 16; + + if ( results.m_pResults ) + { + if ( iQueryID == QUERY_TEXT ) + { + if ( iExtraData == m_CurWorkerTextToken ) + { + ProcessQueryResults_Text( results.m_pResults ); + } + } + else if ( iQueryID == QUERY_GRAPH ) + { + ProcessQueryResults_Graph( results.m_pResults ); + } + else if ( iQueryID == QUERY_WORKER_STATS ) + { + ProcessQueryResults_WorkerStats( results.m_pResults ); + } + + results.m_pResults->Release(); + } + + m_bQueriesInProgress[iQueryID] = false; + } +} + + +void CJobWatchDlg::ProcessQueryResults_WorkerStats( IMySQLRowSet *pSet ) +{ + bool bChange = false; + while ( pSet->NextRow() ) + { + int iColumn = 0; + int workerID = pSet->GetColumnValue_Int( iColumn++ ); + int workerState = pSet->GetColumnValue_Int( iColumn++ ); + int nWorkUnits = pSet->GetColumnValue_Int( iColumn++ ); + unsigned long runningTimeMS = pSet->GetColumnValue_Int( iColumn++ ); + const char *pCurrentStage = pSet->GetColumnValue_String( iColumn++ ); + int iThread0WU = pSet->GetColumnValue_Int( iColumn++ ); + int iThread1WU = pSet->GetColumnValue_Int( iColumn++ ); + int iThread2WU = pSet->GetColumnValue_Int( iColumn++ ); + int iThread3WU = pSet->GetColumnValue_Int( iColumn++ ); + int bIsMaster = pSet->GetColumnValue_Int( iColumn++ ); + const char *pMachineName = pSet->GetColumnValue_String( iColumn ); + + CWorkerInfo *pInfo = FindWorkerByID( workerID ); + if ( pInfo ) + { + if ( workerState != pInfo->m_bConnected || + nWorkUnits != pInfo->m_nWorkUnitsDone || + runningTimeMS != pInfo->m_RunningTimeMS || + stricmp( pCurrentStage, pInfo->m_CurrentStage ) != 0 || + iThread0WU != pInfo->m_ThreadWUs[0] || + iThread1WU != pInfo->m_ThreadWUs[1] || + iThread2WU != pInfo->m_ThreadWUs[2] || + iThread3WU != pInfo->m_ThreadWUs[3] + ) + { + bChange = true; + pInfo->m_bConnected = workerState; + pInfo->m_nWorkUnitsDone = nWorkUnits; + pInfo->m_RunningTimeMS = runningTimeMS; + pInfo->m_CurrentStage = pCurrentStage; + pInfo->m_ThreadWUs[0] = iThread0WU; + pInfo->m_ThreadWUs[1] = iThread1WU; + pInfo->m_ThreadWUs[2] = iThread2WU; + pInfo->m_ThreadWUs[3] = iThread3WU; + } + } + else + { + // Add a new entry. + CWorkerInfo *pInfo = new CWorkerInfo; + pInfo->m_ComputerName = pMachineName; + pInfo->m_bConnected = false; + pInfo->m_nWorkUnitsDone = 0; + pInfo->m_RunningTimeMS = 0; + pInfo->m_JobWorkerID = workerID; + + int index; + if ( bIsMaster ) + { + char tempStr[512]; + Q_snprintf( tempStr, sizeof( tempStr ), "%s [master]", (const char*)pInfo->m_ComputerName ); + index = m_Workers.InsertItem( COLUMN_COMPUTER_NAME, tempStr, NULL ); + } + else + { + index = m_Workers.InsertItem( COLUMN_COMPUTER_NAME, pInfo->m_ComputerName, NULL ); + } + + m_Workers.SetItemData( index, (DWORD)pInfo ); + bChange = true; + } + } + + if ( bChange ) + { + UpdateWorkersList(); + } +} + + +void CJobWatchDlg::ProcessQueryResults_Text( IMySQLRowSet *pSet ) +{ + CUtlVector<char> text; + + while ( pSet->NextRow() ) + { + const char *pTextStr = pSet->GetColumnValue( "text" ).String(); + int len = strlen( pTextStr ); + text.AddMultipleToTail( len, pTextStr ); + + m_CurMessageIndex = pSet->GetColumnValue( "MessageIndex" ).Int32() + 1; + } + + text.AddToTail( 0 ); + FormatAndSendToEditControl( m_TextOutput.GetSafeHwnd(), text.Base() ); +} + + +void CJobWatchDlg::ProcessQueryResults_Graph( IMySQLRowSet *pSet ) +{ + int iMSTime = pSet->GetColumnIndex( "MSSinceJobStart" ); + int iBytesSent = pSet->GetColumnIndex( "BytesSent" ); + int iBytesReceived = pSet->GetColumnIndex( "BytesReceived" ); + + // See if there's anything new. + CUtlVector<CGraphEntry> entries; + + int highest = m_CurGraphTime; + while ( pSet->NextRow() ) + { + CGraphEntry entry; + entry.m_msTime = pSet->GetColumnValue( iMSTime ).Int32(); + entry.m_nBytesSent = pSet->GetColumnValue( iBytesSent ).Int32(); + entry.m_nBytesReceived = pSet->GetColumnValue( iBytesReceived ).Int32(); + entries.AddToTail( entry ); + + highest = max( highest, entry.m_msTime ); + } + + if ( highest > m_CurGraphTime ) + { + m_CurGraphTime = highest; + + m_GraphControl.Clear(); + m_GraphControl.Fill( entries ); + } +} + + +void CJobWatchDlg::OnSize(UINT nType, int cx, int cy) +{ + CIdleDialog::OnSize(nType, cx, cy); + + m_AnchorMgr.UpdateAnchors( this ); +} + + +BOOL CJobWatchDlg::OnNotify(WPARAM wParam, LPARAM lParam, LRESULT* pResult) +{ + NMHDR *pHdr = (NMHDR*)lParam; + if ( pHdr->idFrom == IDC_WORKERS ) + { + if ( pHdr->code == LVN_COLUMNCLICK ) + { + LPNMLISTVIEW pListView = (LPNMLISTVIEW)lParam; + + // Now sort by this column. + g_iSortColumn = max( 0, min( pListView->iSubItem, (int)ARRAYSIZE( g_ColumnInfos ) - 1 ) ); + ResortItems(); + } + } + + return CIdleDialog::OnNotify(wParam, lParam, pResult); +} + +void CJobWatchDlg::OnOdstatechangedWorkers(NMHDR* pNMHDR, LRESULT* pResult) +{ + NMLVODSTATECHANGE* pStateChanged = (NMLVODSTATECHANGE*)pNMHDR; + + if ( !( pStateChanged->uOldState & LVIS_SELECTED ) && ( pStateChanged->uNewState & LVIS_SELECTED ) ) + { + OnSelChangeWorkers(); + } + + *pResult = 0; +} + +void CJobWatchDlg::OnItemchangedWorkers(NMHDR* pNMHDR, LRESULT* pResult) +{ + NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR; + + if ( !( pNMListView->uOldState & LVIS_SELECTED ) && ( pNMListView->uNewState & LVIS_SELECTED ) ) + { + OnSelChangeWorkers(); + } + + *pResult = 0; +} |