bin/geom_setenv.py

   1 #! /usr/bin/env python
   2 #  -*- coding: iso-8859-1 -*-
   3 # Copyright (C) 2007-2014  CEA/DEN, EDF R&D, OPEN CASCADE
   4 #
   5 # This library is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU Lesser General Public
   7 # License as published by the Free Software Foundation; either
   8 # version 2.1 of the License, or (at your option) any later version.
   9 #
  10 # This library is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 # Lesser General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU Lesser General Public
  16 # License along with this library; if not, write to the Free Software
  17 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  18 #
  19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  20 #
  21
  22 import os, sys, string
  23 from salome_utils import getTmpDir, generateFileName, uniteFiles
  24 from setenv import add_path, get_lib_dir, salome_subdir
  25
  26 # -----------------------------------------------------------------------------
  27
  28 def set_env( args ):
  29     """Add to the PATH-variables modules specific paths"""
  30     psep = os.pathsep
  31     python_version="python%d.%d" % sys.version_info[0:2]
  32
  33     tmp_dir = getTmpDir()
  34     env_dir = generateFileName( tmp_dir, prefix="env", with_port=True )
  35     res_dir = os.path.join( os.getenv( "GEOM_ROOT_DIR" ), "share", salome_subdir, "resources", "geom" )
  36
  37     csf_list = ["Plugin", "GEOMDS_Resources", "ShHealing"]
  38
  39     for csf_file in csf_list:
  40        uniteFiles( os.path.join( res_dir, csf_file ), os.path.join( env_dir, csf_file ) )
  41        pass
  42
  43     for csf_string in csf_list:
  44         csf_var = "CSF_" + csf_string + "Defaults"
  45         if not os.getenv( csf_var ):
  46             os.environ[ csf_var ] = env_dir
  47             pass
  48         pass
  49
  50     # find plugins
  51     plugin_list = []
  52     resource_path_list = []
  53     plugins_dir_var = "GEOM_ENGINE_RESOURCES_DIR"
  54     if os.environ.has_key(plugins_dir_var):
  55         # reverse the user's paths list, because the used 'add_path' prepends a new path,
  56         # but we want to append it: [a, b, c] => [c, b, a]
  57         plugins_dirs = os.environ[plugins_dir_var].split(os.pathsep)
  58         plugins_dirs.reverse()
  59         os.environ[plugins_dir_var] = string.join(plugins_dirs, os.pathsep)
  60         pass
  61     for env_var in os.environ.keys():
  62         value = os.environ[env_var]
  63         if env_var[-9:] == "_ROOT_DIR" and value:
  64             plugin_root = value
  65             plugin = env_var[:-9] # plugin name may have wrong case
  66
  67             # look for NAMEOFPlugin.xml file among resource files
  68             resource_dir = os.path.join(plugin_root,"share",salome_subdir,"resources",plugin.lower())
  69             if not os.access( resource_dir, os.F_OK ): continue
  70             for resource_file in os.listdir( resource_dir ):
  71                 if resource_file.endswith( ".xml") and \
  72                    resource_file.lower() == plugin.lower() + ".xml":
  73                     # use "name" attribute of "geom-plugin" as name of plugin in a right case
  74                     from xml.dom.minidom import parse
  75                     try:
  76                         xml_doc = parse( os.path.join( resource_dir, resource_file ))
  77                         plugin_nodes = xml_doc.getElementsByTagName("geom-plugin")
  78                     except:
  79                         continue
  80                     if not plugin_nodes or not plugin_nodes[0].hasAttribute("name"): continue
  81                     plugin = plugin_nodes[0].getAttribute("name")
  82                     if plugin in plugin_list: continue
  83
  84                     # add paths of plugin
  85                     plugin_list.append(plugin)
  86                     if not os.environ.has_key("SALOME_"+plugin+"Resources"):
  87                         resource_path = os.path.join(plugin_root,"share",salome_subdir,"resources",plugin.lower())
  88                         os.environ["SALOME_"+plugin+"Resources"] = resource_path
  89                         resource_path_list.append( resource_path )
  90                         add_path(os.path.join(plugin_root,get_lib_dir(),python_version, "site-packages",salome_subdir), "PYTHONPATH")
  91                         add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "PYTHONPATH")
  92
  93                         if sys.platform == "win32":
  94                             add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "PATH")
  95                             add_path(os.path.join(plugin_root,"bin",salome_subdir), "PYTHONPATH")
  96                         else:
  97                             add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "LD_LIBRARY_PATH")
  98                             add_path(os.path.join(plugin_root,"bin",salome_subdir), "PYTHONPATH")
  99                             add_path(os.path.join(plugin_root,"bin",salome_subdir), "PATH")
 100                             pass
 101                         pass
 102                     pass
 103                 elif resource_file == "ImportExport" and plugin.upper() != "GEOM":
 104                     # add 'ImportExport' plugin file path into variable
 105                     add_path(resource_dir, plugins_dir_var)
 106                     # add plugin's library path into environment
 107                     if sys.platform == "win32":
 108                         add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "PATH")
 109                     else:
 110                         add_path(os.path.join(plugin_root,get_lib_dir(),salome_subdir), "LD_LIBRARY_PATH")
 111                         pass
 112                     pass
 113                 pass
 114     plugin_list.append("GEOMActions")
 115     os.environ["GEOM_PluginsList"] = ":".join(plugin_list)
 116     os.environ["SalomeAppConfig"] = os.environ["SalomeAppConfig"] + psep + psep.join(resource_path_list)
 117
 118     if os.environ.has_key(plugins_dir_var):
 119         # reverse back the plugin paths:
 120         # [f, e, d, c, b, a] => [a, b, c, d, e, f]
 121         plugins_dirs = os.environ[plugins_dir_var].split(os.pathsep)
 122         plugins_dirs.reverse()
 123         os.environ[plugins_dir_var] = string.join(plugins_dirs, os.pathsep)
 124         pass
 125     pass