src/PYHELLO/PYHELLO.py

   1 # Copyright (C) 2007-2014  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, or (at your option) any later version.
  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
  23 # ---
  24 # File   : PYHELLOGUI.py
  25 # Author : Vadim SANDLER, Open CASCADE S.A.S. (vadim.sandler@opencascade.com)
  26 # ---
  27 #
  28 import PYHELLO_ORB__POA
  29 import SALOME_ComponentPy
  30 import SALOME_DriverPy
  31 import SALOMEDS
  32 from SALOME_DataContainerPy import *
  33
  34 from PYHELLO_utils import *
  35
  36 class PYHELLO(PYHELLO_ORB__POA.PYHELLO_Gen,
  37               SALOME_ComponentPy.SALOME_ComponentPy_i,
  38               SALOME_DriverPy.SALOME_DriverPy_i):
  39     """
  40     Construct an instance of PYHELLO module engine.
  41     The class PYHELLO implements CORBA interface PYHELLO_Gen (see PYHELLO_Gen.idl).
  42     It is inherited from the classes SALOME_ComponentPy_i (implementation of
  43     Engines::EngineComponent CORBA interface - SALOME component) and SALOME_DriverPy_i
  44     (implementation of SALOMEDS::Driver CORBA interface - SALOME module's engine).
  45     """
  46     def __init__ ( self, orb, poa, contID, containerName, instanceName,
  47                    interfaceName ):
  48         SALOME_ComponentPy.SALOME_ComponentPy_i.__init__(self, orb, poa,
  49                     contID, containerName, instanceName, interfaceName, 0)
  50         SALOME_DriverPy.SALOME_DriverPy_i.__init__(self, interfaceName)
  51         #
  52         self._naming_service = SALOME_ComponentPy.SALOME_NamingServicePy_i( self._orb )
  53         #
  54         pass
  55
  56     """
  57     Get version information.
  58     """
  59     def getVersion( self ):
  60         import salome_version
  61         return salome_version.getVersion("PYHELLO", True)
  62
  63     """
  64     Generate hello banner.
  65     """
  66     def makeBanner( self, name ):
  67         banner = "Hello %s!" % name
  68         return banner
  69
  70     """
  71     Create object.
  72     """
  73     def createObject( self, study, name ):
  74         self._createdNew = True # used for getModifiedData method
  75         builder = study.NewBuilder()
  76         father  = findOrCreateComponent( study )
  77         object  = builder.NewObject( father )
  78         attr    = builder.FindOrCreateAttribute( object, "AttributeName" )
  79         attr.SetValue( name )
  80         attr    = builder.FindOrCreateAttribute( object, "AttributeLocalID" )
  81         attr.SetValue( objectID() )
  82         pass
  83
  84     """
  85     Dump module data to the Python script.
  86     """
  87     def DumpPython( self, study, isPublished, isMultiFile ):
  88         abuffer = []
  89         names = []
  90         father = study.FindComponent( moduleName() )
  91         if father:
  92             iter = study.NewChildIterator( father )
  93             while iter.More():
  94                 name = iter.Value().GetName()
  95                 if name: names.append( name )
  96                 iter.Next()
  97                 pass
  98             pass
  99         if names:
 100             abuffer += [ "from salome import lcc" ]
 101             abuffer += [ "import PYHELLO_ORB" ]
 102             abuffer += [ "" ]
 103             abuffer += [ "pyhello = lcc.FindOrLoadComponent( 'FactoryServerPy', '%s' )" % moduleName() ]
 104             abuffer += [ "" ]
 105             abuffer += [ "pyhello.createObject( theStudy, '%s' )" % name for name in names ]
 106             abuffer += [ "" ]
 107             pass
 108         if isMultiFile:
 109             abuffer       = [ "  " + s for s in abuffer ]
 110             abuffer[0:0]  = [ "def RebuildData( theStudy ):" ]
 111             abuffer      += [ "  pass" ]
 112         abuffer += [ "\0" ]
 113         return ("\n".join( abuffer ), 1)
 114
 115     """
 116     Import file to restore module data
 117     """
 118     def importData(self, studyId, dataContainer, options):
 119       # get study by Id
 120       obj = self._naming_service.Resolve("myStudyManager")
 121       myStudyManager = obj._narrow(SALOMEDS.StudyManager)
 122       study = myStudyManager.GetStudyByID(studyId)
 123       # create all objects from the imported stream
 124       stream = dataContainer.get()
 125       for objname in stream.split("\n"):
 126         if len(objname) != 0:
 127           self.createObject(study, objname)
 128       self._createdNew = False # to store the modification of the study information later
 129       return ["objects"] # identifier what is in this file
 130
 131     def getModifiedData(self, studyId):
 132       if self._createdNew:
 133         # get study by Id
 134         obj = self._naming_service.Resolve("myStudyManager")
 135         myStudyManager = obj._narrow(SALOMEDS.StudyManager)
 136         study = myStudyManager.GetStudyByID(studyId)
 137         # iterate all objects to get their names and store this information in stream
 138         stream=""
 139         father = study.FindComponent( moduleName() )
 140         if father:
 141             iter = study.NewChildIterator( father )
 142             while iter.More():
 143                 name = iter.Value().GetName()
 144                 stream += name + "\n"
 145                 iter.Next()
 146         # store stream to the temporary file to send it in DataContainer
 147         dataContainer = SALOME_DataContainerPy_i(stream, "", "objects", False, True)
 148         aVar = dataContainer._this()
 149         return [aVar]
 150       return []