nel/tools/logic/logic_editor_dll/StatesView.cpp

   1 // StatesView.cpp : implementation file
   2 //
   3
   4 #include "stdafx.h"
   5 #include "logic_editor.h"
   6 #include "StatesView.h"
   7
   8
   9 #include "Logic_editorDoc.h"
  10 #include "MainFrm.h"
  11 #include "ChildFrm.h"
  12 #include "EditorFormView.h"
  13
  14 #ifdef _DEBUG
  15 #define new DEBUG_NEW
  16 #undef THIS_FILE
  17 static char THIS_FILE[] = __FILE__;
  18 #endif
  19
  20 /////////////////////////////////////////////////////////////////////////////
  21 // CStatesView
  22
  23 IMPLEMENT_DYNCREATE(CStatesView, CTreeView)
  24
  25 CStatesView::CStatesView()
  26 {
  27 }
  28
  29 CStatesView::~CStatesView()
  30 {
  31 }
  32
  33
  34 BEGIN_MESSAGE_MAP(CStatesView, CTreeView)
  35         //{{AFX_MSG_MAP(CStatesView)
  36         ON_WM_KILLFOCUS()
  37         ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelchanged)
  38         //}}AFX_MSG_MAP
  39 END_MESSAGE_MAP()
  40
  41 /////////////////////////////////////////////////////////////////////////////
  42 // CStatesView drawing
  43
  44 void CStatesView::OnDraw(CDC* pDC)
  45 {
  46         CDocument* pDoc = GetDocument();
  47         // TODO: add draw code here
  48 }
  49
  50 /////////////////////////////////////////////////////////////////////////////
  51 // CStatesView diagnostics
  52
  53 #ifdef _DEBUG
  54 void CStatesView::AssertValid() const
  55 {
  56         CTreeView::AssertValid();
  57 }
  58
  59 void CStatesView::Dump(CDumpContext& dc) const
  60 {
  61         CTreeView::Dump(dc);
  62 }
  63 #endif //_DEBUG
  64
  65
  66
  67
  68
  69
  70
  71
  72 void CStatesView::expand(UINT nCode, HTREEITEM hItem)
  73 {
  74         HTREEITEM hElement;
  75
  76         HTreeItemArray pile;
  77         CTreeCtrl &m_tree = GetTreeCtrl();
  78
  79         if (hItem == NULL)
  80                 hElement = m_tree.GetRootItem();
  81         else
  82         {
  83                 m_tree.Expand(hItem,nCode);
  84                 hElement = m_tree.GetChildItem(hItem);
  85         }
  86
  87         pile.RemoveAll();
  88
  89         while ((hElement != NULL) || (pile.GetSize() != 0))
  90         {
  91                 while ((hElement == NULL) && (pile.GetSize()!=0))
  92                 {
  93                         hElement = pile[pile.GetSize()-1];
  94                         pile.RemoveAt(pile.GetSize()-1);
  95                         hElement = m_tree.GetNextSiblingItem(hElement);
  96                 }
  97                 if (hElement)
  98                 {
  99                         pile.Add(hElement);
 100                         m_tree.Expand(hElement,nCode);
 101                         hElement = m_tree.GetChildItem(hElement);
 102                 }
 103         }
 104 }
 105
 106
 107
 108
 109
 110
 111 /////////////////////////////////////////////////////////////////////////////
 112 // CStatesView message handlers
 113
 114 void CStatesView::OnUpdate(CView* pSender, LPARAM lHint, CObject* pHint)
 115 {
 116         /************* TO DO : REDRAW ONLY IF Some states have changed *****/
 117
 118         SetRedraw(FALSE);
 119
 120         CTreeView::OnUpdate(pSender,lHint,pHint);
 121
 122
 123         BOOL bSelection = FALSE;
 124
 125
 126         CLogic_editorDoc *pDoc =  static_cast<CLogic_editorDoc*>(GetDocument());
 127         ASSERT_VALID(pDoc);
 128
 129         CTreeCtrl &m_tree = GetTreeCtrl();
 130
 131         m_tree.DeleteAllItems();
 132         m_mapItemToEvent.RemoveAll();
 133
 134         m_tree.InsertItem(_T("States"));
 135
 136         // get the states map
 137         CState *pState;
 138         CString stateName;
 139         POSITION pos = pDoc->m_states.GetStartPosition();
 140         HTREEITEM hItem;
 141
 142         while (pos != NULL)
 143         {
 144                 pDoc->m_states.GetNextAssoc( pos, stateName, (void*&)pState );
 145                 hItem = m_tree.InsertItem( stateName , m_tree.GetRootItem());
 146
 147                 // if this is the selected state (and there was no selected event), select it)
 148                 if ( m_pSelectedEvent == NULL && m_pSelectedState == pState )
 149                 {
 150                         m_tree.SelectItem( hItem);
 151                         bSelection = TRUE;
 152                 }
 153
 154                 // insert every event as children nodes
 155                 CEvent *pEvent = NULL;
 156                 HTREEITEM hItemCurrent;
 157
 158                 POSITION eventPos = pState->m_evEvents.GetHeadPosition();
 159                 while (eventPos != NULL)
 160                 {
 161                         pEvent = pState->m_evEvents.GetNext( eventPos );
 162                         if (pEvent != NULL)
 163                         {
 164                                 hItemCurrent = m_tree.InsertItem( pEvent->getEventAsString() , hItem);
 165                                 m_mapItemToEvent.SetAt( hItemCurrent, pEvent );
 166                                 // if this is the selected event, select it
 167                                 if ( m_pSelectedEvent == pEvent )
 168                                 {
 169                                         m_tree.SelectItem( hItemCurrent);
 170                                         bSelection = TRUE;
 171                                 }
 172                         }
 173                 }
 174         }
 175
 176         // expand all tree
 177         expand(/*TVE_EXPAND ,NULL*/ );
 178
 179         //
 180         if (!bSelection)
 181         {
 182                 m_pSelectedEvent = NULL;
 183                 m_pSelectedState = NULL;
 184         }
 185
 186         SetRedraw(/*TRUE*/);
 187 }
 188
 189 BOOL CStatesView::PreCreateWindow(CREATESTRUCT& cs)
 190 {
 191         // modify arborescence style
 192         cs.style |=(TVS_HASLINES|TVS_LINESATROOT|TVS_HASBUTTONS/*|TVS_EDITLABELS*/);
 193 //      cs.style &= ~TVS_DISABLEDRAGDROP;
 194
 195         return CTreeView::PreCreateWindow(cs);
 196 }
 197
 198 void CStatesView::OnKillFocus(CWnd* pNewWnd)
 199 {
 200 //      CTreeView::OnKillFocus(pNewWnd);
 201 }
 202
 203 void CStatesView::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult)
 204 {
 205         NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
 206
 207
 208         m_pSelectedEvent = NULL;
 209         m_pSelectedState = NULL;
 210
 211
 212         //
 213         CTreeCtrl &treeCtrl = GetTreeCtrl();
 214
 215         HTREEITEM hItem;
 216         hItem = treeCtrl.GetSelectedItem();
 217
 218         //
 219         if (hItem == treeCtrl.GetRootItem())
 220                 return;
 221
 222         // get the event selected (if any)
 223         if ( ! m_mapItemToEvent.Lookup(hItem, m_pSelectedEvent) )
 224                 m_pSelectedEvent = NULL;
 225         else
 226         {
 227                 // nothing special
 228         }
 229
 230         // get the state (if any)
 231         while ( treeCtrl.GetParentItem( hItem ) != treeCtrl.GetRootItem() )
 232                 hItem = treeCtrl.GetParentItem( hItem );
 233
 234         CString stateName = treeCtrl.GetItemText( hItem );
 235
 236         CLogic_editorDoc *pDoc = static_cast<CLogic_editorDoc *> (GetDocument());
 237
 238         //  to update the fields of the condition page in the form view
 239         CMainFrame *pFrame = (CMainFrame*)AfxGetApp()->m_pMainWnd;
 240
 241         // Get the active MDI child window.
 242         CChildFrame *pChild = (CChildFrame *) pFrame->GetActiveFrame();
 243
 244         CEditorFormView *pFormView = static_cast<CEditorFormView *> ( pChild->m_wndSplitter.GetPane(0,1) );
 245         CStatePage *pStatePage = static_cast<CStatePage*> ( pFormView->m_pPropertySheet->GetPage(3) );
 246
 247         if ( pDoc->m_states.Lookup( stateName, (void*&) m_pSelectedState) )
 248         {
 249                 pFormView->m_pPropertySheet->SetActivePage( pStatePage );
 250
 251                 if ( this->m_pSelectedEvent == NULL || pStatePage->m_pSelectedEvent != this->m_pSelectedEvent)
 252                 {
 253                         pStatePage->m_pSelectedEvent = this->m_pSelectedEvent;
 254                         pStatePage->m_pSelectedState = this->m_pSelectedState;
 255
 256                         pStatePage->Update();
 257                 }
 258         }
 259         else //this->m_pSelectedState == NULL
 260         {
 261                 if (pStatePage->m_pSelectedState != NULL)
 262                 {
 263                         pStatePage->m_pSelectedEvent = this->m_pSelectedEvent; // = NULL
 264                         pStatePage->m_pSelectedState = this->m_pSelectedState; // = NULL
 265
 266                         pStatePage->Update();
 267                 }
 268         }
 269         //
 270         *pResult = 0;
 271 }
 272
 273 void CStatesView::OnInitialUpdate()
 274 {
 275         CTreeView::OnInitialUpdate();
 276
 277         m_pSelectedEvent = NULL;
 278         m_pSelectedState = NULL;
 279 }