return NULL;
// Set the list elements.
- for ( int i = 1, n = sipCpp->size(); i <= n; ++i )
+ for ( int i = 0, n = sipCpp->size(); i < n; ++i )
{
TYPE* t = new TYPE( (*sipCpp)[i] );
return NULL;
}
- PyList_SET_ITEM( l, i - 1, pobj );
+ PyList_SET_ITEM( l, i, pobj );
}
return l;
return NULL;
// Set the list elements.
- for ( int i = 1, n = sipCpp->size(); i <= n; ++i )
+ for ( int i = 0, n = sipCpp->size(); i < n; ++i )
{
PyObject *pobj;
if ( ( pobj = PyFloat_FromDouble( (*sipCpp)[i] ) ) == NULL )
return NULL;
}
- PyList_SET_ITEM( l, i - 1, pobj );
+ PyList_SET_ITEM( l, i, pobj );
}
return l;
return NULL;
// Set the list elements.
- for ( int i = 1, n = sipCpp->size(); i <= n; ++i )
+ for ( int i = 0, n = sipCpp->size(); i < n; ++i )
{
PyObject *pobj;
if ( ( pobj = SIPLong_FromLong( (*sipCpp)[i] ) ) == NULL )
return NULL;
}
- PyList_SET_ITEM( l, i - 1, pobj );
+ PyList_SET_ITEM( l, i, pobj );
}
return l;
return NULL;
// Set the list elements.
- for ( int i = 1, n = sipCpp->size(); i <= n; ++i )
+ for ( int i = 0, n = sipCpp->size(); i < n; ++i )
{
PyObject *pobj;
if ( ( pobj = SIPLong_FromLong( (*sipCpp)[i] ) ) == NULL )
return NULL;
}
- PyList_SET_ITEM( l, i - 1, pobj );
+ PyList_SET_ITEM( l, i, pobj );
}
return l;
* Remove reference boundary polyline object from calculation case.
*/
void RemoveBoundaryPolyline();
+
+ void SetLandCoverMap( HYDROData_LandCoverMap theLandCoverMap ) [void ( const Handle_HYDROData_LandCoverMap& )];
+ %MethodCode
+ Handle(HYDROData_LandCoverMap) aRef =
+ Handle(HYDROData_LandCoverMap)::DownCast( createHandle( a0 ) );
+ if ( !aRef.IsNull() )
+ {
+ Py_BEGIN_ALLOW_THREADS
+ if ( sipSelfWasArg ) {
+ sipCpp->HYDROData_CalculationCase::SetLandCoverMap( aRef );
+ } else {
+ sipCpp->SetLandCoverMap( aRef );
+ }
+ Py_END_ALLOW_THREADS
+ }
+ %End
+ void SetStricklerTable( HYDROData_StricklerTable theStricklerTable ) [void ( const Handle_HYDROData_StricklerTable& )];
+ %MethodCode
+ Handle(HYDROData_StricklerTable) aRef =
+ Handle(HYDROData_StricklerTable)::DownCast( createHandle( a0 ) );
+ if ( !aRef.IsNull() )
+ {
+ Py_BEGIN_ALLOW_THREADS
+ if ( sipSelfWasArg ) {
+ sipCpp->HYDROData_CalculationCase::SetStricklerTable( aRef );
+ } else {
+ sipCpp->SetStricklerTable( aRef );
+ }
+ Py_END_ALLOW_THREADS
+ }
+ %End
/**
* Add new one child region for calculation case.
--- /dev/null
+# -*- coding: utf-8 -*-
+
+import os
+import sys
+import shutil
+
+import salome
+
+from med import medfile
+from med import medmesh
+from med import medfield
+from med import medenum
+
+import HYDROPy
+
+# Get current study id
+salome.salome_init()
+theStudy = salome.myStudy
+theStudyId = salome.myStudyId
+
+"""Bad parameters exception"""
+class BadParamsError(ValueError):
+ pass
+
+"""
+Creates MED file with scalar field associated with the mesh of the calculation case.
+The scalar value associated with the mesh node is the Strickler coefficient of the land cover containing the node.
+case_name: calculation case name in the study
+med_file_name: path to input MED file with mesh on nodes
+output_file_name: path to output MED file with
+med_field_name: field name
+"""
+def assignStrickler(case_name, med_file_name, output_file_name, med_field_name='FRICTION'):
+ # Check calculation case
+ doc = HYDROPy.HYDROData_Document.Document( theStudyId )
+ case = doc.FindObjectByName(case_name)
+ if case is None:
+ 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)
+
+ # 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
+
+ # 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)
+
+
+ # 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)
+
+ # 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)
+
+ 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
