// LeftView.cpp : implementation of the CLeftView class
//

#include "stdafx.h"
#include "DataManager.h"

#include "DataManagerDoc.h"
#include "LeftView.h"
#include "SegmentDialog.h"
#include "MainFrm.h"
#include "DataManagerView.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CLeftView

IMPLEMENT_DYNCREATE(CLeftView, CTreeView)

BEGIN_MESSAGE_MAP(CLeftView, CTreeView)
	//{{AFX_MSG_MAP(CLeftView)
	ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelchanged)
	ON_COMMAND(ID_ADDSEGMENT, OnAddsegment)
	ON_COMMAND(ID_DELSEGMENT, OnDelsegment)
	ON_WM_RBUTTONDOWN()
	ON_COMMAND(ID_REFRESH, OnRefresh)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CLeftView construction/destruction

CLeftView::CLeftView()
{
	// TODO: add construction code here

}

CLeftView::~CLeftView()
{
}

BOOL CLeftView::PreCreateWindow(CREATESTRUCT& cs)
{
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs

	return CTreeView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CLeftView drawing

void CLeftView::OnDraw(CDC* pDC)
{
	CDataManagerDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);

	// TODO: add draw code for native data here
}


void CLeftView::OnInitialUpdate()
{
	CTreeView::OnInitialUpdate();

	// TODO: You may populate your TreeView with items by directly accessing
	//  its tree control through a call to GetTreeCtrl().
	CTreeCtrl& m_tree=GetTreeCtrl();
	if(p_ControlModule==NULL)
		return;
	HTREEITEM hItem;
	for(int i=0;i<p_ControlModule->GetSegmentCount();i++)
	{
		CControlSegment *p_Segment=p_ControlModule->GetSegment(i); 
		hItem=m_tree.InsertItem(p_Segment->GetSegmentName(),
			0,0,TVI_ROOT,TVI_LAST);
		m_tree.SetItemData(hItem,0);
		p_Segment->hItem=hItem; 

		for (int j=0;j<p_Segment->GetUnitCount();j++)
		{
			CControlUnit *m_Unit;
			m_Unit=p_Segment->GetUnit(j);
			CString str1;
			str1=m_Unit->GetUnitName();

			TVINSERTSTRUCT tvItem;
			tvItem.hParent=hItem;
			tvItem.hInsertAfter=TVI_LAST;
			tvItem.item.mask=TVIF_TEXT|TVIF_IMAGE|TVIF_SELECTEDIMAGE;
			tvItem.item.pszText=(char *)(LPCTSTR)str1;
			tvItem.item.iImage=1;
			tvItem.item.iSelectedImage=1;

			HTREEITEM hItem1;
			hItem1=m_tree.InsertItem(&tvItem); 
			m_Unit->hItem=hItem1; 
			if(hItem1!=NULL)
			{
				m_tree.SetItemData(hItem1,1);
			}
		}
	}
}

/////////////////////////////////////////////////////////////////////////////
// CLeftView diagnostics

#ifdef _DEBUG
void CLeftView::AssertValid() const
{
	CTreeView::AssertValid();
}

void CLeftView::Dump(CDumpContext& dc) const
{
	CTreeView::Dump(dc);
}

CDataManagerDoc* CLeftView::GetDocument() // non-debug version is inline
{
	ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CDataManagerDoc)));
	return (CDataManagerDoc*)m_pDocument;
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CLeftView message handlers

BOOL CLeftView::Create(LPCTSTR lpszClassName, LPCTSTR lpszWindowName, DWORD dwStyle, const RECT& rect, CWnd* pParentWnd, UINT nID, CCreateContext* pContext) 
{
	// TODO: Add your specialized code here and/or call the base class
	dwStyle|=TVS_SHOWSELALWAYS;
	dwStyle|=TVS_HASBUTTONS ;
	dwStyle|=TVS_HASLINES ;
	dwStyle|=TVS_LINESATROOT ;
	dwStyle|=TVS_EDITLABELS ;
	return CWnd::Create(lpszClassName, lpszWindowName, dwStyle, rect, pParentWnd, nID, pContext);
}

void CLeftView::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
	// TODO: Add your control notification handler code here

	GetDocument()->UpdateAllViews(this,0,this);

	*pResult = 0;
}

void CLeftView::OnAddsegment() 
{
	// TODO: Add your command handler code here
	CSegmentDialog dlg_Segment;
	if(dlg_Segment.DoModal()==IDOK)
	{
		p_ControlModule->AddSegment(dlg_Segment.m_SegmentName);
	}
}

void CLeftView::OnDelsegment() 
{
	// TODO: Add your command handler code here
	if(AfxMessageBox("刪除此段嗎?",MB_YESNO)==6)
	{
		CTreeCtrl& m_tree=GetTreeCtrl();
		HTREEITEM hItem;	
		hItem=m_tree.GetSelectedItem();
		if (hItem==NULL)
			return;
		
		p_ControlModule->DeleteSegment(
			p_ControlModule->GetSegment(m_tree.GetItemText(hItem))); 
	}
}

void CLeftView::OnRButtonDown(UINT nFlags, CPoint point) 
{
	// TODO: Add your message handler code here and/or call default
	CTreeView::OnRButtonDown(nFlags, point);

	CDataManagerDoc* m_pDoc=(CDataManagerDoc*)((CMainFrame*)
		AfxGetApp()->m_pMainWnd)->GetActiveDocument();
	POSITION pos=m_pDoc->GetFirstViewPosition();
	
	CLeftView* p_TreeView;
	while (pos != NULL) 
	{ 
		CView* pView = m_pDoc->GetNextView(pos); 
		if(pView->IsKindOf(RUNTIME_CLASS(CLeftView)))
		{
			p_TreeView=(CLeftView*)pView;
		}
	}
	CTreeCtrl& m_tree=p_TreeView->GetTreeCtrl();
	
	HTREEITEM hItem=m_tree.GetSelectedItem();
	if(hItem==NULL)
		return;

	int nLevel=m_tree.GetItemData(hItem);  
	if(nLevel==0)
	{
		CMenu PopMenu; 
		CMenu *pMenu; 
		CPoint pt; 
		pt= point; 
     
		PopMenu.LoadMenu(IDR_EDIT); 
		ClientToScreen(&pt); 
		pMenu = PopMenu.GetSubMenu(0); 
		pMenu->TrackPopupMenu(TPM_LEFTALIGN|TPM_RIGHTBUTTON,
			pt.x,pt.y,this); 
	}
}

void CLeftView::OnRefresh() 
{
	// TODO: Add your command handler code here
	CControlSegment		*lpTempSegment;
	CControlUnit		*lpTempUnit;
	CString				TempSegmentName , TempUnitName ,strTemp;

	CTreeCtrl& m_tree=GetTreeCtrl();
	HTREEITEM hItem;	
	hItem=m_tree.GetSelectedItem();
	if (hItem==NULL)
		return;
		
	int nLevel=m_tree.GetItemData(hItem);  
	if(nLevel==0)
	{
		strTemp=m_tree.GetItemText(hItem);
		lpTempSegment=p_ControlModule->GetSegment(strTemp); 
		if(lpTempSegment==NULL)
			return;

		if(!lpTempSegment->m_SegmentSocket->m_bNetConnected)
			return;

		LPMODFRAME pFrame;
		pFrame=new ModFrame;
		pFrame->Content.Format("%s;",strTemp);
		pFrame->nOrder = 5;
		if (lpTempSegment->m_SegmentSocket!=NULL)
			lpTempSegment->m_SegmentSocket->SendFrame(pFrame);
	}
	else if(nLevel==1)
	{
		TempSegmentName=m_tree.GetItemText(m_tree.GetParentItem(hItem));
		lpTempSegment=p_ControlModule->GetSegment(TempSegmentName); 
		if (lpTempSegment==NULL)
			return;
		
		if(!lpTempSegment->m_SegmentSocket->m_bNetConnected)
			return;

		TempUnitName=m_tree.GetItemText(hItem);
		lpTempUnit=lpTempSegment->GetUnit(TempUnitName);
		if(lpTempUnit==NULL)
			return;

		LPMODFRAME pFrame;
		pFrame=new ModFrame;
		pFrame->Content.Format("%s;",lpTempUnit->GetUnitLongName());
		pFrame->nOrder = 7;
		lpTempSegment->m_SegmentSocket->SendFrame(pFrame);
	}
}