1 // GEOM GEOMGUI : GUI for Geometry component
3 // Copyright (C) 2004 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 : GEOMToolsGUI_1.cxx
25 // Author : Sergey ANIKIN
29 #include <PythonConsole_PyConsole.h>
31 #include "GEOMToolsGUI.h"
33 #include "GeometryGUI.h"
34 #include "GEOMToolsGUI_TransparencyDlg.h"
35 #include "GEOMToolsGUI_NbIsosDlg.h" // Method ISOS adjustement
37 #include "GEOM_Actor.h"
40 #include "SALOME_ListIO.hxx"
41 #include "SALOME_ListIteratorOfListIO.hxx"
46 #include <SVTK_ViewModel.h>
47 #include <SVTK_ViewWindow.h>
48 #include <SVTK_View.h>
50 #include <OCCViewer_ViewModel.h>
51 #include <OCCViewer_ViewWindow.h>
53 #include <SUIT_ViewManager.h>
54 #include <SUIT_Application.h>
55 #include <SUIT_Desktop.h>
56 #include <SUIT_ResourceMgr.h>
57 #include <SUIT_Session.h>
58 #include <SUIT_OverrideCursor.h>
59 #include <SUIT_MessageBox.h>
61 #include <SalomeApp_Application.h>
62 #include <SalomeApp_Study.h>
63 #include <SalomeApp_Module.h>
65 #include <LightApp_SelectionMgr.h>
66 #include <LightApp_NameDlg.h>
68 #include <GEOMImpl_Types.hxx>
70 #include "SALOMEDSClient.hxx"
72 #include "utilities.h"
75 #include <AIS_Drawer.hxx>
76 #include <AIS_ListOfInteractive.hxx>
77 #include <AIS_ListIteratorOfListOfInteractive.hxx>
78 #include <Prs3d_IsoAspect.hxx>
79 #include <Prs3d_PointAspect.hxx>
80 #include <Graphic3d_AspectMarker3d.hxx>
83 #include <vtkBMPReader.h>
84 #include <vtkTexture.h>
85 #include <vtkTextureMapToPlane.h>
86 #include <vtkTransformTextureCoords.h>
87 #include <vtkDataSetMapper.h>
88 #include <vtkRenderer.h>
91 #include <qfileinfo.h>
92 #include <qcolordialog.h>
94 #include <qapplication.h>
100 void GEOMToolsGUI::OnSettingsColor()
102 SUIT_Session* sess = SUIT_Session::session();
103 SUIT_ResourceMgr* resMgr = sess->resourceMgr();
104 SUIT_Desktop* desk = sess->activeApplication()->desktop();
106 QColor anInitColor = resMgr->colorValue( "Geometry", "SettingsShadingColor", QColor( "yellow" ) );
108 QColor aDialogColor = QColorDialog::getColor(anInitColor, desk );
109 if( aDialogColor.isValid() )
111 QString type = desk->activeWindow()->getViewManager()->getType();
112 if( type != OCCViewer_Viewer::Type() && type != SVTK_Viewer::Type() )
113 MESSAGE("Settings Color is not supported for current Viewer");
115 resMgr->setValue( "Geometry", "SettingsShadingColor", aDialogColor );
119 void GEOMToolsGUI::OnRename()
121 SALOME_ListIO selected;
122 SalomeApp_Application* app =
123 dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
125 LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
126 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
127 if ( aSelMgr && appStudy ) {
128 aSelMgr->selectedObjects( selected );
129 if ( !selected.IsEmpty() ) {
130 _PTR(Study) aStudy = appStudy->studyDS();
132 bool aLocked = (_PTR(AttributeStudyProperties)(aStudy->GetProperties()))->IsLocked();
134 SUIT_MessageBox::warn1 ( app->desktop(),
135 QObject::tr("WRN_WARNING"),
136 QObject::tr("WRN_STUDY_LOCKED"),
137 QObject::tr("BUT_OK") );
141 bool isAny = false; // is there any appropriate object selected
142 for ( SALOME_ListIteratorOfListIO It( selected ); It.More(); It.Next() ) {
143 Handle(SALOME_InteractiveObject) IObject = It.Value();
145 _PTR(SObject) obj ( aStudy->FindObjectID(IObject->getEntry()) );
146 _PTR(GenericAttribute) anAttr;
148 if ( obj->FindAttribute(anAttr, "AttributeName") ) {
149 _PTR(AttributeName) aName (anAttr);
151 GEOM::GEOM_Object_var anObj =
152 GEOM::GEOM_Object::_narrow(GeometryGUI::ClientSObjectToObject(obj));
153 if (!CORBA::is_nil(anObj)) {
155 QString newName = LightApp_NameDlg::getName( app->desktop(), aName->Value().c_str() );
156 if (!newName.isEmpty()) {
157 aName->SetValue( newName.latin1() ); // rename the SObject
158 IObject->setName( newName.latin1() );// rename the InteractiveObject
159 anObj->SetName( newName.latin1() ); // Rename the corresponding GEOM_Object
160 (dynamic_cast<SalomeApp_Module*>(app->activeModule()))->updateObjBrowser( false );
163 } // if ( name attribute )
168 SUIT_MessageBox::warn1(app->desktop(),
169 QObject::tr("WRN_WARNING"),
170 QObject::tr("GEOM_WRN_NO_APPROPRIATE_SELECTION"),
171 QObject::tr("BUT_OK"));
178 app->updateActions(); //SRN: To update a Save button in the toolbar
181 void GEOMToolsGUI::OnCheckGeometry()
183 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
184 PythonConsole* pyConsole = app->pythonConsole();
187 pyConsole->exec("from GEOM_usinggeom import *");
190 void GEOMToolsGUI::OnAutoColor()
192 QPtrList<SALOME_Prs> aListOfGroups;
194 SALOME_ListIO selected;
195 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
199 LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
200 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
201 if( !aSelMgr || !appStudy )
204 aSelMgr->selectedObjects( selected );
205 if( selected.IsEmpty() )
208 Handle(SALOME_InteractiveObject) anIObject = selected.First();
210 _PTR(Study) aStudy = appStudy->studyDS();
211 _PTR(SObject) aMainSObject( aStudy->FindObjectID( anIObject->getEntry() ) );
212 GEOM::GEOM_Object_var aMainObject = GEOM::GEOM_Object::_narrow(GeometryGUI::ClientSObjectToObject(aMainSObject));
213 if( CORBA::is_nil( aMainObject ) )
216 aMainObject->SetAutoColor( true );
218 QValueList<SALOMEDS::Color> aReservedColors;
220 GEOM_Displayer aDisp (appStudy);
222 SALOME_View* vf = aDisp.GetActiveView();
224 SUIT_ViewWindow* window = app->desktop()->activeWindow();
225 bool isOCC = ( window && window->getViewManager()->getType() == OCCViewer_Viewer::Type() );
226 bool isVTK = ( window && window->getViewManager()->getType() == SVTK_Viewer::Type() );
228 for( _PTR(ChildIterator) it( aStudy->NewChildIterator( aMainSObject ) ); it->More(); it->Next() )
230 _PTR(SObject) aChildSObject( it->Value() );
231 GEOM::GEOM_Object_var aChildObject = GEOM::GEOM_Object::_narrow(GeometryGUI::ClientSObjectToObject(aChildSObject));
232 if( CORBA::is_nil( aChildObject ) )
235 if( aChildObject->GetType() != GEOM_GROUP )
238 SALOMEDS::Color aColor = GEOM_Displayer::getUniqueColor( aReservedColors );
239 aChildObject->SetColor( aColor );
240 aReservedColors.append( aColor );
242 QColor c( (int)( aColor.R * 255.0 ), (int)( aColor.G * 255.0 ), (int)( aColor.B * 255.0 ) );
244 SALOME_Prs* aPrs = vf->CreatePrs( aChildSObject->GetID().c_str() );
248 SVTK_ViewWindow* vtkVW = dynamic_cast<SVTK_ViewWindow*>( window );
251 SVTK_View* aView = vtkVW->getView();
252 SUIT_OverrideCursor();
253 for ( SALOME_ListIteratorOfListIO It( selected ); It.More(); It.Next() )
254 aView->SetColor( It.Value(), c );
258 OCCViewer_Viewer* vm = dynamic_cast<OCCViewer_Viewer*>( window->getViewManager()->getViewModel() );
259 Handle(AIS_InteractiveContext) ic = vm->getAISContext();
261 SOCC_Prs* anOCCPrs = dynamic_cast<SOCC_Prs*>( aPrs );
265 AIS_ListOfInteractive aList;
266 anOCCPrs->GetObjects( aList );
267 if( !aList.Extent() )
270 Handle(AIS_InteractiveObject) io = aList.First();
274 Quantity_Color aQuanColor( c.red() / 255., c.green() / 255., c.blue() / 255., Quantity_TOC_RGB );
276 // Set color for a point
277 Handle(AIS_Drawer) aCurDrawer = io->Attributes();
278 Handle(Prs3d_PointAspect) aCurPointAspect = aCurDrawer->PointAspect();
279 Quantity_Color aCurColor;
280 Standard_Real aCurScale;
281 Aspect_TypeOfMarker aCurTypeOfMarker;
282 aCurPointAspect->Aspect()->Values( aCurColor, aCurTypeOfMarker, aCurScale );
283 aCurDrawer->SetPointAspect( new Prs3d_PointAspect( aCurTypeOfMarker, aQuanColor, aCurScale) );
284 ic->SetLocalAttributes( io, aCurDrawer );
286 io->SetColor( aQuanColor );
287 if ( io->IsKind( STANDARD_TYPE(GEOM_AISShape) ) )
288 Handle(GEOM_AISShape)::DownCast( io )->SetShadingColor( aQuanColor );
290 io->Redisplay( Standard_True );
294 app->updateActions(); //SRN: To update a Save button in the toolbar
297 void GEOMToolsGUI::OnDisableAutoColor()
299 SALOME_ListIO selected;
300 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
304 LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
305 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
306 if( !aSelMgr || !appStudy )
309 aSelMgr->selectedObjects( selected );
310 if( selected.IsEmpty() )
313 Handle(SALOME_InteractiveObject) anIObject = selected.First();
315 _PTR(Study) aStudy = appStudy->studyDS();
316 _PTR(SObject) aMainSObject( aStudy->FindObjectID( anIObject->getEntry() ) );
317 GEOM::GEOM_Object_var aMainObject = GEOM::GEOM_Object::_narrow(GeometryGUI::ClientSObjectToObject(aMainSObject));
318 if( CORBA::is_nil( aMainObject ) )
321 aMainObject->SetAutoColor( false );
325 void GEOMToolsGUI::OnColor()
327 SALOME_ListIO selected;
328 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
329 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
330 if ( app && appStudy ) {
331 LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
333 aSelMgr->selectedObjects( selected );
334 if ( !selected.IsEmpty() ) {
335 SUIT_ViewWindow* window = app->desktop()->activeWindow();
336 bool isOCC = ( window && window->getViewManager()->getType() == OCCViewer_Viewer::Type() );
337 bool isVTK = ( window && window->getViewManager()->getType() == SVTK_Viewer::Type() );
339 SVTK_ViewWindow* vtkVW = dynamic_cast<SVTK_ViewWindow*>( window );
342 SVTK_View* aView = vtkVW->getView();
343 QColor initcolor = aView->GetColor( selected.First() );
344 QColor c = QColorDialog::getColor( initcolor, app->desktop() );
346 SUIT_OverrideCursor();
347 for ( SALOME_ListIteratorOfListIO It( selected ); It.More(); It.Next() ) {
348 aView->SetColor( It.Value(), c );
353 Handle(AIS_InteractiveObject) io = GEOMBase::GetAIS( selected.First() );
354 if ( !io.IsNull() ) {
355 Quantity_Color aColor;
357 QColor initcolor( (int)( aColor.Red() * 255.0 ), (int)( aColor.Green() * 255.0 ), (int)( aColor.Blue() * 255.0 ) );
358 QColor c = QColorDialog::getColor( initcolor, app->desktop() );
360 SUIT_OverrideCursor();
361 aColor = Quantity_Color( c.red() / 255., c.green() / 255., c.blue() / 255., Quantity_TOC_RGB );
362 for ( SALOME_ListIteratorOfListIO It( selected ); It.More(); It.Next() ) {
363 io = GEOMBase::GetAIS( It.Value(), true );
364 if ( !io.IsNull() ) {
365 // Set color for a point
366 OCCViewer_Viewer* vm = dynamic_cast<OCCViewer_Viewer*>( window->getViewManager()->getViewModel() );
367 Handle (AIS_InteractiveContext) ic = vm->getAISContext();
368 Handle(AIS_Drawer) aCurDrawer = io->Attributes();
369 Handle(Prs3d_PointAspect) aCurPointAspect = aCurDrawer->PointAspect();
370 Quantity_Color aCurColor;
371 Standard_Real aCurScale;
372 Aspect_TypeOfMarker aCurTypeOfMarker;
373 aCurPointAspect->Aspect()->Values( aCurColor, aCurTypeOfMarker, aCurScale );
374 aCurDrawer->SetPointAspect( new Prs3d_PointAspect( aCurTypeOfMarker, aColor, aCurScale) );
375 ic->SetLocalAttributes(io, aCurDrawer);
377 io->SetColor( aColor );
378 if ( io->IsKind( STANDARD_TYPE(GEOM_AISShape) ) )
379 Handle(GEOM_AISShape)::DownCast( io )->SetShadingColor( aColor );
381 io->Redisplay( Standard_True );
383 // store color to GEOM_Object
384 _PTR(Study) aStudy = appStudy->studyDS();
385 _PTR(SObject) aSObject( aStudy->FindObjectID( It.Value()->getEntry() ) );
386 GEOM::GEOM_Object_var anObject =
387 GEOM::GEOM_Object::_narrow(GeometryGUI::ClientSObjectToObject(aSObject));
389 SALOMEDS::Color aSColor;
390 aSColor.R = (double)c.red() / 255.0;
391 aSColor.G = (double)c.green() / 255.0;
392 aSColor.B = (double)c.blue() / 255.0;
393 anObject->SetColor( aSColor );
394 anObject->SetAutoColor( false );
398 } // first IO is not null
400 } // if ( selection not empty )
404 app->updateActions(); //SRN: To update a Save button in the toolbar
407 void GEOMToolsGUI::OnTransparency()
409 GEOMToolsGUI_TransparencyDlg dlg( SUIT_Session::session()->activeApplication()->desktop() );
413 void GEOMToolsGUI::OnNbIsos()
415 SUIT_ViewWindow* window = SUIT_Session::session()->activeApplication()->desktop()->activeWindow();
417 bool isOCC = ( window && window->getViewManager()->getType() == OCCViewer_Viewer::Type() );
418 const bool isVTK = ( window && window->getViewManager()->getType() == SVTK_Viewer::Type() );
423 OCCViewer_Viewer* vm = dynamic_cast<OCCViewer_Viewer*>( window->getViewManager()->getViewModel() );
424 Handle (AIS_InteractiveContext) ic = vm->getAISContext();
427 if ( ic->MoreCurrent() ) {
428 Handle(GEOM_AISShape) CurObject = Handle(GEOM_AISShape)::DownCast(ic->Current());
429 Handle(AIS_Drawer) CurDrawer = CurObject->Attributes();
431 int UIso = CurDrawer->UIsoAspect()->Number();
432 int VIso = CurDrawer->VIsoAspect()->Number();
434 GEOMToolsGUI_NbIsosDlg * NbIsosDlg =
435 new GEOMToolsGUI_NbIsosDlg( SUIT_Session::session()->activeApplication()->desktop() );
437 NbIsosDlg->setU( UIso );
438 NbIsosDlg->setV( VIso );
440 if ( NbIsosDlg->exec() ) {
441 SUIT_OverrideCursor();
442 for(; ic->MoreCurrent(); ic->NextCurrent()) {
443 CurObject = Handle(GEOM_AISShape)::DownCast(ic->Current());
444 Handle(AIS_Drawer) CurDrawer = CurObject->Attributes();
446 int nbUIso = NbIsosDlg->getU();
447 int nbVIso = NbIsosDlg->getV();
449 CurDrawer->SetUIsoAspect( new Prs3d_IsoAspect(Quantity_NOC_GRAY75, Aspect_TOL_SOLID, 0.5 , nbUIso) );
450 CurDrawer->SetVIsoAspect( new Prs3d_IsoAspect(Quantity_NOC_GRAY75, Aspect_TOL_SOLID, 0.5 , nbVIso) );
452 ic->SetLocalAttributes(CurObject, CurDrawer);
453 ic->Redisplay(CurObject);
457 } else if(isVTK){ // if is VTKViewer
460 // Warning. It's works incorrect. must be recheked.
462 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
465 LightApp_SelectionMgr* aSelMgr = app->selectionMgr();
468 SALOME_ListIO selected;
469 aSelMgr->selectedObjects( selected );
470 if ( selected.IsEmpty() )
473 SVTK_ViewWindow* vtkVW = dynamic_cast<SVTK_ViewWindow*>( window );
477 SALOME_View* view = GEOM_Displayer::GetActiveView();
479 vtkActorCollection* aCollection = vtkActorCollection::New();
481 for ( SALOME_ListIteratorOfListIO It( selected ); It.More(); It.Next() ) {
482 Handle(SALOME_InteractiveObject) anIObject = It.Value();
483 SALOME_Prs* aPrs = view->CreatePrs( anIObject->getEntry() );
484 SVTK_Prs* vtkPrs = dynamic_cast<SVTK_Prs*>( aPrs );
486 vtkActorCollection* anActors = vtkPrs->GetObjects();
487 anActors->InitTraversal();
488 vtkActor* anAct = anActors->GetNextActor();
489 aCollection->AddItem(anAct);
494 aCollection->InitTraversal();
501 vtkActor* anAct = aCollection->GetNextActor();
502 if (GEOM_Actor* anActor = GEOM_Actor::SafeDownCast(anAct))
503 anActor->GetNbIsos(UIso,VIso);
507 GEOMToolsGUI_NbIsosDlg* NbIsosDlg =
508 new GEOMToolsGUI_NbIsosDlg( SUIT_Session::session()->activeApplication()->desktop() );
510 NbIsosDlg->setU( UIso );
511 NbIsosDlg->setV( VIso );
513 if ( NbIsosDlg->exec() ) {
514 SUIT_OverrideCursor();
516 while( anAct!=NULL ) {
517 if(GEOM_Actor* anActor = GEOM_Actor::SafeDownCast(anAct)){
518 // There are no casting to needed actor.
519 UIso = NbIsosDlg->getU();
520 VIso = NbIsosDlg->getV();
521 int aIsos[2]={UIso,VIso};
522 anActor->SetNbIsos(aIsos);
524 anAct = aCollection->GetNextActor();
531 void GEOMToolsGUI::OnOpen()
534 SALOME_Selection* Sel = SALOME_Selection::Selection(QAD_Application::getDesktop()->getActiveStudy()->getSelection());
535 _PTR(Study) aStudy = QAD_Application::getDesktop()->getActiveStudy()->getStudyDocument();
537 SALOME_ListIteratorOfListIO It(Sel->StoredIObjects());
538 Handle(SALOME_InteractiveObject) anIObject;
539 for(;It.More();It.Next()) {
540 anIObject = It.Value();
541 _PTR(SObject) obj ( aStudy->FindObjectID(anIObject->getEntry()) );
542 _PTR(AttributePersistentRef) aPersist;
543 _PTR(AttributeIOR) anIOR;
545 // this SObject may be GEOM module root SObject
546 _PTR(ChildIterator) anIter ( aStudy->NewChildIterator() );
547 _PTR(GenericAttribute) anAttr;
548 bool useSubItems = false;
549 while (anIter->More() && !useSubItems) {
550 _PTR(SObject) subobj ( anIter->Value() );
551 if (subobj->FindAttribute(anAttr, "AttributePersistentRef")) {
558 obj->FindAttribute(anAttr, "AttributePersistentRef");
560 while(useSubItems?anIter->More():!anAttr->_is_nil()) {
561 if(!obj->FindAttribute(anAttr, "AttributeIOR") &&
562 obj->FindAttribute(anAttr, "AttributePersistentRef")) {
563 _PTR(SComponent) FComp ( obj->GetFatherComponent() );
565 if (FComp->FindAttribute(anAttr, "AttributeName")) {
566 _PTR(AttributeName) aName ( anAttr );
567 QString compName = QAD_Application::getDesktop()->getComponentName(aName->Value().c_str());
568 // parent->loadComponentData(parent->getComponentName(aName->Value()));
569 Engines::Component_var comp ;
570 if ( compName.compare("SUPERV") == 0 ) {
571 comp = QAD_Application::getDesktop()->getEngine( "SuperVisionContainer", compName) ;
574 comp = QAD_Application::getDesktop()->getEngine( "FactoryServer", compName);
575 if ( comp->_is_nil() )
576 comp = QAD_Application::getDesktop()->getEngine( "FactoryServerPy", compName);
579 if (!CORBA::is_nil(comp)) {
580 SALOMEDS::Driver_var driver = SALOMEDS::Driver::_narrow(comp);
581 if (!CORBA::is_nil(driver)) {
582 SALOMEDS::StudyBuilder_var B = dynamic_cast<SALOMEDS_Study*>(aStudy.get())->GetStudy()->NewBuilder();
583 if (!CORBA::is_nil(B)) {
584 B->LoadWith(FComp,driver);
590 MESSAGE("loadComponentData(): Driver is null");
595 MESSAGE("loadComponentData(): Engine is null");
599 // Engines::Component_var comp = QAD_Application::getDesktop()->getEngine("FactoryServer","GEOM");
600 // if (!CORBA::is_nil(comp)) {
601 // SALOMEDS::Driver_var driver = SALOMEDS::Driver::_narrow(comp);
602 // SALOMEDS::StudyBuilder_var aStudyBuilder = aStudy->NewBuilder();
603 // SALOMEDS::SComponent_var SC = aStudy->FindComponent("GEOM");
604 // if (!CORBA::is_nil(SC))
605 // aStudyBuilder->LoadWith(SC,driver);
609 MESSAGE("Component is null");
614 obj.reset( anIter->Value() );
624 void GEOMToolsGUI::OnSelectOnly(int mode)
626 SalomeApp_Application* app = dynamic_cast< SalomeApp_Application* >( SUIT_Session::session()->activeApplication() );
628 SalomeApp_Study* appStudy = dynamic_cast<SalomeApp_Study*>( app->activeStudy() );
629 GEOM_Displayer aDisp (appStudy);
630 aDisp.GlobalSelection(mode);
631 getGeometryGUI()->setLocalSelectionMode(mode);