1 // GEOM GEOMGUI : GUI for Geometry component
3 // Copyright (C) 2003 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 // Lesser General Public License for more details.
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
24 // File : GeometryGUI_Swig.cxx
25 // Author : Nicolas REJNERI, Paul RASCLE
29 #include "GeometryGUI_Swig.hxx"
30 #include "SALOMEGUI_Swig.hxx"
32 #include "GeometryGUI.h"
34 #include "SUIT_Session.h"
35 #include "SalomeApp_Application.h"
36 #include "SalomeApp_Study.h"
37 #include "SalomeApp_ImportOperation.h"
39 #include "OB_Browser.h"
41 #include "OCCViewer_ViewWindow.h"
42 #include "OCCViewer_ViewManager.h"
43 #include "SOCC_ViewModel.h"
45 #include "SVTK_ViewModel.h"
46 #include "SVTK_ViewWindow.h"
47 #include "SVTK_View.h"
48 #include "SVTK_Renderer.h"
50 #include "GEOM_Actor.h"
51 #include "GEOM_Client.hxx"
52 #include "GEOM_AISShape.hxx"
53 #include "GEOM_AssemblyBuilder.h"
54 #include "GEOM_InteractiveObject.hxx"
55 #include "GEOM_Displayer.h"
57 #include "SALOME_Event.hxx"
59 #include "utilities.h"
61 #include "SALOMEDSClient.hxx"
65 #include <TopoDS_Shape.hxx>
66 #include <AIS_ListOfInteractive.hxx>
67 #include <AIS_ListIteratorOfListOfInteractive.hxx>
68 #include <AIS_Drawer.hxx>
71 #include <SALOMEconfig.h>
72 #include CORBA_SERVER_HEADER(GEOM_Gen)
74 #include <vtkRenderer.h>
79 static GEOM_Client ShapeReader;
81 inline OCCViewer_Viewer* GetOCCViewer(SUIT_Application* theApp){
82 SUIT_ViewWindow* window = theApp->desktop()->activeWindow();
83 if(window && window->getViewManager()->getType() == OCCViewer_Viewer::Type()){
84 OCCViewer_ViewWindow* vw = dynamic_cast<OCCViewer_ViewWindow*>( window );
86 OCCViewer_ViewManager* vm = dynamic_cast<OCCViewer_ViewManager*>( vw->getViewManager() );
88 return vm->getOCCViewer();
95 inline SVTK_ViewWindow* GetSVTKViewWindow(SUIT_Application* theApp){
96 SUIT_ViewWindow* window = theApp->desktop()->activeWindow();
97 if(window && window->getViewManager()->getType() == SVTK_Viewer::Type())
98 return dynamic_cast<SVTK_ViewWindow*>( window );
103 GEOM_Swig::GEOM_Swig()
105 // MESSAGE("Constructeur");
108 GEOM_Swig::~GEOM_Swig()
110 // MESSAGE("Destructeur");
113 void GEOM_Swig::createAndDisplayGO (const char* Entry, bool isUpdated)
115 class TEvent: public SALOME_Event
120 TEvent(const char* theEntry, bool toUpdateViewer):
122 myUpdateViewer(toUpdateViewer)
124 virtual void Execute()
126 SUIT_Application* app = SUIT_Session::session()->activeApplication();
129 SalomeApp_Study* ActiveStudy = dynamic_cast<SalomeApp_Study*>(app->activeStudy());
130 if (!ActiveStudy) return;
132 _PTR(Study) aStudy(ActiveStudy->studyDS());
133 _PTR(StudyBuilder) aStudyBuilder = aStudy->NewBuilder();
135 GEOM::GEOM_Gen_var Geom = GeometryGUI::GetGeomGen();
136 if (CORBA::is_nil(Geom)) {
137 GeometryGUI::InitGeomGen();
138 Geom = GeometryGUI::GetGeomGen();
140 if (CORBA::is_nil(Geom))
144 _PTR(SComponent) father = aStudy->FindComponent("GEOM");
147 if (!father->ComponentIOR(aFatherIOR)) {
148 aStudyBuilder->LoadWith(father, SalomeApp_Application::orb()->object_to_string(Geom));
149 father->ComponentIOR(aFatherIOR);
152 _PTR(SObject) obj = aStudy->FindObjectID(myEntry);
157 _PTR(GenericAttribute) anAttr;
158 if (!obj->FindAttribute(anAttr, "AttributeIOR"))
160 _PTR(AttributeIOR) anIOR(anAttr);
161 string anIORValue = anIOR->Value();
163 GEOM::GEOM_Object_var aShape = Geom->GetIORFromString(anIORValue.c_str());
164 TopoDS_Shape Shape = ShapeReader.GetShape(Geom,aShape);
165 if (!Shape.IsNull()) {
166 if (obj->FindAttribute(anAttr, "AttributeName")) {
167 _PTR(AttributeName) aName (anAttr);
168 string aNameValue = aName->Value();
170 /*SUIT_Operation* op = new SalomeApp_ImportOperation (app);
173 _PTR(SObject) newObj1 = aStudyBuilder->NewObject(father);
174 aStudyBuilder->Addreference(newObj1, obj);
175 // commit transaction
177 Handle(GEOM_InteractiveObject) anIO =
178 new GEOM_InteractiveObject (const_cast<char*>(anIORValue.c_str()),
179 const_cast<char*>(aFatherIOR.c_str()),
181 const_cast<char*>( obj->GetID().c_str()));
183 GEOM_Displayer(ActiveStudy).Display(anIO, myUpdateViewer);
184 /*if (SVTK_ViewWindow* aViewWindow = GetSVTKViewWindow(app)) {
185 SVTK_View* aView = aViewWindow->getView();
186 int aMode = aView->GetDisplayMode();
188 vtkActorCollection* theActors =
189 GEOM_AssemblyBuilder::BuildActors(Shape,0,aMode,true);
190 theActors->InitTraversal();
191 while (vtkActor* anActor = theActors->GetNextActor()) {
192 GEOM_Actor* GActor = GEOM_Actor::SafeDownCast(anActor);
193 GActor->setName(const_cast<char*>(aNameValue.c_str()));
195 aView->Display(GActor);
198 } else if (OCCViewer_Viewer* occViewer = GetOCCViewer(app)) {
199 Handle(AIS_InteractiveContext) ic = occViewer->getAISContext();
200 Handle(GEOM_AISShape) aSh =
201 new GEOM_AISShape (Shape,const_cast<char*>(aNameValue.c_str()));
202 aSh->setName(const_cast<char*>(aNameValue.c_str()));
205 ic->AddOrRemoveCurrentObject(aSh,true);
207 // update object browser
208 SalomeApp_Application* app = NULL; //dynamic_cast<SalomeApp_Application*>(app);
210 CAM_Module* module = app->module("Geometry");
211 SalomeApp_Module* appMod = dynamic_cast<SalomeApp_Module*>(module);
213 appMod->updateObjBrowser(true);
220 // MESSAGE("createAndDisplayGO");
221 ProcessVoidEvent(new TEvent (Entry, isUpdated));
223 class TEventUpdateBrowser: public SALOME_Event
226 TEventUpdateBrowser() {}
227 virtual void Execute() {
228 SalomeApp_Application* app =
229 dynamic_cast<SalomeApp_Application*>(SUIT_Session::session()->activeApplication());
231 CAM_Module* module = app->module("Geometry");
232 SalomeApp_Module* appMod = dynamic_cast<SalomeApp_Module*>(module);
233 if (appMod) appMod->updateObjBrowser(true);
239 ProcessVoidEvent(new TEventUpdateBrowser ());
242 void GEOM_Swig::createAndDisplayFitAllGO (const char* Entry)
244 class TEventFitAll: public SALOME_Event
248 virtual void Execute() {
249 SUIT_Application* app = SUIT_Session::session()->activeApplication();
252 if (SVTK_ViewWindow* aViewWindow = GetSVTKViewWindow(app))
254 SVTK_View* aView = aViewWindow->getView();
255 aView->GetRenderer()->OnFitAll();
257 else if (OCCViewer_Viewer* occViewer = GetOCCViewer(app))
259 Handle(V3d_Viewer) aViewer3d = occViewer->getViewer3d();
260 aViewer3d->InitActiveViews();
262 if (aViewer3d->MoreActiveViews())
263 aViewer3d->ActiveView()->FitAll();
268 createAndDisplayGO(Entry);
269 ProcessVoidEvent(new TEventFitAll());
272 void GEOM_Swig::UpdateViewer()
274 class TEventUpdateViewer: public SALOME_Event
277 TEventUpdateViewer() {}
278 virtual void Execute() {
279 SUIT_Application* app = SUIT_Session::session()->activeApplication();
281 SalomeApp_Study* ActiveStudy = dynamic_cast<SalomeApp_Study*>(app->activeStudy());
282 if (!ActiveStudy) return;
284 GEOM_Displayer(ActiveStudy).UpdateViewer();
288 ProcessVoidEvent(new TEventUpdateViewer());
291 int GEOM_Swig::getIndexTopology(const char* SubIOR, const char* IOR)
293 GEOM::GEOM_Gen_var aGeomGen = GeometryGUI::GetGeomGen();
294 if (CORBA::is_nil(aGeomGen))
297 GEOM::GEOM_Object_var aMainShape = aGeomGen->GetIORFromString(IOR);
298 GEOM::GEOM_Object_var aSubShape = aGeomGen->GetIORFromString(SubIOR);
299 if (CORBA::is_nil(aMainShape) || CORBA::is_nil(aSubShape))
302 GEOM::GEOM_IShapesOperations_var anIShapesOperations =
303 aGeomGen->GetIShapesOperations(aMainShape->GetStudyID());
304 if (CORBA::is_nil(anIShapesOperations))
307 return anIShapesOperations->GetTopologyIndex(aMainShape, aSubShape);
310 const char* GEOM_Swig::getShapeTypeString(const char* IOR)
312 TCollection_AsciiString aTypeName ("Shape of unknown type");
314 GEOM::GEOM_Gen_var aGeomGen = GeometryGUI::GetGeomGen();
315 if (!CORBA::is_nil(aGeomGen))
317 GEOM::GEOM_Object_var aShape = aGeomGen->GetIORFromString(IOR);
318 if (!CORBA::is_nil(aShape))
320 GEOM::GEOM_IShapesOperations_var anIShapesOperations =
321 aGeomGen->GetIShapesOperations(aShape->GetStudyID());
322 if (!CORBA::is_nil(anIShapesOperations))
324 aTypeName = anIShapesOperations->GetShapeTypeString(aShape);
329 return CORBA::string_dup(aTypeName.ToCString());
333 const char* GEOM_Swig::getShapeTypeIcon(const char* IOR)
335 GEOM::GEOM_Gen_var Geom = GeometryGUI::GetGeomGen();
336 if ( CORBA::is_nil( Geom ) )
339 GEOM::GEOM_Object_var aShape = Geom->GetIORFromString(IOR);
340 TopoDS_Shape shape = ShapeReader.GetShape(Geom, aShape);
342 if( shape.IsNull() ) {
346 switch (shape.ShapeType() )
348 case TopAbs_COMPOUND:
349 { return "ICON_OBJBROWSER_COMPOUND" ;}
350 case TopAbs_COMPSOLID:
351 { return "ICON_OBJBROWSER_COMPSOLID" ;}
353 { return "ICON_OBJBROWSER_SOLID" ;}
355 { return "ICON_OBJBROWSER_SHELL" ;}
357 { return "ICON_OBJBROWSER_FACE" ;}
359 { return "ICON_OBJBROWSER_WIRE" ;}
361 { return "ICON_OBJBROWSER_EDGE" ;}
363 { return "ICON_OBJBROWSER_VERTEX" ;}
369 void GEOM_Swig::setDisplayMode(const char* theEntry, int theMode, bool isUpdated)
371 class TEvent: public SALOME_Event {
376 TEvent(const char* theEntryArg, int theModeArg, bool theUpdated):
377 myEntry(theEntryArg), myMode(theModeArg), myUpdateViewer(theUpdated)
379 virtual void Execute() {
380 SUIT_Application* anApp = SUIT_Session::session()->activeApplication();
383 Handle(SALOME_InteractiveObject) anIO =
384 new SALOME_InteractiveObject(myEntry.c_str(), "GEOM", "");
386 if (SVTK_ViewWindow* aViewWindow = GetSVTKViewWindow(anApp)) {
387 SVTK_View* aView = aViewWindow->getView();
388 aView->SetDisplayMode(anIO, myMode);
392 else if (OCCViewer_Viewer* occViewer = GetOCCViewer(anApp)) {
393 SOCC_Viewer* soccViewer = dynamic_cast<SOCC_Viewer*>(occViewer);
395 soccViewer->switchRepresentation(anIO, myMode, myUpdateViewer);
400 ProcessVoidEvent(new TEvent (theEntry, theMode, isUpdated));
403 void GEOM_Swig::setColor(const char* theEntry, int red, int green, int blue, bool isUpdated)
405 class TEvent: public SALOME_Event {
412 TEvent(const char* theEntryArg, int theR, int theG, int theB, bool theUpdated):
413 myEntry(theEntryArg), myRed(theR), myGreen(theG), myBlue(theB), myUpdateViewer(theUpdated)
415 virtual void Execute() {
416 SUIT_Application* anApp = SUIT_Session::session()->activeApplication();
419 Handle(SALOME_InteractiveObject) anIO =
420 new SALOME_InteractiveObject(myEntry.c_str(), "GEOM", "");
422 if (SVTK_ViewWindow* aViewWindow = GetSVTKViewWindow(anApp)){
423 SVTK_View* aView = aViewWindow->getView();
424 QColor aColor (myRed, myGreen, myBlue);
425 aView->SetColor(anIO, aColor);
428 } else if (OCCViewer_Viewer* occViewer = GetOCCViewer(anApp)) {
429 Handle(AIS_InteractiveContext) ic = occViewer->getAISContext();
430 AIS_ListOfInteractive List;
431 ic->DisplayedObjects(List);
432 AIS_ListIteratorOfListOfInteractive ite (List);
433 for (; ite.More(); ite.Next()) {
434 Handle(SALOME_InteractiveObject) anObj =
435 Handle(SALOME_InteractiveObject)::DownCast(ite.Value()->GetOwner());
436 if (!anObj.IsNull() && anObj->hasEntry() && anObj->isSame(anIO)) {
437 Quantity_Color CSFColor =
438 Quantity_Color(myRed/255., myGreen/255., myBlue/255., Quantity_TOC_RGB);
439 ite.Value()->SetColor(CSFColor);
440 if (ite.Value()->IsKind(STANDARD_TYPE(GEOM_AISShape)))
441 Handle(GEOM_AISShape)::DownCast(ite.Value())->SetShadingColor(CSFColor);
442 ite.Value()->Redisplay(Standard_True);
451 ProcessVoidEvent(new TEvent(theEntry, red, green, blue, isUpdated));
454 void GEOM_Swig::setTransparency(const char* theEntry, float transp, bool isUpdated)
456 class TEvent: public SALOME_Event {
461 TEvent(const char* theEntryArg, float theParam, bool theUpdated):
462 myEntry(theEntryArg), myParam(theParam), myUpdateViewer(theUpdated)
464 virtual void Execute() {
465 SUIT_Application* anApp = SUIT_Session::session()->activeApplication();
468 Handle(SALOME_InteractiveObject) anIO =
469 new SALOME_InteractiveObject(myEntry.c_str(), "GEOM", "");
471 if (SVTK_ViewWindow* aViewWindow = GetSVTKViewWindow(anApp)) {
472 SVTK_View* aView = aViewWindow->getView();
473 aView->SetTransparency(anIO, myParam);
476 } else if (OCCViewer_Viewer* occViewer = GetOCCViewer(anApp)) {
477 SOCC_Viewer* soccViewer = dynamic_cast<SOCC_Viewer*>(occViewer);
479 soccViewer->setTransparency(anIO, myParam, myUpdateViewer);
484 ProcessVoidEvent(new TEvent (theEntry, transp, isUpdated));
488 class TInitGeomGenEvent: public SALOME_Event {
490 typedef bool TResult;
492 TInitGeomGenEvent() : myResult(false) {}
493 virtual void Execute() {
494 myResult = GeometryGUI::InitGeomGen();
497 bool GEOM_Swig::initGeomGen()
499 return ProcessEvent(new TInitGeomGenEvent());
504 void GEOM_Swig::eraseGO (const char* Entry, bool allWindows)
506 class TEvent: public SALOME_Event
509 bool myFromAllWindows;
511 TEvent(const char* theEntry, bool fromAllWindows):
512 myEntry(theEntry), myFromAllWindows(fromAllWindows)
514 virtual void Execute()
516 SUIT_Application* app = SUIT_Session::session()->activeApplication();
518 SalomeApp_Study* ActiveStudy = dynamic_cast<SalomeApp_Study*>(app->activeStudy());
519 if (!ActiveStudy) return;
521 Handle (SALOME_InteractiveObject) aIO = new SALOME_InteractiveObject(myEntry.c_str(), "GEOM", "");
523 GEOM_Displayer(ActiveStudy).Erase(aIO, true);
524 /* if (myFromAllWindows) {
525 QPtrList<SUIT_ViewWindow> aWindows = app->desktop()->windows();
526 SUIT_ViewWindow* aWin = 0;
527 for (aWin = aWindows.first(); aWin; aWin = aWindows.next()) {
528 EraseObject(aWin, aIO);
531 SUIT_ViewWindow* aWin = app->desktop()->activeWindow();
533 EraseObject(aWin, aIO);
538 void EraseObject(SUIT_ViewWindow* theWin, Handle (SALOME_InteractiveObject) theIO)
540 if (theWin->getViewManager()->getType() == OCCViewer_Viewer::Type()){
541 OCCViewer_ViewWindow* vw = dynamic_cast<OCCViewer_ViewWindow*>( theWin );
543 OCCViewer_ViewManager* vm = dynamic_cast<OCCViewer_ViewManager*>( vw->getViewManager() );
545 SOCC_Viewer* aViewer = dynamic_cast<SOCC_Viewer*>(vm->getOCCViewer());
547 SALOME_Prs* aPrs = aViewer->CreatePrs(myEntry.c_str());
549 SALOME_OCCPrs* aOccPrs = dynamic_cast<SALOME_OCCPrs*>(aPrs);
551 aViewer->Erase(aOccPrs);
558 } else if (theWin->getViewManager()->getType() == SVTK_Viewer::Type()){
559 SVTK_ViewWindow* aViewWindow = dynamic_cast<SVTK_ViewWindow*>( theWin );
561 aViewWindow->Erase(theIO);
567 ProcessVoidEvent(new TEvent(Entry, allWindows));
572 void GEOM_Swig::setDeflection(const char* theEntry, float theDeflect)
574 class TEvent: public SALOME_Event {
578 TEvent(const char* theEntryArg, float theParam):
579 myEntry(theEntryArg), myParam(theParam)
581 virtual void Execute() {
582 SUIT_Application* anApp = SUIT_Session::session()->activeApplication();
585 Handle(SALOME_InteractiveObject) anIO =
586 new SALOME_InteractiveObject(myEntry.c_str(), "GEOM", "");
588 if (SVTK_ViewWindow* aViewWindow = GetSVTKViewWindow(anApp)) {
589 vtkActorCollection* aActors = aViewWindow->getRenderer()->GetActors();
590 aActors->InitTraversal();
591 while (vtkActor* aAct = aActors->GetNextActor()) {
592 if (GEOM_Actor* aGeomActor = dynamic_cast<GEOM_Actor*>(aAct)) {
593 if (aGeomActor->hasIO()) {
594 Handle(SALOME_InteractiveObject) aNextIO = aGeomActor->getIO();
595 if (aNextIO->isSame(anIO)) {
596 aGeomActor->setDeflection(myParam);
597 aViewWindow->Repaint();
603 // aView->SetTransparency(anIO, myParam);
605 } else if (OCCViewer_Viewer* occViewer = GetOCCViewer(anApp)) {
606 Handle(AIS_InteractiveContext) aContext = occViewer->getAISContext();
607 AIS_ListOfInteractive aAISList;
608 aContext->DisplayedObjects(aAISList);
609 AIS_ListIteratorOfListOfInteractive it(aAISList);
610 for (; it.More(); it.Next()) {
611 Handle(SALOME_InteractiveObject) aObj =
612 Handle(SALOME_InteractiveObject)::DownCast(it.Value()->GetOwner());
613 if ((!aObj.IsNull()) && aObj->hasEntry() && aObj->isSame(anIO)) {
614 Handle(AIS_Shape) aShape = Handle(AIS_Shape)::DownCast(it.Value());
615 if (!aShape.IsNull()) {
616 Handle(AIS_Drawer) aDrawer = aShape->Attributes();
617 if (aDrawer.IsNull())
618 aDrawer = new AIS_Drawer();
619 aDrawer->SetDeviationCoefficient(myParam);
620 aShape->SetAttributes(aDrawer);
621 aContext->Redisplay(aShape, true, true);
622 aContext->UpdateCurrentViewer();
631 ProcessVoidEvent(new TEvent (theEntry, theDeflect));