Salome HOME
Example of light module from Salome tutorial
[samples/atomic.git] / src / ATOMICGUI / ATOMICGUI_Selection.cxx
1 //  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 //  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
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.
10 //
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.
15 //
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
19 //
20 //  See http://www.opencascade.org/SALOME/ or email : webmaster.salome@opencascade.org
21
22 #include "ATOMICGUI_Selection.h"
23 #include "LightApp_DataOwner.h"
24
25 #include <LightApp_SelectionMgr.h>
26 #include <qstringlist.h>
27
28 /*! Constructor */
29 ATOMICGUI_Selection::ATOMICGUI_Selection ()
30 : LightApp_Selection()
31 {
32 }
33
34 /*! Destructor */
35 ATOMICGUI_Selection::~ATOMICGUI_Selection()
36 {
37 }
38
39 /*! Initialization of Selection */
40 void ATOMICGUI_Selection::init( const QString& client, LightApp_SelectionMgr* mgr )
41 {
42   if ( mgr ) {
43     SUIT_DataOwnerPtrList sel;
44     mgr->selected( sel);
45     SUIT_DataOwnerPtrList::const_iterator anIt  = sel.begin(),
46                                           aLast = sel.end();
47     for ( ; anIt != aLast; anIt++ ) {
48       QString type = "Unknown";
49       SUIT_DataOwner* owner = (SUIT_DataOwner*)( (*anIt).get() );
50       LightApp_DataOwner* sowner = dynamic_cast<LightApp_DataOwner*>( owner );
51       QStringList es = sowner->entry().split( "_" );
52       if ( es.count() > 0 && es[ 0 ] == "ATOMICGUI" ) {
53         if ( es.count() > 1 ) {
54           if( es[ 1 ] == "root" )
55             type = "Root";
56           else
57             type = "Molecule";
58           if ( es.count() > 2 )
59             type = "Atom";
60         }
61       }
62       myTypes.append( type );
63     }
64   }
65   LightApp_Selection::init( client, mgr );
66 }
67
68 /*! Returns number of objects of supported types. */
69 int ATOMICGUI_Selection::count() const
70 {
71   return myTypes.count();
72 }
73
74 /*! Returns selection parameter value. */
75 QVariant ATOMICGUI_Selection::parameter( const int ind, const QString& p ) const
76 {
77   if ( p == "type" )
78     return myTypes[ ind ];
79   return LightApp_Selection::parameter( ind, p );
80 }