]> SALOME platform Git repositories - modules/geom.git/blob - src/GEOMGUI/GEOMGUI_AnnotationMgr.cxx
Salome HOME
Prepare OCCT version 7.1.0+ for SALOME-8.2.0
[modules/geom.git] / src / GEOMGUI / GEOMGUI_AnnotationMgr.cxx
1 // Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 #include "GEOMGUI_AnnotationMgr.h"
21
22 #include <GEOMGUI_AnnotationAttrs.h>
23 #include <GEOM_Annotation.hxx>
24 #include <GEOM_Client.hxx>
25 #include <GEOM_Constants.h>
26 #include <GEOM_Displayer.h>
27 #include <GEOMGUI_TextTreeWdg.h>
28
29 #include <GeometryGUI.h>
30
31 #include <SalomeApp_Application.h>
32 #include <SalomeApp_Study.h>
33 #include <SALOME_Prs.h>
34
35 #include <SUIT_ResourceMgr.h>
36 #include <SUIT_Session.h>
37 #include <SUIT_Desktop.h>
38 #include <SUIT_ViewWindow.h>
39 #include <SUIT_ViewManager.h>
40
41 #include <SOCC_Prs.h>
42 #include <SOCC_ViewWindow.h>
43
44 #include <Quantity_Color.hxx>
45 #include <TCollection_AsciiString.hxx>
46
47 #include <TopExp.hxx>
48 #include <TopoDS_Shape.hxx>
49 #include <TopTools_IndexedMapOfShape.hxx>
50 #include <gp_Ax3.hxx>
51 #include <AIS_ListIteratorOfListOfInteractive.hxx>
52
53 #include <QFont>
54 #include <QColor>
55
56
57 GEOMGUI_AnnotationMgr::GEOMGUI_AnnotationMgr( SalomeApp_Application* theApplication )
58 : myApplication( theApplication )
59 {
60 }
61
62 QString GEOMGUI_AnnotationMgr::GetEntrySeparator()
63 {
64         return "_annotation:";
65 }
66
67 //================================================================
68 // Function : CreatePresentation
69 // Purpose  :
70 //================================================================
71 SALOME_Prs* GEOMGUI_AnnotationMgr::CreatePresentation( const GEOMGUI_AnnotationAttrs::Properties& theProperty,
72                                                        GEOM::GEOM_Object_ptr theObject,
73                                                        SOCC_Viewer* theView,
74                                                        const QString& theEntry )
75 {
76   SOCC_Viewer* aView = viewOrActiveView( theView );
77   if ( !aView ) {
78     return NULL;
79   }
80
81   Handle ( GEOM_Annotation ) aPresentation = new GEOM_Annotation();
82   if ( !theEntry.isEmpty() ) {
83     // owner should be set to provide selection mechanizm
84     Handle( SALOME_InteractiveObject ) anIO = new SALOME_InteractiveObject();
85     anIO->setEntry( theEntry.toLatin1().constData() );
86     aPresentation->SetOwner( anIO );
87   }
88
89   aPresentation->SetIsScreenFixed( theProperty.IsScreenFixed );
90
91   TopoDS_Shape aShape = GEOM_Client::get_client().GetShape( GeometryGUI::GetGeomGen(), theObject );
92   if ( !aShape.IsNull() ) {
93
94     gp_Ax3 aShapeLCS = gp_Ax3().Transformed( aShape.Location().Transformation() );
95
96     GEOMGUI_AnnotationAttrs::SetupPresentation( aPresentation, theProperty, aShapeLCS );
97
98     if ( theProperty.ShapeType == TopAbs_SHAPE ) {
99
100       aPresentation->SetHilightShape( aShape );
101     }
102     else if ( theProperty.ShapeIndex > 0 ) {
103
104       TopTools_IndexedMapOfShape aSubShapeMap;
105       TopExp::MapShapes( aShape, static_cast<TopAbs_ShapeEnum>( theProperty.ShapeType ), aSubShapeMap );
106       if ( theProperty.ShapeIndex <= aSubShapeMap.Extent() ) {
107
108         aPresentation->SetHilightShape( aSubShapeMap( theProperty.ShapeIndex ) );
109       }
110     }
111   }
112
113   setDisplayProperties( aPresentation, aView, getEntry( theObject ).c_str() );
114
115   // add Prs to preview
116   SUIT_ViewWindow* vw = getApplication()->desktop()->activeWindow();
117   SOCC_Prs* aPrs =
118       dynamic_cast<SOCC_Prs*>( ( aView )->CreatePrs( 0 ) );
119
120   if ( aPrs )
121     aPrs->AddObject( aPresentation );
122
123   return aPrs;
124 }
125
126 bool GEOMGUI_AnnotationMgr::IsDisplayed( const QString& theEntry, const int theIndex, SOCC_Viewer* theView ) const
127 {
128   SOCC_Viewer* aView = viewOrActiveView( theView );
129   if ( !aView || !myVisualized.contains( aView ) )
130     return false;
131
132   EntryToAnnotations anEntryToAnnotation = myVisualized[aView];
133   if ( !anEntryToAnnotation.contains( theEntry ) )
134     return false;
135
136   AnnotationToPrs anAnnotationToPrs = anEntryToAnnotation[theEntry];
137   if ( !anAnnotationToPrs.contains( theIndex ) )
138     return false;
139
140   return true;
141 }
142
143 //=======================================================================
144 // function : GEOMGUI_AnnotationMgr::Display
145 // purpose  : Displays annotation shape presentation in view. It creates an annotation presentation
146 // and stores it in an internal container
147 //=======================================================================
148 void GEOMGUI_AnnotationMgr::Display( const QString& theEntry, const int theIndex, SOCC_Viewer* theView,
149                                      const bool isStoreViewState, const bool isUpdateViewer )
150 {
151   SOCC_Viewer* aView = viewOrActiveView( theView );
152   if ( !aView )
153     return;
154
155   if ( IsDisplayed( theEntry, theIndex, aView ) )
156     return;
157
158   GEOMGUI_AnnotationAttrs::Properties aProperty;
159   GEOM::GEOM_Object_ptr anObject;
160   getObject( theEntry, theIndex, anObject, aProperty );
161
162   // display presentation in the viewer
163   QString anEntry = QString("%1%2%3").arg(theEntry).arg(GetEntrySeparator()).arg(theIndex);
164   SALOME_Prs* aPrs = CreatePresentation( aProperty, anObject, aView, anEntry );
165   ((SALOME_View*)aView)->Display( getDisplayer(), aPrs );
166   if ( isUpdateViewer )
167     getDisplayer()->UpdateViewer();
168
169   EntryToAnnotations anEntryToMap;
170   if ( myVisualized.contains( aView ) )
171     anEntryToMap = myVisualized[aView];
172
173   // store displayed parameters to an internal container
174   AnnotationToPrs anAnnotationToPrsMap;
175   if ( anEntryToMap.contains( theEntry ) )
176     anAnnotationToPrsMap = anEntryToMap[theEntry];
177   anAnnotationToPrsMap[theIndex] = aPrs;
178   anEntryToMap[theEntry] = anAnnotationToPrsMap;
179   myVisualized[aView] = anEntryToMap;
180
181   if ( isStoreViewState ) {
182     // change persistent for the entry: set visible state in true for indices which presentations are shown
183     storeVisibleState( theEntry, theView );
184     storeFixedPosition( theEntry, theView );
185   }
186 }
187
188 void GEOMGUI_AnnotationMgr::Redisplay( const QString& theEntry, const int theIndex,
189                                        const GEOMGUI_AnnotationAttrs::Properties& theProperty )
190 {
191   SUIT_Session* ses = SUIT_Session::session();
192   SUIT_Application* app = ses->activeApplication();
193   if ( app )
194   {
195     SUIT_Desktop* desk = app->desktop();
196     QList<SUIT_ViewWindow*> wnds = desk->windows();
197     SUIT_ViewWindow* wnd;
198     QListIterator<SUIT_ViewWindow*> it( wnds );
199     while ( it.hasNext() && (wnd = it.next()) )
200     {
201       SUIT_ViewManager* vman = wnd->getViewManager();
202       if ( vman )
203       {
204         SUIT_ViewModel* vmodel = vman->getViewModel();
205         if ( vmodel )
206         {
207           SOCC_Viewer* aView = dynamic_cast<SOCC_Viewer*>(vmodel);
208           if ( aView )
209             Redisplay( theEntry, theIndex, theProperty, aView );
210         }
211       }
212     }
213   }
214 }
215
216 void GEOMGUI_AnnotationMgr::Redisplay( const QString& theEntry, const int theIndex,
217                                        const GEOMGUI_AnnotationAttrs::Properties& theProperty,
218                                        SOCC_Viewer* theView )
219 {
220   SOCC_Viewer* aView = viewOrActiveView( theView );
221   if ( !aView )
222     return;
223
224   if ( !myVisualized.contains( aView ) )
225     return;
226
227   EntryToAnnotations anEntryToAnnotation = myVisualized[aView];
228   if ( !anEntryToAnnotation.contains( theEntry ) )
229     return;
230
231   AnnotationToPrs anAnnotationToPrs = anEntryToAnnotation[theEntry];
232   if ( !anAnnotationToPrs.contains( theIndex ) )
233     return;
234
235   GEOMGUI_AnnotationAttrs::Properties aProperty;
236   GEOM::GEOM_Object_ptr anObject;
237   getObject( theEntry, theIndex, anObject, aProperty );
238   TopoDS_Shape aShape = GEOM_Client::get_client().GetShape( GeometryGUI::GetGeomGen(), anObject );
239   gp_Ax3 aShapeLCS = gp_Ax3().Transformed( aShape.Location().Transformation() );
240
241   // erase presentation from the viewer
242   SALOME_Prs* aPrs = anAnnotationToPrs[theIndex];
243   SOCC_Prs* anOCCPrs = dynamic_cast<SOCC_Prs*>( aPrs );
244   if ( anOCCPrs ) {
245     AIS_ListOfInteractive anIOs;
246     anOCCPrs->GetObjects( anIOs );
247     AIS_ListIteratorOfListOfInteractive anIter( anIOs );
248
249     for ( ; anIter.More(); anIter.Next() ) {
250       Handle(AIS_InteractiveObject) aPrs = anIter.Value();
251       Handle(GEOM_Annotation) aPresentation = Handle(GEOM_Annotation)::DownCast( aPrs );
252
253       GEOMGUI_AnnotationAttrs::SetupPresentation( aPresentation, theProperty, aShapeLCS );
254       aView->getAISContext()->Redisplay( aPresentation, Standard_True );
255     }
256   }
257 }
258
259 void GEOMGUI_AnnotationMgr::Erase( const QString& theEntry, const int theIndex, SOCC_Viewer* theView,
260                                    const bool isUpdateViewer )
261 {
262   SOCC_Viewer* aView = viewOrActiveView( theView );
263   if ( !aView )
264     return;
265
266   if ( !myVisualized.contains( aView ) )
267     return;
268
269   EntryToAnnotations anEntryToAnnotation = myVisualized[aView];
270   if ( !anEntryToAnnotation.contains( theEntry ) )
271     return;
272
273   AnnotationToPrs anAnnotationToPrs = anEntryToAnnotation[theEntry];
274   if ( !anAnnotationToPrs.contains( theIndex ) )
275     return;
276
277   // erase presentation from the viewer
278   SALOME_Prs* aPrs = anAnnotationToPrs[theIndex];
279   ((SALOME_View*)aView)->Erase( getDisplayer(), aPrs );
280   if ( isUpdateViewer )
281     getDisplayer()->UpdateViewer();
282
283   // remove displayed parameters from an internal container
284   anAnnotationToPrs.remove( theIndex );
285   anEntryToAnnotation[theEntry] = anAnnotationToPrs;
286   if (anAnnotationToPrs.isEmpty()) {
287     anEntryToAnnotation.remove( theEntry );
288   }
289   else {
290     anEntryToAnnotation[theEntry] = anAnnotationToPrs;
291   }
292   myVisualized[aView] = anEntryToAnnotation;
293
294   // change persistent for the entry: set visible state in true for indices which presentations are shown
295   storeVisibleState( theEntry, theView );
296 }
297
298 void GEOMGUI_AnnotationMgr::DisplayVisibleAnnotations( const QString& theEntry, SOCC_Viewer* theView,
299                                                        const bool isUpdateViewer )
300 {
301   SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( getApplication()->activeStudy() );
302   _PTR(SObject) aSObj = aStudy->studyDS()->FindObjectID( theEntry.toStdString() );
303   const Handle(GEOMGUI_AnnotationAttrs) aShapeAnnotations = GEOMGUI_AnnotationAttrs::FindAttributes( aSObj );
304   if ( !aShapeAnnotations.IsNull() ) {
305     const int aCount = aShapeAnnotations->GetNbAnnotation();
306     std::vector<bool> isVisible( aCount );
307     for ( int anIndex = 0; anIndex < aCount; ++anIndex )
308     {
309       isVisible[anIndex] = aShapeAnnotations->GetIsVisible( anIndex );
310     }
311     for ( int anIndex = 0; anIndex < aCount; ++anIndex )
312     {
313       if ( isVisible[anIndex] )
314         Display( theEntry, anIndex, theView, true, isUpdateViewer );
315     }
316   }
317 }
318
319 void GEOMGUI_AnnotationMgr::EraseVisibleAnnotations( const QString& theEntry, SOCC_Viewer* theView,
320                                                      const bool isUpdateViewer )
321 {
322   SOCC_Viewer* aView = viewOrActiveView( theView );
323   if ( !myVisualized.contains( aView ) )
324     return;
325
326   EntryToAnnotations anEntryToAnnotation = myVisualized[aView];
327   if ( !anEntryToAnnotation.contains( theEntry ) )
328     return;
329   AnnotationToPrs anAnnotationToPrs = anEntryToAnnotation[theEntry];
330
331   SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( getApplication()->activeStudy() );
332   _PTR(SObject) aSObj = aStudy->studyDS()->FindObjectID( theEntry.toStdString() );
333   const Handle(GEOMGUI_AnnotationAttrs) aShapeAnnotations = GEOMGUI_AnnotationAttrs::FindAttributes( aSObj );
334
335   const int aCount = aShapeAnnotations->GetNbAnnotation();
336   for ( int anIndex = 0; anIndex < aCount; ++anIndex )
337   {
338     if ( !anAnnotationToPrs.contains( anIndex ) )
339       continue;
340
341     // erase presentation from the viewer
342     SALOME_Prs* aPrs = anAnnotationToPrs[anIndex];
343     ((SALOME_View*)aView)->Erase( getDisplayer(), aPrs );
344   }
345   if ( isUpdateViewer )
346     getDisplayer()->UpdateViewer();
347   anEntryToAnnotation.remove( theEntry );
348   myVisualized[aView] = anEntryToAnnotation;
349 }
350
351 //=======================================================================
352 // function : GEOMGUI_AnnotationMgr::EraseRemovedAnnotation
353 // purpose  : Method to update internal maps after removing an
354 //  annotation from the object.
355 //=======================================================================
356 void GEOMGUI_AnnotationMgr::EraseRemovedAnnotation( const QString& theEntry, const int theIndex )
357 {
358   QMap<SOCC_Viewer*, EntryToAnnotations>::iterator aEntryMapIt = myVisualized.begin();
359   for ( ; aEntryMapIt != myVisualized.end(); ++aEntryMapIt ) {
360     SOCC_Viewer* aView = aEntryMapIt.key();
361     EntryToAnnotations& anEntryToAnnotation = aEntryMapIt.value();
362     if ( !anEntryToAnnotation.contains( theEntry ) )
363       continue;
364
365     AnnotationToPrs aUpdatedToPrs;
366     AnnotationToPrs& anAnnotationToPrs = anEntryToAnnotation[theEntry];
367     AnnotationToPrs::iterator anAnnotationIt = anAnnotationToPrs.begin();
368     for ( ; anAnnotationIt != anAnnotationToPrs.end(); ++anAnnotationIt ) {
369
370       const int aIndex = anAnnotationIt.key();
371       SALOME_Prs* aPrs = anAnnotationIt.value();
372       if ( aIndex > theIndex ) {
373         aUpdatedToPrs[aIndex - 1] = aPrs;
374       }
375       else if ( aIndex != theIndex ) {
376         aUpdatedToPrs[aIndex] = aPrs;
377       }
378       else {
379         ((SALOME_View*)aView)->Erase( getDisplayer(), aPrs );
380       }
381     }
382
383     anAnnotationToPrs = aUpdatedToPrs;
384   }
385   getDisplayer()->UpdateViewer();
386 }
387
388 //=======================================================================
389 // function : GEOMGUI_AnnotationMgr::UpdateVisibleAnnotations
390 // purpose  : 
391 //=======================================================================
392 void GEOMGUI_AnnotationMgr::UpdateVisibleAnnotations( const QString& theEntry, SOCC_Viewer* theView )
393 {
394   SOCC_Viewer* aView = viewOrActiveView( theView );
395   if ( !myVisualized.contains( aView ) )
396     return;
397
398   EntryToAnnotations& anEntryToAnnotation = myVisualized[aView];
399   if ( !anEntryToAnnotation.contains( theEntry ) )
400     return;
401
402   AnnotationToPrs& anAnnotationToPrs = anEntryToAnnotation[theEntry];
403   AnnotationToPrs::iterator anIt = anAnnotationToPrs.begin();
404   for (; anIt != anAnnotationToPrs.end(); ++anIt ) {
405     SOCC_Prs* aPrs =
406       dynamic_cast<SOCC_Prs*> (anIt.value());
407
408     GEOMGUI_AnnotationAttrs::Properties aProperty;
409     GEOM::GEOM_Object_ptr anObject;
410     getObject( theEntry, anIt.key(), anObject, aProperty );
411     TopoDS_Shape aShape = GEOM_Client::get_client().GetShape( GeometryGUI::GetGeomGen(), anObject );
412     gp_Ax3 aShapeLCS = gp_Ax3().Transformed( aShape.Location().Transformation() );
413
414     AIS_ListOfInteractive aIObjects;
415     aPrs->GetObjects( aIObjects );
416     AIS_ListOfInteractive::Iterator aIOIt( aIObjects );
417     for ( ; aIOIt.More(); aIOIt.Next() ) {
418
419       Handle(GEOM_Annotation) aPresentation =
420         Handle(GEOM_Annotation)::DownCast( aIOIt.Value() );
421
422       if ( aPresentation.IsNull() )
423         continue;
424
425       if ( !aShape.IsNull() ) {
426
427         gp_Ax3 aShapeLCS = gp_Ax3().Transformed( aShape.Location().Transformation() );
428         GEOMGUI_AnnotationAttrs::SetupPresentation( aPresentation, aProperty, aShapeLCS );
429         if ( aProperty.ShapeType == TopAbs_SHAPE ) {
430           aPresentation->SetHilightShape( aShape );
431         }
432         else if ( aProperty.ShapeIndex > 0 ) {
433           TopTools_IndexedMapOfShape aSubShapeMap;
434           TopExp::MapShapes( aShape, static_cast<TopAbs_ShapeEnum>( aProperty.ShapeType ), aSubShapeMap );
435           if ( aProperty.ShapeIndex <= aSubShapeMap.Extent() ) {
436             aPresentation->SetHilightShape( aSubShapeMap( aProperty.ShapeIndex ) );
437           }
438         }
439       }
440
441       setDisplayProperties( aPresentation, aView, theEntry );
442
443       aView->getAISContext()->Redisplay( aPresentation, Standard_True );
444     }
445   }
446   getDisplayer()->UpdateViewer();
447 }
448
449 void GEOMGUI_AnnotationMgr::DisplayAllAnnotations( SOCC_Viewer* theView )
450 {
451   SOCC_Viewer* aView = viewOrActiveView( theView );
452   if ( !myVisualized.contains( aView ) )
453     return;
454
455   GeometryGUI* aModule = dynamic_cast<GeometryGUI*>( getApplication()->activeModule() );
456   GEOMGUI_TextTreeWdg* aTextWidget = aModule->GetTextTreeWdg();
457   QList<QString> anEntries = aTextWidget->getAllEntries( GEOMGUI_TextTreeWdg::AnnotationShape );
458
459   SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( getApplication()->activeStudy() );
460   for ( int i = 0, aCount = anEntries.size(); i < aCount; i++ ) {
461      QString anEntry = anEntries[i];
462
463     _PTR(SObject) aSObj = aStudy->studyDS()->FindObjectID( anEntry.toStdString() );
464     if ( !aSObj )
465       continue;
466
467     const Handle(GEOMGUI_AnnotationAttrs) aShapeAnnotations = GEOMGUI_AnnotationAttrs::FindAttributes( aSObj );
468     if ( aShapeAnnotations.IsNull() )
469       continue;
470
471     int anAnnotationsCount = aShapeAnnotations->GetNbAnnotation();
472     for ( int anIndex = 0; anIndex < anAnnotationsCount; ++anIndex )
473     {
474       Display( anEntry, anIndex, aView, false, false );
475     }
476     getDisplayer()->UpdateViewer();
477     storeVisibleState( anEntry, aView );
478     storeFixedPosition( anEntry, aView );
479   }
480 }
481
482 void GEOMGUI_AnnotationMgr::EraseAllAnnotations( SOCC_Viewer* theView )
483 {
484   SOCC_Viewer* aView = viewOrActiveView( theView );
485   if ( !myVisualized.contains( aView ) )
486     return;
487
488   GeometryGUI* aModule = dynamic_cast<GeometryGUI*>( getApplication()->activeModule() );
489   GEOMGUI_TextTreeWdg* aTextWidget = aModule->GetTextTreeWdg();
490   QList<QString> anEntries = aTextWidget->getAllEntries( GEOMGUI_TextTreeWdg::AnnotationShape );
491
492   for ( int i = 0, aCount = anEntries.size(); i < aCount; i++ ) {
493     QString anEntry = anEntries[i];
494     EraseVisibleAnnotations( anEntry, aView, false );
495     storeVisibleState( anEntry, aView );
496   }
497   getDisplayer()->UpdateViewer();
498 }
499
500 void GEOMGUI_AnnotationMgr::SetPreviewStyle( const QString& theEntry, const int theIndex, const bool theIsPreview )
501 {
502   SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
503   const QColor aFontColor = aResMgr->colorValue( "Geometry", "shape_annotation_font_color", QColor( 255, 255, 255 ) );
504   const QColor aLineColor = aResMgr->colorValue( "Geometry", "shape_annotation_line_color", QColor( 255, 255, 255 ) );
505
506   const Quantity_Color aOcctFontColor( aFontColor.redF(), aFontColor.greenF(), aFontColor.blueF(), Quantity_TOC_RGB );
507   const Quantity_Color aOcctLineColor( aLineColor.redF(), aLineColor.greenF(), aLineColor.blueF(), Quantity_TOC_RGB );
508
509   QMap<SOCC_Viewer*, EntryToAnnotations>::Iterator aViewIt = myVisualized.begin();
510   for (; aViewIt != myVisualized.end(); ++aViewIt ) {
511
512     Handle(GEOM_Annotation) aPresentation = getAISPresentation ( theEntry, theIndex, aViewIt.key() );
513     if ( aPresentation.IsNull() ) {
514       continue;
515     }
516
517     if ( theIsPreview ) {
518       aPresentation->SetTextColor( Quantity_NOC_VIOLET );
519       aPresentation->SetLineColor( Quantity_NOC_VIOLET );
520       aPresentation->SetDepthCulling( Standard_False );
521     }
522     else {
523       aPresentation->SetTextColor( aOcctFontColor );
524       aPresentation->SetLineColor( aOcctLineColor );
525       aPresentation->SetDepthCulling( Standard_True );
526     }
527   }
528   getDisplayer()->UpdateViewer();
529 }
530
531 Handle(SALOME_InteractiveObject) GEOMGUI_AnnotationMgr::FindInteractiveObject( const QString& theEntry,
532                                                                                const int theIndex,
533                                                                                SOCC_Viewer* theView ) const
534 {
535   Handle(SALOME_InteractiveObject) anIO;
536
537   SOCC_Viewer* aView = viewOrActiveView( theView );
538   if ( !myVisualized.contains( aView ) )
539     return anIO;
540
541   EntryToAnnotations anEntryToAnnotation = myVisualized[aView];
542   if ( !anEntryToAnnotation.contains( theEntry ) )
543     return anIO;
544
545   AnnotationToPrs anAnnotationToPrs = anEntryToAnnotation[theEntry];
546   if ( !anAnnotationToPrs.contains(theIndex) )
547     return anIO;
548
549   SALOME_Prs* aPrs = anAnnotationToPrs[theIndex];
550   SOCC_Prs* anOCCPrs = dynamic_cast<SOCC_Prs*>( aPrs );
551   if ( !anOCCPrs )
552     return anIO;
553
554   AIS_ListOfInteractive anIOs;
555   anOCCPrs->GetObjects( anIOs );
556   AIS_ListIteratorOfListOfInteractive anIter( anIOs );
557   for ( ; anIter.More() && anIO.IsNull(); anIter.Next() ) {
558     Handle(AIS_InteractiveObject) aPrs = anIter.Value();
559     if ( aPrs->GetOwner() )
560       anIO = Handle(SALOME_InteractiveObject)::DownCast( aPrs->GetOwner() );
561   }
562   return anIO;
563 }
564
565 int GEOMGUI_AnnotationMgr::FindAnnotationIndex( Handle(SALOME_InteractiveObject) theIO,
566                                                 SOCC_Viewer* theView )
567 {
568   int anIndex = -1;
569
570   SOCC_Viewer* aView = viewOrActiveView( theView );
571   if ( !myVisualized.contains( aView ) )
572     return anIndex;
573
574   QString anEntry = theIO->getEntry();
575
576   EntryToAnnotations anEntryToAnnotation = myVisualized[aView];
577   if ( !anEntryToAnnotation.contains( anEntry ) )
578     return anIndex;
579
580   AnnotationToPrs anAnnotationToPrs = anEntryToAnnotation[anEntry];
581   //typedef QMap<int, SALOME_Prs*> AnnotationToPrs;
582   AnnotationToPrs::const_iterator anIt = anAnnotationToPrs.begin(),
583                                   aLast = anAnnotationToPrs.end();
584   for (; anIt != aLast && anIndex < 0; anIt++) {
585     SALOME_Prs* aPrs = anIt.value();
586     SOCC_Prs* anOCCPrs = dynamic_cast<SOCC_Prs*>( aPrs );
587     if ( !anOCCPrs )
588       continue;
589
590     AIS_ListOfInteractive anIOs;
591     anOCCPrs->GetObjects( anIOs );
592     AIS_ListIteratorOfListOfInteractive anIter( anIOs );
593     for ( ; anIter.More() && anIndex < 0; anIter.Next() ) {
594       Handle(AIS_InteractiveObject) aPrs = anIter.Value();
595       if ( aPrs->GetOwner() ) {
596         Handle(SALOME_InteractiveObject) aPrsOwner = Handle(SALOME_InteractiveObject)::DownCast(aPrs->GetOwner());
597         if ( aPrsOwner == theIO )
598           anIndex = anIt.key();
599       }
600     }
601   }
602
603   return anIndex;
604 }
605
606 //=======================================================================
607 // function : GEOMGUI_AnnotationMgr::RemoveView
608 // purpose  : 
609 //=======================================================================
610 void GEOMGUI_AnnotationMgr::RemoveView( SOCC_Viewer* theView )
611 {
612   if ( !theView && myVisualized.contains( theView ) )
613     myVisualized.remove( theView );
614 }
615
616 QString GEOMGUI_AnnotationMgr::getDisplayedIndicesInfo( const QString& theEntry, SOCC_Viewer* theView ) const
617 {
618   QString aDisplayedIndices;
619
620   SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( getApplication()->activeStudy() );
621   _PTR(SObject) aSObj = aStudy->studyDS()->FindObjectID( theEntry.toStdString() );
622   if ( !aSObj )
623   {
624     return aDisplayedIndices;
625   }
626   const Handle(GEOMGUI_AnnotationAttrs) aShapeAnnotations = GEOMGUI_AnnotationAttrs::FindAttributes( aSObj );
627   if ( !aShapeAnnotations.IsNull() )
628   {
629     const int aCount = aShapeAnnotations->GetNbAnnotation();
630     QStringList anIndices;
631     for ( int anIndex = 0; anIndex < aCount; ++anIndex )
632     {
633       if (IsDisplayed( theEntry, anIndex, theView ) )
634         anIndices.append( QString::number(anIndex) );
635     }
636     aDisplayedIndices = anIndices.join(";");
637   }
638   return aDisplayedIndices;
639 }
640
641 void GEOMGUI_AnnotationMgr::setDisplayedIndicesInfo( const QString& theEntry, SOCC_Viewer* theView,
642                                                      const QString theIndicesInfo )
643 {
644   if ( theIndicesInfo.isEmpty() )
645     return;
646
647   QStringList anIndices = theIndicesInfo.split( ";" );
648   for ( int i = 0, aCount = anIndices.size(); i < aCount; i++ ) {
649     Display( theEntry, anIndices[i].toInt(), theView );
650   }
651 }
652
653 GEOM_Displayer* GEOMGUI_AnnotationMgr::getDisplayer() const
654 {
655   LightApp_Module* aModule = dynamic_cast<LightApp_Module*>( getApplication()->activeModule() );
656   return dynamic_cast<GEOM_Displayer*>( aModule->displayer() );
657 }
658
659 SOCC_Viewer* GEOMGUI_AnnotationMgr::viewOrActiveView( SOCC_Viewer* theView ) const
660 {
661   SOCC_Viewer* aView = theView;
662   if ( !aView ) {
663     SalomeApp_Application* anApp = getApplication();
664     SUIT_ViewWindow* anActiveWindow = anApp->desktop()->activeWindow();
665     if ( anActiveWindow ) {
666       aView = dynamic_cast<SOCC_Viewer*>( anActiveWindow->getViewManager()->getViewModel() );
667     }
668   }
669   return aView;
670 }
671
672 QString GEOMGUI_AnnotationMgr::makeAnnotationEntry( const QString& theEntry, const int theIndex )
673 {
674   return QString("%1%2%3").arg(theEntry).arg(GetEntrySeparator()).arg(theIndex);
675 }
676
677 bool GEOMGUI_AnnotationMgr::getIndexFromEntry( const QString& theEntry, QString& theObjEntry, int& theIndex )
678 {
679   QStringList aSplit = theEntry.split( GetEntrySeparator() );
680   if ( aSplit.size() < 2 )
681     return false;
682
683   bool isOk = true;
684   theObjEntry = aSplit.at( 0 );
685   theIndex = aSplit.at( 1 ).toInt( &isOk );
686   return isOk;
687 }
688
689 void GEOMGUI_AnnotationMgr::getObject( const QString& theEntry, const int theIndex,
690                                        GEOM::GEOM_Object_ptr& theObject,
691                                        GEOMGUI_AnnotationAttrs::Properties& theProperty )
692 {
693   SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( getApplication()->activeStudy() );
694   _PTR(SObject) aSObj = aStudy->studyDS()->FindObjectID( theEntry.toStdString() );
695   const Handle(GEOMGUI_AnnotationAttrs) aShapeAnnotations = GEOMGUI_AnnotationAttrs::FindAttributes( aSObj );
696   if ( !aShapeAnnotations.IsNull() ) {
697     aShapeAnnotations->GetProperties( theIndex, theProperty );
698
699     theObject = GEOM::GEOM_Object::_narrow( GeometryGUI::ClientSObjectToObject(aSObj) );
700   }
701 }
702
703 void GEOMGUI_AnnotationMgr::storeFixedPosition( const QString& theEntry, SOCC_Viewer* theView )
704 {
705   SOCC_Viewer* aView = viewOrActiveView( theView );
706   if ( !aView || !myVisualized.contains( aView ) )
707     return;
708
709   SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( getApplication()->activeStudy() );
710   _PTR(SObject) aSObj = aStudy->studyDS()->FindObjectID( theEntry.toStdString() );
711   const Handle(GEOMGUI_AnnotationAttrs) aShapeAnnotations = GEOMGUI_AnnotationAttrs::FindAttributes( aSObj );
712   if ( aShapeAnnotations.IsNull() )
713     return;
714
715
716   EntryToAnnotations anEntryToAnnotation = myVisualized[aView];
717   AnnotationToPrs anAnnotationToPrs;
718   if ( anEntryToAnnotation.contains( theEntry ) )
719     anAnnotationToPrs = anEntryToAnnotation[theEntry];
720
721   AnnotationToPrs::iterator anIt = anAnnotationToPrs.begin();
722   for (; anIt != anAnnotationToPrs.end(); ++anIt ) {
723     int anIndex = anIt.key();
724     bool isFixedAnnotation = aShapeAnnotations->GetIsScreenFixed( anIndex );
725     if ( !isFixedAnnotation )
726       continue;
727
728     SOCC_Prs* aPrs = dynamic_cast<SOCC_Prs*> (anIt.value());
729     Handle(GEOM_Annotation) anAnnotationPresentation;
730
731     AIS_ListOfInteractive aIObjects;
732     aPrs->GetObjects( aIObjects );
733     AIS_ListOfInteractive::Iterator aIOIt( aIObjects );
734     for ( ; aIOIt.More(); aIOIt.Next() ) {
735       anAnnotationPresentation = Handle(GEOM_Annotation)::DownCast( aIOIt.Value() );
736       if ( !anAnnotationPresentation.IsNull() )
737         break;
738     }
739     if ( !anAnnotationPresentation.IsNull() )
740       aShapeAnnotations->SetPosition( anIndex, anAnnotationPresentation->GetPosition() );
741   }
742 }
743
744 void GEOMGUI_AnnotationMgr::storeVisibleState( const QString& theEntry, SOCC_Viewer* theView )
745 {
746   SOCC_Viewer* aView = viewOrActiveView( theView );
747   if ( !aView || !myVisualized.contains( aView ) )
748     return;
749
750   EntryToAnnotations anEntryToAnnotation = myVisualized[aView];
751   AnnotationToPrs anAnnotationToPrs;
752   if ( anEntryToAnnotation.contains( theEntry ) )
753     anAnnotationToPrs = anEntryToAnnotation[theEntry];
754
755
756   SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( getApplication()->activeStudy() );
757   _PTR(SObject) aSObj = aStudy->studyDS()->FindObjectID( theEntry.toStdString() );
758   const Handle(GEOMGUI_AnnotationAttrs) aShapeAnnotations = GEOMGUI_AnnotationAttrs::FindAttributes( aSObj );
759   if ( !aShapeAnnotations.IsNull() ) {
760     const int aCount = aShapeAnnotations->GetNbAnnotation();
761     for ( int anIndex = 0; anIndex < aCount; ++anIndex ) {
762       bool aVisible = anAnnotationToPrs.contains( anIndex );
763       aShapeAnnotations->SetIsVisible( anIndex, aVisible );
764     }
765   }
766 }
767
768 //=======================================================================
769 // function : GEOMGUI_AnnotationMgr::getEntry
770 // purpose  : 
771 //=======================================================================
772 std::string GEOMGUI_AnnotationMgr::getEntry( const GEOM::GEOM_Object_ptr theObject )
773 {
774   SUIT_Session* session = SUIT_Session::session();
775   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
776   if ( app )
777   {
778     CORBA::String_var IOR = app->orb()->object_to_string( theObject );
779     if ( strcmp(IOR.in(), "") != 0 )
780     {
781       SalomeApp_Study* study = ( SalomeApp_Study* )app->activeStudy();
782       _PTR(SObject) SO ( study->studyDS()->FindObjectIOR( std::string(IOR) ) );
783       if ( SO )
784         return SO->GetID();
785     }
786   }
787   return std::string();
788 }
789
790 //=======================================================================
791 // function : GEOMGUI_AnnotationMgr::getName
792 // purpose  : 
793 //=======================================================================
794 std::string GEOMGUI_AnnotationMgr::getName( const GEOM::GEOM_Object_ptr theObject )
795 {
796   SUIT_Session* session = SUIT_Session::session();
797   SalomeApp_Application* app = dynamic_cast<SalomeApp_Application*>( session->activeApplication() );
798   if ( app )
799   {
800     CORBA::String_var IOR = app->orb()->object_to_string( theObject );
801     if ( strcmp(IOR.in(), "") != 0 )
802     {
803       SalomeApp_Study* study = ( SalomeApp_Study* )app->activeStudy();
804       _PTR(SObject) aSObj ( study->studyDS()->FindObjectIOR( std::string(IOR) ) );
805
806       _PTR(GenericAttribute) anAttr;
807
808       if ( aSObj && aSObj->FindAttribute( anAttr, "AttributeName") )
809       {
810         _PTR(AttributeName) aNameAttr( anAttr );
811         return aNameAttr->Value();
812       }
813     }
814   }
815   return std::string();
816 }
817
818 //=======================================================================
819 // function : GEOMGUI_AnnotationMgr::setDisplayProperties
820 // purpose  : 
821 //=======================================================================
822 void GEOMGUI_AnnotationMgr::setDisplayProperties( const Handle(GEOM_Annotation)& thePrs,
823                                                   SOCC_Viewer* theView,
824                                                   const QString& theEntry )
825 {
826   SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
827   const QFont  aFont      = aResMgr->fontValue( "Geometry", "shape_annotation_font", QFont( "Y14.5M-2009", 24 ) );
828   const QColor aFontColor = aResMgr->colorValue( "Geometry", "shape_annotation_font_color", QColor( 0, 0, 127 ) );
829   const QColor aLineColor = aResMgr->colorValue( "Geometry", "shape_annotation_line_color", QColor( 0, 0, 127 ) );
830   const double aLineWidth = aResMgr->doubleValue( "Geometry", "shape_annotation_line_width", 1.0 );
831   const int aLineStyle    = aResMgr->integerValue( "Geometry", "shape_annotation_line_style", 0 );
832   const bool isAutoHide   = aResMgr->booleanValue( "Geometry", "shape_annotation_autohide", false );
833
834   const Quantity_Color aOcctFontColor( aFontColor.redF(), aFontColor.greenF(), aFontColor.blueF(), Quantity_TOC_RGB );
835   const Quantity_Color aOcctLineColor( aLineColor.redF(), aLineColor.greenF(), aLineColor.blueF(), Quantity_TOC_RGB );
836   const Standard_Real aFontHeight = aFont.pixelSize() != -1 ? aFont.pixelSize() : aFont.pointSize();
837
838   thePrs->SetFont( TCollection_AsciiString( aFont.family().toLatin1().data() ) );
839   thePrs->SetTextHeight( aFontHeight );
840   thePrs->SetTextColor( Quantity_Color( aFontColor.redF(), aFontColor.greenF(), aFontColor.blueF(), Quantity_TOC_RGB ) );
841   thePrs->SetLineColor( Quantity_Color( aLineColor.redF(), aLineColor.greenF(), aLineColor.blueF(), Quantity_TOC_RGB ) );
842   thePrs->SetLineWidth( aLineWidth );
843   thePrs->SetLineStyle( static_cast<Aspect_TypeOfLine>( aLineStyle ) );
844   thePrs->SetAutoHide( isAutoHide ? Standard_True : Standard_False );
845   thePrs->SetDepthCulling( Standard_True );
846
847   SOCC_Viewer* aView = viewOrActiveView( theView );
848   if ( aView && !theEntry.isEmpty() ) {
849
850     SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>( getApplication()->activeStudy() );
851     int aMgrId = dynamic_cast< SUIT_ViewModel* >( aView )->getViewManager()->getGlobalId();
852     QVariant aVal = aStudy->getObjectProperty( aMgrId, theEntry, GEOM::propertyName( GEOM::TopLevel ), QVariant() );
853     bool isBringToFront = aVal.isValid() ? aVal.toBool() : false;
854     if( isBringToFront ) {
855       thePrs->SetZLayer( Graphic3d_ZLayerId_Topmost );
856     }
857     else {
858       thePrs->SetDefaultZLayer();
859     }
860   }
861   else {
862     thePrs->SetDefaultZLayer();
863   }
864 }
865
866 //=======================================================================
867 // function : GEOMGUI_AnnotationMgr::getAISPresentation
868 // purpose  : 
869 //=======================================================================
870 Handle(GEOM_Annotation) GEOMGUI_AnnotationMgr::getAISPresentation ( const QString& theEntry,
871                                                                     const int theIndex,
872                                                                     SOCC_Viewer* theView )
873 {
874   if ( !myVisualized.contains( theView ) ) {
875     return Handle(GEOM_Annotation)();
876   }
877
878   EntryToAnnotations& aEntryToAnnotation = myVisualized[theView];
879   if ( !aEntryToAnnotation.contains( theEntry ) ) {
880     return Handle(GEOM_Annotation)();
881   }
882
883   AnnotationToPrs& aAnnotationToPrs = aEntryToAnnotation[theEntry];
884   if ( !aAnnotationToPrs.contains( theIndex ) ) {
885     return Handle(GEOM_Annotation)();
886   }
887
888   SALOME_Prs* aPrs = aAnnotationToPrs[theIndex];
889
890   // set or unset preview style for the presentation
891   AIS_ListOfInteractive aIObjects;
892   ((SOCC_Prs*)aPrs)->GetObjects( aIObjects );
893   AIS_ListOfInteractive::Iterator aIOIt( aIObjects );
894   for ( ; aIOIt.More(); aIOIt.Next() ) {
895     return Handle(GEOM_Annotation)::DownCast( aIOIt.Value() );
896   }
897
898   return Handle(GEOM_Annotation)();
899 }