Create mesh in batch mode / Macros

Hi Fabian,

It's easy to create geometries and meshes in the python console which
appear in salome desktop ... But il also possible (with some work ) in
pure batch mode (I mean in a terminal via a command line)

To test theses functionalities, save the following script in a file named
"as_you_want.py". You can test it in salome python console ( do an
execfile "as_you_want.py" and you can test it in "pure batch" (with
"python as_you_want.py". You should have created two files
("box_hexa.med" and "box_tetra.med"

Caution : There is a regression in salome version >= 2.2.6 the
ExportMED line crash the application. This regression has been
sent to SMESH team and should be corrected soon !

Cheers,

                    E.A.

#########################################

# ----------
# Tip to detect if we are in salome embedded python console
# Indeed, in this context the __import__ function has been
# modified ... It is just a tip ... and it uses a non
# documented feature !
# ----------

try:
    __import_____module__ = __import__.__module__
except:
    __import_____module__ = '__builtin__'
    pass

if __import_____module__ == "import_hook":
    in_tui = 1
    print "Embedded python"
else:
    in_tui = 0
    print "External python ('batch mode')"
    pass

# -------------
# Class SalomeSession to lanuch the services of salome
# in 'batch mode' then to close this services at the
# end of the script ... Of course, this class should be
# defined elsewhere but for the script to be self-consistent
# I've put it here
# -------------

class SalomeSession(object):
   
    def getKRD(self):
        import os
        KERNEL_ROOT_DIR = os.getenv("KERNEL_ROOT_DIR"
        if KERNEL_ROOT_DIR is None:
            msg  = '\n\n'
            msg += 'KERNEL_ROOT_DIR not defined ...\n'
            msg += 'Be sure you are in salome environement.\n'
            raise Exception(msg)
        return KERNEL_ROOT_DIR
   
    def __init__(self, modules):
        #
        # Save (copy) the real user arguments
        #
        import sys
        argv_ini = sys.argv[:]
        #
        # Do like a runSalome --modules=... --logger -- terminal
        #
        import sys
        sys.argv  = ['bin/salome/runSalome.py']
        sys.argv += ['--modules=%s'%(",".join(modules))]
        sys.argv += ['--logger']
        sys.argv += ['--terminal']
        sys.argv += ['--standalone=pyContainer']
        #
        # Find a free port for naming service
        #
        port = self.searchFreePort()
        # --
        # E.A. : sys.path.insert(0, KERNEL_ROOT_DIR+"/bin/salome"
        # because there is multiple runSalome.py in V2_2_4 distribution
        # This will be removed after ...
        #
        KERNEL_ROOT_DIR = self.getKRD()
        sys.path.insert(0, KERNEL_ROOT_DIR+"/bin/salome"
        #
        import runSalome
        self.runSalome = runSalome
        clt, args = runSalome.main()
        runSalome.clt, runSalome.args = clt, args
        self.port = runSalome.args['port']
        self.naming_service = clt
        self.orb = clt.orb
        from LifeCycleCORBA import LifeCycleCORBA
        lcc = LifeCycleCORBA(clt.orb)
        self.lcc = lcc
        #
        # revert to the user args
        #
        import sys
        sys.argv = argv_ini
        #
        return
   
    def searchFreePort(self):
        print "Searching a free port for naming service:",
        NSPORT=2810
        limit=NSPORT
        limit=limit+100
        while 1:
            print "%s "%(NSPORT),
            import os
            status = os.system("netstat -ltn | grep -E :%s"%(NSPORT))
            if status:
                home = os.environ['HOME']
                hostname = os.environ['HOSTNAME']
                os.environ['OMNIORB_CONFIG'] = '%s/.omniORB_%s.cfg'%(home, NSPORT)
                f = open(os.environ['OMNIORB_CONFIG'], "w"
                f.write("ORBInitRef NameService=corbaname::%s:%s\n"%(hostname, NSPORT))
                f.close()
                print "- Ok"
                break
            if NSPORT == limit:
                msg  = ""
                msg += "I Can't find a free port to launch omniNames\n"
                msg += "I suggest you to kill the running servers and try again.\n"
                raise msg
            NSPORT=NSPORT+1
            pass
        return NSPORT
   
    def __del__(self):
        try:
            port = self.port
        except:
            return
        from os import system
        system("killSalomeWithPort.py %s"%(port))
        return
   
    pass

#
# Import modules
#

if in_tui:
    import salome
    def updateStudy():
        salome.sg.updateObjBrowser(1)
        return
    from geompy import *
else:
    salome = SalomeSession(modules=['GEOM','SMESH'])
    def updateStudy():
        return
    from batchmode_geompy import *
    pass

#
# Construction of the geometrical box
#

box = MakeBox(0.0, 0.0, 0.0, 100.0, 100.0, 100.0)
addToStudy(box, "box"
updateStudy()

#
# Construction of the mesh hexa
#

import StdMeshers
import SMESH
smesh = salome.lcc.FindOrLoadComponent("FactoryServer", "SMESH"
smesh.SetCurrentStudy(myStudy)

mesh = smesh.CreateMesh(box)
ior  = salome.orb.object_to_string(mesh)
sobj = myStudy.FindObjectIOR(ior)
attr = sobj.FindAttribute("AttributeName"[1]
attr.SetValue("BOX_HEXA"

hyp1d = smesh.CreateHypothesis("Regular_1D", "libStdMeshersEngine.so"
hyp2d = smesh.CreateHypothesis("Quadrangle_2D", "libStdMeshersEngine.so"
hyp3d = smesh.CreateHypothesis("Hexa_3D", "libStdMeshersEngine.so"

algo1d = smesh.CreateHypothesis("NumberOfSegments", "libStdMeshersEngine.so"
algo1d.SetNumberOfSegments(10)

for tool in [hyp1d, hyp2d, hyp3d, algo1d]:
    mesh.AddHypothesis(box, tool)
    pass

b = smesh.Compute(mesh, box)
if in_tui:
    smeshgui = salome.ImportComponentGUI("SMESH"
    smeshgui.Init(salome.myStudyId)
    smeshgui.SetMeshIcon(salome.ObjectToID(mesh), b )
    pass

updateStudy()

file_name = "box_hexa.med"
from os import system
system("rm -f %s > /dev/null 2>&1"%(file_name))
mesh.ExportMED(file_name, 0)


#
# Construction of the mesh tetra
#

import StdMeshers
import SMESH
smesh = salome.lcc.FindOrLoadComponent("FactoryServer", "SMESH"
smesh.SetCurrentStudy(myStudy)

mesh = smesh.CreateMesh(box)
ior  = salome.orb.object_to_string(mesh)
sobj = myStudy.FindObjectIOR(ior)
attr = sobj.FindAttribute("AttributeName"[1]
attr.SetValue("BOX_TETRA"

hyp1d = smesh.CreateHypothesis("Regular_1D", "libStdMeshersEngine.so"
hyp2d = smesh.CreateHypothesis("MEFISTO_2D", "libStdMeshersEngine.so"
hyp3d = smesh.CreateHypothesis("NETGEN_3D", "libNETGENEngine.so"

algo1d = smesh.CreateHypothesis("NumberOfSegments", "libStdMeshersEngine.so"
algo1d.SetNumberOfSegments(10)
algo2d = smesh.CreateHypothesis("MaxElementArea", "libStdMeshersEngine.so"
algo2d.SetMaxElementArea(1000.0)
algo3d = smesh.CreateHypothesis("MaxElementVolume", "libStdMeshersEngine.so"
algo3d.SetMaxElementVolume(100000.0)

for tool in [hyp1d, hyp2d, hyp3d, algo1d, algo2d, algo3d]:
    mesh.AddHypothesis(box, tool)
    pass

b = smesh.Compute(mesh, box)
if in_tui:
    smeshgui = salome.ImportComponentGUI("SMESH"
    smeshgui.Init(salome.myStudyId)
    smeshgui.SetMeshIcon(salome.ObjectToID(mesh), b )
    pass

updateStudy()

file_name = "box_tetra.med"
from os import system
system("rm -f %s > /dev/null 2>&1"%(file_name))
mesh.ExportMED(file_name, 0)

#########################################