src/SalomeApp/SalomeApp_OBSelector.cxx

   1 #include "SalomeApp_OBSelector.h"
   2
   3 #include "SalomeApp_DataOwner.h"
   4 #include "SalomeApp_DataObject.h"
   5
   6 #include <OB_Browser.h>
   7
   8 #include <SUIT_DataObjectIterator.h>
   9
  10 /*!
  11   Constructor
  12 */
  13 SalomeApp_OBSelector::SalomeApp_OBSelector( OB_Browser* ob, SUIT_SelectionMgr* mgr )
  14 : SUIT_Selector( mgr, ob ),
  15   myBrowser( ob )
  16 {
  17   if ( myBrowser ) {
  18     connect( myBrowser, SIGNAL( selectionChanged() ), this, SLOT( onSelectionChanged() ) );
  19   }
  20 }
  21
  22 /*!
  23   Destructor
  24 */
  25 SalomeApp_OBSelector::~SalomeApp_OBSelector()
  26 {
  27 }
  28
  29 /*!
  30   Gets browser.
  31 */
  32 OB_Browser* SalomeApp_OBSelector::browser() const
  33 {
  34   return myBrowser;
  35 }
  36
  37 /*!
  38   Gets selection.
  39 */
  40 void SalomeApp_OBSelector::getSelection( SUIT_DataOwnerPtrList& list ) const
  41 {
  42   if ( !myBrowser )
  43     return;
  44
  45   DataObjectList objlist;
  46   myBrowser->getSelected( objlist );
  47   for ( DataObjectListIterator it( objlist ); it.current(); ++it )
  48   {
  49     SalomeApp_DataObject* obj = dynamic_cast<SalomeApp_DataObject*>( it.current() );
  50     if ( obj )
  51     {
  52       Handle( SALOME_InteractiveObject ) aSObj = new SALOME_InteractiveObject
  53         ( obj->entry(), obj->componentDataType(), obj->name() );
  54       list.append( SUIT_DataOwnerPtr( new SalomeApp_DataOwner( aSObj  ) ) );
  55     }
  56   }
  57 }
  58
  59 /*!Sets selection.*/
  60 void SalomeApp_OBSelector::setSelection( const SUIT_DataOwnerPtrList& list )
  61 {
  62   if ( !myBrowser )
  63     return;
  64
  65   QMap<QString, SalomeApp_DataObject*> map;
  66   fillEntries( map );
  67
  68   DataObjectList objList;
  69   for ( SUIT_DataOwnerPtrList::const_iterator it = list.begin(); it != list.end(); ++it )
  70   {
  71     const SalomeApp_DataOwner* owner = dynamic_cast<const SalomeApp_DataOwner*>( (*it).operator->() );
  72     if ( owner && map.contains( owner->entry() ) )
  73       objList.append( map[owner->entry()] );
  74   }
  75
  76   myBrowser->setSelected( objList );
  77 }
  78
  79 /*!On selection changed.*/
  80 void SalomeApp_OBSelector::onSelectionChanged()
  81 {
  82   selectionChanged();
  83 }
  84
  85 /*!Fill entries.*/
  86 void SalomeApp_OBSelector::fillEntries( QMap<QString, SalomeApp_DataObject*>& entires )
  87 {
  88   entires.clear();
  89
  90   if ( !myBrowser )
  91     return;
  92
  93   for ( SUIT_DataObjectIterator it( myBrowser->getRootObject(),
  94                                     SUIT_DataObjectIterator::DepthLeft ); it.current(); ++it )
  95   {
  96     SalomeApp_DataObject* obj = dynamic_cast<SalomeApp_DataObject*>( it.current() );
  97     if ( obj )
  98       entires.insert( obj->entry(), obj );
  99   }
 100 }