src/ATOMGEN/ATOMGEN_Data.py

   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 import ATOMGEN_ORB__POA
  23
  24 class Atom(ATOMGEN_ORB__POA.Atom):
  25     """
  26     Atom interface implementation
  27     """
  28     def __init__(self, name, x, y, z):
  29         """
  30         Constructor
  31         """
  32         self.name = name
  33         self.x = x
  34         self.y = y
  35         self.z = z
  36
  37     def getName(self):
  38         """
  39         Gets the name of the atom
  40         """
  41         return self.name
  42
  43     def getX(self):
  44         """
  45         Gets x coordinate of the atom
  46         """
  47         return self.x
  48
  49     def getY(self):
  50         """
  51         Gets y coordinate of the atom
  52         """
  53         return self.y
  54
  55     def getZ(self):
  56         """
  57         Gets z coordinate of the atom
  58         """
  59         return self.z
  60
  61 class Molecule(ATOMGEN_ORB__POA.Molecule):
  62     """
  63     Atomic configuration interface implementatio
  64     """
  65     def __init__(self, name):
  66         """
  67         Constructor
  68         """
  69         self.name = name
  70         self.atoms = []
  71         pass
  72
  73     def getName(self):
  74         """
  75         Gets the name of the configuration
  76         """
  77         return self.name
  78
  79     def getNbAtoms(self):
  80         """
  81         GEts the number of the atoms
  82         """
  83         return len(self.atoms)
  84
  85     def addAtom( self, atom ):
  86         """
  87         Adds an atom
  88         """
  89         self.atoms.append( atom )
  90         pass
  91
  92     def getAtom(self, index):
  93         """
  94         Gets the atom by index
  95         """
  96         if index < 0 or index > len(self.atoms)-1:
  97             raise Exception( "Bad index" )
  98         return self.atoms[ index ]._this()