+ Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
+ if( !aDocument.IsNull() )
+ {
+ HYDROData_Iterator anIterator( aDocument );
+ for( ; anIterator.More(); anIterator.Next() )
+ {
+ Handle(HYDROData_Entity) anObject = anIterator.Current();
+ if ( !IsObjectHasPresentation( anObject ) )
+ continue;
+
+ theSeq.Append( anObject );
+ }
+ }
+}
+
+HYDROGUI_Prs* HYDROGUI_Tool::GetPresentation( const Handle(HYDROData_Entity)& theObj,
+ const GraphicsView_ObjectList& theObjects )
+{
+ if( !theObj.IsNull() )
+ {
+ GraphicsView_ObjectListIterator anIter( theObjects );
+ while( anIter.hasNext() )
+ {
+ if( HYDROGUI_Prs* aPrs = dynamic_cast<HYDROGUI_Prs*>( anIter.next() ) )
+ {
+ Handle(HYDROData_Entity) anObj = aPrs->getObject();
+ if( IsEqual( anObj, theObj ) )
+ return aPrs;
+ }
+ }
+ }
+ return NULL;
+}
+
+GraphicsView_ObjectList HYDROGUI_Tool::GetPrsList( GraphicsView_ViewPort* theViewPort )
+{
+ GraphicsView_ObjectList aList;
+ if( theViewPort )
+ {
+ GraphicsView_ObjectListIterator anIter( theViewPort->getObjects() );
+ while( anIter.hasNext() )
+ if( HYDROGUI_Prs* aPrs = dynamic_cast<HYDROGUI_Prs*>( anIter.next() ) )
+ aList.append( aPrs );
+ }
+ return aList;
+}
+
+HYDROData_SequenceOfObjects HYDROGUI_Tool::GetSelectedObjects( HYDROGUI_Module* theModule )
+{
+ HYDROData_SequenceOfObjects aSeq;
+
+ HYDROGUI_DataModel* aModel = theModule->getDataModel();
+
+ SUIT_SelectionMgr* aSelectionMgr = theModule->getApp()->selectionMgr();
+ SUIT_DataOwnerPtrList anOwners;
+ aSelectionMgr->selected( anOwners );
+
+ QStringList aCollectedNameList; // to avoid duplication
+ foreach( SUIT_DataOwner* aSUITOwner, anOwners )
+ {
+ if( LightApp_DataOwner* anOwner = dynamic_cast<LightApp_DataOwner*>( aSUITOwner ) )
+ {
+ Handle(HYDROData_Entity) anObject = aModel->objectByEntry( anOwner->entry() );
+ if( !anObject.IsNull() )
+ {
+ QString aName = anObject->GetName();
+ if( !aCollectedNameList.contains( aName ) )
+ {
+ aSeq.Append( anObject );
+ aCollectedNameList.append( aName );
+ }
+ }
+ }
+ }
+ return aSeq;
+}
+
+Handle(HYDROData_Entity) HYDROGUI_Tool::GetSelectedObject( HYDROGUI_Module* theModule )
+{
+ HYDROData_SequenceOfObjects aSeq = GetSelectedObjects( theModule );
+ if( !aSeq.IsEmpty() )
+ return aSeq.First();
+ return NULL;
+}
+
+HYDROData_SequenceOfObjects HYDROGUI_Tool::GetGeometryObjects(
+ HYDROGUI_Module* theModule )
+{
+ Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
+
+ HYDROData_SequenceOfObjects aResSeq;
+
+ HYDROData_Iterator anIter( aDocument );
+ for ( ; anIter.More(); anIter.Next() )
+ {
+ Handle(HYDROData_Entity) anObj = anIter.Current();
+ if ( !HYDROData_Tool::IsGeometryObject( anObj ) )
+ continue;
+
+ aResSeq.Append( anObj );
+ }
+
+ return aResSeq;
+}
+
+ObjectKind HYDROGUI_Tool::GetSelectedPartition( HYDROGUI_Module* theModule )
+{
+ HYDROGUI_DataModel* aModel = theModule->getDataModel();
+
+ SUIT_SelectionMgr* aSelectionMgr = theModule->getApp()->selectionMgr();
+ SUIT_DataOwnerPtrList anOwners;
+ aSelectionMgr->selected( anOwners );
+
+ if( anOwners.size() != 1 )
+ return KIND_UNKNOWN;
+
+ if( LightApp_DataOwner* anOwner = dynamic_cast<LightApp_DataOwner*>( anOwners.first().operator->() ) )
+ {
+ QString anEntry = anOwner->entry();
+ QString aPrefix = HYDROGUI_DataObject::entryPrefix();
+ if( anEntry.left( aPrefix.length() ) == aPrefix )
+ {
+ anEntry.remove( aPrefix );
+ for( ObjectKind anObjectKind = KIND_UNKNOWN + 1; anObjectKind <= KIND_LAST; anObjectKind++ )
+ if( HYDROGUI_DataModel::tr( HYDROGUI_DataModel::partitionName( anObjectKind ).toAscii() ) == anEntry )
+ return anObjectKind;
+ }
+ }
+ return KIND_UNKNOWN;
+}
+
+QStringList HYDROGUI_Tool::GetSelectedGeomObjects( HYDROGUI_Module* theModule,
+ QList<GEOM::shape_type> theTypes )
+{
+ QStringList anEntryList;
+
+ // Get active SalomeApp_Study
+ SalomeApp_Study* aStudy = NULL;
+ if ( theModule && theModule->getApp() ) {
+ aStudy = dynamic_cast<SalomeApp_Study*>( theModule->getApp()->activeStudy() );
+ }
+ if ( !aStudy ) {
+ return anEntryList;
+ }
+
+ // Get selection
+ SUIT_SelectionMgr* aSelectionMgr = theModule->getApp()->selectionMgr();
+ SUIT_DataOwnerPtrList anOwners;
+ aSelectionMgr->selected( anOwners );
+
+ // Check if the selected objects belong to GEOM and have a shape
+ foreach( SUIT_DataOwner* aSUITOwner, anOwners )
+ {
+ if( LightApp_DataOwner* anOwner = dynamic_cast<LightApp_DataOwner*>( aSUITOwner ) )
+ {
+ QString anEntry = anOwner->entry();
+ _PTR(SObject) aSObject( aStudy->studyDS()->FindObjectID(qPrintable(anEntry)) );
+ if (aSObject) {
+ _PTR(SComponent) aSComponent = aSObject->GetFatherComponent();
+ if ( aSComponent && aSComponent->ComponentDataType() == "GEOM" ) {
+ GEOM::GEOM_Object_var aGeomObj =
+ GEOMBase::GetObjectFromIOR( aSObject->GetIOR().c_str() );
+
+ if ( !aGeomObj->_is_nil() && aGeomObj->IsShape() &&
+ theTypes.contains( aGeomObj->GetShapeType() ) ) {
+ anEntryList << anEntry;
+ }
+ }
+ }
+ }
+ }
+
+ return anEntryList;
+}
+
+Handle(HYDROData_Entity) HYDROGUI_Tool::FindObjectByName( HYDROGUI_Module* theModule,
+ const QString& theName,
+ const ObjectKind theObjectKind )
+{
+ Handle(HYDROData_Entity) aResObj;
+
+ Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
+ if ( !aDocument.IsNull() )
+ aResObj = aDocument->FindObjectByName( theName, theObjectKind );
+
+ return aResObj;
+}
+
+HYDROData_SequenceOfObjects HYDROGUI_Tool::FindObjectsByNames( HYDROGUI_Module* theModule,
+ const QStringList& theNames,
+ const ObjectKind theObjectKind )
+{
+ HYDROData_SequenceOfObjects aResSeq;
+
+ Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
+ if ( !aDocument.IsNull() )
+ aResSeq = aDocument->FindObjectsByNames( theNames, theObjectKind );
+
+ return aResSeq;
+}
+
+QString HYDROGUI_Tool::GenerateObjectName( HYDROGUI_Module* theModule,
+ const QString& thePrefix,
+ const QStringList& theUsedNames,
+ const bool theIsTryToUsePurePrefix)
+{
+ Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( theModule->getStudyId() );
+ return HYDROData_Tool::GenerateObjectName( aDocument, thePrefix, theUsedNames, theIsTryToUsePurePrefix );
+}
+
+size_t HYDROGUI_Tool::GetActiveViewId( HYDROGUI_Module* theModule,
+ const QString& theViewId )
+{
+ size_t aViewId = 0;
+ SUIT_ViewManager* aViewMgr = theModule->getApp()->activeViewManager();
+ if( !aViewMgr || ( !theViewId.isEmpty() && aViewMgr->getType() != theViewId ) )
+ return aViewId;
+
+ if( SUIT_ViewModel* aViewer = aViewMgr->getViewModel() )
+ aViewId = (size_t)aViewer;
+ return aViewId;
+}
+
+size_t HYDROGUI_Tool::GetActiveGraphicsViewId( HYDROGUI_Module* theModule )
+{
+ return GetActiveViewId( theModule, GraphicsView_Viewer::Type() );
+}
+
+size_t HYDROGUI_Tool::GetActiveOCCViewId( HYDROGUI_Module* theModule )
+{
+ return GetActiveViewId( theModule, OCCViewer_Viewer::Type() );
+}
+
+QList<size_t> getViewIdList( HYDROGUI_Module* theModule,
+ const QString& theViewId )
+{
+ QList<size_t> aList;
+ ViewManagerList aViewMgrs;
+ theModule->getApp()->viewManagers( theViewId, aViewMgrs );
+ QListIterator<SUIT_ViewManager*> anIter( aViewMgrs );
+ while( anIter.hasNext() )
+ {
+ if( SUIT_ViewManager* aViewMgr = anIter.next() )
+ {
+ if( SUIT_ViewModel* aViewer = aViewMgr->getViewModel() )
+ aList.append( (size_t)aViewer );
+ }
+ }
+ return aList;
+}
+
+QList<size_t> HYDROGUI_Tool::GetGraphicsViewIdList( HYDROGUI_Module* theModule )
+{
+ return getViewIdList( theModule, GraphicsView_Viewer::Type() );
+}
+
+QList<size_t> HYDROGUI_Tool::GetOCCViewIdList( HYDROGUI_Module* theModule )
+{
+ return getViewIdList( theModule, OCCViewer_Viewer::Type() );
+}
+
+void HYDROGUI_Tool::setOCCActionShown( OCCViewer_ViewFrame* theViewFrame,
+ const int theActionId,
+ const bool isShown )
+{
+ if ( !theViewFrame )