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()
271 type(logm).NaiveFetch = LogManagerNaiveFetch
272 type(logm).Fetch = LogManagerFetch
273 type(logm).DumpInFile = LogManagerDumpInFile
274 type(logm).LoadFromFile = LogManagerLoadFromFile
277 naming_service.Register(KernelLogger.myLogger(),"/Logger")
279 from NamingService import NamingService
280 if iorfakensfile is not None:
281 with open(iorfakensfile,"w") as iorfakensf:
282 iorfakensf.write(NamingService.IOROfNS())
284 def salome_init_without_session_attached(path=None, embedded=False):
286 Configuration SSL inside a python interpretor launched in the SALOME_Container_No_NS_Serv.
287 In this configuration, a local FakeNamingService is created and remote objects are stored in it.
288 lcc is pointing to the FakeNamingService above.
290 salome_init_without_session_common(path,embedded)
291 global lcc,cm,dsm,esm,rm,logm
293 orb=CORBA.ORB_init([''])
296 nsAbroad = orb.string_to_object( KernelBasis.getIOROfEmbeddedNS() )
298 CM_NAME_IN_NS = "/ContainerManager"
299 cm = orb.string_to_object( nsAbroad.Resolve(CM_NAME_IN_NS).decode() )
300 type(cm).SetOverrideEnvForContainersSimple = ContainerManagerSetOverrideEnvForContainersSimple
301 naming_service.Register(cm,CM_NAME_IN_NS)
302 RM_NAME_IN_NS = "/ResourcesManager"
303 rm = orb.string_to_object( nsAbroad.Resolve(RM_NAME_IN_NS).decode() )
304 naming_service.Register(rm,RM_NAME_IN_NS)
306 from LifeCycleCORBA import LifeCycleCORBASSL
307 lcc = LifeCycleCORBASSL()
308 DSM_NAME_IN_NS = "/DataServerManager"
309 dsm = orb.string_to_object( nsAbroad.Resolve(DSM_NAME_IN_NS).decode() )
310 naming_service.Register(dsm,DSM_NAME_IN_NS)
312 ESM_NAME_IN_NS = "/ExternalServers"
313 esm = orb.string_to_object( nsAbroad.Resolve(ESM_NAME_IN_NS).decode() )
314 naming_service.Register(esm,ESM_NAME_IN_NS)
316 LOGM_NAME_IN_NS = "/LogManager"
317 logm = orb.string_to_object( nsAbroad.Resolve(LOGM_NAME_IN_NS).decode() )
318 naming_service.Register(logm,LOGM_NAME_IN_NS)
320 def salome_init_with_session(path=None, embedded=False):
322 Performs only once SALOME general purpose initialisation for scripts.
324 orb reference to CORBA
325 lcc a LifeCycleCorba instance
326 naming_service a naming service instance
327 cm reference to the container manager
328 esm reference to external server manager
329 dsm reference to shared dataserver manager
330 modulcat reference to modulecatalog instance
331 sg access to SALOME GUI (when linked with IAPP GUI)
332 myStudy active study itself (CORBA reference)
333 myStudyName active study name
335 global salome_initial
336 global orb, lcc, naming_service, cm, esm, dsm, modulcat
338 global myStudy, myStudyName
340 KernelBasis.setSSLMode(False)
344 sg = salome_iapp_init(embedded)
345 orb, lcc, naming_service, cm, esm, dsm, modulcat = salome_kernel_init()
346 myStudy, myStudyName = salome_study_init(path)
349 except RuntimeError as inst:
350 # wait a little to avoid trace mix
354 print("salome.salome_init_with_session():", x)
356 ============================================
357 May be there is no running SALOME session
358 salome.salome_init() is intended to be used
359 within an already running session
360 ============================================
365 global salome_initial, myStudy, myStudyName, lcc
367 # study can be clear either from GUI or directly with salome.myStudy.Clear()
374 myStudy, myStudyName = None, None
375 lcc = None # to salome_init to rebuild all in case of salome_init after salome_close
377 if KernelBasis.getSSLMode() and not KernelBasis.getGUIMode():
379 KernelDS.KillGlobalSessionInstance()
381 KernelSDS.KillCPythonHelper()
387 orb = CORBA.ORB_init()
388 ns0 = orb.resolve_initial_references("NameService")
389 return ns0._narrow(CosNaming.NamingContext)
391 def salome_walk_on_containers(ns,root):
398 cont,obj = it.next_one()
400 if obj.binding_name[0].kind == "object":
402 corbaObj = ns.resolve(obj.binding_name)
403 if isinstance(corbaObj,Engines._objref_Container):
404 yield corbaObj,(root,obj.binding_name[0].id)
406 father = ns.resolve([obj.binding_name[0]])
407 for elt,elt2 in salome_walk_on_containers(father,root+[obj.binding_name[0].id]):
413 def salome_shutdown_containers_with_session():
416 li = [elt for elt in salome_walk_on_containers(ns,[""])]
417 print("Number of containers in NS : {}".format(len(li)))
418 for cont,(root,cont_name) in li:
423 ref_in_ns = "/".join(root+[cont_name])
424 naming_service.Destroy_Name(ref_in_ns)
425 print("Number of containers in NS after clean : {}".format( len( list(salome_walk_on_containers(ns,[""])) )))
427 def retrieve_containers_in_ns():
428 return [elt for elt in naming_service.repr() if "/Containers/" == elt[:12]]
430 def salome_shutdown_containers_without_session():
431 containersEntries = retrieve_containers_in_ns()
432 for containerEntry in containersEntries:
433 cont = naming_service.Resolve(containerEntry)
439 def salome_shutdown_containers():
441 if KernelBasis.getSSLMode():
442 salome_shutdown_containers_without_session()
444 salome_shutdown_containers_with_session()
446 class SessionContextManager:
450 def __exit__(self, type, value, traceback):
453 def ContainerManagerSetOverrideEnvForContainersSimple(self,env):
454 envEff = [ Engines.KeyValPairString(key=k,val=v) for k,v in env ]
455 return self.SetOverrideEnvForContainers( envEff )
457 def LogManagerNaiveFetch(self):
459 Fetch data from server with multiple CORBA invokations.
461 import SALOME_ContainerHelper
462 return [SALOME_ContainerHelper.ContainerLogInfoClt(elt) for elt in self.listOfContainerLogs()]
464 def LogManagerFetch(self,clearMemory = False):
466 Fetch data from server in one shot mode.
468 from SALOME_ContainerHelper import unserializeLogManager
469 return unserializeLogManager( self.getAllStruct(clearMemory) )
471 def LogManagerDumpInFile(self,fileName,clearMemory = False):
472 with open(fileName,"wb") as f:
473 f.write( self.getAllStruct( clearMemory ) )
475 def LogManagerLoadFromFile(self,fileName):
476 from SALOME_ContainerHelper import unserializeLogManager
477 with open(fileName,"rb") as f:
479 return unserializeLogManager( data )
481 #to expose all objects to pydoc