1 // OCC_3dBaseDoc.cpp: implementation of the OCC_3dBaseDoc class.
3 //////////////////////////////////////////////////////////////////////
7 #include "OCC_3dBaseDoc.h"
9 #include "OCC_3dView.h"
11 #include <res\OCC_Resource.h>
12 #include "ImportExport/ImportExport.h"
13 #include "AISDialogs.h"
14 #include <AIS_ListOfInteractive.hxx>
15 #include <AIS_ListIteratorOfListOfInteractive.hxx>
16 #include <TColStd_ListIteratorOfListOfInteger.hxx>
17 #include <TColStd_ListOfInteger.hxx>
18 #include <TopoDS_Shape.hxx>
20 BEGIN_MESSAGE_MAP(OCC_3dBaseDoc, OCC_BaseDoc)
21 //{{AFX_MSG_MAP(OCC_3dBaseDoc)
22 ON_COMMAND(ID_FILE_IMPORT_BREP, OnFileImportBrep)
23 ON_COMMAND(ID_FILE_EXPORT_BREP, OnFileExportBrep)
24 ON_COMMAND(ID_OBJECT_ERASE, OnObjectErase)
25 ON_UPDATE_COMMAND_UI(ID_OBJECT_ERASE, OnUpdateObjectErase)
26 ON_COMMAND(ID_OBJECT_COLOR, OnObjectColor)
27 ON_UPDATE_COMMAND_UI(ID_OBJECT_COLOR, OnUpdateObjectColor)
28 ON_COMMAND(ID_OBJECT_SHADING, OnObjectShading)
29 ON_UPDATE_COMMAND_UI(ID_OBJECT_SHADING, OnUpdateObjectShading)
30 ON_COMMAND(ID_OBJECT_WIREFRAME, OnObjectWireframe)
31 ON_UPDATE_COMMAND_UI(ID_OBJECT_WIREFRAME, OnUpdateObjectWireframe)
32 ON_COMMAND(ID_OBJECT_TRANSPARENCY, OnObjectTransparency)
33 ON_UPDATE_COMMAND_UI(ID_OBJECT_TRANSPARENCY, OnUpdateObjectTransparency)
34 ON_COMMAND(ID_OBJECT_MATERIAL, OnObjectMaterial)
35 ON_UPDATE_COMMAND_UI(ID_OBJECT_MATERIAL, OnUpdateObjectMaterial)
36 ON_COMMAND(ID_OBJECT_DISPLAYALL, OnObjectDisplayall)
37 ON_UPDATE_COMMAND_UI(ID_OBJECT_DISPLAYALL, OnUpdateObjectDisplayall)
38 ON_COMMAND(ID_OBJECT_REMOVE, OnObjectRemove)
39 ON_UPDATE_COMMAND_UI(ID_OBJECT_REMOVE, OnUpdateObjectRemove)
42 ON_COMMAND_EX_RANGE(ID_OBJECT_MATERIAL_BRASS,ID_OBJECT_MATERIAL_DEFAULT, OnObjectMaterialRange)
43 ON_UPDATE_COMMAND_UI_RANGE(ID_OBJECT_MATERIAL_BRASS,ID_OBJECT_MATERIAL_DEFAULT, OnUpdateObjectMaterialRange)
46 ON_COMMAND(ID_OBJECT_RAY_TRACING,OnObjectRayTracing)
47 ON_COMMAND(ID_OBJECT_SHADOWS,OnObjectShadows)
48 ON_COMMAND(ID_OBJECT_REFLECTIONS,OnObjectReflections)
49 ON_COMMAND(ID_OBJECT_ANTI_ALIASING,OnObjectAntiAliasing)
51 ON_UPDATE_COMMAND_UI(ID_OBJECT_RAY_TRACING, OnUpdateV3dButtons)
52 ON_UPDATE_COMMAND_UI(ID_OBJECT_SHADOWS, OnUpdateV3dButtons)
53 ON_UPDATE_COMMAND_UI(ID_OBJECT_REFLECTIONS, OnUpdateV3dButtons)
54 ON_UPDATE_COMMAND_UI(ID_OBJECT_ANTI_ALIASING, OnUpdateV3dButtons)
58 //////////////////////////////////////////////////////////////////////
59 // Construction/Destruction
60 //////////////////////////////////////////////////////////////////////
62 OCC_3dBaseDoc::OCC_3dBaseDoc()
67 Handle(Graphic3d_GraphicDriver) aGraphicDriver = ((OCC_App*)AfxGetApp())->GetGraphicDriver();
69 myViewer = new V3d_Viewer (aGraphicDriver);
70 myViewer->SetDefaultLights();
71 myViewer->SetLightOn();
72 myAISContext = new AIS_InteractiveContext (myViewer);
74 myRayTracingIsOn = false;
75 myRaytracedShadowsIsOn = true;
76 myRaytracedReflectionsIsOn = false;
77 myRaytracedAntialiasingIsOn = false;
80 OCC_3dBaseDoc::~OCC_3dBaseDoc()
84 //-----------------------------------------------------------------------------------------
86 //-----------------------------------------------------------------------------------------
87 void OCC_3dBaseDoc::DragEvent (const Standard_Integer theMouseX,
88 const Standard_Integer theMouseY,
89 const Standard_Integer theState,
90 const Handle(V3d_View)& theView)
92 // TheState == -1 button down
94 // TheState == 1 button up
96 static Standard_Integer aStartDragX = 0;
97 static Standard_Integer aStartDragY = 0;
103 aStartDragX = theMouseX;
104 aStartDragY = theMouseY;
109 myAISContext->UpdateCurrentViewer();
114 myAISContext->Select (aStartDragX, aStartDragY,
115 theMouseX, theMouseY,
116 theView, Standard_True);
122 //-----------------------------------------------------------------------------------------
124 //-----------------------------------------------------------------------------------------
125 void OCC_3dBaseDoc::InputEvent (const Standard_Integer theMouseX,
126 const Standard_Integer theMouseY,
127 const Handle(V3d_View)& theView)
129 myAISContext->MoveTo (theMouseX, theMouseY, theView, Standard_False);
130 myAISContext->Select (Standard_True);
133 //-----------------------------------------------------------------------------------------
135 //-----------------------------------------------------------------------------------------
136 void OCC_3dBaseDoc::MoveEvent (const Standard_Integer theMouseX,
137 const Standard_Integer theMouseY,
138 const Handle(V3d_View)& theView)
140 myAISContext->MoveTo (theMouseX, theMouseY, theView, Standard_True);
143 //-----------------------------------------------------------------------------------------
145 //-----------------------------------------------------------------------------------------
146 void OCC_3dBaseDoc::ShiftMoveEvent (const Standard_Integer theMouseX,
147 const Standard_Integer theMouseY,
148 const Handle(V3d_View)& theView)
150 myAISContext->MoveTo (theMouseX, theMouseY, theView, Standard_True);
153 //-----------------------------------------------------------------------------------------
155 //-----------------------------------------------------------------------------------------
156 void OCC_3dBaseDoc::ShiftDragEvent (const Standard_Integer theMouseX,
157 const Standard_Integer theMouseY,
158 const Standard_Integer theState,
159 const Handle(V3d_View)& theView)
161 // TheState == -1 button down
162 // TheState == 0 move
163 // TheState == 1 button up
165 static Standard_Integer aStartDragX = 0;
166 static Standard_Integer aStartDragY = 0;
171 aStartDragX = theMouseX;
172 aStartDragY = theMouseY;
178 myAISContext->ShiftSelect (aStartDragX, aStartDragY,
179 theMouseX, theMouseY,
180 theView, Standard_True);
184 //-----------------------------------------------------------------------------------------
186 //-----------------------------------------------------------------------------------------
187 void OCC_3dBaseDoc::ShiftInputEvent (const Standard_Integer /*theMouseX*/,
188 const Standard_Integer /*theMouseY*/,
189 const Handle(V3d_View)& /*theView*/)
191 myAISContext->ShiftSelect (Standard_True);
194 //-----------------------------------------------------------------------------------------
196 //-----------------------------------------------------------------------------------------
197 void OCC_3dBaseDoc::Popup (const Standard_Integer theMouseX,
198 const Standard_Integer theMouseY,
199 const Handle(V3d_View)& theView)
201 // Base check which context menu to call
202 if (!myPopupMenuNumber)
204 myAISContext->InitSelected();
205 if (myAISContext->MoreSelected())
210 VERIFY(menu.LoadMenu(IDR_Popup3D));
211 CMenu* pPopup = menu.GetSubMenu(myPopupMenuNumber);
213 ASSERT(pPopup != NULL);
214 if (myPopupMenuNumber == 1) // more than 1 object.
216 bool OneOrMoreInShading = false;
217 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
218 if (myAISContext->IsDisplayed(myAISContext->SelectedInteractive(),1)) OneOrMoreInShading=true;
219 if(!OneOrMoreInShading)
220 pPopup->EnableMenuItem(5, MF_BYPOSITION | MF_DISABLED | MF_GRAYED);
223 POINT winCoord = { theMouseX , theMouseY };
224 Handle(WNT_Window) aWNTWindow=
225 Handle(WNT_Window)::DownCast(theView->Window());
226 ClientToScreen ( (HWND)(aWNTWindow->HWindow()),&winCoord);
227 pPopup->TrackPopupMenu (TPM_LEFTALIGN | TPM_RIGHTBUTTON,
233 void OCC_3dBaseDoc::Fit()
235 CMDIFrameWnd *pFrame = (CMDIFrameWnd*)AfxGetApp()->m_pMainWnd;
236 CMDIChildWnd *pChild = (CMDIChildWnd *) pFrame->GetActiveFrame();
237 OCC_3dView *pView = (OCC_3dView *) pChild->GetActiveView();
241 int OCC_3dBaseDoc::OnFileImportBrep_WithInitDir (const wchar_t* )
243 if (CImportExport::ReadBREP (myAISContext) == 1)
249 void OCC_3dBaseDoc::OnFileImportBrep()
251 OnFileImportBrep_WithInitDir (NULL);
254 void OCC_3dBaseDoc::OnFileExportBrep()
255 { CImportExport::SaveBREP(myAISContext);}
257 void OCC_3dBaseDoc::OnObjectColor()
259 Handle(AIS_InteractiveObject) Current ;
262 myAISContext->InitSelected();
263 Current = myAISContext->SelectedInteractive();
264 if ( Current->HasColor () ) {
265 Quantity_Color CSFColor;
266 myAISContext->Color (Current, CSFColor);
267 MSColor = RGB (CSFColor.Red()*255.,CSFColor.Green()*255.,CSFColor.Blue()*255.);
270 MSColor = RGB (255,255,255) ;
273 CColorDialog dlgColor(MSColor);
274 if (dlgColor.DoModal() == IDOK)
276 MSColor = dlgColor.GetColor();
277 Quantity_Color CSFColor (GetRValue(MSColor)/255., GetGValue(MSColor)/255., GetBValue(MSColor)/255., Quantity_TOC_RGB);
278 for (;myAISContext->MoreSelected ();myAISContext->NextSelected ())
279 myAISContext->SetColor (myAISContext->SelectedInteractive(), CSFColor, Standard_False);
280 myAISContext->UpdateCurrentViewer();
283 void OCC_3dBaseDoc::OnUpdateObjectColor(CCmdUI* pCmdUI)
285 bool OneOrMoreIsShadingOrWireframe = false;
286 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
287 if (myAISContext->IsDisplayed(myAISContext->SelectedInteractive(),0)
288 ||myAISContext->IsDisplayed(myAISContext->SelectedInteractive(),1))
289 OneOrMoreIsShadingOrWireframe=true;
290 pCmdUI->Enable (OneOrMoreIsShadingOrWireframe);
293 void OCC_3dBaseDoc::OnObjectErase()
295 myAISContext->EraseSelected (Standard_True);
297 void OCC_3dBaseDoc::OnUpdateObjectErase(CCmdUI* pCmdUI)
299 bool OneOrMoreIsDisplayed = false;
300 for (myAISContext->InitSelected(); myAISContext->MoreSelected(); myAISContext->NextSelected())
302 if (myAISContext->IsDisplayed (myAISContext->SelectedInteractive()))
303 OneOrMoreIsDisplayed = true;
305 pCmdUI->Enable (OneOrMoreIsDisplayed);
308 void OCC_3dBaseDoc::OnObjectWireframe()
310 for(myAISContext->InitSelected();myAISContext->MoreSelected();myAISContext->NextSelected())
311 myAISContext->SetDisplayMode (myAISContext->SelectedInteractive(), 0, Standard_False);
312 myAISContext->UpdateCurrentViewer();
314 void OCC_3dBaseDoc::OnUpdateObjectWireframe(CCmdUI* pCmdUI)
316 bool OneOrMoreInShading = false;
317 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
318 if (myAISContext->IsDisplayed(myAISContext->SelectedInteractive(),1)) OneOrMoreInShading=true;
319 pCmdUI->Enable (OneOrMoreInShading);
322 void OCC_3dBaseDoc::OnObjectShading()
324 for(myAISContext->InitSelected();myAISContext->MoreSelected();myAISContext->NextSelected())
325 myAISContext->SetDisplayMode (myAISContext->SelectedInteractive(), 1, Standard_False);
326 myAISContext->UpdateCurrentViewer();
329 void OCC_3dBaseDoc::OnUpdateObjectShading(CCmdUI* pCmdUI)
331 bool OneOrMoreInWireframe = false;
332 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
333 if (myAISContext->IsDisplayed(myAISContext->SelectedInteractive(),0)) OneOrMoreInWireframe=true;
334 pCmdUI->Enable (OneOrMoreInWireframe);
337 void OCC_3dBaseDoc::OnObjectMaterial()
339 CDialogMaterial DialBox(myAISContext);
341 //CMDIFrameWnd *pFrame = (CMDIFrameWnd*)AfxGetApp()->m_pMainWnd;
342 //CMDIChildWnd *pChild = (CMDIChildWnd *) pFrame->GetActiveFrame();
343 //OCC_3dView *pView = (OCC_3dView *) pChild->GetActiveView();
346 void OCC_3dBaseDoc::OnUpdateObjectMaterial(CCmdUI* pCmdUI)
348 bool OneOrMoreInShading = false;
349 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
350 if (myAISContext->IsDisplayed(myAISContext->SelectedInteractive(),1)) OneOrMoreInShading=true;
351 pCmdUI->Enable (OneOrMoreInShading);
354 BOOL OCC_3dBaseDoc::OnObjectMaterialRange(UINT nID)
356 // the range ID_OBJECT_MATERIAL_BRASS to ID_OBJECT_MATERIAL_SILVER is
357 // continue with the same values as enumeration Type Of Material
358 Standard_Real aTransparency;
360 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ()){
361 aTransparency = myAISContext->SelectedInteractive()->Transparency();
362 myAISContext->SetMaterial (myAISContext->SelectedInteractive(),(Graphic3d_NameOfMaterial)(nID-ID_OBJECT_MATERIAL_BRASS), Standard_False);
363 myAISContext->SetTransparency (myAISContext->SelectedInteractive(),aTransparency, Standard_False);
365 myAISContext->UpdateCurrentViewer();
370 void OCC_3dBaseDoc::OnUpdateObjectMaterialRange(CCmdUI* pCmdUI)
372 bool OneOrMoreInShading = false;
373 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
374 if (myAISContext->IsDisplayed(myAISContext->SelectedInteractive(),1)) OneOrMoreInShading=true;
375 pCmdUI->Enable (OneOrMoreInShading);
376 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
377 if (myAISContext->SelectedInteractive()->Material() - (pCmdUI->m_nID - ID_OBJECT_MATERIAL_BRASS) == 0)
382 void OCC_3dBaseDoc::OnObjectTransparency()
384 CDialogTransparency DialBox(myAISContext);
386 //CMDIFrameWnd *pFrame = (CMDIFrameWnd*)AfxGetApp()->m_pMainWnd;
387 //CMDIChildWnd *pChild = (CMDIChildWnd *) pFrame->GetActiveFrame();
388 //OCC_3dView *pView = (OCC_3dView *) pChild->GetActiveView();
392 void OCC_3dBaseDoc::OnUpdateObjectTransparency(CCmdUI* pCmdUI)
394 bool OneOrMoreInShading = false;
395 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
396 if (myAISContext->IsDisplayed(myAISContext->SelectedInteractive(),1)) OneOrMoreInShading=true;
397 pCmdUI->Enable (OneOrMoreInShading);
401 void OCC_3dBaseDoc::OnObjectDisplayall()
403 myAISContext->DisplayAll (Standard_True);
406 void OCC_3dBaseDoc::OnUpdateObjectDisplayall(CCmdUI* pCmdUI)
408 AIS_ListOfInteractive aList;
409 myAISContext->ObjectsInside (aList);
410 for (AIS_ListIteratorOfListOfInteractive aLI (aList);aLI.More();aLI.Next())
412 if (!myAISContext->IsDisplayed (aLI.Value()))
414 pCmdUI->Enable (true);
418 pCmdUI->Enable (false);
421 void OCC_3dBaseDoc::OnObjectRemove()
423 for(myAISContext->InitSelected();myAISContext->MoreSelected();myAISContext->InitSelected())
424 myAISContext->Remove(myAISContext->SelectedInteractive(),Standard_True);
427 void OCC_3dBaseDoc::OnUpdateObjectRemove(CCmdUI* pCmdUI)
429 bool OneOrMoreIsDisplayed = false;
430 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
431 if (myAISContext->IsDisplayed(myAISContext->SelectedInteractive())) OneOrMoreIsDisplayed=true;
432 pCmdUI->Enable (OneOrMoreIsDisplayed);
435 void OCC_3dBaseDoc::SetMaterial(Graphic3d_NameOfMaterial Material)
437 for (myAISContext->InitSelected();myAISContext->MoreSelected ();myAISContext->NextSelected ())
438 myAISContext->SetMaterial (myAISContext->SelectedInteractive(),
439 (Graphic3d_NameOfMaterial)(Material), Standard_False);
440 myAISContext->UpdateCurrentViewer();
445 void OCC_3dBaseDoc::OnObjectRayTracing()
447 myRayTracingIsOn = !myRayTracingIsOn;
448 if(!myRayTracingIsOn)
450 myRaytracedShadowsIsOn = false;
451 myRaytracedReflectionsIsOn = false;
452 myRaytracedAntialiasingIsOn = false;
454 OnObjectRayTracingAction();
457 void OCC_3dBaseDoc::OnObjectShadows()
459 myRaytracedShadowsIsOn = !myRaytracedShadowsIsOn;
460 OnObjectRayTracingAction();
463 void OCC_3dBaseDoc::OnObjectReflections()
465 myRaytracedReflectionsIsOn = !myRaytracedReflectionsIsOn;
466 OnObjectRayTracingAction();
469 void OCC_3dBaseDoc::OnObjectAntiAliasing()
471 myRaytracedAntialiasingIsOn = !myRaytracedAntialiasingIsOn;
472 OnObjectRayTracingAction();
474 void OCC_3dBaseDoc::OnUpdateV3dButtons (CCmdUI* pCmdUI)
476 if (pCmdUI->m_nID == ID_OBJECT_RAY_TRACING)
478 pCmdUI->SetCheck(myRayTracingIsOn);
480 pCmdUI->Enable(myRayTracingIsOn);
481 if (pCmdUI->m_nID == ID_OBJECT_SHADOWS)
482 pCmdUI->SetCheck(myRaytracedShadowsIsOn);
483 if (pCmdUI->m_nID == ID_OBJECT_REFLECTIONS)
484 pCmdUI->SetCheck(myRaytracedReflectionsIsOn);
485 if (pCmdUI->m_nID == ID_OBJECT_ANTI_ALIASING)
486 pCmdUI->SetCheck(myRaytracedAntialiasingIsOn);
489 // Common function to change raytracing params and redraw view
490 void OCC_3dBaseDoc::OnObjectRayTracingAction()
492 myAISContext->CurrentViewer()->InitActiveViews();
493 Handle(V3d_View) aView = myAISContext->CurrentViewer()->ActiveView();
494 Graphic3d_RenderingParams& aParams = aView->ChangeRenderingParams();
495 if (myRayTracingIsOn)
496 aParams.Method = Graphic3d_RM_RAYTRACING;
498 aParams.Method = Graphic3d_RM_RASTERIZATION;
499 aParams.IsShadowEnabled = myRaytracedShadowsIsOn;
500 aParams.IsReflectionEnabled = myRaytracedReflectionsIsOn;
501 aParams.IsAntialiasingEnabled = myRaytracedAntialiasingIsOn;
502 myAISContext->UpdateCurrentViewer();