1 // VISU VISUGUI : GUI of VISU component
3 // Copyright (C) 2005 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.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
24 // File : VisuGUI_Tools.cxx
25 // Author : Sergey Anikin
29 #include "VisuGUI_Tools.h"
31 #include "VISU_Gen_i.hh"
32 #include "VISU_Prs3d_i.hh"
33 #include "VISU_Result_i.hh"
34 #include "VISU_Table_i.hh"
35 #include "VISU_Mesh_i.hh"
36 #include "VISU_ViewManager_i.hh"
38 #include "VISU_Actor.h"
40 #include "SalomeApp_Module.h"
41 #include "SalomeApp_Study.h"
42 #include "SalomeApp_Application.h"
43 #include "SalomeApp_SelectionMgr.h"
45 #include "SALOME_ListIO.hxx"
46 #include "SALOME_ListIteratorOfListIO.hxx"
48 #include "SVTK_ViewWindow.h"
49 #include "SVTK_ViewModel.h"
50 #include "SVTK_Functor.h"
52 #include "VTKViewer_Algorithm.h"
54 #include "SPlot2d_ViewModel.h"
55 #include "Plot2d_ViewFrame.h"
57 #include "SUIT_Session.h"
58 #include "SUIT_MessageBox.h"
60 #include <vtkRenderer.h>
61 #include <vtkActorCollection.h>
64 //=============================================================================
68 GetDesktop(const CAM_Module* theModule)
70 return theModule->application()->desktop();
73 SalomeApp_SelectionMgr*
74 GetSelectionMgr(const SalomeApp_Module* theModule)
76 return theModule->getApp()->selectionMgr();
80 GetAppStudy(const CAM_Module* theModule)
83 dynamic_cast<SalomeApp_Study*>(theModule->application()->activeStudy());
87 GetCStudy(const SalomeApp_Study* theStudy)
89 return theStudy->studyDS();
93 IsStudyLocked( _PTR(Study) theStudy )
96 return theStudy->GetProperties()->IsLocked();
101 CheckLock( _PTR(Study) theStudy )
103 if(IsStudyLocked(theStudy))
104 throw std::runtime_error(QObject::tr("WRN_STUDY_LOCKED").latin1());
109 GetVisuGen(const CAM_Module* theModule)
111 static VISU_Gen_i* aGen = NULL;
113 SALOME_LifeCycleCORBA aLCC(SalomeApp_Application::namingService());
114 Engines::Component_var aComponent = aLCC.FindOrLoad_Component("FactoryServer","VISU");
115 VISU_Gen_var aVISU = VISU_Gen::_narrow(aComponent);
116 if(!CORBA::is_nil(aVISU)){
117 aGen = VISU_Gen_i::GetVisuGenImpl();
118 aGen->SetCurrentStudy(GetDSStudy(GetCStudy(GetAppStudy(theModule))));
122 throw std::runtime_error(QObject::tr("ERR_CANT_FIND_VISU_COMPONENT").latin1());
126 SALOME_MED::MED_Gen_var
129 static SALOME_MED::MED_Gen_var aGen;
130 if(CORBA::is_nil(aGen)){
131 SALOME_LifeCycleCORBA aLCC(SalomeApp_Application::namingService());
132 Engines::Component_var aComponent = aLCC.FindOrLoad_Component("FactoryServer","MED");
133 aGen = SALOME_MED::MED_Gen::_narrow(aComponent);
135 if(CORBA::is_nil(aGen))
136 throw std::runtime_error(QObject::tr("ERR_CANT_FIND_MED_COMPONENT").latin1());
141 VISU::Storable::TRestoringMap getMapOfValue (_PTR(SObject) theSObject)
143 VISU::Storable::TRestoringMap aMap;
145 _PTR(GenericAttribute) anAttr;
146 if (theSObject->FindAttribute(anAttr, "AttributeComment")) {
147 _PTR(AttributeComment) aComment (anAttr);
148 std::string aValue = aComment->Value();
149 QString aString (aValue.c_str());
150 VISU::Storable::StrToMap(aString, aMap);
156 QString getValue (_PTR(SObject) theSObject, QString theKey)
159 VISU::Storable::TRestoringMap aMap = getMapOfValue(theSObject);
161 aStr = VISU::Storable::FindValue(aMap, theKey.latin1());
165 //************************************************************
168 GetSelectedObj(const SalomeApp_Study* theStudy,
169 const QString& theEntry,
170 VISU::Storable::TRestoringMap* theMap)
172 if (!theStudy || theEntry.isEmpty())
173 return CORBA::Object::_nil();
175 _PTR(Study) aStudy = GetCStudy(theStudy);
176 _PTR(SObject) aSObject = aStudy->FindObjectID(theEntry.latin1());
178 _PTR(GenericAttribute) anAttr;
179 if (theMap && aSObject->FindAttribute(anAttr,"AttributeComment")) {
180 _PTR(AttributeComment) aComment (anAttr);
181 std::string aValue = aComment->Value();
182 QString aString (aValue.c_str());
183 VISU::Storable::StrToMap(aString, *theMap);
185 return VISU::ClientSObjectToObject(aSObject);
187 return CORBA::Object::_nil();
191 GetSelectedObj(const SalomeApp_Module* theModule,
192 Handle(SALOME_InteractiveObject)* theIO,
193 VISU::Storable::TRestoringMap* theMap)
195 SalomeApp_SelectionMgr* aSelectionMgr = GetSelectionMgr(theModule);
196 SALOME_ListIO aListIO;
197 aSelectionMgr->selectedObjects(aListIO);
198 SALOME_ListIteratorOfListIO anIter(aListIO);
200 Handle(SALOME_InteractiveObject) anIO = anIter.Value();
204 return GetSelectedObj( GetAppStudy(theModule), anIO->getEntry(), theMap );
206 return CORBA::Object::_nil();
210 Add(SalomeApp_SelectionMgr* theSelectionMgr,
211 const Handle(SALOME_InteractiveObject)& theIO)
213 SALOME_ListIO aListIO;
214 theSelectionMgr->selectedObjects(aListIO);
215 aListIO.Append(theIO);
216 theSelectionMgr->setSelectedObjects(aListIO);
220 Remove(SalomeApp_SelectionMgr* theSelectionMgr,
221 const Handle(SALOME_InteractiveObject)& theIO)
223 if (theIO.IsNull()) return;
224 SALOME_ListIO aListIO, aNewListIO;
225 theSelectionMgr->selectedObjects(aListIO);
226 SALOME_ListIteratorOfListIO anIter (aListIO);
227 for (; anIter.More(); anIter.Next()) {
228 Handle(SALOME_InteractiveObject) anIO = anIter.Value();
229 if (!anIO->isSame(theIO)) {
230 aNewListIO.Append(theIO);
233 theSelectionMgr->setSelectedObjects(aNewListIO);
237 * \brief Check, if "Delete" popup-menu can be put on current selection
239 * \param theModule - is used to access SelectionManager, Study and VISU_Gen
240 * \retval bool - returns TRUE if all currently selected objects are removable
243 IsRemovableSelected (const SalomeApp_Module* theModule)
245 SalomeApp_SelectionMgr* aSelectionMgr = GetSelectionMgr(theModule);
246 SALOME_ListIO aListIO;
247 aSelectionMgr->selectedObjects(aListIO);
249 if (aListIO.Extent() < 1)
252 _PTR(Study) aStudy = GetCStudy(GetAppStudy(theModule));
253 if (!aStudy) return false;
255 // In some cases single selection can have its own popup-menu item for deletion
256 /*if (aListIO.Extent() == 1) {
257 Handle(SALOME_InteractiveObject) anIO = aListIO.First();
258 _PTR(SObject) aSObject = aStudy->FindObjectID(anIO->getEntry());
260 VISU::VISUType aType = (VISU::VISUType)getValue(aSObject, "myType").toInt();
261 if (aType == VISU::TVIEW3D) {
262 return false; // special case
267 SALOME_ListIteratorOfListIO anIter (aListIO);
268 for (; anIter.More(); anIter.Next()) {
269 Handle(SALOME_InteractiveObject) anIO = anIter.Value();
270 if (anIO->hasEntry()) {
271 _PTR(SObject) aSObject = aStudy->FindObjectID(anIO->getEntry());
272 VISU::Storable::TRestoringMap pMap;
274 CORBA::Object_var aCORBAObject = VISU::ClientSObjectToObject(aSObject);
275 if (!CORBA::is_nil(aCORBAObject)) {
276 VISU::RemovableObject_var aRemovableObj = VISU::RemovableObject::_narrow(aCORBAObject);
277 if (CORBA::is_nil(aRemovableObj)) {
278 // Not removable CORBA object
282 // Can be removed, if lays directly under VISU
283 // (first sub-level) or is a child of such an object
284 string aNAME, aVisuNAME = GetVisuGen(theModule)->ComponentDataType();
285 _PTR(GenericAttribute) anAttr;
286 _PTR(AttributeComment) aComment;
288 bool isUnderVISU = false;
289 _PTR(SObject) aFatherSObject = aSObject->GetFather();
290 if (aFatherSObject->FindAttribute(anAttr, "AttributeComment")) {
291 _PTR(AttributeComment) aComment (anAttr);
292 aNAME = aComment->Value();
293 if (aNAME == aVisuNAME) {
298 // Not directly under VISU component, check father
299 aCORBAObject = VISU::ClientSObjectToObject(aFatherSObject);
300 if (!CORBA::is_nil(aCORBAObject)) {
306 aFatherSObject = aFatherSObject->GetFather();
307 if (aFatherSObject->FindAttribute(anAttr, "AttributeComment")) {
308 _PTR(AttributeComment) aComment (anAttr);
309 aNAME = aComment->Value();
310 if (aNAME == aVisuNAME) {
315 // Father is not directly under VISU component
326 //************************************************************
330 ErasePrs (const SalomeApp_Module* theModule,
331 CORBA::Object_ptr theObject, bool theUpdate)
333 if (!CORBA::is_nil(theObject)) {
334 VISU::Base_var aBase = VISU::Base::_narrow(theObject);
335 if (CORBA::is_nil(aBase)) return;
336 VISU::VISUType aType = aBase->GetType();
340 if (VISU::Curve_i* aCurve = dynamic_cast<VISU::Curve_i*>(VISU::GetServant(aBase).in()))
341 PlotCurve(theModule, aCurve, VISU::eErase );
344 case VISU::TCONTAINER:
346 if (VISU::Container_i* aContainer = dynamic_cast<VISU::Container_i*>(VISU::GetServant(aBase).in()))
347 PlotContainer(theModule, aContainer, VISU::eErase );
352 if (VISU::Table_i* aTable = dynamic_cast<VISU::Table_i*>(VISU::GetServant(aBase).in()))
353 PlotTable(theModule, aTable, VISU::eErase );
358 if (VISU::Prs3d_i* aPrsObject = dynamic_cast<VISU::Prs3d_i*>(VISU::GetServant(aBase).in())) {
359 ErasePrs3d(theModule, aPrsObject);
361 if (SVTK_ViewWindow* vw = GetViewWindow(theModule))
371 DeleteSObject (SalomeApp_Module* theModule,
372 _PTR(Study) theStudy,
373 _PTR(SObject) theSObject)
375 _PTR(ChildIterator) aChildIter = theStudy->NewChildIterator(theSObject);
376 for (aChildIter->InitEx(true); aChildIter->More(); aChildIter->Next()) {
377 _PTR(SObject) aChildSObject = aChildIter->Value();
378 CORBA::Object_var aChildObj = VISU::ClientSObjectToObject(aChildSObject);
379 ErasePrs(theModule, aChildObj);
382 CORBA::Object_var anObj = VISU::ClientSObjectToObject(theSObject);
383 if (!CORBA::is_nil(anObj)) {
384 ErasePrs(theModule, anObj);
386 VISU::RemovableObject_var aRemovableObject = VISU::RemovableObject::_narrow(anObj);
387 if (!CORBA::is_nil(aRemovableObject)) {
388 aRemovableObject->RemoveFromStudy();
391 // Remove aSObject together with all its sub-objects
392 VISU::RemoveFromStudy(theSObject,
393 false, // remove not only IOR attribute, but Object With Children
394 false); // not Destroy() sub-objects
399 DeletePrs3d(SalomeApp_Module* theModule,
400 VISU::Prs3d_i* thePrs,
401 const Handle(SALOME_InteractiveObject)& theIO)
405 if (CheckLock(GetCStudy(GetAppStudy(theModule))))
407 SALOMEDS::SObject_var aSObject = thePrs->GetSObject();
408 CORBA::String_var anEntry = aSObject->GetID();
409 SalomeApp_SelectionMgr* aSelectionMgr = GetSelectionMgr(theModule);
410 Remove(aSelectionMgr,theIO);
411 TViewWindows aViewWindows = GetViews(theModule);
412 for(int i = 0, iEnd = aViewWindows.size(); i < iEnd; i++){
413 SVTK_ViewWindow* aView = aViewWindows[i];
414 if(VISU_Actor* anActor = FindActor(aView,anEntry.in())){
415 aView->RemoveActor(anActor);
419 thePrs->RemoveFromStudy();
423 ErasePrs3d(const SalomeApp_Module* theModule,
424 VISU::Prs3d_i* thePrs)
426 if ( SVTK_ViewWindow* vw = GetViewWindow( theModule ) ){
427 VISU_Actor* anVISUActor = FindActor( vw, thePrs );
429 anVISUActor->VisibilityOff();
434 //************************************************************
435 // Presentation management
438 ChangeRepresentation (const SalomeApp_Module* theModule,
439 VISU::PresentationType theType)
441 SUIT_ViewWindow* aView = GetActiveView(theModule, VTKViewer_Viewer::Type());
443 SVTK_ViewWindow* vw = (SVTK_ViewWindow*) aView;
445 Handle(SALOME_InteractiveObject) anIO;
446 CORBA::Object_var anObject = GetSelectedObj(theModule, &anIO);
447 if (CORBA::is_nil(anObject)) return;
448 PortableServer::ServantBase_var aServant = VISU::GetServant(anObject);
449 if (!aServant.in()) return;
451 VISU::Prs3d_i* aPrs3d = dynamic_cast<VISU::Prs3d_i*>(aServant.in());
453 if (VISU_Actor* anActor = GetActor(aPrs3d, vw)) {
456 if (anActor->IsShrunk())
459 anActor->SetShrink();
462 if (VISU::Mesh_i* aMesh = dynamic_cast<VISU::Mesh_i*>(aPrs3d)) {
463 aMesh->SetPresentationType(theType);
464 RecreateActor(theModule, aMesh);
466 anActor->SetRepresentation(theType);
474 //************************************************************
478 CheckTimeStamp(const SalomeApp_Module* theModule,
479 _PTR(SObject)& theSObject,
480 Handle(SALOME_InteractiveObject)* theIO)
482 Handle(SALOME_InteractiveObject) anIO;
483 CORBA::Object_var anObject = GetSelectedObj(theModule, &anIO);
486 if (!anIO.IsNull() && anIO->hasEntry()){
487 _PTR(Study) aStudy = GetCStudy(GetAppStudy(theModule));
488 theSObject = aStudy->FindObjectID(anIO->getEntry());
489 QString aValue = getValue(theSObject,"myType");
490 if (aValue.toInt() == int(VISU::TTIMESTAMP))
493 SUIT_MessageBox::warn1(GetDesktop(theModule),
494 QObject::tr("WRN_VISU"),
495 QObject::tr("WRN_NO_AVAILABLE_DATA"),
496 QObject::tr("BUT_OK") );
501 CheckResult(const SalomeApp_Module* theModule,
502 _PTR(SObject) theSource,
503 VISU::Result_var& theResult)
505 _PTR(SObject) aSObj = theSource->GetFather();
509 aSObj = aSObj->GetFather();
513 aSObj = aSObj->GetFather();
517 CORBA::Object_var anObject = VISU::ClientSObjectToObject(aSObj);
518 if (CORBA::is_nil(anObject)) {
519 aSObj = aSObj->GetFather();
522 anObject = VISU::ClientSObjectToObject(aSObj);
525 if (CORBA::is_nil(anObject))
528 theResult = VISU::Result::_narrow(anObject);
529 VISU::Result_i* pResult = dynamic_cast<VISU::Result_i*>(VISU::GetServant(anObject).in());
531 SUIT_MessageBox::warn1(GetDesktop(theModule),
532 QObject::tr("WRN_VISU"),
533 QObject::tr("WRN_NO_AVAILABLE_DATA"),
534 QObject::tr("BUT_OK"));
538 //************************************************************
541 SUIT_ViewWindow* GetActiveView(const SalomeApp_Module* theModule, QString theType)
543 if(SalomeApp_Application* anApp = theModule->getApp()){
544 if(SUIT_ViewManager* aViewManager = anApp->activeViewManager()){
545 if (!theType.isNull()) {
546 if (aViewManager->getType() != theType)
549 return aViewManager->getActiveView();
555 //************************************************************
559 GetViews(const SalomeApp_Module* theModule)
561 TViewWindows aViewWindows;
562 if(SalomeApp_Application* anApp = theModule->getApp()){
563 ViewManagerList aViewManagerList;
564 anApp->viewManagers(SVTK_Viewer::Type(),aViewManagerList);
565 QPtrListIterator<SUIT_ViewManager> anIter(aViewManagerList);
566 while(SUIT_ViewManager* aViewManager = anIter.current()){
567 QPtrVector<SUIT_ViewWindow> aViews = aViewManager->getViews();
568 for(int i = 0, iEnd = aViews.size(); i < iEnd; i++){
569 if(SUIT_ViewWindow* aViewWindow = aViews.at(i))
570 if(SVTK_ViewWindow* aView = dynamic_cast<SVTK_ViewWindow*>(aViewWindow))
571 aViewWindows.push_back(aView);
580 GetViewWindow(const SalomeApp_Module* theModule, const bool theCreate )
582 if(SalomeApp_Application* anApp = theModule->getApp()){
583 if(SUIT_ViewManager* aViewManager = anApp->getViewManager( SVTK_Viewer::Type(), theCreate )){
584 if(SUIT_ViewWindow* aViewWindow = aViewManager->getActiveView()){
585 return dynamic_cast<SVTK_ViewWindow*>(aViewWindow);
596 SalomeApp_Application* anApp = dynamic_cast<SalomeApp_Application*>
597 (SUIT_Session::session()->activeApplication());
599 if (SUIT_ViewManager* aViewManager = anApp->activeViewManager()) {
600 if (aViewManager->getType() == SVTK_Viewer::Type()) {
601 if (SUIT_ViewWindow* aViewWindow = aViewManager->getActiveView()) {
602 return dynamic_cast<SVTK_ViewWindow*>(aViewWindow);
611 PublishInView(const SalomeApp_Module* theModule,
612 VISU::Prs3d_i* thePrs)
614 VISU_Actor* aActor = NULL;
617 if(SVTK_ViewWindow* aView = GetViewWindow(theModule)){
618 QApplication::setOverrideCursor( Qt::waitCursor );
620 if(aActor = thePrs->CreateActor())
621 aView->AddActor(aActor);
622 }catch(std::exception& exc){
623 SUIT_MessageBox::warn1(GetDesktop(theModule),
624 QObject::tr("WRN_VISU"),
625 QObject::tr("ERR_CANT_CREATE_ACTOR"),
626 QObject::tr("BUT_OK"));
628 QApplication::restoreOverrideCursor();
634 UpdateViewer(const SalomeApp_Module* theModule,
635 VISU::Prs3d_i* thePrs,
638 SVTK_ViewWindow* vw = GetViewWindow( theModule );
639 if (!vw) return NULL;
641 vtkRenderer *aRen = vw->getRenderer();
642 vtkActorCollection *anActColl = aRen->GetActors();
645 VISU_Actor* anVISUActor = NULL;
646 VISU_Actor* aResActor = NULL;
647 for(anActColl->InitTraversal(); (anActor = anActColl->GetNextActor()) != NULL; ){
648 if(!SALOME_Actor::SafeDownCast(anActor))
650 if(anActor->IsA("VISU_Actor")){
651 anVISUActor = VISU_Actor::SafeDownCast(anActor);
652 VISU::Prs3d_i* aPrs = anVISUActor->GetPrs3d();
653 if(aPrs == NULL) continue;
654 if (thePrs == aPrs) {
655 aResActor = anVISUActor->GetParent();
656 thePrs->UpdateActor(aResActor);
657 aResActor->VisibilityOn();
659 } else if (theDispOnly) {
660 anVISUActor->GetParent()->VisibilityOff();
663 } else if (theDispOnly && anActor->GetVisibility()) {
664 anActor->VisibilityOff();
671 anVISUActor = PublishInView( theModule, thePrs );
676 RepaintViewWindows (const SalomeApp_Module* theModule,
677 const Handle(SALOME_InteractiveObject)& theIObject)
679 TViewWindows aViewWindows;
680 if (SalomeApp_Application* anApp = theModule->getApp()) {
681 ViewManagerList aViewManagerList;
682 anApp->viewManagers(SVTK_Viewer::Type(),aViewManagerList);
683 QPtrListIterator<SUIT_ViewManager> anIter (aViewManagerList);
684 while (SUIT_ViewManager* aViewManager = anIter.current()) {
685 QPtrVector<SUIT_ViewWindow> aViews = aViewManager->getViews();
686 for (int i = 0, iEnd = aViews.size(); i < iEnd; i++) {
687 if (SUIT_ViewWindow* aViewWindow = aViews.at(i)) {
688 if (SVTK_ViewWindow* vw = dynamic_cast<SVTK_ViewWindow*>(aViewWindow)) {
689 if (vw->isVisible(theIObject)) {
690 vw->getRenderer()->ResetCameraClippingRange();
692 vw->highlight(theIObject, true, true);
703 FindActor(SVTK_ViewWindow* theViewWindow,
704 const char* theEntry)
707 if(vtkRenderer* aRenderer = theViewWindow->getRenderer()){
708 if(vtkActorCollection* aCollection = aRenderer->GetActors()){
709 if(VISU_Actor* anActor = Find<VISU_Actor>(aCollection,TIsSameEntry<VISU_Actor>(theEntry))){
710 return anActor->GetParent();
718 FindActor(SVTK_ViewWindow* theViewWindow,
719 VISU::Prs3d_i* thePrs)
721 SALOMEDS::SObject_var aSObject = thePrs->GetSObject();
722 CORBA::String_var anEntry = aSObject->GetID();
723 return FindActor(theViewWindow,anEntry.in());
727 RecreateActor (const SalomeApp_Module* theModule,
728 VISU::Prs3d_i* thePrs)
730 QApplication::setOverrideCursor(Qt::waitCursor);
732 SALOMEDS::SObject_var aSObject = thePrs->GetSObject();
733 CORBA::String_var anEntry = aSObject->GetID();
738 TViewWindows aViewWindows = GetViews(theModule);
739 for (int i = 0, iEnd = aViewWindows.size(); i < iEnd; i++) {
740 SVTK_ViewWindow* aView = aViewWindows[i];
741 if (VISU_Actor* anActor = FindActor(aView, anEntry.in())) {
742 thePrs->UpdateActor(anActor);
745 } catch (std::runtime_error& ex) {
747 QApplication::restoreOverrideCursor();
748 SUIT_MessageBox::warn1 (GetDesktop(theModule), QObject::tr("WRN_VISU"),
749 QObject::tr("ERR_CANT_BUILD_PRESENTATION") + " " + QObject::tr(ex.what()),
750 QObject::tr("BUT_OK"));
752 TViewWindows aViewWindows = GetViews(theModule);
753 for (int i = 0, iEnd = aViewWindows.size(); i < iEnd; i++) {
754 SVTK_ViewWindow* aView = aViewWindows[i];
755 if (VISU_Actor* anActor = FindActor(aView, anEntry.in())) {
756 aView->RemoveActor(anActor);
762 QApplication::restoreOverrideCursor();
765 //************************************************************
769 GetPlot2dViewer(const SalomeApp_Module* theModule, const bool theCreate)
771 if(SalomeApp_Application* anApp = theModule->getApp()){
772 if(SUIT_ViewManager* aViewManager = anApp->getViewManager( Plot2d_Viewer::Type(), theCreate )){
773 return dynamic_cast<SPlot2d_Viewer*>(aViewManager->getViewModel());
779 // Internal function used by several public functions below
781 UpdateCurve(VISU::Curve_i* theCurve,
782 Plot2d_ViewFrame* aPlot,
783 SPlot2d_Curve* plotCurve,
786 if ( theDisplaying == VISU::eErase ) {
788 aPlot->eraseCurve( plotCurve, false );
790 else if ( theDisplaying == VISU::eDisplay || theDisplaying == VISU::eDisplayOnly ) {
792 plotCurve->setHorTitle( theCurve->GetHorTitle().c_str() );
793 //plotCurve->setVerTitle( ( theCurve->GetVerTitle().c_str() ) );
794 plotCurve->setVerTitle( theCurve->GetName() );
795 plotCurve->setHorUnits( theCurve->GetHorUnits().c_str() );
796 plotCurve->setVerUnits( theCurve->GetVerUnits().c_str() );
799 int nbPoints = theCurve->GetData( xList, yList );
800 if ( nbPoints > 0 && xList && yList ) {
801 plotCurve->setData( xList, yList, nbPoints );
803 if ( !theCurve->IsAuto() ) {
804 plotCurve->setLine( (Plot2d_Curve::LineType)theCurve->GetLine(), theCurve->GetLineWidth() );
805 plotCurve->setMarker( (Plot2d_Curve::MarkerType)theCurve->GetMarker() );
806 SALOMEDS::Color color = theCurve->GetColor();
807 plotCurve->setColor( QColor( (int)(color.R*255.), (int)(color.G*255.), (int)(color.B*255.) ) );
809 plotCurve->setAutoAssign( theCurve->IsAuto() );
810 aPlot->displayCurve( plotCurve, false );
813 Plot2d_Curve* crv = theCurve->CreatePresentation();
815 aPlot->displayCurve( crv, false );
816 theCurve->SetLine( (VISU::Curve::LineType)crv->getLine(), crv->getLineWidth() );
817 theCurve->SetMarker( (VISU::Curve::MarkerType)crv->getMarker());
818 SALOMEDS::Color newColor;
819 newColor.R = crv->getColor().red()/255.;
820 newColor.G = crv->getColor().green()/255.;
821 newColor.B = crv->getColor().blue()/255.;
822 theCurve->SetColor( newColor );
823 crv->setAutoAssign( theCurve->IsAuto() );
830 PlotTable(const SalomeApp_Module* theModule,
831 VISU::Table_i* table,
834 SPlot2d_Viewer* aView = GetPlot2dViewer( theModule, true ); // create if necessary
837 Plot2d_ViewFrame* aPlot = aView->getActiveViewFrame();
841 if ( theDisplaying == VISU::eDisplayOnly )
843 QList<Plot2d_Curve> clist;
844 aPlot->getCurves( clist );
845 _PTR(Study) aStudy = GetCStudy( GetAppStudy( theModule ) );
846 _PTR(SObject) TableSO = aStudy->FindObjectID( table->GetEntry().latin1() );
848 _PTR(ChildIterator) Iter = aStudy->NewChildIterator( TableSO );
849 for ( ; Iter->More(); Iter->Next() ) {
850 CORBA::Object_var childObject = VISU::ClientSObjectToObject( Iter->Value() );
851 if( !CORBA::is_nil( childObject ) ) {
852 CORBA::Object_ptr aCurve = VISU::Curve::_narrow( childObject );
853 if( !CORBA::is_nil( aCurve ) ) {
854 VISU::Curve_i* theCurve = dynamic_cast<VISU::Curve_i*>(VISU::GetServant(aCurve).in());
855 SPlot2d_Curve* plotCurve = 0;
856 SPlot2d_Curve* tmpCurve;
857 for ( int i = 0; i < clist.count(); i++ ) {
858 tmpCurve = dynamic_cast<SPlot2d_Curve*>( clist.at( i ) );
859 if (tmpCurve && tmpCurve->hasIO() &&
860 !strcmp(tmpCurve->getIO()->getEntry(), theCurve->GetEntry())) {
861 plotCurve = tmpCurve;
866 UpdateCurve( theCurve, aPlot, plotCurve, theDisplaying );
868 if ( theDisplaying == VISU::eErase && plotCurve ) {
869 clist.remove( plotCurve );
879 PlotCurve(const SalomeApp_Module* theModule,
880 VISU::Curve_i* theCurve,
883 SPlot2d_Viewer* aView = GetPlot2dViewer( theModule, true );
886 Plot2d_ViewFrame* aPlot = aView->getActiveViewFrame();
890 // if ( theDisplaying == VISU::eDisplayOnly )
891 // aPlot->EraseAll();
892 QList<Plot2d_Curve> clist;
893 aPlot->getCurves( clist );
894 SPlot2d_Curve* plotCurve = 0;
895 SPlot2d_Curve* tmpCurve;
896 for (int i = 0; i < clist.count(); i++) {
897 tmpCurve = dynamic_cast<SPlot2d_Curve*>(clist.at(i));
898 if (tmpCurve && tmpCurve->hasIO() &&
899 !strcmp(tmpCurve->getIO()->getEntry(), theCurve->GetEntry())) {
900 plotCurve = tmpCurve;
901 } else if (theDisplaying == VISU::eDisplayOnly) {
902 aPlot->eraseCurve(clist.at(i));
906 UpdateCurve(theCurve, aPlot, plotCurve, theDisplaying);
912 PlotContainer(const SalomeApp_Module* theModule,
913 VISU::Container_i* container,
916 SPlot2d_Viewer* aView = GetPlot2dViewer( theModule, true );
919 Plot2d_ViewFrame* aPlot = aView->getActiveViewFrame();
923 if ( theDisplaying == VISU::eDisplayOnly )
925 QList<Plot2d_Curve> clist;
926 aPlot->getCurves( clist );
927 if ( container->GetNbCurves() > 0 ) {
928 int nbCurves = container->GetNbCurves();
929 for ( int k = 1; k <= nbCurves; k++ ) {
930 VISU::Curve_i* theCurve = container->GetCurve( k );
931 if ( theCurve && theCurve->IsValid() ) {
932 SPlot2d_Curve* plotCurve = dynamic_cast<SPlot2d_Curve*>
933 (aView->getCurveByIO(new SALOME_InteractiveObject (theCurve->GetEntry(), "", "")));
935 UpdateCurve( theCurve, aPlot, plotCurve, theDisplaying );
937 if ( plotCurve && theDisplaying == VISU::eErase ) {
938 clist.remove( plotCurve );
947 CreatePlot(SalomeApp_Module* theModule,
948 _PTR(SObject) theTableSO)
950 if ( IsSObjectTable(theTableSO) ) {
951 CORBA::Object_var aTable = VISU::ClientSObjectToObject(theTableSO);
952 CORBA::Object_var aContainer = GetVisuGen( theModule )->CreateContainer();
953 if ( !CORBA::is_nil( aTable ) && !CORBA::is_nil( aContainer ) ) {
954 VISU::Table_i* pTable = dynamic_cast<VISU::Table_i*>(VISU::GetServant(aTable).in());
955 VISU::Container_i* pContainer = dynamic_cast<VISU::Container_i*>(VISU::GetServant(aContainer).in());
957 if ( pContainer && pTable ) {
958 for ( int i = 2; i <= pTable->GetNbRows(); i++ ) {
959 CORBA::Object_var aNewCurve = GetVisuGen( theModule )->CreateCurve( pTable->_this(), 1, i );
960 if( !CORBA::is_nil( aNewCurve ) ) {
961 VISU::Curve_i* pCrv = dynamic_cast<VISU::Curve_i*>( VISU::GetServant(aNewCurve).in() );
963 pContainer->AddCurve( pCrv->_this() );
967 PlotContainer( theModule, pContainer, VISU::eDisplay );
973 //************************************************************
976 void CreateMesh (const SalomeApp_Module* theModule,
977 const Handle(SALOME_InteractiveObject)& theIO)
979 _PTR(Study) aStudy = GetCStudy(GetAppStudy(theModule));
980 //if (CheckLock(aStudy))
983 _PTR(SObject) aResultSObj = aStudy->FindObjectID(theIO->getEntry());
986 VISU::Result_var aResult;
987 VISU::Result_i* pResult = CheckResult(theModule, aResultSObj, aResult);
991 Storable::TRestoringMap aMap = getMapOfValue(aResultSObj);
993 string aComment = Storable::FindValue(aMap,"myComment",&isExist).latin1();
997 CORBA::Object_var aMesh;
998 string aMeshName = Storable::FindValue(aMap,"myMeshName").latin1();
1003 if (aComment == "ENTITY") {
1004 VISU::Entity anEntity = (VISU::Entity)Storable::FindValue(aMap,"myId").toInt();
1005 if (VISU::Mesh_i::IsPossible(pResult,aMeshName.c_str(),anEntity))
1006 aMesh = GetVisuGen(theModule)->MeshOnEntity(aResult,aMeshName.c_str(),anEntity);
1007 } else if (aComment == "FAMILY") {
1008 VISU::Entity anEntity = (VISU::Entity)Storable::FindValue(aMap,"myEntityId").toInt();
1009 string aFamilyName = Storable::FindValue(aMap,"myName").latin1();
1010 if (VISU::Mesh_i::IsPossible(pResult,aMeshName.c_str(),anEntity,aFamilyName.c_str()))
1011 aMesh = GetVisuGen(theModule)->FamilyMeshOnEntity
1012 (aResult,aMeshName.c_str(),anEntity,aFamilyName.c_str());
1013 } else if (aComment == "GROUP") {
1014 string aGroupName = Storable::FindValue(aMap,"myName").latin1();
1015 if (VISU::Mesh_i::IsPossible(pResult,aMeshName.c_str(),aGroupName.c_str()))
1016 aMesh = GetVisuGen(theModule)->GroupMesh(aResult,aMeshName.c_str(),aGroupName.c_str());
1020 MESSAGE("VisuGUI::CreateMesh() - CREATE MESH");
1024 QApplication::restoreOverrideCursor();
1025 VISU::Mesh_i* pPresent = NULL;
1026 if (!CORBA::is_nil(aMesh))
1027 pPresent = dynamic_cast<VISU::Mesh_i*>(VISU::GetServant(aMesh).in());
1028 if (pPresent == NULL) {
1029 SUIT_MessageBox::warn1 (GetDesktop(theModule),
1030 QObject::tr("WRN_VISU"),
1031 QObject::tr("ERR_CANT_BUILD_PRESENTATION"),
1032 QObject::tr("BUT_OK"));
1036 if (SVTK_ViewWindow* aView = GetViewWindow(theModule)) {
1042 PublishInView(theModule, pPresent);
1046 MESSAGE("VisuGUI::CreateMesh() - DISPLAY MESH");
1049 theModule->application()->putInfo(QObject::tr("INF_DONE"));
1050 // Make "Save" button active
1051 theModule->getApp()->updateActions();
1052 } catch (std::runtime_error& exc) {
1054 SUIT_MessageBox::warn1 (GetDesktop(theModule),
1055 QObject::tr("WRN_VISU"),
1056 QObject::tr("ERR_CANT_CREATE_ACTOR") + " " + QObject::tr(exc.what()),
1057 QObject::tr("BUT_OK"));
1062 // ========================================================================================
1063 // GetPrs3dList: find list of presentations for the given object
1064 // ========================================================================================
1065 std::vector<VISU::Prs3d_i*> GetPrs3dList (const SalomeApp_Module* theModule,
1066 const Handle(SALOME_InteractiveObject)& theIO)
1068 std::vector<VISU::Prs3d_i*> aList;
1069 if (!theIO.IsNull() && theIO->hasEntry()) {
1070 _PTR(Study) aCStudy = GetCStudy(GetAppStudy(theModule));
1071 _PTR(SObject) aSObject = aCStudy->FindObjectID(theIO->getEntry());
1072 aList = GetPrs3dList(theModule, aSObject);
1077 std::vector<VISU::Prs3d_i*> GetPrs3dList (const SalomeApp_Module* theModule,
1078 _PTR(SObject) theObject)
1080 std::vector<VISU::Prs3d_i*> aList; int k = 0;
1084 _PTR(Study) aCStudy = GetCStudy(GetAppStudy(theModule));
1086 CORBA::Object_var aCORBAObject = VISU::ClientSObjectToObject(theObject);
1087 if (!CORBA::is_nil(aCORBAObject)) {
1088 VISU::Base_var aVisuObj = VISU::Base::_narrow(aCORBAObject);
1089 if (!CORBA::is_nil(aVisuObj)) {
1090 VISU::VISUType aType = aVisuObj->GetType();
1092 case VISU::TSCALARMAP:
1093 case VISU::TISOSURFACE:
1094 case VISU::TDEFORMEDSHAPE:
1095 case VISU::TCUTPLANES:
1096 case VISU::TCUTLINES:
1097 case VISU::TVECTORS:
1098 case VISU::TSTREAMLINES:
1101 PortableServer::ServantBase_var aServant = VISU::GetServant(aCORBAObject);
1102 if (aServant.in()) {
1103 VISU::Prs3d_i* aPrsObject = dynamic_cast<VISU::Prs3d_i*>(aServant.in());
1105 aList[k] = aPrsObject;
1112 _PTR(ChildIterator) anIter = aCStudy->NewChildIterator(theObject);
1113 _PTR(SObject) aTimeStamp;
1114 anIter->Next(); // First is reference on support
1115 for (; anIter->More(); anIter->Next()) {
1116 aTimeStamp = anIter->Value();
1117 if (!aTimeStamp) continue;
1118 std::vector<VISU::Prs3d_i*> aSubList = GetPrs3dList(theModule, aTimeStamp);
1119 if (!aSubList.empty()) {
1120 int n = aSubList.size();
1122 for (int i = 0; i < n; i++) {
1123 aList[k] = aSubList[i];
1133 _PTR(GenericAttribute) anAttr;
1134 if (theObject->FindAttribute(anAttr, "AttributeComment")) {
1135 _PTR(AttributeComment) aComment (anAttr);
1136 string aComm = aComment->Value();
1137 QString strIn (aComm.c_str());
1138 VISU::Storable::TRestoringMap pMap;
1139 VISU::Storable::StrToMap(strIn, pMap);
1141 VISU::VISUType aType =
1142 (VISU::VISUType)VISU::Storable::FindValue(pMap,"myType",&isExist).toInt();
1147 _PTR(ChildIterator) anIter = aCStudy->NewChildIterator(theObject);
1148 _PTR(SObject) aTimeStamp;
1149 anIter->Next(); // First is reference on support
1150 for (; anIter->More(); anIter->Next()) {
1151 aTimeStamp = anIter->Value();
1152 if (!aTimeStamp) continue;
1153 std::vector<VISU::Prs3d_i*> aSubList = GetPrs3dList(theModule, aTimeStamp);
1154 if (!aSubList.empty()) {
1155 int n = aSubList.size();
1157 for (int i = 0; i < n; i++) {
1158 aList[k] = aSubList[i];
1165 case VISU::TTIMESTAMP:
1167 _PTR(ChildIterator) anIter = aCStudy->NewChildIterator(theObject);
1169 for (; anIter->More(); anIter->Next()) {
1170 aPrs = anIter->Value();
1171 if (!aPrs) continue;
1172 std::vector<VISU::Prs3d_i*> aSubList = GetPrs3dList(theModule, aPrs);
1173 if (!aSubList.empty()) {
1174 int n = aSubList.size();
1176 for (int i = 0; i < n; i++) {
1177 aList[k] = aSubList[i];