src/KERNEL_PY/__init__.py

   1 #  -*- coding: iso-8859-1 -*-
   2 # Copyright (C) 2007-2021  CEA/DEN, EDF R&D, OPEN CASCADE
   3 #
   4 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   6 #
   7 # This library is free software; you can redistribute it and/or
   8 # modify it under the terms of the GNU Lesser General Public
   9 # License as published by the Free Software Foundation; either
  10 # version 2.1 of the License, or (at your option) any later version.
  11 #
  12 # This library is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 # Lesser General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU Lesser General Public
  18 # License along with this library; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  20 #
  21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  22 #
  23
  24 #  File   : salome.py renamed as __init__.py for python packaging (gboulant)
  25 #  Author : Paul RASCLE, EDF
  26 #  Module : SALOME
  27 #
  28 """
  29 Module salome gives access to Salome resources.
  30
  31 variables:
  32
  33   - salome.orb             : CORBA
  34   - salome.naming_service  : instance of naming Service class
  35       - methods:
  36           - Resolve(name)  : find a CORBA object (ior) by its pathname
  37           - Register(name) : register a CORBA object under a pathname
  38
  39   - salome.lcc             : instance of lifeCycleCORBA class
  40       - methods:
  41           - FindOrLoadComponent(server,name) :
  42                            obtain an Engine (CORBA object)
  43                            or launch the Engine if not found,
  44                            with a Server name and an Engine name
  45
  46   - salome.sg              : salome object to communicate with the graphical user interface (if any)
  47       - methods:
  48          - updateObjBrowser():
  49
  50          - SelectedCount():      returns number of selected objects
  51          - getSelected(i):       returns entry of selected object number i
  52          - getAllSelected():     returns list of entry of selected objects
  53          - AddIObject(Entry):    select an existing Interactive object
  54          - RemoveIObject(Entry): remove object from selection
  55          - ClearIObjects():      clear selection
  56
  57          - Display(*Entry):
  58          - DisplayOnly(Entry):
  59          - Erase(Entry):
  60          - DisplayAll():
  61          - EraseAll():
  62
  63          - IDToObject(Entry):    returns CORBA reference from entry
  64
  65   - salome.myStudyName     : active Study Name
  66   - salome.myStudy         : the active Study itself (CORBA ior)
  67       - methods : defined in SALOMEDS.idl
  68
  69 """
  70 ## @package salome
  71 # Module salome gives access to Salome resources.
  72 #
  73 #  \param salome.orb             : CORBA orb object
  74 #  \param salome.naming_service  : instance of naming Service class (SALOME_NamingServicePy::SALOME_NamingServicePy_i)
  75 #  \param salome.lcc             : instance of lifeCycleCORBA class (SALOME_LifeCycleCORBA)
  76 #  \param salome.sg              : Salome object to communicate with the graphical user interface, if running (see interface in salome_iapp::SalomeOutsideGUI)
  77 #  \param salome.myStudyName     : active Study Name
  78 #  \param salome.myStudy         : the active Study (interface SALOMEDS::Study)
  79
  80 #
  81 # ==========================================================================
  82 #
  83 # The function extend_path is used here to aggregate in a single
  84 # virtual python package all the python sub-packages embedded in each
  85 # SALOME modules (python "namespace" pattern).
  86 #
  87 ROOT_PYTHONPACKAGE_NAME="salome"
  88 #
  89 # This root package name is expected to be found as a directory in
  90 # some paths of the sys.path variable, especially the paths
  91 # <MODULE_ROOT_DIR>/lib/pythonX.Y/site-packages/salome where are
  92 # installed the python files. These paths are theorically appended by
  93 # the SALOME main runner and should be in the sys.path at this point
  94 # of the application. The extend_path is looking then for directories
  95 # of the type:
  96 #
  97 # <MODULE_ROOT_DIR>/lib/pythonX.Y/site-packages/salome/<ROOT_PYTHONPACKAGE_NAME>
  98 #
  99 # And append them to the sys.path. These directories are supposed to
 100 # be the pieces to be aggregated as a single virtual python package.
 101 #
 102 import os, sys
 103 from salome_utils import verbose
 104
 105 MATCH_ENDING_PATTERN="site-packages" + os.path.sep + "salome"
 106
 107 def extend_path(pname):
 108     for dir in sys.path:
 109         if not isinstance(dir, str) or not os.path.isdir(dir) or not dir.endswith(MATCH_ENDING_PATTERN):
 110             continue
 111         subdir = os.path.join(dir, pname)
 112         # XXX This may still add duplicate entries to path on
 113         # case-insensitive filesystems
 114         if os.path.isdir(subdir) and subdir not in __path__:
 115             if verbose(): print("INFO - The directory %s is appended to sys.path" % subdir)
 116             __path__.append(subdir)
 117
 118 extend_path(ROOT_PYTHONPACKAGE_NAME)
 119 # ==========================================================================
 120 #
 121
 122 from salome_kernel import *
 123 from salome_study import *
 124 from salome_iapp import *
 125 import salome_study
 126
 127 #
 128 # The next block is workaround for the problem of shared symbols loading for the extension modules (e.g. SWIG-generated)
 129 # that causes RTTI unavailable in some cases. To solve this problem, sys.setdlopenflags() function is used.
 130 # Depending on the Python version and platform, the dlopen flags can be defined in the dl, DLFUN or ctypes module.
 131 #
 132 import sys
 133 flags = None
 134 if not flags:
 135     try:
 136         # dl module can be unavailable
 137         import dl
 138         flags = dl.RTLD_NOW | dl.RTLD_GLOBAL
 139     except:
 140         pass
 141     pass
 142 if not flags:
 143     try:
 144         # DLFCN module can be unavailable
 145         import DLFCN
 146         flags = DLFCN.RTLD_NOW | DLFCN.RTLD_GLOBAL
 147     except:
 148         pass
 149     pass
 150 if not flags:
 151     try:
 152         # ctypes module can be unavailable
 153         import ctypes
 154         flags = ctypes.RTLD_GLOBAL
 155     except:
 156         pass
 157     pass
 158
 159 # Disable -> bug with scipy, seems very dangerous to do that
 160 #if flags:
 161 #    sys.setdlopenflags(flags)
 162 #    pass
 163
 164 orb, lcc, naming_service, cm, sg, esm, dsm, modulcat = None,None,None,None,None,None,None,None
 165 myStudy, myStudyName = None,None
 166
 167 salome_initial=True
 168
 169 __EMB_SERVANT_ENV_VAR_NAME = "SALOME_EMB_SERVANT"
 170
 171 def standalone():
 172     import os
 173     os.environ[__EMB_SERVANT_ENV_VAR_NAME] = "1"
 174
 175 def salome_init(path=None, embedded=False):
 176     import os
 177     if __EMB_SERVANT_ENV_VAR_NAME in os.environ:
 178         salome_init_without_session()
 179     else:
 180         import KernelBasis
 181         if KernelBasis.getSSLMode():
 182             salome_init_without_session()
 183         else:
 184             salome_init_with_session(path, embedded)
 185
 186 class StandAloneLifecyle:
 187     def __init__(self, containerManager, resourcesManager):
 188         self._cm = containerManager
 189         self._rm = resourcesManager
 190
 191     def FindOrLoadComponent(self,contName,moduleName):
 192         global orb
 193         import importlib
 194         builder_name = moduleName + "_SalomeSessionless"
 195         moduleObj = importlib.import_module(builder_name)
 196         result, orb = moduleObj.buildInstance(orb)
 197         return result
 198         #raise RuntimeError("Undealed situation cont = {} module = {}".format(contName,moduleName))
 199
 200     def getContainerManager(self):
 201       return self._cm
 202
 203     def getResourcesManager(self):
 204       return self._rm
 205
 206 def salome_init_without_session():
 207     global lcc,naming_service,myStudy,orb,modulcat,sg,cm
 208     import KernelBasis
 209     KernelBasis.setSSLMode(True)
 210     import KernelDS
 211     myStudy = KernelDS.myStudy()
 212     import CORBA
 213     orb=CORBA.ORB_init([''])
 214     import KernelModuleCatalog
 215     import SALOME_ModuleCatalog
 216     from salome_kernel import list_of_catalogs_regarding_environement
 217     modulcat = KernelModuleCatalog.myModuleCatalog( list_of_catalogs_regarding_environement() )
 218     import KernelLauncher
 219     from NamingService import NamingService
 220     cm = KernelLauncher.myContainerManager()
 221     lcc = StandAloneLifecyle(cm, KernelLauncher.myResourcesManager())
 222     # activate poaManager to accept co-localized CORBA calls.
 223     poa = orb.resolve_initial_references("RootPOA")
 224     poaManager = poa._get_the_POAManager()
 225     poaManager.activate()
 226     sg = SalomeOutsideGUI()
 227     salome_study_init_without_session()
 228     naming_service = NamingService()
 229
 230 def salome_init_with_session(path=None, embedded=False):
 231     """
 232     Performs only once SALOME general purpose initialisation for scripts.
 233     Provides:
 234     orb             reference to CORBA
 235     lcc             a LifeCycleCorba instance
 236     naming_service  a naming service instance
 237     cm              reference to the container manager
 238     esm             reference to external server manager
 239     dsm             reference to shared dataserver manager
 240     modulcat        reference to modulecatalog instance
 241     sg              access to SALOME GUI (when linked with IAPP GUI)
 242     myStudy         active study itself (CORBA reference)
 243     myStudyName     active study name
 244     """
 245     global salome_initial
 246     global orb, lcc, naming_service, cm, esm, dsm, modulcat
 247     global sg
 248     global myStudy, myStudyName
 249     import KernelBasis
 250     KernelBasis.setSSLMode(False)
 251     try:
 252         if salome_initial:
 253             salome_initial=False
 254             sg = salome_iapp_init(embedded)
 255             orb, lcc, naming_service, cm, esm, dsm, modulcat = salome_kernel_init()
 256             myStudy, myStudyName = salome_study_init(path)
 257             pass
 258         pass
 259     except RuntimeError as inst:
 260         # wait a little to avoid trace mix
 261         import time
 262         time.sleep(0.2)
 263         x = inst
 264         print("salome.salome_init():", x)
 265         print("""
 266         ============================================
 267         May be there is no running SALOME session
 268         salome.salome_init() is intended to be used
 269         within an already running session
 270         ============================================
 271         """)
 272         raise
 273
 274 def salome_close():
 275     global salome_initial, myStudy, myStudyName
 276     try:
 277         # study can be clear either from GUI or directly with salome.myStudy.Clear()
 278         myStudy.Clear()
 279     except:
 280         pass
 281     salome_initial=True
 282     salome_iapp_close()
 283     salome_study_close()
 284     myStudy, myStudyName = None, None
 285     import KernelDS
 286     KernelDS.KillGlobalSessionInstance()
 287     pass
 288
 289 def salome_NS():
 290     import CORBA
 291     import CosNaming
 292     orb = CORBA.ORB_init()
 293     ns0 = orb.resolve_initial_references("NameService")
 294     return ns0._narrow(CosNaming.NamingContext)
 295
 296 def salome_walk_on_containers(ns,root):
 297     import CosNaming
 298     it = ns.list(0)[1]
 299     if not it:
 300         return
 301     cont = True
 302     while cont:
 303         cont,obj = it.next_one()
 304         if cont:
 305             if obj.binding_name[0].kind == "object":
 306                 import Engines
 307                 corbaObj = ns.resolve(obj.binding_name)
 308                 if isinstance(corbaObj,Engines._objref_Container):
 309                     yield corbaObj,(root,obj.binding_name[0].id)
 310             else:
 311                 father = ns.resolve([obj.binding_name[0]])
 312                 for elt,elt2 in salome_walk_on_containers(father,root+[obj.binding_name[0].id]):
 313                     yield elt,elt2
 314             pass
 315         pass
 316     pass
 317
 318 def salome_shutdown_containers():
 319     salome_init()
 320     ns=salome_NS()
 321     li = [elt for elt in salome_walk_on_containers(ns,[""])]
 322     print("Number of containers in NS : {}".format(len(li)))
 323     for cont,(root,cont_name) in li:
 324         try:
 325             cont.Shutdown()
 326         except:
 327             pass
 328         ref_in_ns = "/".join(root+[cont_name])
 329         naming_service.Destroy_Name(ref_in_ns)
 330     print("Number of containers in NS after clean : {}".format( len( list(salome_walk_on_containers(ns,[""])) )))
 331
 332
 333 #to expose all objects to pydoc
 334 __all__=dir()