diff --git a/visu.py b/visu.py index 89c1a21..9c4b2da 100644 --- a/visu.py +++ b/visu.py @@ -166,7 +166,7 @@ def warped_surface_rectilinear(x,y,z,data,axis): grid['warp'] = data.transpose().ravel() return grid.warp_by_scalar(normal=normal) -def warped_surface_uniform(origin,spacing,data,axis): +def warped_surface_uniform(origin,spacing,data,axis,factor=1.0): import numpy as np import pyvista as pv assert data.shape[axis]==1 @@ -177,6 +177,6 @@ def warped_surface_uniform(origin,spacing,data,axis): grid.origin = origin grid.spacing = spacing grid['warp'] = data.transpose().ravel() - return grid.warp_by_scalar(normal=normal) + return grid.warp_by_scalar(normal=normal,factor=factor) #def extend_structuredgrid(sg,bounds,rep,axis):