src/GEOMGUI/GeometryGUI_Swig.cxx

   1 using namespace std;
   2 //  File      : GeometryGUI_Swig.cxx
   3 //  Created   : Tue Mar 26 15:08:38 2002
   4 //  Author    : Nicolas REJNERI, Paul RASCLE
   5 //  Project   : SALOME
   6 //  Module    : GEOM
   7 //  Copyright : Open CASCADE, EDF 2002
   8 //  $Header$
   9
  10 #include "GeometryGUI_Swig.hxx"
  11 #include "utilities.h"
  12
  13 #include "QAD_Application.h"
  14 #include "QAD_Desktop.h"
  15 #include "QAD_Study.h"
  16 #include "QAD_StudyFrame.h"
  17 #include "QAD_RightFrame.h"
  18 #include "SALOMEGUI_ImportOperation.h"
  19
  20 #include "OCCViewer_Viewer3d.h"
  21 #include "VTKViewer_RenderWindowInteractor.h"
  22 #include "VTKViewer_ViewFrame.h"
  23
  24 #include "GEOM_Actor.h"
  25 #include "GEOM_Client.hxx"
  26 #include "GEOM_AISShape.hxx"
  27 #include "GEOM_AssemblyBuilder.h"
  28 #include "GEOM_InteractiveObject.hxx"
  29
  30 #include <TopExp_Explorer.hxx>
  31 #include <TopTools_MapOfShape.hxx>
  32 #include <BRepAdaptor_Surface.hxx>
  33 #include <BRepAdaptor_Curve.hxx>
  34 #include <GeomAbs_CurveType.hxx>
  35 #include <GeomAbs_SurfaceType.hxx>
  36 #include <TopoDS.hxx>
  37 #include <TopoDS_Edge.hxx>
  38 #include <TopoDS_Face.hxx>
  39
  40 static GEOM_Client ShapeReader;
  41
  42 GeometryGUI_Swig::GeometryGUI_Swig()
  43 {
  44   // MESSAGE("Constructeur");
  45 }
  46
  47 GeometryGUI_Swig::~GeometryGUI_Swig()
  48 {
  49   // MESSAGE("Destructeur");
  50 }
  51
  52 void GeometryGUI_Swig::createAndDisplayGO(const char* Entry)
  53 {
  54   // MESSAGE("createAndDisplayGO");
  55   QAD_Study* ActiveStudy = QAD_Application::getDesktop()->getActiveStudy();
  56   SALOMEDS::Study_var aStudy = ActiveStudy->getStudyDocument();
  57   SALOMEDS::StudyBuilder_var aStudyBuilder = aStudy->NewBuilder();
  58
  59   Engines::Component_var comp = QAD_Application::getDesktop()->getEngine("FactoryServer", "Geometry");
  60   GEOM::GEOM_Gen_var Geom = GEOM::GEOM_Gen::_narrow(comp);
  61
  62   Standard_CString Fatherior = "";
  63   SALOMEDS::SComponent_var father = aStudy->FindComponent("GEOM");
  64   aStudyBuilder->DefineComponentInstance( father, Geom );
  65   father->ComponentIOR( Fatherior );
  66
  67   SALOMEDS::SObject_var fatherSF = aStudy->FindObjectID(ActiveStudy->getActiveStudyFrame()->entry());
  68
  69   SALOMEDS::SObject_var obj = aStudy->FindObjectID(Entry);
  70   SALOMEDS::GenericAttribute_var anAttr;
  71   SALOMEDS::AttributeName_var    aName;
  72   SALOMEDS::AttributeIOR_var     anIOR;
  73   // Create new actor
  74   if ( !obj->FindAttribute(anAttr, "AttributeIOR"))
  75     return;
  76   anIOR = SALOMEDS::AttributeIOR::_narrow(anAttr);
  77   GEOM::GEOM_Shape_var aShape = Geom->GetIORFromString(anIOR->Value());
  78   TopoDS_Shape Shape = ShapeReader.GetShape(Geom,aShape);
  79
  80   if ( !obj->_is_nil() ) {
  81     if (obj->FindAttribute(anAttr, "AttributeName")) {
  82       aName = SALOMEDS::AttributeName::_narrow(anAttr);
  83       // open transaction
  84       QAD_Operation* op = new SALOMEGUI_ImportOperation( ActiveStudy );
  85       op->start();
  86
  87       SALOMEDS::SObject_var newObj1 = aStudyBuilder->NewObject(fatherSF);
  88       aStudyBuilder->Addreference(newObj1, obj);
  89       // commit transaction
  90       op->finish();
  91
  92       if ( ActiveStudy->getActiveStudyFrame()->getTypeView() == VIEW_VTK )  { // VTK
  93         //vtkQGLRenderWindowInteractor* myRenderInter = ActiveStudy->getActiveStudyFrame()->getRightFrame()->getVTKView()->getRWInteractor();
  94         VTKViewer_RenderWindowInteractor* myRenderInter= ((VTKViewer_ViewFrame*)ActiveStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame())->getRWInteractor();
  95         int themode = myRenderInter->GetDisplayMode();
  96
  97         vtkActorCollection* theActors =
  98           GEOM_AssemblyBuilder::BuildActors(Shape,0,themode,Standard_True);
  99         theActors->InitTraversal();
 100         vtkActor* anActor = (vtkActor*)theActors->GetNextActor();
 101         while(!(anActor==NULL)) {
 102           GEOM_Actor* GActor = GEOM_Actor::SafeDownCast( anActor );
 103           Handle(GEOM_InteractiveObject) IO = new GEOM_InteractiveObject(anIOR->Value(),
 104                                                                          Fatherior,
 105                                                                          "GEOM");
 106           IO->setEntry(obj->GetID());
 107           GActor->setIO( IO );
 108           GActor->setName( aName->Value() );
 109
 110           myRenderInter->Display(GActor);
 111           anActor = (vtkActor*)theActors->GetNextActor();
 112         }
 113         myRenderInter->Update();
 114       }
 115       else if ( ActiveStudy->getActiveStudyFrame()->getTypeView() == VIEW_OCC ) // OCC
 116         {
 117           OCCViewer_Viewer3d* v3d = ((OCCViewer_ViewFrame*)ActiveStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame())->getViewer();
 118           //      QAD_Viewer3d* v3d = ActiveStudy->getActiveStudyFrame()->getViewerOCC();
 119           Handle(AIS_InteractiveContext) ic = v3d->getAISContext();
 120           Handle(GEOM_AISShape) aSh = new GEOM_AISShape(Shape, aName->Value());
 121           Handle(GEOM_InteractiveObject) IO = new GEOM_InteractiveObject(anIOR->Value(),
 122                                                                          Fatherior,
 123                                                                          "GEOM");
 124
 125           IO->setEntry(obj->GetID());
 126           aSh->setIO( IO );
 127           aSh->setName( aName->Value() );
 128           ic->Display (aSh);
 129           ic->AddOrRemoveCurrentObject(aSh, true);
 130         }
 131     }
 132   }
 133   ActiveStudy->updateObjBrowser( true );
 134 }
 135
 136 int  GeometryGUI_Swig::getIndexTopology(const char* SubIOR, const char* IOR)
 137 {
 138   Engines::Component_var comp = QAD_Application::getDesktop()->getEngine("FactoryServer", "Geometry");
 139   GEOM::GEOM_Gen_var  Geom = GEOM::GEOM_Gen::_narrow(comp);
 140
 141   GEOM::GEOM_Shape_var aMainShape = Geom->GetIORFromString(IOR);
 142   TopoDS_Shape shape        = ShapeReader.GetShape(Geom, aMainShape);
 143
 144   GEOM::GEOM_Shape_var aSubShape = Geom->GetIORFromString(SubIOR);
 145   TopoDS_Shape subshape    = ShapeReader.GetShape(Geom, aSubShape);
 146
 147   TopExp_Explorer Exp ( shape, subshape.ShapeType() );
 148   int index = 1;
 149   TopTools_MapOfShape M;
 150   while ( Exp.More() ) {
 151     if ( M.Add(Exp.Current()) ) {
 152       if ( Exp.Current().IsSame(subshape) )
 153         return index;
 154       index++;
 155     }
 156     Exp.Next();
 157   }
 158   return -1;
 159 }
 160
 161 const char* GeometryGUI_Swig::getShapeTypeString(const char* IOR)
 162 {
 163   Engines::Component_var comp = QAD_Application::getDesktop()->getEngine("FactoryServer", "Geometry");
 164   GEOM::GEOM_Gen_var  Geom = GEOM::GEOM_Gen::_narrow(comp);
 165
 166   GEOM::GEOM_Shape_var aShape = Geom->GetIORFromString(IOR);
 167   TopoDS_Shape shape    = ShapeReader.GetShape(Geom, aShape);
 168
 169   if( shape.IsNull() ) {
 170     return "Null Shape" ;
 171   }
 172
 173   switch (shape.ShapeType() )
 174   {
 175   case TopAbs_COMPOUND:
 176     { return "Compound" ;}
 177   case  TopAbs_COMPSOLID:
 178     { return "Compound Solid" ;}
 179   case TopAbs_SOLID:
 180     { return "Solid" ;}
 181   case TopAbs_SHELL:
 182     { return "Shell" ;}
 183   case TopAbs_FACE:
 184     {
 185       BRepAdaptor_Surface surf(TopoDS::Face(shape));
 186       if ( surf.GetType() == GeomAbs_Plane ) {
 187         return "Plane" ;
 188       } else if ( surf.GetType() == GeomAbs_Cylinder ) {
 189         return "Cylindrical Face" ;
 190       } else if ( surf.GetType() == GeomAbs_Sphere ) {
 191         return "Spherical Face" ;
 192       } else if ( surf.GetType() == GeomAbs_Torus ) {
 193         return "Toroidal Face" ;
 194       } else if ( surf.GetType() == GeomAbs_Cone ) {
 195         return "Conical Face" ;
 196       } else {
 197         return "GEOM::FACE" ;
 198       }
 199     }
 200   case TopAbs_WIRE:
 201     { return "Wire" ;}
 202   case TopAbs_EDGE:
 203     {
 204       BRepAdaptor_Curve curv(TopoDS::Edge(shape));
 205       if ( curv.GetType() == GeomAbs_Line ) {
 206         if ( (Abs(curv.FirstParameter()) >= 1E6 ) ||
 207              (Abs(curv.LastParameter()) >= 1E6 )) {
 208           return "Line" ;
 209         } else
 210           return "Edge" ;
 211       } else if ( curv.GetType() == GeomAbs_Circle ) {
 212         if ( curv.IsClosed() )
 213           return "Circle" ;
 214         else
 215           return "Arc" ;
 216       } else {
 217         return "Edge" ;
 218       }
 219     }
 220   case TopAbs_VERTEX:
 221     { return "Vertex" ;}
 222   case TopAbs_SHAPE:
 223     { return "Shape" ;}
 224   }
 225 }
 226
 227
 228 const char* GeometryGUI_Swig::getShapeTypeIcon(const char* IOR)
 229 {
 230   Engines::Component_var comp = QAD_Application::getDesktop()->getEngine("FactoryServer", "Geometry");
 231   GEOM::GEOM_Gen_var  Geom = GEOM::GEOM_Gen::_narrow(comp);
 232
 233   GEOM::GEOM_Shape_var aShape = Geom->GetIORFromString(IOR);
 234
 235   if( aShape->_is_nil() ) {
 236     return "None" ;
 237   }
 238
 239   switch (aShape->ShapeType() )
 240     {
 241     case TopAbs_COMPOUND:
 242       { return "ICON_OBJBROWSER_COMPOUND" ;}
 243     case  TopAbs_COMPSOLID:
 244       { return "ICON_OBJBROWSER_COMPSOLID" ;}
 245     case TopAbs_SOLID:
 246       { return "ICON_OBJBROWSER_SOLID" ;}
 247     case TopAbs_SHELL:
 248       { return "ICON_OBJBROWSER_SHELL" ;}
 249     case TopAbs_FACE:
 250       { return "ICON_OBJBROWSER_FACE" ;}
 251     case TopAbs_WIRE:
 252       { return "ICON_OBJBROWSER_WIRE" ;}
 253     case TopAbs_EDGE:
 254       { return "ICON_OBJBROWSER_EDGE" ;}
 255     case TopAbs_VERTEX:
 256       { return "ICON_OBJBROWSER_VERTEX" ;}
 257     }
 258   return "None";
 259 }
 260
 261 void GeometryGUI_Swig::setDisplayMode(const char* Entry, int mode)
 262 {
 263   QAD_Study* myStudy = QAD_Application::getDesktop()->getActiveStudy();
 264   Handle(SALOME_InteractiveObject) IO =
 265     myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame()->FindIObject( Entry );
 266
 267   if ( myStudy->getActiveStudyFrame()->getTypeView() == VIEW_VTK )  { // VTK
 268     VTKViewer_RenderWindowInteractor* myRenderInter =
 269       ((VTKViewer_ViewFrame*)myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame())->getRWInteractor();
 270
 271     myRenderInter->SwitchRepresentation(IO, mode);
 272     myRenderInter->Update();
 273   }
 274   else if ( myStudy->getActiveStudyFrame()->getTypeView() == VIEW_OCC ) // OCC
 275     {
 276       OCCViewer_Viewer3d* v3d = ((OCCViewer_ViewFrame*)myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame())->getViewer();
 277       v3d->SwitchRepresentation(IO, mode);
 278     }
 279 }
 280
 281 void GeometryGUI_Swig::setColor(const char* Entry, int red, int green, int blue)
 282 {
 283   QAD_Study* myStudy = QAD_Application::getDesktop()->getActiveStudy();
 284   QColor c = QColor (red, green, blue);
 285   Handle(SALOME_InteractiveObject) IO =
 286     myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame()->FindIObject( Entry );
 287
 288   if ( myStudy->getActiveStudyFrame()->getTypeView() == VIEW_VTK )  { // VTK
 289     VTKViewer_RenderWindowInteractor* myRenderInter =
 290       ((VTKViewer_ViewFrame*)myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame())->getRWInteractor();
 291     myRenderInter->SetColor(IO,c);
 292     myRenderInter->Update();
 293   }
 294   else if ( myStudy->getActiveStudyFrame()->getTypeView() == VIEW_OCC ) // OCC
 295     {
 296       OCCViewer_Viewer3d* v3d = ((OCCViewer_ViewFrame*)myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame())->getViewer();
 297       v3d->SetColor(IO,c);
 298     }
 299 }
 300
 301 void GeometryGUI_Swig::setTransparency(const char* Entry, float transp)
 302 {
 303   QAD_Study* myStudy = QAD_Application::getDesktop()->getActiveStudy();
 304   Handle(SALOME_InteractiveObject) IO =
 305     myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame()->FindIObject( Entry );
 306
 307   if ( myStudy->getActiveStudyFrame()->getTypeView() == VIEW_VTK )  { // VTK
 308     VTKViewer_RenderWindowInteractor* myRenderInter =
 309       ((VTKViewer_ViewFrame*)myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame())->getRWInteractor();
 310     myRenderInter->SetTransparency(IO,transp);
 311     myRenderInter->Update();
 312   }
 313   else if ( myStudy->getActiveStudyFrame()->getTypeView() == VIEW_OCC ) // OCC
 314     {
 315       OCCViewer_Viewer3d* v3d = ((OCCViewer_ViewFrame*)myStudy->getActiveStudyFrame()->getRightFrame()->getViewFrame())->getViewer();
 316       v3d->SetTransparency(IO,transp);
 317     }
 318 }