Category

Development Platform
Home > SourceCode/Document > Graph program > View Code
SplineEdit.cpp
Package: solidgraph_sources.zip　[view]
Upload User: kairuinn
Upload Date: 2009-02-07
Package Size: 2922k
Code Size: 11k
Category:

Graph program

Development Platform:

Visual C++

SplineEdit.cpp：Code Content
  1. #include "stdafx.h"
  3. #include "SplineEdit.h"
  5. #include "..//resource.h"
  6. #include <math.h>
  7. #include <float.h>
  10. SplineEditCommand::SplineEditCommand(sgCSpline* es, IApplicationInterface* appI):
  11. m_editable_spline(es)
  12. , m_clone_spline(NULL)
  13. , m_geo(NULL)
  14. , m_app(appI)
  15. ,m_sel_point_panel(NULL)
  16. , m_get_point_panel(NULL)
  17. , m_choise_pnt(-1)
  18. , m_step(0)
  19. , m_was_started(false)
  20. {
  21. ASSERT(es);
  22. ASSERT(m_app);
  24. SWITCH_RESOURCE
  25. m_bitmap = new CBitmap;
  26. m_bitmap->LoadBitmap(IDB_SPL_EDIT);
  27. }
  29. SplineEditCommand::~SplineEditCommand()
  30. {
  31. m_app->GetCommandPanel()->RemoveAllDialogs();
  32. m_app->GetViewPort()->InvalidateViewPort();
  33. if (m_bitmap)
  34. {
  35. if (m_bitmap->m_hObject)
  36. m_bitmap->DeleteObject();
  37. delete m_bitmap;
  38. }
  39. if (m_clone_spline)
  40. sgDeleteObject(m_clone_spline);
  41. /*if (m_geo)
  42. SG_SPLINE::DeleteSplineGeo(m_geo);*/
  43. }
  45. void     SplineEditCommand::SendCommanderMessage(ICommander::COMMANDER_MESSAGE mes, 
  46.    void* params) 
  47. {
  48. if (mes==ICommander::CM_SWITCH_ROLLUP_DIALOG)
  49. {
  50. ASSERT(params!=NULL);
  51. int   newActiveDlg = *(reinterpret_cast<int*>(params));
  53. ASSERT(newActiveDlg<=1);
  55. m_step = (unsigned int)newActiveDlg;
  56. if (newActiveDlg==0)
  57. {
  58. SWITCH_RESOURCE
  59. m_app->GetCommandPanel()->EnableRadio(1,false);
  60. m_message.LoadString(IDS_CHOISE_SPLINE_PNT);
  61. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  62. }
  63. m_app->GetViewPort()->InvalidateViewPort();
  64. return;
  65. }
  66. if (mes==ICommander::CM_UPDATE_COMMAND_PANEL)
  67. {
  68. ASSERT(params==NULL);
  69. if (m_step==0)
  70. {
  71. m_choise_pnt = m_sel_point_panel->GetCurrentPoint();
  72. m_app->GetViewPort()->InvalidateViewPort();
  73. }
  74. }
  75. }
  77. bool    SplineEditCommand::PreTranslateMessage(MSG* pMsg)
  78. {
  79. /*if (pMsg->message==WM_KEYUP||
  80. pMsg->message==WM_CHAR)
  81. return false;*/
  83. if (pMsg->message==WM_KEYUP||pMsg->message==WM_KEYDOWN || 
  84. pMsg->message==WM_CHAR)
  85. {
  86. if (pMsg->wParam==VK_RETURN)
  87. {
  88. OnEnter();
  89. return true;
  90. }
  91. if (pMsg->wParam==VK_ESCAPE)
  92. {
  93. m_app->StopCommander();
  94. return true;
  95. }
  96. switch(m_step) 
  97. {
  98. case 0:
  99. if (m_sel_point_panel)
  100. {
  101. m_sel_point_panel->GetWindow()->SendMessage(pMsg->message,
  102. pMsg->wParam,
  103. pMsg->lParam);
  104. }
  105. break;
  106. case 1:
  107. if (m_get_point_panel)
  108. {
  109. m_get_point_panel->GetWindow()->SendMessage(pMsg->message,
  110. pMsg->wParam,
  111. pMsg->lParam);
  112. }
  113. break;
  114. default:
  115. break;
  116. }
  117. if (pMsg->message==WM_KEYDOWN)
  118. return false;
  119. else 
  120. return true;
  121. }
  122. else
  123. {
  124. if (pMsg->hwnd == m_app->GetViewPort()->GetWindow()->m_hWnd)
  125. {
  126. switch(pMsg->message) 
  127. {
  128. case WM_MOUSEMOVE:
  129. MouseMove(pMsg->wParam,GET_X_LPARAM(pMsg->lParam),GET_Y_LPARAM(pMsg->lParam));
  130. return true;
  131. case WM_LBUTTONUP:
  132. LeftClick(pMsg->wParam,GET_X_LPARAM(pMsg->lParam),GET_Y_LPARAM(pMsg->lParam));
  133. return true;
  134. default:
  135. return false;
  136. }
  137. }
  138. }
  139. return false;
  140. }
  142. void  SplineEditCommand::Start()
  143. {
  144. ASSERT(m_editable_spline);
  146. SWITCH_RESOURCE
  147. CString lab;
  148. lab.LoadString(IDS_REDACTION);
  149. m_app->StartCommander(lab);
  151. if (m_clone_spline)
  152. sgDeleteObject(m_clone_spline);
  154. m_clone_spline = reinterpret_cast<sgCSpline*>(m_editable_spline->Clone());
  155. m_geo = const_cast<SG_SPLINE*>(m_clone_spline->GetGeometry());
  157. NewPanels();
  158. m_app->GetViewPort()->SetEditableObject(m_editable_spline);
  159. m_was_started = true;
  160. }
  162. void   SplineEditCommand::NewPanels()
  163. {
  164. SWITCH_RESOURCE
  166. CString lab;
  168. lab.LoadString(IDS_CHOISE_SPLINE_PNT);
  169. m_app->GetCommandPanel()->RemoveAllDialogs();
  170. m_sel_point_panel = 
  171. reinterpret_cast<ISelectPointPanel*>(m_app->GetCommandPanel()->
  172. AddDialog(IBaseInterfaceOfGetDialogs::SELECT_POINT_DLG,
  173. lab,true));
  174. if (m_geo)
  175. {
  176. int ssz = (m_geo->IsClosed())?(m_geo->GetKnotsCount()-1):(m_geo->GetKnotsCount());
  177. for (int i=0;i<ssz;i++)
  178. m_sel_point_panel->AddPoint(m_geo->GetKnots()[i].x,
  179. m_geo->GetKnots()[i].y,
  180. m_geo->GetKnots()[i].z);
  181. }
  183. lab.LoadString(IDS_NEW_COORDINATES);
  185. m_get_point_panel = 
  186. reinterpret_cast<IGetPointPanel*>(m_app->GetCommandPanel()->
  187. AddDialog(IBaseInterfaceOfGetDialogs::GET_POINT_DLG,
  188. lab,true));
  190. m_app->GetCommandPanel()->SetActiveRadio(0);
  191. m_step=0;
  192. m_choise_pnt =0;
  193. }
  195. void  SplineEditCommand::MouseMove(unsigned int nFlags,int pX,int pY)
  196. {
  197. if (!m_was_started)
  198. return;
  200. SWITCH_RESOURCE
  201. if (m_step==0)
  202. {
  203. ASSERT(m_sel_point_panel);
  204. ASSERT(m_editable_spline);
  205. ASSERT(m_geo);
  206. double    winX1,winY1,winZ1;
  207. double    minDist = FLT_MAX;
  208. m_choise_pnt = -1;
  209. for (int i=0;i<m_geo->GetKnotsCount();i++)
  210. {
  211. m_app->GetViewPort()->ProjectWorldPoint(m_geo->GetKnots()[i],winX1,winY1,winZ1);
  212. double d1;
  213. d1 = sqrt((winX1-pX)*(winX1-pX)+(winY1-pY)*(winY1-pY));
  214. if (d1<minDist)
  215. {
  216. minDist = d1;
  217. m_choise_pnt = i;
  218. }
  219. }
  220. if (m_choise_pnt>=0)
  221. {
  222. if (m_sel_point_panel)
  223. m_sel_point_panel->SetCurrentPoint(m_choise_pnt);
  224. }
  225. }
  226. else
  227. {
  228. if (m_choise_pnt<0)
  229. {
  230. ASSERT(0);
  231. return;
  232. }
  233. ASSERT(m_get_point_panel);
  234. IViewPort::GET_SNAP_IN in_arg;
  235. in_arg.scrX = pX;
  236. in_arg.scrY = pY;
  237. in_arg.snapType = SNAP_SYSTEM;
  238. in_arg.XFix = m_get_point_panel->IsXFixed();
  239. in_arg.YFix = m_get_point_panel->IsYFixed();
  240. in_arg.ZFix = m_get_point_panel->IsZFixed();
  241. double tmpFl[3];
  242. m_get_point_panel->GetPoint(tmpFl[0],tmpFl[1],tmpFl[2]);
  243. in_arg.FixPoint.x = tmpFl[0];
  244. in_arg.FixPoint.y = tmpFl[1];
  245. in_arg.FixPoint.z = tmpFl[2];
  247. IViewPort::GET_SNAP_OUT out_arg;
  248. m_app->GetViewPort()->GetWorldPointAfterSnap(in_arg,out_arg);
  249. m_cur_pnt = out_arg.result_point;
  250. m_get_point_panel->SetPoint(m_cur_pnt.x,m_cur_pnt.y,m_cur_pnt.z);
  251. m_geo->MoveKnot(m_choise_pnt,m_cur_pnt);
  252. }
  253. m_app->GetViewPort()->InvalidateViewPort();
  255. }
  257. void  SplineEditCommand::LeftClick(unsigned int nFlags,int pX,int pY)
  258. {
  259. if (!m_was_started)
  260. return;
  262. SWITCH_RESOURCE
  264. if (m_step==0)
  265. {
  266. if (m_choise_pnt>=0)
  267. {
  268. m_step++;
  269. m_app->GetCommandPanel()->SetActiveRadio(m_step);
  270. }
  271. else
  272. {
  273. ASSERT(0);
  274. }
  275. }
  276. else
  277. {
  278. if (m_choise_pnt<0)
  279. {
  280. ASSERT(0);
  281. return;
  282. }
  283. CString mes;
  285. if ((m_choise_pnt>0 && sgSpaceMath::PointsDistance((m_geo->GetKnots()[m_choise_pnt]),
  286. (m_geo->GetKnots()[m_choise_pnt-1]))<0.001) ||
  287. (m_choise_pnt<m_geo->GetKnotsCount()-1 && sgSpaceMath::PointsDistance((m_geo->GetKnots()[m_choise_pnt]),
  288. (m_geo->GetKnots()[m_choise_pnt+1]))<0.001))
  289. {
  290. m_message.LoadString(IDS_ERROR_SPL_KNOT_AS_PREV);
  291. m_app->PutMessage(IApplicationInterface::MT_ERROR,
  292. m_message);
  293. return;
  294. }
  295. sgCSpline* spl = sgCreateSpline(*m_geo);
  296. if (!spl)
  297. {
  298. if (m_clone_spline)
  299. {
  300. sgDeleteObject(m_clone_spline);
  301. m_clone_spline = NULL;
  302. }
  303. //m_geo = SG_SPLINE::CreateSplineGeo();
  304. m_message.LoadString(IDS_CREARE_SPL_ERROR);
  305. m_app->PutMessage(IApplicationInterface::MT_ERROR,
  306. m_message);
  307. m_app->GetViewPort()->InvalidateViewPort();
  308. m_app->StopCommander();
  309. return;
  310. }
  311. sgGetScene()->StartUndoGroup();
  312. sgGetScene()->DetachObject(m_editable_spline);
  313. sgGetScene()->AttachObject(spl);
  314. sgGetScene()->EndUndoGroup();
  315. m_app->CopyAttributes(*spl,*m_editable_spline);
  316. m_app->GetViewPort()->InvalidateViewPort();
  317. m_app->StopCommander();
  318. }
  320. }
  323. void  SplineEditCommand::Draw()
  324. {
  326. if (!m_was_started)
  327. return;
  329. if (!m_geo)
  330. return;
  332. float pC[3];
  333. if (m_choise_pnt>=0)
  334. {
  335. m_app->GetViewPort()->GetPainter()->GetUserColorPoints(pC[0],pC[1],pC[2]);
  336. m_app->GetViewPort()->GetPainter()->SetCurColor(pC[0],pC[1],pC[2]);
  337. m_app->GetViewPort()->GetPainter()->DrawPoint(m_geo->GetKnots()[m_choise_pnt]);
  338. }
  340. m_app->GetViewPort()->GetPainter()->GetUserColorLines(pC[0],pC[1],pC[2]);
  341. m_app->GetViewPort()->GetPainter()->SetCurColor(pC[0],pC[1],pC[2]);
  342. m_app->GetViewPort()->GetPainter()->DrawSplineFrame(m_geo);
  343. m_app->GetViewPort()->GetPainter()->SetCurColor(1.0,0.0,0.0);
  344. m_app->GetViewPort()->GetPainter()->DrawSpline(m_geo);
  345. }
  348. void  SplineEditCommand::OnEnter()
  349. {
  350. if (!m_was_started)
  351. return;
  352. SWITCH_RESOURCE
  354. if (m_step==0)
  355. {
  356. if (m_choise_pnt>=0)
  357. {
  358. m_step++;
  359. m_app->GetCommandPanel()->SetActiveRadio(m_step);
  360. }
  361. else
  362. {
  363. ASSERT(0);
  364. }
  365. }
  366. else
  367. {
  368. ASSERT(m_get_point_panel);
  370. m_get_point_panel->GetPoint(m_cur_pnt.x,m_cur_pnt.y, m_cur_pnt.z);
  372. if (m_choise_pnt<0)
  373. {
  374. ASSERT(0);
  375. return;
  376. }
  378. m_geo->MoveKnot(m_choise_pnt,m_cur_pnt);
  379. CString mes;
  381. if ((m_choise_pnt>0 && sgSpaceMath::PointsDistance((m_geo->GetKnots()[m_choise_pnt]),
  382. (m_geo->GetKnots()[m_choise_pnt-1]))<0.001) ||
  383. (m_choise_pnt<m_geo->GetKnotsCount()-1 && sgSpaceMath::PointsDistance((m_geo->GetKnots()[m_choise_pnt]),
  384. (m_geo->GetKnots()[m_choise_pnt+1]))<0.001))
  385. {
  386. m_message.LoadString(IDS_ERROR_SPL_KNOT_AS_PREV);
  387. m_app->PutMessage(IApplicationInterface::MT_ERROR,
  388. m_message);
  389. return;
  390. }
  392. sgCSpline* spl = sgCreateSpline(*m_geo);
  393. if (!spl)
  394. {
  395. if (m_clone_spline)
  396. {
  397. sgDeleteObject(m_clone_spline);
  398. m_clone_spline = NULL;
  399. }
  400. //m_geo = SG_SPLINE::CreateSplineGeo();
  401. m_message.LoadString(IDS_CREARE_SPL_ERROR);
  402. m_app->PutMessage(IApplicationInterface::MT_ERROR,
  403. m_message);
  404. m_app->GetViewPort()->InvalidateViewPort();
  405. m_app->StopCommander();
  406. return;
  407. }
  408. sgGetScene()->StartUndoGroup();
  409. sgGetScene()->DetachObject(m_editable_spline);
  410. sgGetScene()->AttachObject(spl);
  411. sgGetScene()->EndUndoGroup();
  412. m_app->CopyAttributes(*spl,*m_editable_spline);
  413. m_app->GetViewPort()->InvalidateViewPort();
  414. m_app->StopCommander();
  415. }
  416. }
  419. unsigned int  SplineEditCommand::GetItemsCount()
  420. {
  421. if (!m_was_started)
  422. return 1;
  423. else
  424. return 0;
  425. }
  427. void         SplineEditCommand::GetItem(unsigned int itemID, CString& itSrt)
  428. {
  429. if (m_was_started)
  430. {
  431. ASSERT(0);
  432. return;
  433. }
  435. SWITCH_RESOURCE
  436. if(itemID==0) 
  437. {
  438. itSrt.LoadString(IDS_MOVE_SPLINE_KNOT);
  439. }
  440. else
  441. ASSERT(0);
  442. }
  445. void     SplineEditCommand::GetItemState(unsigned int itemID, bool& enbl, bool& checked)
  446. {
  447. if (m_was_started)
  448. {
  449. ASSERT(0);
  450. return;
  451. }
  453. enbl = true;
  454. checked = false;
  457. }
  459. HBITMAP   SplineEditCommand::GetItemBitmap(unsigned int)
  460. {
  461. /*if (!m_was_started)
  462. return (HBITMAP)m_bitmap;
  463. else*/
  464. return NULL;
  465. }
  467. void         SplineEditCommand::Run(unsigned int itemID)
  468. {
  469. if (!m_was_started)
  470. {
  471. if (itemID==0 || itemID==1)
  472. {
  473. //m_scenario = itemID;
  474. Start();
  475. }
  476. else
  477. {
  478. ASSERT(0);
  479. }
  480. }
  481. /*else
  482. m_scenario = itemID;*/
  483. }