Left side tree navigation updates - still need sorting.

NREL · Oct 11, 2023 · 7e5f7ec · 7e5f7ec
1 parent bab9710
commit 7e5f7ec
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 17 deletions.
diff --git a/src/casewin.cpp b/src/casewin.cpp
@@ -145,9 +145,11 @@ BEGIN_EVENT_TABLE( CaseWindow, wxSplitterWindow )
 	EVT_MENU(ID_PVUNCERTAINTY, CaseWindow::OnCommand)
 	EVT_MENU( ID_MACRO, CaseWindow::OnCommand )
 	EVT_LISTBOX( ID_INPUTPAGELIST, CaseWindow::OnCommand )
-    EVT_DATAVIEW_SELECTION_CHANGED(ID_TechTree, CaseWindow::OnTechTree)
-    EVT_DATAVIEW_ITEM_START_EDITING(ID_TechTree, CaseWindow::OnTreeActivated)
-    EVT_DATAVIEW_ITEM_ACTIVATED(ID_TechTree, CaseWindow::OnTreeActivated)
+    EVT_DATAVIEW_SELECTION_CHANGED(ID_TechTree, CaseWindow::OnTree)
+	EVT_DATAVIEW_ITEM_COLLAPSING(ID_TechTree, CaseWindow::OnTreeCollapsing)
+
+//    EVT_DATAVIEW_ITEM_START_EDITING(ID_TechTree, CaseWindow::OnTreeActivated)
+//    EVT_DATAVIEW_ITEM_ACTIVATED(ID_TechTree, CaseWindow::OnTreeActivated)
     //EVT_LISTBOX( ID_TechTree, CaseWindow::OnCommand)
 	EVT_BUTTON( ID_EXCL_BUTTON, CaseWindow::OnCommand )
     EVT_LISTBOX( ID_EXCL_RADIO, CaseWindow::OnCommand)
@@ -626,8 +628,37 @@ bool CaseWindow::GenerateReport( wxString pdffile, wxString templfile, VarValue
 	return false;
 }
 
-void CaseWindow::OnTechTree(wxDataViewEvent&)
+void CaseWindow::OnTree(wxDataViewEvent &evt)
 {
+	m_pageFlipper->SetSelection(0);
+	wxDataViewItem dvi = evt.GetItem();
+	if (!dvi.IsOk())
+		return;
+
+	if (m_navigationMenu->IsContainer(dvi))
+	{
+		if (m_navigationMenu->IsExpanded(dvi))
+			m_navigationMenu->Collapse(dvi);
+		else
+			m_navigationMenu->Expand(dvi);
+		if (m_currentSelection.IsOk()) {// keep current selection 
+			m_navigationMenu->SetCurrentItem(m_currentSelection);
+			return;
+		}
+		else {// select first child
+			m_currentSelection = m_navigationMenu->GetNthChild(dvi, 0);
+			m_navigationMenu->SetCurrentItem(m_currentSelection);
+		}
+	}
+	else {
+		m_currentSelection = evt.GetItem();
+	}
+	wxString title = m_navigationMenu->GetItemText(m_currentSelection);
+	SwitchToInputPage(title);
+
+
+
+	/*
     m_pageFlipper->SetSelection(0);
     if (m_navigationMenu->IsContainer(m_navigationMenu->GetCurrentItem()))
     {
@@ -674,11 +705,6 @@ void CaseWindow::OnTechTree(wxDataViewEvent&)
         //m_navigationMenu->Update();
         //wxDataViewItemArray dvia;
 
-        /*
-        m_navigationMenu->GetModel()->GetChildren(m_navigationMenu->GetCurrentItem(), dvia);
-        if (m_navigationMenu->GetItemText(dvia[0]) != L"")
-            SwitchToInputPage(m_navigationMenu->GetItemText(dvia[0]));
-            */
         
     }
     else {
@@ -693,13 +719,21 @@ void CaseWindow::OnTechTree(wxDataViewEvent&)
         
     }
     //m_navigationMenu->Update();
-
+    */
 }
 
-void CaseWindow::OnTreeActivated(wxDataViewEvent& evt)
+void CaseWindow::OnTreeCollapsing(wxDataViewEvent& evt)
 {
-	wxString here = "yes";
-    evt.Veto();
+	wxDataViewItem dvi = evt.GetItem();
+	if (dvi.IsOk() && m_navigationMenu->IsContainer(dvi)) {
+		auto selectedDVI = m_navigationMenu->GetCurrentItem();
+		if (selectedDVI.IsOk()) {
+			for (size_t i = 0; i < m_navigationMenu->GetChildCount(dvi); i++) {
+				if (selectedDVI == m_navigationMenu->GetNthChild(dvi, i))
+					evt.Veto();
+			}
+		}
+	}
 }
 
 void CaseWindow::OnCommand( wxCommandEvent &evt )
@@ -1482,7 +1516,7 @@ void CaseWindow::UpdateConfiguration()
 		m_navigationMenu->SetCurrentItem(dvic[0]);
 		SwitchToInputPage(m_navigationMenu->GetItemText(m_navigationMenu->GetCurrentItem()));
 	}
-	m_previousPage = (m_navigationMenu->GetCurrentItem());
+	m_currentSelection = (m_navigationMenu->GetCurrentItem());
 
 	// check for orphaned notes and if any found add to first page per Github issue 796
 	CheckAndUpdateNotes(inputPageHelpContext);

diff --git a/src/casewin.h b/src/casewin.h
@@ -151,7 +151,7 @@ class CaseWindow : public wxSplitterWindow, CaseEventListener
     wxStaticText* m_finLabel;
 	wxMetroButton *m_simButton, *m_resultsButton;
 	wxMetroDataViewTreeCtrl *m_navigationMenu;
-    wxDataViewItem m_previousPage;
+    wxDataViewItem m_currentSelection;
 
     // to allow switching case configurations with P50/P90 and PVUncertainty
     wxGridSizer *m_szsims;
@@ -169,8 +169,8 @@ class CaseWindow : public wxSplitterWindow, CaseEventListener
 	wxString m_lastPageNoteId;
 
 	void OnCommand( wxCommandEvent & );
-    void OnTechTree(wxDataViewEvent&);
-    void OnTreeActivated(wxDataViewEvent &evt );
+    void OnTree(wxDataViewEvent&);
+    void OnTreeCollapsing(wxDataViewEvent &evt );
 	virtual void OnCaseEvent( Case *, CaseEvent & );
 	void OnSubNotebookPageChanged( wxNotebookEvent &evt );