]> SALOME platform Git repositories - modules/hydro.git/blob - src/HYDROTools/interpolS.py
Salome HOME
Merge branch 'BR_H2018_3' into BR_2018_V8_5
[modules/hydro.git] / src / HYDROTools / interpolS.py
1 # -*- coding: utf-8 -*-
2
3 """
4 assignStrickler
5 """
6 __revision__ = "V2.02"
7
8 import os
9
10 import salome
11
12 import HYDROPy
13
14 import numpy as np
15 import MEDLoader as ml
16 import MEDCoupling as mc
17
18 # Get current study id
19 salome.salome_init()
20 theStudy = salome.myStudy
21 theStudyId = salome.myStudyId
22
23 class BadParamsError(ValueError):
24   """Bad parameters exception"""
25   pass
26
27
28 def assignStrickler(nomCas, input_file_name, output_file_name="", med_field_name='BOTTOM FRICTION', verbose=False):
29   """
30   assignStrickler creates the scalar field associated with the mesh of the calculation case that
31   represents the Strickler coefficient of the land cover.
32
33   In:
34     nomCas: Calculation Case Name in module HYDRO
35     input_file_name: med file name corresponding to the HYDRO case
36     output_file_name: med file name with the Strickler coefficient; default is the same as fichierMaillage
37     med_field_name: name of the field of the Strickler coefficient
38                      default value is 'BOTTOM FRICTION'.
39   Out:
40   """
41   erreur = 0
42   message = ""
43
44   while not erreur:
45
46     if verbose:
47       print "nomCas:", nomCas
48       print "input_file_name:", input_file_name
49       print "output_file_name:", output_file_name
50       print "med_field_name:", med_field_name
51
52 # 1. Controls
53 # 1.1. Check calculation case
54     doc = HYDROPy.HYDROData_Document.Document(theStudyId)
55     case = doc.FindObjectByName(nomCas)
56     if case is None:
57       raise BadParamsError("Calculation case '%s' not found" % nomCas)
58
59 # 1.2. Check input MED file
60     if not os.path.exists(input_file_name):
61       raise BadParamsError("Input file '%s' not exists" % input_file_name)
62
63 # 2. The output file
64 # 2.1. If output MED file is not defined, it is equal to the input file
65     if not output_file_name:
66       output_file_name = input_file_name
67
68 # 2.2. Copy input file to the output path, if the output path is different from the input path
69     if output_file_name != input_file_name:
70       import shutil
71       shutil.copyfile(input_file_name, output_file_name)
72 #
73 # 3. Reads the mesh
74 #
75     meshMEDFileRead = ml.MEDFileMesh.New(output_file_name)
76 #
77 # 4. Gets the information about the nodes
78 #
79     nbnodes = meshMEDFileRead.getNumberOfNodes()
80     if verbose:
81       print "Number of nodes: %d" % nbnodes
82 #
83     coords = meshMEDFileRead.getCoords()
84     #print "coords =\n", coords
85 #
86 # 5. Calculation of the Strickler coefficient
87     values = list()
88     x_coords = coords[:, 0]
89     y_coords = coords[:, 1]
90     #print "x_coords =\n", x_coords
91     #print "y_coords =\n", y_coords
92     values = case.GetStricklerCoefficientForPoints(x_coords, y_coords, 0.0, True)
93 #
94 # 6. Field MED file
95 #
96     coeff = mc.DataArrayDouble(np.asfarray(values, dtype='float'))
97     coeff.setInfoOnComponents(["Strickler [SI]"])
98     #print "coeff =\n", coeff
99     if verbose:
100       print ".. Ecriture du Strickler sous le nom '"+med_field_name+"'"
101     fieldOnNodes = ml.MEDCouplingFieldDouble(ml.ON_NODES)
102     fieldOnNodes.setName(med_field_name)
103     fieldOnNodes.setMesh(meshMEDFileRead.getMeshAtLevel(0))
104     fieldOnNodes.setArray(coeff)
105 #   Ces valeurs d'instants sont mises pour assurer la lecture par TELEMAC
106 #   instant = 0.0
107 #   numero d'itération : 0
108 #   pas de numero d'ordre (-1)
109     fieldOnNodes.setTime(0.0, 0, -1)
110 #
111     fMEDFile_ch_d = ml.MEDFileField1TS()
112     fMEDFile_ch_d.setFieldNoProfileSBT(fieldOnNodes)
113     fMEDFile_ch_d.write(output_file_name, 0)
114 #
115     break
116
117   if erreur:
118     print message
119
120   return