Category

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

Graph program

Development Platform:

Visual C++

SplineSurf.cpp：Code Content
  1. #include "stdafx.h"
  3. #include "SplineSurf.h"
  6. #include <math.h>
  7. #include <float.h>
  8. #include "..//resource.h"
  11. int     spline_surface_index = 1;
  13. SplineSurfaceCommand::SplineSurfaceCommand(IApplicationInterface* appI):
  14. m_app(appI)
  15. {
  16. ASSERT(m_app);
  17. m_step = 0;
  18. m_cur_obj = NULL;
  20. m_get_object_panel = NULL;
  22. m_objs.clear();
  23. m_coef_on_curve.clear();
  25. m_need_regime = NEED_OBJECT;
  26. m_inQuestionRegime = false;
  27. }
  29. SplineSurfaceCommand::~SplineSurfaceCommand()
  30. {
  31. if (sgGetScene()->GetSelectedObjectsList()->GetCount()>0)
  32. {
  33. sgCObject*  curObj = sgGetScene()->GetSelectedObjectsList()->GetHead();
  34. while (curObj) 
  35. {
  36. curObj->Select(false);
  37. curObj = sgGetScene()->GetSelectedObjectsList()->GetNext(curObj);
  38. }
  39. }
  40. m_app->GetCommandPanel()->RemoveAllDialogs();
  41. m_app->GetViewPort()->InvalidateViewPort();
  42. }
  45. bool    SplineSurfaceCommand::PreTranslateMessage(MSG* pMsg)
  46. {
  47. if (m_inQuestionRegime)
  48. return false;
  50. if (pMsg->message==WM_KEYUP||pMsg->message==WM_KEYDOWN || 
  51. pMsg->message==WM_CHAR)
  52. {
  53. if (pMsg->wParam==VK_RETURN)
  54. {
  55. OnEnter();
  56. return true;
  57. }
  58. if (pMsg->wParam==VK_ESCAPE)
  59. {
  60. m_app->StopCommander();
  61. return true;
  62. }
  64. if (m_get_object_panel)
  65. m_get_object_panel->GetWindow()->SendMessage(pMsg->message,
  66. pMsg->wParam,
  67. pMsg->lParam);
  69. if (pMsg->message==WM_KEYDOWN)
  70. return false;
  71. else 
  72. return true;
  73. }
  74. else
  75. {
  76. if (pMsg->hwnd == m_app->GetViewPort()->GetWindow()->m_hWnd)
  77. {
  78. switch(pMsg->message) 
  79. {
  80. case WM_MOUSEMOVE:
  81. MouseMove(pMsg->wParam,GET_X_LPARAM(pMsg->lParam),GET_Y_LPARAM(pMsg->lParam));
  82. return true;
  83. case WM_LBUTTONDOWN:
  84. LeftClick(pMsg->wParam,GET_X_LPARAM(pMsg->lParam),GET_Y_LPARAM(pMsg->lParam));
  85. return true;
  86. default:
  87. return false;
  88. }
  89. }
  90. }
  91. return false;
  92. }
  94. void     SplineSurfaceCommand::SendCommanderMessage(ICommander::COMMANDER_MESSAGE mes, 
  95.    void* params) 
  96. {
  97. /*if (mes==ICommander::CM_SWITCH_ROLLUP_DIALOG)
  98. {
  99. ASSERT(params!=NULL);
  100. int   newActiveDlg = *(reinterpret_cast<int*>(params));
  102. ASSERT(newActiveDlg<=1);
  103. m_step = (unsigned int)newActiveDlg;
  104. for (unsigned int i=m_step+1;i<=1;i++)
  105. {
  106. m_app->GetCommandPanel()->EnableRadio(i,false);
  107. m_objs[i]=NULL;
  108. }
  109. SWITCH_RESOURCE
  111. m_message.LoadString(IDS_COON_SEL_1_CONT);
  113. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  114. return;
  115. }
  116. if (mes==ICommander::CM_SELECT_OBJECT)
  117. {
  118. ASSERT(params!=NULL);
  119. ASSERT(m_step==0 || m_step==1);
  120. sgCObject* so = (sgCObject*)params;
  121. if (so!=m_app->GetViewPort()->GetHotObject())
  122. {
  123. m_app->GetViewPort()->SetHotObject(so);
  124. m_cur_obj = so;
  125. }
  126. m_app->GetViewPort()->InvalidateViewPort();
  127. }*/
  128. }
  132. static   bool isObjAddToList(sgCObject* obj)
  133. {
  134. SG_OBJECT_TYPE ot = obj->GetType();
  135. if (ot==SG_OT_LINE ||
  136. ot==SG_OT_ARC ||
  137. ot==SG_OT_CIRCLE)
  138. return true;
  139. if (ot==SG_OT_SPLINE)
  140. {
  141. sgCSpline* spl = reinterpret_cast<sgCSpline*>(obj);
  142. if (!spl->IsSelfIntersecting())
  143. return true;
  144. else
  145. return false;
  146. }
  147. if (ot==SG_OT_CONTOUR)
  148. {
  149. sgCContour* cntr = reinterpret_cast<sgCContour*>(obj);
  150. if (!cntr->IsSelfIntersecting())
  151. return true;
  152. else
  153. return false;
  154. }
  155. return false;
  156. }
  159. void  SplineSurfaceCommand::Start()
  160. {
  161. SWITCH_RESOURCE
  163. m_app->GetCommandPanel()->RemoveAllDialogs();
  165. m_message.LoadString(IDS_TOOLTIP_FIVETH);
  167. m_app->StartCommander(m_message);
  169. m_message.LoadString(IDS_LS_FIR_CONT);
  170. m_get_object_panel = reinterpret_cast<IGetObjectsPanel*>(m_app->
  171. GetCommandPanel()->
  172. AddDialog(IBaseInterfaceOfGetDialogs::GET_OBJECTS_DLG,m_message,true));
  174. if (m_get_object_panel)
  175. {
  176. m_get_object_panel->SetMultiselectMode(true);
  177. m_get_object_panel->FillList(isObjAddToList);
  178. }
  180. m_step=0;
  181. m_app->GetCommandPanel()->SetActiveRadio(0);
  182. m_app->GetCommandPanel()->EnableRadio(0,false);
  184. m_message.LoadString(IDS_LS_SEL_FIR_CONT);
  185. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  187. }
  189. void   SplineSurfaceCommand::NeedObject(int pX, int pY)
  190. {
  191. int snapSz = m_app->GetViewPort()->GetSnapSize();
  193. sgCObject* ho = m_app->GetViewPort()->GetTopObject(
  194. m_app->GetViewPort()->GetHitsInRect(CRect(pX-snapSz, pY-snapSz,
  195. pX+snapSz, pY+snapSz)));
  196. if (ho && isObjAddToList(ho))
  197. {
  198. m_app->GetViewPort()->SetHotObject(ho);
  199. if (m_get_object_panel)
  200. m_get_object_panel->SelectObject(ho,true);
  201. }
  202. else
  203. {
  204. m_app->GetViewPort()->SetHotObject(NULL);
  205. if (m_get_object_panel)
  206. m_get_object_panel->SelectObject(NULL,true);
  207. }
  208. m_app->GetViewPort()->InvalidateViewPort();
  209. }
  211. void  SplineSurfaceCommand::NeedOneOfEndPoint(int pX, int pY)
  212. {
  213. SG_POINT begP,endP;
  214. begP = m_points_on_curve[m_points_on_curve.size()-1][0];
  215. endP = m_points_on_curve[m_points_on_curve.size()-1][m_points_on_curve[m_points_on_curve.size()-1].size()-1];
  217. SG_POINT begP_proj,endP_proj;
  218. m_app->GetViewPort()->ProjectWorldPoint(begP,begP_proj.x,begP_proj.y,begP_proj.z);
  219. m_app->GetViewPort()->ProjectWorldPoint(endP,endP_proj.x,endP_proj.y,endP_proj.z);
  221. double d1;
  222. d1 = sqrt((begP_proj.x-pX)*(begP_proj.x-pX)+
  223. (begP_proj.y-pY)*(begP_proj.y-pY));
  224. double d2;
  225. d2 = sqrt((endP_proj.x-pX)*(endP_proj.x-pX)+
  226. (endP_proj.y-pY)*(endP_proj.y-pY));
  228. if (d1<d2)
  229. m_inverse_curve[m_inverse_curve.size()-1] = false;
  230. else
  231. m_inverse_curve[m_inverse_curve.size()-1] = true;
  233. m_app->GetViewPort()->InvalidateViewPort();
  234. }
  236. void  SplineSurfaceCommand::NeedPointOnCurve(int pX, int pY)
  237. {
  238. double dis = FLT_MAX;
  239. size_t pInd = 0;
  240. size_t sz = m_temp_buffer.size();
  241. SG_POINT P_proj;
  242. double d2;
  243. m_inverse_curve[m_inverse_curve.size()-1] = false;
  244. for (size_t i =0;i<sz;i++)
  245. {
  246. m_app->GetViewPort()->ProjectWorldPoint(m_temp_buffer[i].pnt,
  247. P_proj.x,P_proj.y,P_proj.z);
  248. d2 = sqrt((P_proj.x-pX)*(P_proj.x-pX)+
  249. (P_proj.y-pY)*(P_proj.y-pY));
  250. if (d2<dis)
  251. {
  252. dis=d2;
  253. pInd = i;
  254. m_coef_on_curve[m_coef_on_curve.size()-1] = m_temp_buffer[i].coef;
  255. }
  256. }
  258. m_points_on_curve[m_points_on_curve.size()-1].clear();
  260. m_points_on_curve[m_points_on_curve.size()-1].reserve(COUNT_OF_POINTS_ON_CURVE_2+1);
  261. for (double i=0.0;i<=1.0;i+=1.0/COUNT_OF_POINTS_ON_CURVE_2)
  262. {
  263. double istCoef = m_coef_on_curve[m_coef_on_curve.size()-1]+i;
  264. if (istCoef>1.0)
  265. istCoef-=1.0;
  266. m_points_on_curve[m_points_on_curve.size()-1].push_back(m_objs[m_objs.size()-1]->GetPointFromCoefficient(istCoef));
  267. }
  268. m_app->GetViewPort()->InvalidateViewPort();
  269. }
  271. void  SplineSurfaceCommand::NeedDirection(int pX, int pY)
  272. {
  273. SG_POINT begP,endP;
  274. begP = m_points_on_curve[m_points_on_curve.size()-1][1];
  275. endP = m_points_on_curve[m_points_on_curve.size()-1][m_points_on_curve[m_points_on_curve.size()-1].size()-1];
  277. SG_POINT begP_proj,endP_proj;
  278. m_app->GetViewPort()->ProjectWorldPoint(begP,begP_proj.x,begP_proj.y,begP_proj.z);
  279. m_app->GetViewPort()->ProjectWorldPoint(endP,endP_proj.x,endP_proj.y,endP_proj.z);
  281. double d1;
  282. d1 = sqrt((begP_proj.x-pX)*(begP_proj.x-pX)+
  283. (begP_proj.y-pY)*(begP_proj.y-pY));
  284. double d2;
  285. d2 = sqrt((endP_proj.x-pX)*(endP_proj.x-pX)+
  286. (endP_proj.y-pY)*(endP_proj.y-pY));
  288. if (d1<d2)
  289. m_inverse_curve[m_inverse_curve.size()-1] = false;
  290. else
  291. m_inverse_curve[m_inverse_curve.size()-1] = true;
  293. m_app->GetViewPort()->InvalidateViewPort();
  294. }
  297. void  SplineSurfaceCommand::MouseMove(unsigned int nFlags,int pX,int pY)
  298. {
  299. switch(m_need_regime) 
  300. {
  301. case NEED_OBJECT:
  302. NeedObject(pX,pY);
  303. break;
  304. case NEED_ONE_OF_END_POINT:
  305. NeedOneOfEndPoint(pX,pY);
  306. break;
  307. case NEED_POINT_ON_CURVE:
  308. NeedPointOnCurve(pX,pY);
  309. break;
  310. case NEED_DIRECTION:
  311. NeedDirection(pX,pY);
  312. break;
  313. }
  314. }
  316. void  SplineSurfaceCommand::LeftClick(unsigned int nFlags,int pX,int pY)
  317. {
  318. SWITCH_RESOURCE
  320. switch(m_need_regime) 
  321. {
  322. case NEED_OBJECT:
  323. m_cur_obj=m_app->GetViewPort()->GetHotObject();
  324. if (m_cur_obj)
  325. {
  326. m_cur_obj->Select(true);
  327. sgC2DObject* ooo = reinterpret_cast<sgC2DObject*>(m_cur_obj);
  328. m_objs.push_back(ooo);
  330. std::vector<SG_POINT>  pnts;
  331. pnts.reserve(COUNT_OF_POINTS_ON_CURVE_2+1);
  332. for (double i=0.0;i<=1.0;i+=1.0/COUNT_OF_POINTS_ON_CURVE_2)
  333. pnts.push_back(ooo->GetPointFromCoefficient(i));
  334. m_points_on_curve.push_back(pnts);
  336. m_inverse_curve.push_back(false);
  337. m_coef_on_curve.push_back(0.0);
  339. if (m_objs.size()==1)
  340. {
  341. m_message.LoadString(IDS_SELECT_NEXT_CONTOUR);
  342. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  343. return;
  344. }
  346. if (!ooo->IsClosed())
  347. {
  348. m_need_regime = NEED_ONE_OF_END_POINT;
  349. m_message.LoadString(IDS_SEL_CURVE_DIR);
  350. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  351. m_app->GetViewPort()->InvalidateViewPort();
  352. return;
  353. }
  354. else
  355. {
  356. m_temp_buffer.clear();
  357. m_temp_buffer.reserve(COUNT_OF_POINTS_ON_CURVE_1+1);
  358. for (double i=0.0;i<=1.0;i+=1.0/COUNT_OF_POINTS_ON_CURVE_1)
  359. {
  360. POINT_AND_HER_COEF tmpS;
  361. tmpS.coef = i;
  362. tmpS.pnt = ooo->GetPointFromCoefficient(i);
  363. m_temp_buffer.push_back(tmpS);
  364. }
  365. m_need_regime = NEED_POINT_ON_CURVE;
  366. m_message.LoadString(IDS_SEL_HREBET_POINT);
  367. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  368. return;
  369. }
  370. }
  371. else
  372. {
  373. m_message.LoadString(IDS_ERROR_OBJ_NOT_SEL);
  374. m_app->PutMessage(IApplicationInterface::MT_ERROR,m_message);
  375. return;
  376. }
  377. break;
  378. case NEED_ONE_OF_END_POINT:
  379. if (m_inverse_curve[m_inverse_curve.size()-1])
  380. m_objs[m_objs.size()-1]->ChangeOrient();
  381. m_need_regime=NEED_OBJECT;
  382. m_message.LoadString(IDS_SELECT_NEXT_CONTOUR);
  383. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  384. m_app->GetViewPort()->InvalidateViewPort();
  385. return;
  386. case NEED_POINT_ON_CURVE:
  387. m_need_regime = NEED_DIRECTION;
  388. m_message.LoadString(IDS_SEL_CURVE_DIR);
  389. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  390. m_temp_buffer.clear();
  391. m_app->GetViewPort()->InvalidateViewPort();
  392. return;
  393. case NEED_DIRECTION:
  394. {
  395. if (m_inverse_curve[m_inverse_curve.size()-1])
  396. m_objs[m_objs.size()-1]->ChangeOrient();
  398. m_need_regime=NEED_OBJECT;
  399. m_message.LoadString(IDS_SELECT_NEXT_CONTOUR);
  400. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  401. m_app->GetViewPort()->InvalidateViewPort();
  402. return;
  403. }
  404. break;
  405. }
  406. }
  409. void  SplineSurfaceCommand::Draw()
  410. {
  411. size_t curves_cnt = m_points_on_curve.size();
  412. if (curves_cnt==0)
  413. return;
  416. float pC[3];
  417. m_app->GetViewPort()->GetPainter()->GetUserColorPoints(pC[0],pC[1],pC[2]);
  418. m_app->GetViewPort()->GetPainter()->SetCurColor(pC[0],pC[1],pC[2]);
  419. size_t poF = m_points_on_curve[0].size();
  420. for (size_t i =0;i<poF;i++)
  421. m_app->GetViewPort()->GetPainter()->DrawPoint(m_points_on_curve[0][i]);
  422. if (m_points_on_curve.size()==1)
  423. return;
  425. for (size_t i=1;i<curves_cnt;i++)
  426. {
  427. m_app->GetViewPort()->GetPainter()->GetUserColorPoints(pC[0],pC[1],pC[2]);
  428. m_app->GetViewPort()->GetPainter()->SetCurColor(pC[0],pC[1],pC[2]);
  429. for (size_t j =0;j<poF;j++)
  430. {
  431. m_app->GetViewPort()->GetPainter()->DrawPoint(m_points_on_curve[i-1][j]);
  432. m_app->GetViewPort()->GetPainter()->DrawPoint(m_points_on_curve[i][j]);
  433. }
  434. m_app->GetViewPort()->GetPainter()->GetUserColorLines(pC[0],pC[1],pC[2]);
  435. m_app->GetViewPort()->GetPainter()->SetCurColor(pC[0],pC[1],pC[2]);
  436. for (size_t j =0;j<poF;j++)
  437. {
  438. SG_LINE ln;
  439. if (m_inverse_curve[i-1])
  440. ln.p1 = m_points_on_curve[i-1][poF-j-1];
  441. else
  442. ln.p1 = m_points_on_curve[i-1][j];
  444. if (m_inverse_curve[i])
  445. ln.p2 = m_points_on_curve[i][poF-j-1];
  446. else
  447. ln.p2 = m_points_on_curve[i][j];
  448. m_app->GetViewPort()->GetPainter()->DrawLine(ln);
  449. }
  450. }
  451. }
  454. void  SplineSurfaceCommand::OnEnter()
  455. {
  456. /*SWITCH_RESOURCE
  457. m_cur_obj=m_app->GetViewPort()->GetHotObject();*/
  458. }
  460. unsigned int  SplineSurfaceCommand::GetItemsCount()
  461. {
  462. return 1;
  463. }
  465. void         SplineSurfaceCommand::GetItem(unsigned int itemID, CString& itSrt)
  466. {
  467. SWITCH_RESOURCE
  468. if (itemID==0) 
  469. {
  470. itSrt.LoadString(IDS_END_COM);
  471. }
  472. else
  473. {
  474. ASSERT(0);
  475. }
  476. }
  478. void     SplineSurfaceCommand::GetItemState(unsigned int itemID, bool& enbl, bool& checked)
  479. {
  480. if (m_objs.size()>=3)
  481. enbl = true;
  482. else
  483. enbl = false;
  484. checked = false;
  485. }
  487. HBITMAP   SplineSurfaceCommand::GetItemBitmap(unsigned int)
  488. {
  489. return NULL;
  490. }
  492. void         SplineSurfaceCommand::Run(unsigned int itemID)
  493. {
  494. SWITCH_RESOURCE
  495. if (itemID==0)
  496. {
  497. bool  cl = false;
  499. if (m_objs[0]->IsPlane(NULL,NULL) && m_objs[0]->IsClosed() &&
  500. m_objs[m_objs.size()-1]->IsPlane(NULL,NULL) && m_objs[m_objs.size()-1]->IsClosed())
  501. {
  502. m_message.LoadString(IDS_IS_CLOSE);
  503. CString qe;
  504. qe.LoadString(IDS_QUESTION);
  505. m_inQuestionRegime = true;
  506. if (MessageBox(m_app->GetViewPort()->GetWindow()->m_hWnd,
  507. m_message,qe,MB_YESNO|MB_ICONQUESTION)==IDYES)
  508. cl = true;
  509. else
  510. cl = false;
  511. m_inQuestionRegime = false;
  512. }
  514. sgCObject* aaa = sgSurfaces::SplineSurfaceFromSections((const sgC2DObject**)(&m_objs[0]),
  515. (const double*)(&m_coef_on_curve[0]),m_objs.size(),cl);
  517. if (!aaa)
  518. {
  519. m_message.LoadString(IDS_ERR_CANT_CREATE);
  520. m_app->PutMessage(IApplicationInterface::MT_ERROR,m_message);
  521. return;
  522. }
  524. CString nm;
  525. nm.LoadString(IDS_SPLINE_NAME);
  526. CString nmInd;
  527. nmInd.Format("%i",spline_surface_index);
  528. nm+=nmInd;
  529. aaa->SetName(nm);
  531. aaa->SetUserGeometry("{8809183F-18CC-49ee-8B5E-9570D3AF3A6A}",0,NULL);
  533. sgGetScene()->StartUndoGroup();
  534. sgGetScene()->AttachObject(aaa);
  535. m_app->ApplyAttributes(aaa);
  536. sgGetScene()->EndUndoGroup();
  537. spline_surface_index++;
  539. if (sgGetScene()->GetSelectedObjectsList()->GetCount()>0)
  540. {
  541. sgCObject*  curObj = sgGetScene()->GetSelectedObjectsList()->GetHead();
  542. while (curObj) 
  543. {
  544. curObj->Select(false);
  545. curObj = sgGetScene()->GetSelectedObjectsList()->GetNext(curObj);
  546. }
  547. }
  549. for (size_t i=0;i<m_points_on_curve.size();i++)
  550.   m_points_on_curve[i].clear();
  551. m_points_on_curve.clear();
  553. m_temp_buffer.clear();
  555. m_inverse_curve.clear();
  556. m_coef_on_curve.clear();
  558. m_step=0;
  559. m_app->GetCommandPanel()->SetActiveRadio(0);
  560. m_app->GetCommandPanel()->EnableRadio(0,false);
  562. m_message.LoadString(IDS_LS_SEL_FIR_CONT);
  563. m_app->PutMessage(IApplicationInterface::MT_MESSAGE,m_message);
  565. }
  566. else
  567. {
  568. ASSERT(0);
  569. }
  570. }