@@ -213,22 +213,27 @@ void ForceMenuRebuild(void) {
213213
214214void SwitchView (int mode ) {
215215 g_viewMode = mode ;
216- ShowWindow (g_hwndQuery , mode == 0 ? SW_SHOW : SW_HIDE );
217- if (g_hwndLineNum ) ShowWindow (g_hwndLineNum , (mode == 0 && g_showLineNumbers ) ? SW_SHOW : SW_HIDE );
218- ShowWindow (g_hwndResult , mode == 1 && !g_gridView ? SW_SHOW : SW_HIDE );
219- if (g_hwndGrid ) ShowWindow (g_hwndGrid , mode == 1 && g_gridView ? SW_SHOW : SW_HIDE );
220- if (g_hwndSchema ) ShowWindow (g_hwndSchema , mode == 2 ? SW_SHOW : SW_HIDE );
216+ ShowWindow (g_hwndQuery , mode == VIEW_QUERY ? SW_SHOW : SW_HIDE );
217+ if (g_hwndLineNum ) ShowWindow (g_hwndLineNum , (mode == VIEW_QUERY && g_showLineNumbers ) ? SW_SHOW : SW_HIDE );
218+ ShowWindow (g_hwndResult , mode == VIEW_RESULT && !g_gridView ? SW_SHOW : SW_HIDE );
219+ if (g_hwndGrid ) ShowWindow (g_hwndGrid , mode == VIEW_RESULT && g_gridView ? SW_SHOW : SW_HIDE );
220+ if (g_hwndSchema ) ShowWindow (g_hwndSchema , mode == VIEW_SCHEMA ? SW_SHOW : SW_HIDE );
221221
222222 /* Update context menu first, before toolbar changes */
223223 UpdateContextMenu (mode );
224224
225+ /* Update view toggle buttons */
226+ SendMessage (g_hwndCB , TB_CHECKBUTTON , IDM_VIEWQUERY , mode == VIEW_QUERY );
227+ SendMessage (g_hwndCB , TB_CHECKBUTTON , IDM_VIEWRESULT , mode == VIEW_RESULT );
228+ SendMessage (g_hwndCB , TB_CHECKBUTTON , IDM_VIEWSCHEMA , mode == VIEW_SCHEMA );
229+
225230 /* Swap button bitmap/state based on view mode */
226- if (mode == 0 ) {
231+ if (mode == VIEW_QUERY ) {
227232 /* Query view: exec-at-cursor toggle */
228233 SendMessage (g_hwndCB , TB_CHANGEBITMAP , IDM_EXECATCURSOR , TB_EXECAT );
229234 SendMessage (g_hwndCB , TB_ENABLEBUTTON , IDM_EXECATCURSOR , TRUE);
230235 SendMessage (g_hwndCB , TB_CHECKBUTTON , IDM_EXECATCURSOR , g_execAtCursor );
231- } else if (mode == 1 ) {
236+ } else if (mode == VIEW_RESULT ) {
232237 /* Results view: grid toggle - checked = grid mode */
233238 SendMessage (g_hwndCB , TB_CHANGEBITMAP , IDM_EXECATCURSOR , TB_GRID );
234239 SendMessage (g_hwndCB , TB_ENABLEBUTTON , IDM_EXECATCURSOR , g_lastResultRows > 0 );
@@ -240,14 +245,14 @@ void SwitchView(int mode) {
240245 SendMessage (g_hwndCB , TB_CHECKBUTTON , IDM_EXECATCURSOR , g_showSizes );
241246 }
242247
243- if (mode == 0 ) {
248+ if (mode == VIEW_QUERY ) {
244249 SetFocus (g_hwndQuery );
245250 UpdateLineCount ();
246251 UpdateLineNumbers ();
247- } else if (mode == 1 ) {
252+ } else if (mode == VIEW_RESULT ) {
248253 SetFocus (g_gridView ? g_hwndGrid : g_hwndResult );
249254 SendMessageW (g_hwndStatus , SB_SETTEXTW , 1 , (LPARAM )g_lastResultStatus );
250- } else if (mode == 2 ) {
255+ } else if (mode == VIEW_SCHEMA ) {
251256 wchar_t statusBuf [64 ];
252257 SetFocus (g_hwndSchema );
253258 RefreshSchema ();
0 commit comments