itop = pftools.computeTopSurfaceIndices(top) ## Compute the surface storage ps = pftools.dataPointExtent(input0) + (3, ) xx = np.reshape(vton(output.GetPoints().GetData()), ps) sus = np.sum(pftools.computeSurfaceStorage(ext, itop, xx, pressure)) arr = ntov(sus) arr.SetName('surface storage') arr.GetInformation().Set(e2r.GLOBAL_TEMPORAL_VARIABLE(), 1) output.GetFieldData().AddArray(arr) ## Compute the surface runoff slope = np.reshape(slope, top.shape + (2, )) mannings = np.reshape(mannings, top.shape) sro = np.sum( pftools.computeSurfaceRunoff(top, xx, pressure, slope, mannings)) arr = ntov(sro) arr.SetName('surface runoff') arr.GetInformation().Set(e2r.GLOBAL_TEMPORAL_VARIABLE(), 1) output.GetFieldData().AddArray(arr) return 1 if __name__ == "__main__": from paraview.detail.pythonalgorithm import get_plugin_xmls from xml.dom.minidom import parseString for xml in get_plugin_xmls(globals()): dom = parseString(xml) print(dom.toprettyxml(" ", "\n"))
print("Pretend work done!") return 1 def test_PythonSuperquadricSource(): src = PythonSuperquadricSource() src.Update() npts = src.GetOutputDataObject(0).GetNumberOfPoints() assert npts > 0 src.SetThetaResolution(50) src.SetPhiResolution(50) src.Update() assert src.GetOutputDataObject(0).GetNumberOfPoints() > npts def test_PythonCSVReader(fname): reader = PythonCSVReader() reader.SetFileName(fname) reader.Update() assert reader.GetOutputDataObject(0).GetNumberOfRows() > 0 if __name__ == "__main__": #test_PythonSuperquadricSource() #test_PythonCSVReader("/tmp/data.csv") from paraview.detail.pythonalgorithm import get_plugin_xmls from xml.dom.minidom import parseString for xml in get_plugin_xmls(globals()): dom = parseString(xml) print(dom.toprettyxml(" ","\n"))