def generateWarpedMesh(
mesh_folder,
mesh_basename,
images,
structure,
deformation,
evolution,
verbose=0):
myVTK.myPrint(verbose, "*** generateWarpedMesh ***")
mesh = myVTK.readUGrid(
filename=mesh_folder+"/"+mesh_basename+".vtk",
verbose=verbose-1)
n_points = mesh.GetNumberOfPoints()
n_cells = mesh.GetNumberOfCells()
if os.path.exists(mesh_folder+"/"+mesh_basename+"-WithLocalBasis.vtk"):
ref_mesh = myVTK.readUGrid(
filename=mesh_folder+"/"+mesh_basename+"-WithLocalBasis.vtk",
verbose=verbose-1)
else:
ref_mesh = None
farray_disp = myVTK.createFloatArray(
name="displacement",
n_components=3,
n_tuples=n_points,
verbose=verbose-1)
mesh.GetPointData().AddArray(farray_disp)
mapping = Mapping(images, structure, deformation, evolution)
X = numpy.empty(3)
x = numpy.empty(3)
U = numpy.empty(3)
if ("zfill" not in images.keys()):
images["zfill"] = len(str(images["n_frames"]))
for k_frame in xrange(images["n_frames"]):
t = images["T"]*float(k_frame)/(images["n_frames"]-1) if (images["n_frames"]>1) else 0.
mapping.init_t(t)
for k_point in xrange(n_points):
mesh.GetPoint(k_point, X)
mapping.x(X, x)
U = x - X
farray_disp.SetTuple(k_point, U)
myVTK.computeStrainsFromDisplacements(
mesh=mesh,
disp_array_name="displacement",
ref_mesh=ref_mesh,
verbose=verbose-1)
myVTK.writeUGrid(
ugrid=mesh,
filename=mesh_folder+"/"+mesh_basename+"_"+str(k_frame).zfill(images["zfill"])+".vtk",
verbose=verbose-1)
def generateWarpedMesh(mesh_folder,
mesh_basename,
images,
structure,
deformation,
evolution,
verbose=0):
myVTK.myPrint(verbose, "*** generateWarpedMesh ***")
mesh = myVTK.readUGrid(filename=mesh_folder + "/" + mesh_basename + ".vtk",
verbose=verbose - 1)
n_points = mesh.GetNumberOfPoints()
n_cells = mesh.GetNumberOfCells()
if os.path.exists(mesh_folder + "/" + mesh_basename +
"-WithLocalBasis.vtk"):
ref_mesh = myVTK.readUGrid(filename=mesh_folder + "/" + mesh_basename +
"-WithLocalBasis.vtk",
verbose=verbose - 1)
else:
ref_mesh = None
farray_disp = myVTK.createFloatArray(name="displacement",
n_components=3,
n_tuples=n_points,
verbose=verbose - 1)
mesh.GetPointData().AddArray(farray_disp)
mapping = Mapping(images, structure, deformation, evolution)
X = numpy.empty(3)
x = numpy.empty(3)
U = numpy.empty(3)
if ("zfill" not in images.keys()):
images["zfill"] = len(str(images["n_frames"]))
for k_frame in xrange(images["n_frames"]):
t = images["T"] * float(k_frame) / (images["n_frames"] - 1) if (
images["n_frames"] > 1) else 0.
mapping.init_t(t)
for k_point in xrange(n_points):
mesh.GetPoint(k_point, X)
mapping.x(X, x)
U = x - X
farray_disp.SetTuple(k_point, U)
myVTK.computeStrainsFromDisplacements(mesh=mesh,
disp_array_name="displacement",
ref_mesh=ref_mesh,
verbose=verbose - 1)
myVTK.writeUGrid(ugrid=mesh,
filename=mesh_folder + "/" + mesh_basename + "_" +
str(k_frame).zfill(images["zfill"]) + ".vtk",
verbose=verbose - 1)
def writeDataSet(dataset, filename, verbose=0):
mypy.my_print(verbose, "*** writeDataSet: " + filename + " ***")
file_ext = filename[-3:]
if (file_ext == "vti") or ((file_ext == "vtk")
and dataset.IsA("vtkImageData")):
myvtk.writeImage(image=dataset, filename=filename, verbose=verbose - 1)
elif (file_ext == "vtp") or ((file_ext == "vtk")
and dataset.IsA("vtkPolyData")):
myvtk.writePData(pdata=dataset, filename=filename, verbose=verbose - 1)
elif (file_ext == "stl"):
myvtk.writeSTL(pdata=dataset, filename=filename, verbose=verbose - 1)
elif (file_ext == "vts") or ((file_ext == "vtk")
and dataset.IsA("vtkStructuredGrid")):
myvtk.writeSGrid(sgrid=dataset, filename=filename, verbose=verbose - 1)
elif (file_ext == "vtu") or ((file_ext == "vtk")
and dataset.IsA("vtkUnstructuredGrid")):
myvtk.writeUGrid(ugrid=dataset, filename=filename, verbose=verbose - 1)
else:
assert 0, "Wrong extention (" + file_ext + "). Extention must be vtk, vti, vtp, stl, vts, vtu. Aborting."
### ###
### University of California at San Francisco (UCSF), USA ###
### Swiss Federal Institute of Technology (ETH), Zurich, Switzerland ###
### École Polytechnique, Palaiseau, France ###
### ###
########################################################################
from builtins import range
import argparse
import myPythonLibrary as mypy
import myVTKPythonLibrary as myvtk
########################################################################
if (__name__ == "__main__"):
parser = argparse.ArgumentParser()
parser.add_argument("inp_filename", type=str)
args = parser.parse_args()
mesh = myvtk.readAbaqusMeshFromINP(
mesh_filename=args.inp_filename,
elem_types="all",
verbose=1)
myvtk.writeUGrid(
ugrid=mesh,
filename=args.inp_filename.replace(".inp", ".vtk"),
verbose=1)
#coding=utf8
########################################################################
### ###
### Created by Martin Genet, 2012-2017 ###
### ###
### University of California at San Francisco (UCSF), USA ###
### Swiss Federal Institute of Technology (ETH), Zurich, Switzerland ###
### École Polytechnique, Palaiseau, France ###
### ###
########################################################################
import argparse
import myPythonLibrary as mypy
import myVTKPythonLibrary as myvtk
########################################################################
if (__name__ == "__main__"):
parser = argparse.ArgumentParser()
parser.add_argument('vtu_filename', type=str)
args = parser.parse_args()
assert (args.vtu_filename.endswith(".vtu"))
mesh = myvtk.readUGrid(
filename=args.vtu_filename)
myvtk.writeUGrid(
ugrid=mesh,
filename=args.vtu_filename.replace(".vtu", ".vtk"))
#coding=utf8
########################################################################
### ###
### Created by Martin Genet, 2012-2016 ###
### ###
### University of California at San Francisco (UCSF), USA ###
### Swiss Federal Institute of Technology (ETH), Zurich, Switzerland ###
### École Polytechnique, Palaiseau, France ###
### ###
########################################################################
import argparse
import myPythonLibrary as mypy
import myVTKPythonLibrary as myvtk
########################################################################
if (__name__ == "__main__"):
parser = argparse.ArgumentParser()
parser.add_argument('vtk_filename', type=str)
args = parser.parse_args()
assert (args.vtk_filename.endswith(".vtk"))
mesh = myvtk.readUGrid(
filename=args.vtk_filename)
myvtk.writeUGrid(
ugrid=mesh,
filename=args.vtk_filename.replace(".vtk", ".vtu"))
########################################################################
def pdata2ugrid(pdata, verbose=0):
mypy.my_print(verbose, "*** pdata2ugrid ***")
filter_append = vtk.vtkAppendFilter()
if (vtk.vtkVersion.GetVTKMajorVersion() >= 6):
filter_append.SetInputData(pdata)
else:
filter_append.SetInput(pdata)
filter_append.Update()
ugrid = filter_append.GetOutput()
return ugrid
########################################################################
if (__name__ == "__main__"):
parser = argparse.ArgumentParser()
parser.add_argument('pdata_filename', type=str)
args = parser.parse_args()
assert (args.ugrid_filename.endswith(".vtp"))
pdata = myvtk.readPData(filename=args.pdata_filename)
ugrid = myvtk.pdata2ugrid(pdata=pdata)
myvtk.writeUGrid(ugrid=ugrid,
filename=args.pdata_filename.replace(".vtp", ".vtu"))
mesh = myvtk.readUGrid(
filename=args.mesh_in_filename,
verbose=args.verbose)
elif (args.mesh_in_filename.endswith(".vtp") or args.mesh_in_filename.endswith(".stl")):
mesh = myvtk.readPData(
filename=args.mesh_in_filename,
verbose=args.verbose)
else:
assert (0), "Input file must be .vtk, .vtu, or .stl. Aborting."
myvtk.moveMeshWithWorldMatrix(
mesh=mesh,
M=I2I,
verbose=args.verbose)
if (args.mesh_out_filename == None):
args.mesh_out_filename = args.mesh_in_filename.replace(args.image_in, args.image_out)
if (args.mesh_out_filename.endswith(".vtk") or args.mesh_out_filename.endswith(".vtu")):
myvtk.writeUGrid(
ugrid=mesh,
filename=args.mesh_out_filename,
verbose=args.verbose)
elif (args.mesh_out_filename.endswith(".stl")):
myvtk.writeSTL(
pdata=mesh,
filename=args.mesh_out_filename,
verbose=args.verbose)
else:
assert (0), "Output file must be .vtk, .vtu, or .stl. Aborting."
########################################################################
def pdata2ugrid(pdata, verbose=0):
mypy.my_print(verbose, "*** pdata2ugrid ***")
filter_append = vtk.vtkAppendFilter()
if vtk.vtkVersion.GetVTKMajorVersion() >= 6:
filter_append.SetInputData(pdata)
else:
filter_append.SetInput(pdata)
filter_append.Update()
ugrid = filter_append.GetOutput()
return ugrid
########################################################################
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("pdata_filename", type=str)
args = parser.parse_args()
assert args.ugrid_filename.endswith(".vtp")
pdata = myvtk.readPData(filename=args.pdata_filename)
ugrid = myvtk.pdata2ugrid(pdata=pdata)
myvtk.writeUGrid(ugrid=ugrid, filename=args.pdata_filename.replace(".vtp", ".vtu"))
### ###
### Created by Martin Genet, 2012-2016 ###
### ###
### University of California at San Francisco (UCSF), USA ###
### Swiss Federal Institute of Technology (ETH), Zurich, Switzerland ###
### École Polytechnique, Palaiseau, France ###
### ###
########################################################################
import argparse
import myPythonLibrary as mypy
import myVTKPythonLibrary as myvtk
########################################################################
if (__name__ == "__main__"):
parser = argparse.ArgumentParser()
parser.add_argument('inp_filename', type=str)
args = parser.parse_args()
mesh = myvtk.readAbaqusMeshFromINP(
mesh_filename=args.inp_filename,
elem_types="all",
verbose=1)
myvtk.writeUGrid(
ugrid=mesh,
filename=args.inp_filename.replace(".inp", ".vtk"),
verbose=1)
mesh = myvtk.readUGrid(
filename=args.mesh_in_filename,
verbose=args.verbose)
elif (args.mesh_in_filename.endswith(".vtp") or args.mesh_in_filename.endswith(".stl")):
mesh = myvtk.readPData(
filename=args.mesh_in_filename,
verbose=args.verbose)
else:
assert (0), "Input file must be .vtk, .vtu, or .stl. Aborting."
myvtk.moveMeshWithWorldMatrix(
mesh=mesh,
M=I2I,
verbose=args.verbose)
if (args.mesh_out_filename == None):
args.mesh_out_filename = args.mesh_in_filename.replace(args.image_in+"Coords", args.image_out+"Coords")
if (args.mesh_out_filename.endswith(".vtk") or args.mesh_out_filename.endswith(".vtu")):
myvtk.writeUGrid(
ugrid=mesh,
filename=args.mesh_out_filename,
verbose=args.verbose)
elif (args.mesh_out_filename.endswith(".stl")):
myvtk.writeSTL(
pdata=mesh,
filename=args.mesh_out_filename,
verbose=args.verbose)
else:
assert (0), "Output file must be .vtk, .vtu, or .stl. Aborting."
