1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2023 CEA, EDF, OPEN CASCADE
4 # Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
24 # File : salome.py renamed as __init__.py for python packaging (gboulant)
25 # Author : Paul RASCLE, EDF
29 Module salome gives access to Salome resources.
34 - salome.naming_service : instance of naming Service class
36 - Resolve(name) : find a CORBA object (ior) by its pathname
37 - Register(name) : register a CORBA object under a pathname
39 - salome.lcc : instance of lifeCycleCORBA class
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
46 - salome.sg : salome object to communicate with the graphical user interface (if any)
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
63 - IDToObject(Entry): returns CORBA reference from entry
65 - salome.myStudyName : active Study Name
66 - salome.myStudy : the active Study itself (CORBA ior)
67 - methods : defined in SALOMEDS.idl
71 # Module salome gives access to Salome resources.
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)
81 # ==========================================================================
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).
87 ROOT_PYTHONPACKAGE_NAME="salome"
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
97 # <MODULE_ROOT_DIR>/lib/pythonX.Y/site-packages/salome/<ROOT_PYTHONPACKAGE_NAME>
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.
103 from salome_utils import verbose
105 MATCH_ENDING_PATTERN="site-packages" + os.path.sep + "salome"
107 def extend_path(pname):
109 if not isinstance(dir, str) or not os.path.isdir(dir) or not dir.endswith(MATCH_ENDING_PATTERN):
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)
118 extend_path(ROOT_PYTHONPACKAGE_NAME)
119 # ==========================================================================
122 from salome_kernel import *
123 from salome_study import *
124 from salome_iapp import *
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.
136 # dl module can be unavailable
138 flags = dl.RTLD_NOW | dl.RTLD_GLOBAL
144 # DLFCN module can be unavailable
146 flags = DLFCN.RTLD_NOW | DLFCN.RTLD_GLOBAL
152 # ctypes module can be unavailable
154 flags = ctypes.RTLD_GLOBAL
159 # Disable -> bug with scipy, seems very dangerous to do that
161 # sys.setdlopenflags(flags)
164 orb, lcc, naming_service, cm, sg, esm, dsm, logm, modulcat, rm = None,None,None,None,None,None,None,None,None,None
165 myStudy, myStudyName = None,None
174 KernelBasis.setSSLMode(False)
176 def salome_init(path=None, embedded=False, iorfakensfile=None, forced=False):
178 Initialize SALOME client process (that can also be server).
179 3 modes of initialization exists:
180 - SSL mode (see salome_init_without_session)
181 - SSL mode attached in the context of python execution inside SALOME_Container_No_NS_Serv server (typically YACS)
182 - Classical mode (see salome_init_with_session)
183 :param iorfakensfile: filename inside which IOR of fake NS will be written
184 :param forced: tell if the multi-initialization protection mecanism of salome_init must be skiped of not
185 (typically in the context where a path to a study is given whereas a previous initialisation without it was done)
188 if lcc is not None:# multi-initialization protection mecanism is based on lcc global var
190 PATH_TO_STUDY_FILE_TO_INITIATE = "PATH_TO_STUDY_FILE_TO_INITIATE"
192 if KernelBasis.getSSLMode():
193 if KernelBasis.getIOROfEmbeddedNS() == "":
195 # make runSalome.py -t study.hdf toto.py
196 if path is None and PATH_TO_STUDY_FILE_TO_INITIATE in os.environ:
197 path = os.environ[PATH_TO_STUDY_FILE_TO_INITIATE]
198 salome_init_without_session(path, embedded, iorfakensfile)
200 salome_init_without_session_attached(path, embedded)
202 salome_init_with_session(path, embedded)
204 def salome_init_without_session_common(path=None, embedded=False):
205 from ORBConfigFile import writeORBConfigFileSSL
206 OMNIORB_USER_PATH = "OMNIORB_USER_PATH"
207 def RemoveOmniorbConfigFile():
209 if "OMNIORB_CONFIG" in os.environ:
210 fileToRemove = os.environ["OMNIORB_CONFIG"]
211 if os.path.exists(fileToRemove):
212 os.unlink(fileToRemove)
214 if OMNIORB_USER_PATH in os.environ:
216 writeORBConfigFileSSL(os.environ[OMNIORB_USER_PATH],kwargs={"with_pid":True})
217 atexit.register(RemoveOmniorbConfigFile)
219 global lcc,naming_service,myStudy,myStudyName,orb,modulcat,sg
221 KernelBasis.setSSLMode(True)
223 myStudy = KernelDS.myStudy()
225 orb=CORBA.ORB_init([''])
226 import KernelModuleCatalog
227 import SALOME_ModuleCatalog
228 from salome_kernel import list_of_catalogs_regarding_environement
229 modulcat = KernelModuleCatalog.myModuleCatalog( list_of_catalogs_regarding_environement() )
231 poa = orb.resolve_initial_references("RootPOA")
232 poaManager = poa._get_the_POAManager()
233 poaManager.activate()
235 sg = salome_iapp_init(embedded)
236 salome_study_init_without_session(path)
238 from NamingService import NamingService
239 naming_service = NamingService()
240 myStudyName = myStudy.Name
242 def salome_init_without_session(path=None, embedded=False, iorfakensfile=None):
244 Force creation of all servants needed by SALOME session in the current process.
245 A Fake NamingService is created storing reference of all servants in the current process.
247 salome_init_without_session_common(path,embedded)
248 global lcc,cm,dsm,esm,rm,logm
249 import KernelLauncher
250 cm = KernelLauncher.myContainerManager()
251 type(cm).SetOverrideEnvForContainersSimple = ContainerManagerSetOverrideEnvForContainersSimple
252 rm = KernelLauncher.myResourcesManager()
253 from LifeCycleCORBA import LifeCycleCORBASSL
254 lcc = LifeCycleCORBASSL()
255 # create a FactoryServer Container servant
256 import KernelContainer
257 KernelContainer.myContainer()
258 # activate poaManager to accept co-localized CORBA calls.
259 from KernelSDS import GetDSMInstance
261 if hasattr(sys, 'argv'):
265 dsm = GetDSMInstance(argv)
266 # esm inherits from SALOME_CPythonHelper singleton already initialized by GetDSMInstance
267 # esm inherits also from SALOME_ResourcesManager creation/initialization (concerning SingleThreadPOA POA) when KernelLauncher.GetContainerManager() has been called
268 esm = KernelLauncher.GetExternalServer()
270 logm = KernelLauncher.myLogManager()
273 naming_service.Register(KernelLogger.myLogger(),"/Logger")
275 from NamingService import NamingService
276 if iorfakensfile is not None:
277 with open(iorfakensfile,"w") as iorfakensf:
278 iorfakensf.write(NamingService.IOROfNS())
280 def salome_init_without_session_attached(path=None, embedded=False):
282 Configuration SSL inside a python interpretor launched in the SALOME_Container_No_NS_Serv.
283 In this configuration, a local FakeNamingService is created and remote objects are stored in it.
284 lcc is pointing to the FakeNamingService above.
286 salome_init_without_session_common(path,embedded)
287 global lcc,cm,dsm,esm,rm,logm
289 orb=CORBA.ORB_init([''])
292 nsAbroad = orb.string_to_object( KernelBasis.getIOROfEmbeddedNS() )
294 CM_NAME_IN_NS = "/ContainerManager"
295 cm = orb.string_to_object( nsAbroad.Resolve(CM_NAME_IN_NS).decode() )
296 type(cm).SetOverrideEnvForContainersSimple = ContainerManagerSetOverrideEnvForContainersSimple
297 naming_service.Register(cm,CM_NAME_IN_NS)
298 RM_NAME_IN_NS = "/ResourcesManager"
299 rm = orb.string_to_object( nsAbroad.Resolve(RM_NAME_IN_NS).decode() )
300 naming_service.Register(rm,RM_NAME_IN_NS)
302 from LifeCycleCORBA import LifeCycleCORBASSL
303 lcc = LifeCycleCORBASSL()
304 DSM_NAME_IN_NS = "/DataServerManager"
305 dsm = orb.string_to_object( nsAbroad.Resolve(DSM_NAME_IN_NS).decode() )
306 naming_service.Register(dsm,DSM_NAME_IN_NS)
308 ESM_NAME_IN_NS = "/ExternalServers"
309 esm = orb.string_to_object( nsAbroad.Resolve(ESM_NAME_IN_NS).decode() )
310 naming_service.Register(esm,ESM_NAME_IN_NS)
312 LOGM_NAME_IN_NS = "/LogManager"
313 logm = orb.string_to_object( nsAbroad.Resolve(LOGM_NAME_IN_NS).decode() )
314 naming_service.Register(logm,LOGM_NAME_IN_NS)
316 def salome_init_with_session(path=None, embedded=False):
318 Performs only once SALOME general purpose initialisation for scripts.
320 orb reference to CORBA
321 lcc a LifeCycleCorba instance
322 naming_service a naming service instance
323 cm reference to the container manager
324 esm reference to external server manager
325 dsm reference to shared dataserver manager
326 modulcat reference to modulecatalog instance
327 sg access to SALOME GUI (when linked with IAPP GUI)
328 myStudy active study itself (CORBA reference)
329 myStudyName active study name
331 global salome_initial
332 global orb, lcc, naming_service, cm, esm, dsm, modulcat
334 global myStudy, myStudyName
336 KernelBasis.setSSLMode(False)
340 sg = salome_iapp_init(embedded)
341 orb, lcc, naming_service, cm, esm, dsm, modulcat = salome_kernel_init()
342 myStudy, myStudyName = salome_study_init(path)
345 except RuntimeError as inst:
346 # wait a little to avoid trace mix
350 print("salome.salome_init_with_session():", x)
352 ============================================
353 May be there is no running SALOME session
354 salome.salome_init() is intended to be used
355 within an already running session
356 ============================================
361 global salome_initial, myStudy, myStudyName, lcc
363 # study can be clear either from GUI or directly with salome.myStudy.Clear()
370 myStudy, myStudyName = None, None
371 lcc = None # to salome_init to rebuild all in case of salome_init after salome_close
373 if KernelBasis.getSSLMode() and not KernelBasis.getGUIMode():
375 KernelDS.KillGlobalSessionInstance()
377 KernelSDS.KillCPythonHelper()
383 orb = CORBA.ORB_init()
384 ns0 = orb.resolve_initial_references("NameService")
385 return ns0._narrow(CosNaming.NamingContext)
387 def salome_walk_on_containers(ns,root):
394 cont,obj = it.next_one()
396 if obj.binding_name[0].kind == "object":
398 corbaObj = ns.resolve(obj.binding_name)
399 if isinstance(corbaObj,Engines._objref_Container):
400 yield corbaObj,(root,obj.binding_name[0].id)
402 father = ns.resolve([obj.binding_name[0]])
403 for elt,elt2 in salome_walk_on_containers(father,root+[obj.binding_name[0].id]):
409 def salome_shutdown_containers_with_session():
412 li = [elt for elt in salome_walk_on_containers(ns,[""])]
413 print("Number of containers in NS : {}".format(len(li)))
414 for cont,(root,cont_name) in li:
419 ref_in_ns = "/".join(root+[cont_name])
420 naming_service.Destroy_Name(ref_in_ns)
421 print("Number of containers in NS after clean : {}".format( len( list(salome_walk_on_containers(ns,[""])) )))
423 def retrieve_containers_in_ns():
424 return [elt for elt in naming_service.repr() if "/Containers/" == elt[:12]]
426 def salome_shutdown_containers_without_session():
427 containersEntries = retrieve_containers_in_ns()
428 for containerEntry in containersEntries:
429 cont = naming_service.Resolve(containerEntry)
435 def salome_shutdown_containers():
437 if KernelBasis.getSSLMode():
438 salome_shutdown_containers_without_session()
440 salome_shutdown_containers_with_session()
442 class SessionContextManager:
446 def __exit__(self, type, value, traceback):
449 def ContainerManagerSetOverrideEnvForContainersSimple(self,env):
450 envEff = [ Engines.KeyValPairString(key=k,val=v) for k,v in env ]
451 return self.SetOverrideEnvForContainers( envEff )
453 #to expose all objects to pydoc