src/ATOMSOLVGUI/ATOMSOLVGUI_DataModel.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.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21
  22 #include "ATOMSOLVGUI_DataModel.h"
  23 #include "ATOMSOLVGUI_DataObject.h"
  24 #include "ATOMSOLVGUI.h"
  25
  26 #include <LightApp_Study.h>
  27
  28 #include <CAM_Module.h>
  29 #include <CAM_Application.h>
  30
  31 /*! Constructor */
  32 ATOMSOLVGUI_DataModel::ATOMSOLVGUI_DataModel( CAM_Module* module )
  33   : LightApp_DataModel( module )
  34 {
  35 }
  36
  37 /*! Destructor */
  38 ATOMSOLVGUI_DataModel::~ATOMSOLVGUI_DataModel()
  39 {
  40 }
  41
  42 /*! Called on update of the structure of Data Objects */
  43 void ATOMSOLVGUI_DataModel::build()
  44 {
  45   ATOMSOLVGUI_ModuleObject* modelRoot = dynamic_cast<ATOMSOLVGUI_ModuleObject*>( root() );
  46   if( !modelRoot )  {  // root is not set yet
  47     modelRoot = new ATOMSOLVGUI_ModuleObject( this, 0 );
  48     setRoot( modelRoot );
  49   }
  50
  51   // create 'molecule' objects under model root object and 'atom' objects under 'molecule'-s
  52   ATOMSOLV_ORB::ATOMSOLV_Gen_var engine = ATOMSOLVGUI::GetATOMSOLVGen();
  53   if ( !CORBA::is_nil( engine ) ) {
  54     const int studyID = getStudy()->id();
  55     ATOMSOLV_ORB::TMoleculeList_var molecules;
  56     if ( !engine->getData( studyID, molecules ) ) {
  57       //      printf( " -- DataModel: getData() returned false! \n" );
  58       return;
  59     }
  60     for ( int i = 0, n = molecules->length(); i < n; i++ ) {
  61       ATOMSOLVGUI_DataObject* molDO = new ATOMSOLVGUI_DataObject ( modelRoot, i );
  62       const ATOMSOLV_ORB::TMolecule& mol = molecules[i];
  63       const int atoms = mol.molecule->getNbAtoms();
  64       for ( int j = 0; j < atoms; j++ )
  65         new ATOMSOLVGUI_DataObject ( molDO, i, j );
  66     }
  67   }
  68
  69   //  root()->dump();
  70 }