update after merging trhe branches CEA_V3_0_x, OCC_V3_1_0_a1_x, and the main

[modules/med.git] / src / MedCorba_Swig / batchmode_medcorba_test.py

diff --git a/src/MedCorba_Swig/batchmode_medcorba_test.py b/src/MedCorba_Swig/batchmode_medcorba_test.py
index cbd5273e1982ef433ed373a779d36f397cb8aabd..55d2acafaa1819cbea7874c98119cf60f22c500a 100755 (executable)
--- a/src/MedCorba_Swig/batchmode_medcorba_test.py
+++ b/src/MedCorba_Swig/batchmode_medcorba_test.py
@@ -17,8 +17,8 @@ from random import *
 
 import os
 
-filePath=os.environ["MED_ROOT_DIR"]
-filePath=filePath+"/share/salome/resources/"
+filePath=os.environ["KERNEL_ROOT_DIR"]
+filePath=filePath+"/examples/"
 
 #==============================================================================
 
@@ -61,28 +61,87 @@ def AnalyzeField(field):
 
 #==============================================================================
 
-def getMeshObjectFromStudy(number):
-    mySO = batchmode_salome.myStudy.FindObject("MEDMESH")
-    mysub = mySO.FindSubObject(number)[1]
-    if mysub:
-        Builder = batchmode_salome.myStudy.NewBuilder()
-        anAttr = Builder.FindOrCreateAttribute(mysub, "AttributeIOR")
+def getMeshObjectFromStudy(meshName):
+    objNameInStudy = "/Med/MEDMESH/"+meshName
+    mySO = batchmode_salome.myStudy.FindObjectByPath(objNameInStudy)
+    if (mySO == None) :
+        print "ERROR: ",objNameInStudy," cannot be found in the Study"
+        return mySO
+    else:
+        anAttr = mySO.FindAttribute("AttributeIOR")[1]
         obj = batchmode_salome.orb.string_to_object(anAttr.Value())
         myObj = obj._narrow(SALOME_MED.MESH)
+        if (myObj == None) :
+            print "ERROR: ",objNameInStudy," has been found in the Study but with the wrong type"
         return myObj
+
+
+#==============================================================================
+
+def getFieldDoubleObjectFromStudy(number,subnumber):
+    mySO = batchmode_salome.myStudy.FindObject("MEDFIELD")
+    mysub = mySO.FindSubObject(number)[1]
+    if mysub:
+        mysubsub = mysub.FindSubObject(subnumber)[1]
+        if mysubsub:
+            Builder = batchmode_salome.myStudy.NewBuilder()
+            anAttr = Builder.FindOrCreateAttribute(mysubsub, "AttributeIOR")
+            obj = batchmode_salome.orb.string_to_object(anAttr.Value())
+            myObj = obj._narrow(SALOME_MED.FIELDDOUBLE)
+            return myObj
+        else:
+            print "ERROR: No FieldDouble Object stored in this Study"
+            return None
     else:
-        print "ERROR: No Mesh Object stored in this Study"
+        print "ERROR: No FieldDouble Object stored in this Study"
         return None
 
+
 #==============================================================================
 
-def getMedObjectFromStudy():
-    mySO = batchmode_salome.myStudy.FindObject("Objet MED")
-    Builder = batchmode_salome.myStudy.NewBuilder()
-    anAttr = Builder.FindOrCreateAttribute(mySO, "AttributeIOR")
-    obj = batchmode_salome.orb.string_to_object(anAttr.Value())
-    myObj = obj._narrow(SALOME_MED.MED)
-    return myObj
+def getFieldIntObjectFromStudy(number,subnumber):
+    mySO = batchmode_salome.myStudy.FindObject("MEDFIELD")
+    mysub = mySO.FindSubObject(number)[1]
+    if mysub:
+        mysubsub = mysub.FindSubObject(subnumber)[1]
+        if mysubsub:
+            Builder = batchmode_salome.myStudy.NewBuilder()
+            anAttr = Builder.FindOrCreateAttribute(mysubsub, "AttributeIOR")
+            obj = batchmode_salome.orb.string_to_object(anAttr.Value())
+            myObj = obj._narrow(SALOME_MED.FIELDINT)
+            return myObj
+        else:
+            print "ERROR: No FieldInt Object stored in this Study"
+            return None
+    else:
+        print "ERROR: No FieldInt Object stored in this Study"
+        return None
+
+#==============================================================================
+
+def getMedObjectFromStudy(file):
+    objNameInStudy = "MED_OBJECT_FROM_FILE_"+file
+    compNameInStudy= "MED"
+    listOfSO = batchmode_salome.myStudy.FindObjectByName(objNameInStudy,
+                                                         compNameInStudy)
+    listLength = len(listOfSO)
+    if (listLength == 0) :
+        print "ERROR: ",objNameInStudy," cannot be found in the Study under the component ",compNameInStudy
+        return None
+    elif (listLength > 1) :
+        print "ERROR: there are more than one instance of ",objNameInStudy," in the Study under the component ",compNameInStudy
+        return None
+    mySO = listOfSO[0]
+    if (mySO == None) :
+        print "ERROR: ",objNameInStudy," cannot be found in the Study"
+        return mySO
+    else:
+        anAttr = mySO.FindAttribute("AttributeIOR")[1]
+        obj = batchmode_salome.orb.string_to_object(anAttr.Value())
+        myObj = obj._narrow(SALOME_MED.MED)
+        if (myObj == None) :
+            print "ERROR: ",objNameInStudy," has been found in the Study but with the wrong type"
+        return myObj
 
 #==============================================================================
 #
@@ -105,8 +164,9 @@ studyCurrentId = batchmode_salome.myStudyId
 print "We are working in the study ",studyCurrent," with the ID ",studyCurrentId
 print ""
 
+fileName = "cube_hexa8_quad4.med"
 #medFile = "carre_en_quad4_seg2.med"
-medFile = filePath + "cube_hexa8_quad4.med"
+medFile = filePath + fileName
 
 print "Loading of the Med Component"
 print ""
@@ -115,25 +175,27 @@ med_comp = batchmode_salome.lcc.FindOrLoadComponent("FactoryServer", "MED")
 
 med_comp.readStructFile(medFile,studyCurrent)
 
-med_obj = getMedObjectFromStudy()
+med_obj = getMedObjectFromStudy(fileName)
 
 nbMeshes = med_obj.getNumberOfMeshes()
 
 nbFields = med_obj.getNumberOfFields()
 
+meshNames = med_obj.getMeshNames()
+
 print ""
 print "The med file ",medFile," has ",nbMeshes," Meshe(s) and ",nbFields," Field(s)"
 print ""
 
-meshcorba = getMeshObjectFromStudy(1)
+meshName = meshNames[0]
 
-name = meshcorba.getName()
+meshcorba = getMeshObjectFromStudy(meshName)
 
 nbNodes = meshcorba.getNumberOfNodes()
 
 spaceDim = meshcorba.getSpaceDimension()
 
-print "The mesh from the Study is ",name,".It is a ",spaceDim,"-D mesh and it has ",nbNodes," Nodes"
+print "The mesh from the Study is ",meshName,".It is a ",spaceDim,"-D mesh and it has ",nbNodes," Nodes"
 print ""
 
 for entity in [SALOME_MED.MED_NODE,SALOME_MED.MED_CELL,SALOME_MED.MED_FACE,SALOME_MED.MED_EDGE]: