#!/usr/bin/env python
-# -*- coding: iso-8859-1 -*-
+# -*- coding: utf-8 -*-
+# Copyright (C) 2013-2016 CEA/DEN, EDF R&D
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+#
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+#
+
"""
Lancement d'un calcul ASTER
"""
-__revision__ = "V5.2"
+__revision__ = "V5.9"
#
import sys
import os
# opt1_defaut, opt2_defaut : les valeurs par defaut de ces options, selon les cas
opt1_defaut = {}
opt2_defaut = {}
-# liste_cas_tr : liste des cas transitoires
-liste_cas_tr = []
-#
-nro_couche = 1
-opt1_defaut["couche"] = str(nro_couche)
-nro_adap = 0
-opt2_defaut["couche"] = str(nro_adap)
-#
-liste_cas_tr.append("tet_aster_ther_tr")
-opt1_defaut["tet_aster_ther_tr"] = "0"
-opt2_defaut["tet_aster_ther_tr"] = "25"
-#
-liste_cas_tr.append("ssnp142d")
-opt1_defaut["ssnp142d"] = "0.0"
-opt2_defaut["ssnp142d"] = "4.0"
-#
-liste_cas_tr.append("Riete")
-opt1_defaut["Riete"] = "0.0"
-opt2_defaut["Riete"] = "8.0"
#
# ==== Fin de personnalisation ====
#
#
"""
Mode d'emploi :
----------------
+==============
+
+Cette procedure lance un calcul Aster. Avant de lancer ce script, il faut avoir cree un repertoire pour le calcul. Apres un premier calcul, qui aura ete lance traditionnellement, on aura copie le fichier '*export' sous le nom 'calcul.ref.export' dans ce repertoire.
-Cette procedure lance un calcul Aster. Le mode de lancement est le suivant :
+Le mode de lancement est le suivant :
-ScriptAster --rep_calc=rep_calc --num=nro --mesh_file=mesh_file [-dump] [-h|-help] [-v] [-v_max] options
+ScriptAster --rep_calc=rep_calc --mesh_file=mesh_file --num=nro [-dump] [-tr [--opt1=inst_deb] [--opt2=inst_fin]] [-h|-help] [-v] [-v_max]
Arguments obligatoires :
+------------------------
--rep_calc=rep_calc : le repertoire du calcul.
+--mesh_file=mesh_file : le fichier contenant le maillage sur lequel on veut calculer. Le nom est absolu ou relatif par rapport au repertoire de calcul ou a $HOME.
--num=nro : le numero du calcul, depuis 0.
---mesh_file=mesh_file : le fichier contenant le maillage sur lequel on veut calculer.
Options supplementaires, selon les cas :
-. Pour un cas transitoire :
-Les numeros des instants varient de 0 a N. On fournit le numero ou l'instant ; le jeu de commandes fait le tri entre les deux options.
+----------------------------------------
+. Pour un cas transitoire avec changement des instants de debut et fin :
+On doit faire le choix -tr
+Les numeros des instants varient de 0 a N. On fournit le numero ou l'instant selon les commandes.
--opt1=inst_deb : le numero ou l'instant initial ; si absent, on prend 0.
--opt2=inst_fin : le numero ou l'instant final ; si absent, on va jusqu'au bout du transitoire.
. Pour l'excavation :
Aucune option supplementaire.
Arguments optionnels :
-
---wait=attente : temps d'attente en secondes entre deux examens de l'etat d'un calcul batch ; si absent, on prend 10.
-
+----------------------
+--tr=1 : le calcul est transitoire et on change les instants ; par defaut, pas de changement.
+--wait=wait : temps d'attente en secondes entre deux examens de l'etat d'un calcul batch ; si absent, on prend 10.
-dump : produit le fichier dump du fichier med de resultats ; par defaut, pas de dump.
-h|-help : produit l'aide, quels que soient les autres arguments
-v : mode verbeux simple
-v_max : mode verbeux intensif
-Exemple :
-./ScriptAster.py --rep_calc=`pwd` --num=0 --mesh_file=maill.00.med -dump
-./ScriptAster.py --rep_calc=/scratch/D68518/HOMARD_SVN/trunk/training/tet_aster_ther --num=2 --mesh_file=/scratch/D68518/Salome/Adapt/resu/maill.02.med
+Exemples :
+----------
+./ScriptAster.py --rep_calc=`pwd` --mesh_file=maill.00.med --num=0 -dump
+./ScriptAster.py --rep_calc=/scratch/D68518/HOMARD_SVN/trunk/training/tet_aster_ther --mesh_file=/scratch/D68518/Salome/Adapt/resu/maill.02.med --num=2
"""
#
#====
# 1. Decodage des arguments
#
self.rep_calc = None
- self.num = None
+ self.numero = None
self.mesh_file = None
self.opt1 = None
self.opt2 = None
self.version = None
+ self.tr = 0
#
for argu in liste_arg :
#
self.numero = int(l_aux[1])
elif l_aux[0] == "--mesh_file" :
self.mesh_file = l_aux[1]
+ elif l_aux[0] == "--tr" :
+ self.tr = l_aux[1]
elif l_aux[0] == "--opt1" :
self.opt1 = l_aux[1]
elif l_aux[0] == "--opt2" :
elif saux == "-DUMP" :
self.dump = 1
#
-# 2. Les paramètres de base
+# 2. Les paramètres de base
#
self.dico = {}
self.nomfic_export = None
self.nomcas = ""
+ self.numero_str = None
#
if self.verbose_max :
nom_fonction = __name__ + "/__init__"
print "\nDans " + nom_fonction + ","
- print ". rep_calc :", self.rep_calc
- print ". numero :", self.numero
- print ". mesh_file :", self.mesh_file
- print ". opt1 :", self.opt1
- print ". opt2 :", self.opt2
- print ". attente :", self.attente
+ print ". rep_calc :", self.rep_calc
+ print ". mesh_file :", self.mesh_file
+ print ". numero :", self.numero
+ print ". opt1 :", self.opt1
+ print ". opt2 :", self.opt2
+ print ". attente :", self.attente
#
#========================= Fin de la fonction ===================================
#
"""
Preparation d'un calcul
"""
-#
- messages_erreur = { 0 : None }
#
nom_fonction = __name__ + "/prepa_calcul"
blabla = "\nDans " + nom_fonction + ","
#
if self.verbose_max :
print blabla
- print ". numero :", self.numero
- print ". mesh_file :", self.mesh_file
- print ". rep_calc :", self.rep_calc
+ print ". rep_calc :", self.rep_calc
+ print ". mesh_file :", self.mesh_file
+ print ". tr :", self.tr
#
while not erreur :
#
# 3. Modifications du fichier de commandes
# 3.1. Pour un cas transitoire
#
- if self.nomcas in liste_cas_tr :
+ if self.tr :
erreur, message_erreur = self.modif_cas_transitoire()
if erreur :
break
#
messages_erreur = { 0 : None,
-1 : "Quel repertoire de calcul ?",
- -2 : "Quel fichier de maillage ?",
- -3 : "Ce repertoire est inconnu.",
- -4 : "Ce fichier est inconnu.",
- -5 : "Quel temps d'attente ?",
- -6 : "Quel numero de calcul ?",
- -7 : "Numeros de pas de temps invalides.",
- -8 : "Numero de couche invalide.",
- -9 : "Numero d'adaptation invalide." }
+ -3 : "Quel fichier de maillage ?",
+ -4 : "Ce repertoire est inconnu.",
+ -5 : "Ce fichier est inconnu.",
+ -8 : "Quel temps d'attente ?",
+ -9 : "Quel numero de calcul ?",
+ -20 : "Numeros de pas de temps invalides.",
+ -30 : "Numero de couche invalide.",
+ -31 : "Numero d'adaptation invalide." }
#
nom_fonction = __name__ + "/controle_argument"
blabla = "\nDans " + nom_fonction + ","
#
if self.verbose_max :
print blabla
- print ". rep_calc :", self.rep_calc
- print ". mesh_file :", self.mesh_file
- print ". numero :", self.numero
- print ". attente :", self.attente
- print ". opt1 :", self.opt1
- print ". opt2 :", self.opt2
+ print ". rep_calc :", self.rep_calc
+ print ". mesh_file :", self.mesh_file
+ print ". numero :", self.numero
+ print ". tr :", self.tr
+ print ". opt1 :", self.opt1
+ print ". opt2 :", self.opt2
+ print ". attente :", self.attente
#
while not erreur :
#
# 1. Les repertoire et fichier
+# 1.1. Il y a bien eu une donnee
#
if self.rep_calc == None :
erreur = -1
elif self.mesh_file == None :
- erreur = -2
+ erreur = -3
if erreur < 0 :
self.message_info += "Relancer avec -h pour avoir le mode d'emploi."
break
+#
+# 1.2. Le repertoire de calcul
#
if os.environ.has_key("HOME") :
HOME = os.environ ["HOME"]
self.rep_calc = os.path.join(HOME, self.rep_calc[2:])
if not os.path.isdir(self.rep_calc) :
self.message_info += "Repertoire " + self.rep_calc
- erreur = -3
+ erreur = -4
break
else :
- aux = os.path.join(os.getcwd(),self.rep_calc)
+ aux = os.path.join(os.getcwd(), self.rep_calc)
self.rep_calc = os.path.normpath(aux)
#
- if ( self.mesh_file[:1] == "~" ) :
- self.mesh_file = os.path.join(HOME, self.mesh_file[2:])
- if not os.path.isfile(self.mesh_file) :
- self.message_info += "Fichier " + self.mesh_file
- erreur = -4
- break
- else :
- aux = os.path.join(os.getcwd(),self.mesh_file)
- self.mesh_file = os.path.normpath(aux)
+# 1.3. Les fichiers
+#
+ fic = self.mesh_file
+#
+ if ( fic[:1] == "~" ) :
+ fic = os.path.join(HOME, fic[2:])
+ if not os.path.isfile(fic) :
+ aux = os.path.join(self.rep_calc, fic)
+ if not os.path.isfile(aux) :
+ self.message_info += "Fichier : " + fic
+ erreur = -5
+ break
+ else :
+ fic = os.path.normpath(aux)
+ aux = os.path.join(os.getcwd(), fic)
+#
+ self.mesh_file = os.path.normpath(aux)
#
# 2. On en deduit le cas
#
try :
iaux = int(self.attente)
except :
- erreur = -5
+ erreur = -8
break
self.attente = iaux
#
# 4. Le numero du calcul
#
if self.numero == None :
- erreur = -6
+ erreur = -9
+#
+ if ( self.numero < 100 ) :
+ self.numero_str = "%02d" % self.numero
+ elif ( self.numero < 1000 ) :
+ self.numero_str = "%03d" % self.numero
+ elif ( self.numero < 10000 ) :
+ self.numero_str = "%04d" % self.numero
+ else :
+ self.numero_str = "%d" % self.numero
#
# 5. Options speciales pour les cas transitoires et pour l'excavation
#
- if ( self.nomcas in liste_cas_tr ) or ( self.nomcas[:6] == "couche" ) :
+ if ( self.tr or ( self.nomcas[:6] == "couche" ) ) :
#
if self.opt1 == None :
self.opt1 = opt1_defaut[self.nomcas]
+ #print ". opt1 defaut :", self.opt1
#
if self.opt2 == None :
self.opt2 = opt2_defaut[self.nomcas]
+ #print ". opt2 defaut :", self.opt2
#
# 5.1. Pour un cas transitoire
#
- if self.nomcas in liste_cas_tr :
+ if self.tr :
#
iaux1 = None
daux1 = None
iaux1 = int(self.opt1)
except :
daux1 = float(self.opt1)
+ if ( iaux1 == 0 ) :
+ daux1 = 0.
#
iaux2 = None
daux2 = None
except :
daux2 = float(self.opt2)
#
- if ( ( daux1 == None ) and ( daux2 == None ) ) :
+ if ( ( daux1 == None ) or ( daux2 == None ) ) :
+ #print "------------ ( daux1 == None ) or ( daux2 == None ) ------------"
#print "iaux1, iaux2 =", iaux1, iaux2
if iaux2 < iaux1 :
- erreur = -7
- elif ( ( iaux1 == None ) and ( iaux2 == None ) ) :
+ erreur = -20
+ elif ( ( iaux1 == None ) or ( iaux2 == None ) ) :
+ #print "------------ ( iaux1 == None ) or ( iaux2 == None ) ------------"
#print "daux1, daux2 =", daux1, daux2
if daux2 < daux1 :
- erreur = -7
+ erreur = -20
else :
- erreur = -7
+ erreur = -20
#
if erreur :
self.message_info += "opt1 = " + self.opt1
try :
iaux1 = int(self.opt1)
except :
- erreur = -8
+ erreur = -30
#
if ( ( iaux1 < 1 ) or ( iaux1 > 20 ) ) :
#print "iaux1 =", iaux1
- erreur = -8
+ erreur = -30
#
iaux2 = None
try :
iaux2 = int(self.opt2)
except :
- erreur = -9
+ erreur = -31
#
if ( iaux2 < 0 ) :
#print "iaux1 =", iaux1
- erreur = -9
+ erreur = -31
#
if erreur :
self.message_info += "nro_mail = " + self.numero
if self.verbose_max :
print ". rep_calc :", self.rep_calc
print ". mesh_file :", self.mesh_file
+ print ". opt1 :", self.opt1
+ print ". opt2 :", self.opt2
#
return erreur, message_erreur
#
#
if self.verbose_max :
print blabla
- print ". numero :", self.numero
- print ". mesh_file :", self.mesh_file
+ print ". numero :", self.numero
+ print ". mesh_file :", self.mesh_file
#
while not erreur :
#
# 1. Lecture du fichier export original
#
- nomfic_export = os.path.join(self.rep_calc, "calcul.ref.export")
- fic = open (nomfic_export, "r")
+ fic_export_ref = os.path.join(self.rep_calc, "calcul.ref.export")
+ fic = open (fic_export_ref, "r")
les_lignes = fic.readlines()
fic.close()
#
nomfic = "calcul"
if self.nomcas[:6] == "couche" :
nomfic += ".%02d" % self.nro_couche
- nomfic += ".%03d.export" % self.numero
+ nomfic += "." + self.numero_str + ".export"
self.nomfic_export = os.path.join(self.rep_calc, nomfic)
+ if self.verbose_max :
+ print ". nouveau fic_export :", self.nomfic_export
fic = open (self.nomfic_export, "w")
#
# 3. Exploration des lignes
# 3.2. Decodage des composants de la ligne pour un fichier
#
if ligne[0:2] == "F " :
+#
+ chgt = False
#
laux = ligne.split()
#print laux
typfic = laux[1]
+ statut = laux[3]
#print typfic
#
nomfic_0 = laux[2]
#
# 3.2.1. Si c'est le fichier de commandes, mise a jour du nom du repertoire
#
- if typfic == "comm" :
+ if ( typfic == "comm" ) :
#
+ if self.verbose_max :
+ print ". Commandes : mise a jour du nom du repertoire"
+ chgt = True
nomfic_l_0 = os.path.basename(nomfic_0)
nomfic = os.path.join(self.rep_calc, nomfic_l_0)
#
-# 3.2.2. Si ce n'est pas le fichier de commandes, mise a jour du nom
+# 3.2.2. Si c'est un fichier de resultats ou le fichier de maillage, mise a jour du nom
#
- else :
+ elif ( ( statut == "R" ) or ( typfic == "mmed" ) ) :
+ if self.verbose_max :
+ print ". Mise a jour du nom"
+ chgt = True
#
# 3.2.2.1. Le fichier de maillage est celui passe en argument
#
if typfic == "mmed" :
nomfic = self.mesh_file
#
-# 3.2.2.2. Pour les autres fichiers, on recupère le nom initial
+# 3.2.2.2. Pour les autres fichiers, on recupère le nom initial
#
else :
nomfic_l_0 = os.path.basename(nomfic_0)
laux1 = nomfic_l_0.split(".")
+ #print "laux1 =", laux1
#
# 3.2.2.2.1. Base des noms de fichiers si excavation
#
if laux1[0] == "couche_n" :
saux0 = "couche_%02d" % self.nro_couche
else :
- saux0 = laux1[0]
+ iaux = len(laux1[-1]) + 1
+ saux0 = nomfic_l_0[:-iaux]
+ #print "saux0 =", saux0
#
# 3.2.2.2.2. Indicage des fichiers
#
- if len(laux1) >= 3 :
- iaux = len(laux1[1])
- if iaux == 2 :
- saux1 = "%02d" % self.numero
- elif iaux == 3 :
- saux1 = "%03d" % self.numero
- elif iaux == 4 :
- saux1 = "%04d" % self.numero
- else :
- saux1 = "%d" % self.numero
- nomfic_l = saux0 + "." + saux1 + "." + laux1[2]
- else :
- nomfic_l = saux0 + "." + laux1[1]
-#
+ nomfic_l = saux0 + "." + self.numero_str + "." + laux1[-1]
nomfic_l_1 = os.path.dirname(nomfic_0)
nomfic = os.path.join(nomfic_l_1, nomfic_l)
+ #print " ==> ", nomfic
#
- ligne_bis = laux[0] + " " + laux[1] + " " + saux
- ligne_bis += nomfic + " "
- ligne_bis += laux[3] + " " + laux[4] + "\n"
+ if chgt :
+ ligne_bis = laux[0] + " " + laux[1] + " " + saux
+ ligne_bis += nomfic + " "
+ ligne_bis += laux[3] + " " + laux[4] + "\n"
#
# 3.2.2.3. On detruit preventivement les fichiers des resultats
#
- if self.nomcas[:6] != "couche" :
- if laux[3] == "R" :
+ if ( statut == "R" ) :
+ if ( self.nomcas[:6] != "couche" ) :
if os.path.isfile(nomfic) :
os.remove(nomfic)
#
nomfic_l_0 = os.path.basename(nomfic_0)
laux1 = nomfic_l_0.split(".")
if len(laux1) >= 3 :
- iaux = len(laux1[1])
- if iaux == 2 :
- saux1 = "%02d" % self.numero
- elif iaux == 3 :
- saux1 = "%03d" % self.numero
- elif iaux == 4 :
- saux1 = "%04d" % self.numero
- else :
- saux1 = "%d" % self.numero
- nomfic_l = laux1[0] + "." + saux1 + "." + laux1[2]
+ nomfic_l = laux1[0] + "." + self.numero_str + "." + laux1[2]
elif len(laux1) >= 2 :
if laux1[0] == "couche_nm1" :
saux0 = "couche_%02d" % (self.nro_couche-1)
#if self.verbose_max :
#print commande_base
#
- fic_caract = tempfile.mktemp()
- fic_caract_2 = tempfile.mktemp()
+ t_aux = tempfile.mkstemp()
+ fic_caract = t_aux[1]
+ t_aux = tempfile.mkstemp()
+ fic_caract_2 = t_aux[1]
#
# 3. Lancement
# 3.1. Commande finale
# 2. Commande de l'examen de l'etat du job,
#
fic_etat = os.path.join(self.rep_calc, self.nomjob+".etat")
- fic_etat_2 = tempfile.mktemp()
+ t_aux = tempfile.mkstemp()
+ fic_etat_2 = t_aux[1]
commande_base = os.path.join(self.aster_root, "bin", "as_run")
commande_base += " --actu " + numjob + " " + self.nomjob + " " + self.mode
if self.verbose_max :
chaine = "V_TEST"
nuocc = 1
erreur, message_erreur, info = self.post_aster_1 ( nomfic, chaine, nuocc )
- if erreur :
+ if ( erreur > 0 ) :
break
- self.message_info += info[:-1]
#
# 1.2. Details
#
- info = info.replace(chaine, " ")
- laux = info[:-1].split()
- aux = laux[0]
- if ( "D" in aux ) :
- aux = aux.replace("D", "E")
- dico_resu[chaine] = float(aux)
+ if ( erreur == 0 ) :
+ self.message_info += info[:-1]
+ info = info.replace(chaine, " ")
+ laux = info[:-1].split()
+ aux = laux[0]
+ if ( "D" in aux ) :
+ aux = aux.replace("D", "E")
+ dico_resu[chaine] = float(aux)
+ else :
+ erreur = 0
+ message_erreur = " "
#
# 2. Exploration du fichier mess
# 2.1. Que chercher ?
else :
nuocc = 1
erreur, message_erreur, info = self.post_aster_1 ( nomfic, chaine, nuocc )
- if erreur :
+ if ( erreur > 0 ) :
break
- self.message_info += info[:-1]
#
# 2.3. Details
#
- if chaine == "INSTANT" :
- l_aux = info[:-1].split()
- lg_aux = len(l_aux)
- for iaux in range(lg_aux) :
- if ( "ORDRE" in l_aux[iaux] ) :
- if l_aux[iaux+1] == ":" :
- jaux = iaux+2
- else :
- jaux = iaux+1
- ordre = int(l_aux[jaux])
- dico_resu["ORDRE"] = ordre
- dico_resu["PAS_DE_TEMPS"] = ordre
- elif chaine in ( "NOMBRE DE NOEUDS", "NOMBRE DE MAILLES" ) :
- l_aux = info[:-1].split(chaine)
- dico_resu[chaine] = int(l_aux[1])
+ if ( erreur == 0 ) :
+ self.message_info += info[:-1]
+ if chaine == "INSTANT" :
+ l_aux = info[:-1].split()
+ lg_aux = len(l_aux)
+ for iaux in range(lg_aux) :
+ if ( "ORDRE" in l_aux[iaux] ) :
+ if l_aux[iaux+1] == ":" :
+ jaux = iaux+2
+ else :
+ jaux = iaux+1
+ ordre = int(l_aux[jaux])
+ dico_resu["ORDRE"] = ordre
+ dico_resu["PAS_DE_TEMPS"] = ordre
+ elif chaine in ( "NOMBRE DE NOEUDS", "NOMBRE DE MAILLES" ) :
+ l_aux = info[:-1].split(chaine)
+ dico_resu[chaine] = int(l_aux[1])
+ else :
+ erreur = 0
+ message_erreur = " "
#
if erreur :
break
nomfic = nom du fichier a decoder
chaine = chaine a chercher
nuocc = numero de l'occurence a chercher, 0 si toutes
+Retour :
+codret = 0 : tout va bien
+ 1 : le fichier de resultats est absent
+ -1 : la chaine est absente
+message_erreur = "" : tout va bien
+ != "" si probleme
+info = la ou les lignes recherchees
"""
#
messages_erreur = { 0 : None,
+ -1 : "La chaine est absente.",
1 : "Ce fichier est inconnu." }
#
nom_fonction = __name__ + "/post_aster_1"
blabla = "\nDans " + nom_fonction + " :"
if self.verbose_max :
- print blabla, "chaine =", chaine, ", nuocc =", nuocc
+ print blabla, "nomfic =", nomfic, "chaine =", chaine, ", nuocc =", nuocc
#
+ trouve = False
erreur = 0
message_erreur = " "
info = ""
#
else :
#
- self.message_info += "\n"
iaux = 0
for ligne in les_lignes :
if chaine in ligne :
iaux += 1
if ( ( nuocc == 0 ) or ( iaux == nuocc ) ) :
info += ligne
+ if ( not trouve ) :
+ self.message_info += "\n"
+ trouve = True
#
break
#
- if self.verbose_max :
- print blabla
- print ". erreur :", erreur
+ if ( not trouve ) :
+ erreur = -1
+#
+ if ( self.verbose_max or ( erreur>0 ) ) :
+ print blabla, "nomfic =", nomfic, "chaine =", chaine, ", nuocc =", nuocc
+ print ". erreur =", erreur
#
if erreur :
message_erreur = messages_erreur[erreur]
if self.verbose_max :
print blabla
print ". erreur :", erreur
+#
+ os.remove(nomfic_donn)
#
if erreur :
message_erreur = messages_erreur[erreur]
