From bc893863fe289f967cc5b58f978822baa69aa558 Mon Sep 17 00:00:00 2001 From: Paul RASCLE Date: Wed, 22 May 2019 14:56:49 +0200 Subject: [PATCH] tabulations... --- src/HYDROTools/interpolS.py | 34 +++++++++++++++++----------------- 1 file changed, 17 insertions(+), 17 deletions(-) diff --git a/src/HYDROTools/interpolS.py b/src/HYDROTools/interpolS.py index a2740eb..c116ebd 100644 --- a/src/HYDROTools/interpolS.py +++ b/src/HYDROTools/interpolS.py @@ -35,42 +35,42 @@ def assignStrickler(case_name, med_file_name, output_file_name, med_field_name=' doc = HYDROPy.HYDROData_Document.Document( theStudyId ) case = doc.FindObjectByName(case_name) if case is None: - raise BadParamsError("Calculation case '%s' not found" % case_name) - + raise BadParamsError("Calculation case '%s' not found" % case_name) + # Check input MED file if not os.path.exists(med_file_name): - raise BadParamsError("Input file '%s' not exists" % med_file_name) + raise BadParamsError("Input file '%s' not exists" % med_file_name) # Copy input file to the output path, if the output path is empty - use the input path for the output file_path = med_file_name if output_file_name and (output_file_name != file_path): - shutil.copyfile(med_file_name, output_file_name) - file_path = output_file_name + shutil.copyfile(med_file_name, output_file_name) + file_path = output_file_name # Open input MED file fid = medfile.MEDfileOpen(file_path, medenum.MED_ACC_RDEXT) # Get mesh info mesh_name, sdim, mdim, meshtype, desc, dtunit, sort, nstep, repere, axisname, axisunit = medmesh.MEDmeshInfo(fid, 1) - + nb_nodes, chgt, trsf = medmesh.MEDmeshnEntity(fid, mesh_name, medenum.MED_NO_DT, medenum.MED_NO_IT, - medenum.MED_NODE, medenum.MED_NONE, - medenum.MED_COORDINATE, medenum.MED_NO_CMODE) - - + medenum.MED_NODE, medenum.MED_NONE, + medenum.MED_COORDINATE, medenum.MED_NO_CMODE) + + # Create field if nb_nodes > 0: # Get node coordinates coords = medfile.MEDFLOAT(nb_nodes*sdim) medmesh.MEDmeshNodeCoordinateRd(fid, mesh_name, medenum.MED_NO_DT, medenum.MED_NO_IT, - medenum.MED_FULL_INTERLACE, coords) - + medenum.MED_FULL_INTERLACE, coords) + # Get list of Strickler coefficient values for the nodes values = [] x_coords = coords[0::sdim] y_coords = coords[1::sdim] values = case.GetStricklerCoefficientForPoints(x_coords, y_coords, 0.0, True) - + # Write the values to the field if len(values) > 0: values = medfile.MEDFLOAT(values) @@ -78,12 +78,12 @@ def assignStrickler(case_name, med_file_name, output_file_name, med_field_name=' comp = "strickler coeff " medfield.MEDfieldCr(fid, med_field_name, medfile.MED_FLOAT64, 1, comp, '', '', mesh_name) - + medfield.MEDfieldValueWr(fid, med_field_name, medenum.MED_NO_DT, medenum.MED_NO_IT, 0.0, medenum.MED_NODE, medenum.MED_NONE, medenum.MED_FULL_INTERLACE, medenum.MED_ALL_CONSTITUENT, len(values), values) - + print("MED field '%s' on %s nodes has been created." % (med_field_name, nb_nodes)) - + # close MED file - medfile.MEDfileClose(fid) \ No newline at end of file + medfile.MEDfileClose(fid) -- 2.39.2