Custom update interval option

Author: aubymori

main.cpp

@@ -1914,6 +1914,69 @@ void ExecuteShutdownMenuOption(int iID)
 }
 
 
+/*++ CustomIntervalDlgProc
+
+Routine Description:
+
+    Dialog procedure for Set Custom Interval dialog
+
+Revision History:
+
+      Nov-02-24 aubymori  Created
+
+--*/
+INT_PTR CALLBACK CustomIntervalDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
+{
+    switch (uMsg)
+    {
+        case WM_INITDIALOG:
+            SendDlgItemMessageW(hwndDlg, 
+                                IDC_CUSTOM_INTERVAL_SPINNER, 
+                                UDM_SETRANGE32,
+                                0, 
+                                MAXINT32);
+            SendDlgItemMessageW(hwndDlg, 
+                                IDC_CUSTOM_INTERVAL_SPINNER, 
+                                UDM_SETPOS32,
+                                0, 
+                                g_Options.m_dwTimerInterval);
+            SetDlgItemInt(hwndDlg, 
+                        IDC_CUSTOM_INTERVAL_INPUT, 
+                        g_Options.m_dwTimerInterval, 
+                        FALSE);
+            return TRUE;
+        case WM_CLOSE:
+            EndDialog(hwndDlg, IDCANCEL);
+            return TRUE;
+        case WM_COMMAND:
+            switch (wParam)
+            {
+                case IDOK:
+                {
+                    DWORD dwNewInterval = GetDlgItemInt(hwndDlg,
+                                                       IDC_CUSTOM_INTERVAL_INPUT,
+                                                       NULL,
+                                                       FALSE);
+
+                    // don't need to update if same time
+                    if (dwNewInterval == g_Options.m_dwTimerInterval)
+                        EndDialog(hwndDlg, IDCANCEL);
+
+                    g_Options.m_dwTimerInterval = dwNewInterval;
+                    g_Options.m_usUpdateSpeed = US_CUSTOM;
+                    EndDialog(hwndDlg, IDOK);
+                    break;
+                }
+                case IDCANCEL:
+                    EndDialog(hwndDlg, IDCANCEL);
+                    break;
+            }
+            return TRUE;
+        default:
+            return FALSE;
+    }
+}
+
 /*++ MainWnd_OnCommand
 
 Routine Description:
@@ -2269,6 +2332,22 @@ void MainWnd_OnCommand(HWND hwnd, int id)
         }
         break;
 
+    case IDM_CUSTOM:
+        if (IDOK == DialogBoxParam(g_hInstance, 
+                                MAKEINTRESOURCE(IDD_CUSTOM_INTERVAL),
+                                g_hMainWnd, 
+                                CustomIntervalDlgProc, 
+                                NULL))
+        {
+            KillTimer(g_hMainWnd, 0);
+            if (g_Options.m_dwTimerInterval)
+            {
+                SetTimer(g_hMainWnd, 0, g_Options.m_dwTimerInterval, NULL);
+            }
+            UpdateMenuStates();
+        }
+        break;
+
     case IDM_ABOUT:
         {
             //

resource.h

@@ -269,6 +269,9 @@
 #define IDS_SHUTDOWN_REMOTE_OTHERUSERS  4051
 #define IDS_SHUTDOWN_OTHERUSERS         4052
 #define IDS_RESTART_OTHERUSERS          4053
+#define IDD_CUSTOM_INTERVAL             4054
+#define IDC_CUSTOM_INTERVAL_INPUT       4055
+#define IDC_CUSTOM_INTERVAL_SPINNER     4056
 
 //
 //  5000-5999 reserved for dynamic creation of CPU graphs
@@ -410,6 +413,7 @@
 #define IDM_NORMAL                      40023
 #define IDM_LOW                         40024
 #define IDM_PAUSED                      40025
+#define IDM_CUSTOM                      40026
 #define IDM_PROC_DEBUG                  40027
 #define IDM_PROC_TERMINATE              40028
 

taskmgr.h

@@ -166,9 +166,10 @@ typedef enum
     US_NORMAL,
     US_LOW,
     US_PAUSED,
+    US_CUSTOM,
 } UPDATESPEED;
 #define US_FIRST IDM_HIGH
-#define US_LAST  IDM_PAUSED
+#define US_LAST  IDM_CUSTOM
 
 
 

taskmgr.rc

@@ -427,6 +427,25 @@ BEGIN
     PUSHBUTTON      "&Help",ID_HELP,164,86,50,14
 END
 
+IDD_CUSTOM_INTERVAL DIALOGEX 50, 50, 184, 70
+STYLE DS_MODALFRAME | WS_POPUP | WS_VISIBLE | WS_CAPTION | WS_SYSMENU
+CAPTION "Set Custom Update Interval"
+FONT 8, "MS Shell Dlg 2"
+BEGIN
+    LTEXT           "Input the amount of time, in milliseconds, that Task Manager should wait between updates.",
+                    -1,7,6,170,19
+    EDITTEXT        IDC_CUSTOM_INTERVAL_INPUT,58,30,68,12,ES_NUMBER
+
+    CONTROL         "", IDC_CUSTOM_INTERVAL_SPINNER, 
+                    UPDOWN_CLASS, WS_CHILD | WS_VISIBLE
+                    | UDS_SETBUDDYINT | UDS_AUTOBUDDY 
+                    | UDS_ALIGNRIGHT | UDS_ARROWKEYS | UDS_NOTHOUSANDS,
+                    125, 30, 11, 12
+
+    DEFPUSHBUTTON   "OK",IDOK,73,50,50,14,WS_GROUP
+    PUSHBUTTON      "Cancel",IDCANCEL,127,50,50,14
+END
+
 
 /////////////////////////////////////////////////////////////////////////////
 //
@@ -457,6 +476,7 @@ BEGIN
             MENUITEM "&Normal",                     IDM_NORMAL
             MENUITEM "&Low",                        IDM_LOW
             MENUITEM "&Paused",                     IDM_PAUSED
+            MENUITEM "&Custom...",                     IDM_CUSTOM
         END
         MENUITEM SEPARATOR
         MENUITEM "Lar&ge Icons",                IDM_LARGEICONS
@@ -560,6 +580,7 @@ BEGIN
             MENUITEM "&Normal",                     IDM_NORMAL
             MENUITEM "&Low",                        IDM_LOW
             MENUITEM "&Paused",                     IDM_PAUSED
+            MENUITEM "&Custom...",                     IDM_CUSTOM
         END
         MENUITEM SEPARATOR
         POPUP "&CPU History"
@@ -608,6 +629,7 @@ BEGIN
             MENUITEM "&Normal",                     IDM_NORMAL
             MENUITEM "&Low",                        IDM_LOW
             MENUITEM "&Paused",                     IDM_PAUSED
+            MENUITEM "&Custom...",                     IDM_CUSTOM
         END
         MENUITEM SEPARATOR
         MENUITEM "&Select Columns...",          IDM_PROCCOLS
@@ -658,6 +680,7 @@ BEGIN
             MENUITEM "&Normal",                     IDM_NORMAL
             MENUITEM "&Low",                        IDM_LOW
             MENUITEM "&Paused",                     IDM_PAUSED
+            MENUITEM "&Custom...",                     IDM_CUSTOM
         END
         MENUITEM SEPARATOR
         MENUITEM "&Select Columns...",          IDM_USERCOLS
@@ -728,6 +751,7 @@ BEGIN
             MENUITEM "&Normal",                 IDM_NORMAL
             MENUITEM "&Low",                    IDM_LOW
             MENUITEM "&Paused",                 IDM_PAUSED
+            MENUITEM "&Custom...",                 IDM_CUSTOM
         END
         MENUITEM SEPARATOR  
         POPUP "&Network Adapter History"
@@ -902,6 +926,7 @@ BEGIN
     IDM_NORMAL              "Updates the display every two seconds"
     IDM_LOW                 "Updates the display every four seconds"
     IDM_PAUSED              "Display does not automatically update"
+    IDM_CUSTOM              "Set a custom update interval"
     IDM_PROC_DEBUG          "Attaches the debugger to this process"
     IDM_PROC_TERMINATE      "Removes the process from the system"
     IDM_PROC_REALTIME       "Sets process to the REALTIME priority class"