src/salome_hydro/generate_interpolz.py

   1
   2 import os
   3 hydro_solver_root = os.path.join(os.environ['HYDROSOLVER_ROOT_DIR'], 'lib', 'python2.7', 'site-packages', 'salome', 'salome', 'hydro')
   4
   5 def replace( lines, pattern, subst ):
   6         for i in range( 0, len( lines ) ):
   7                 line = lines[i]
   8                 if pattern in lines[i]:
   9                         if isinstance( subst, list ):
  10                                 del lines[i]
  11                                 for s in subst:
  12                                         new_line = line.replace( pattern, repr(s) )
  13                                         lines.insert( i, new_line )
  14                         elif isinstance( subst, dict ):
  15                                 del lines[i]
  16                                 name = pattern[1:-1]
  17                                 lines.insert( i, "%s = {}\n" % name )
  18                                 keys = subst.keys()
  19                                 keys.sort()
  20                                 i = i+1
  21                                 for k in keys:
  22                                         v = subst[k]
  23                                         lines.insert( i, "%s[%s] = %s\n" % (name, repr(k), repr(v) ) )
  24                                         i = i+1
  25                         else:
  26                                 new_line = line.replace( pattern, repr(subst) )
  27                                 lines[i] = new_line
  28
  29
  30 def generate( path, calc_case, med_file, med_groups_regions, z_undef, interp ):
  31         f = open( path, "w" )
  32         tf = open( hydro_solver_root+"/interpolz.template", "r" )
  33         templ = tf.readlines()
  34
  35         replace( templ, "<case_name_from_dlg>", calc_case )
  36         replace( templ, "<MED_file_path_from_dlg>", med_file )
  37         replace( templ, "<dictGroupRegion>", med_groups_regions )
  38         replace( templ, "<z_undef_value_from_dlg>", z_undef )
  39         replace( templ, "<interpolation_method_value_from_dlg>", interp )
  40
  41         for line in templ:
  42                 f.write( line )
  43         f.close()
  44
  45
  46 if __name__=='__main__':
  47     generate( "test_interpolz.py", "case_1", "mesh.med", {"group_1":"reg_1", "group_2":"reg_2"}, -9999, "linear" )