def Execute(self):
if (self.Mesh == None):
self.PrintError('Error: no Mesh.')
if (self.PointId == None):
self.PrintError('Error: no PointId.')
if (self.ArrayName == None):
self.PrintError('Error: no ArrayName.')
vectorFromComponents = vmtkmeshvectorfromcomponents.vmtkMeshVectorFromComponents(
)
vectorFromComponents.Mesh = self.Mesh
vectorFromComponents.VectorArrayName = self.ArrayName
vectorFromComponents.ComponentsArrayNames = self.ComponentsArrayNames
vectorFromComponents.RemoveComponentArrays = True
vectorFromComponents.Execute()
dataArray = self.Mesh.GetPointData().GetArray(self.ArrayName)
dataValue = vtk.vtkMath.Norm((dataArray.GetTuple(self.PointId)[0],
dataArray.GetTuple(self.PointId)[1],
dataArray.GetTuple(self.PointId)[2]))
self.PrintLog('')
logLine = 'PointId: %d\n' % self.PointId
logLine += ' Array: %s\n' % self.ArrayName
logLine += ' Components[x,y,z]: %s,%s,%s\n' % (dataArray.GetTuple(
self.PointId)[0], dataArray.GetTuple(
self.PointId)[1], dataArray.GetTuple(self.PointId)[2])
logLine += ' Magnitude: %f\n' % dataValue
self.PrintLog(logLine)
def Execute(self):
if (self.Mesh == None):
self.PrintError('Error: no Mesh.')
if (self.PointId == None):
self.PrintError('Error: no PointId.')
if (self.ArrayName == None):
self.PrintError('Error: no ArrayName.')
vectorFromComponents = vmtkmeshvectorfromcomponents.vmtkMeshVectorFromComponents()
vectorFromComponents.Mesh = self.Mesh
vectorFromComponents.VectorArrayName = self.ArrayName
vectorFromComponents.ComponentsArrayNames = self.ComponentsArrayNames
vectorFromComponents.RemoveComponentArrays = True
vectorFromComponents.Execute()
dataArray = self.Mesh.GetPointData().GetArray(self.ArrayName)
dataValue = vtk.vtkMath.Norm((dataArray.GetTuple(self.PointId)[0],dataArray.GetTuple(self.PointId)[1],dataArray.GetTuple(self.PointId)[2]))
self.PrintLog('')
logLine = 'PointId: %d\n' % self.PointId
logLine += ' Array: %s\n' % self.ArrayName
logLine += ' Components[x,y,z]: %s,%s,%s\n' % (dataArray.GetTuple(self.PointId)[0],dataArray.GetTuple(self.PointId)[1],dataArray.GetTuple(self.PointId)[2])
logLine += ' Magnitude: %f\n' % dataValue
self.PrintLog(logLine)
def Execute(self):
if (self.InputDirectoryName == None):
self.PrintError('Error: no directory.')
if (self.Pattern == None):
self.PrintError('Error: no pattern.')
if (self.FirstTimeStep == None):
self.PrintError('Error: no first timestep.')
if (self.LastTimeStep == None):
self.PrintError('Error: no last timestep.')
if (self.VelocityComponentsArrayNames == None):
self.PrintError('Error: no VelocityComponentsArrayNames.')
for root, dirs, files in os.walk(self.InputDirectoryName):
if root == self.InputDirectoryName:
fileList = [x for x in files if not (x.startswith('.'))]
timeIndexList = range(self.FirstTimeStep, self.LastTimeStep + 1,
self.IntervalTimeStep)
reader = vmtkmeshreader.vmtkMeshReader()
#if self.VelocityVector or self.WsrVector:
if self.WsrVector:
vectorFromComponents = vmtkmeshvectorfromcomponents.vmtkMeshVectorFromComponents(
)
u_name = self.VelocityComponentsArrayNames.split(' ')[0]
v_name = self.VelocityComponentsArrayNames.split(' ')[1]
w_name = self.VelocityComponentsArrayNames.split(' ')[2]
if self.Wsr:
taux_name = self.WsrComponentsArrayNames.split(' ')[0]
tauy_name = self.WsrComponentsArrayNames.split(' ')[1]
tauz_name = self.WsrComponentsArrayNames.split(' ')[2]
field = vtk.vtkFieldData()
field.AllocateArrays(1)
timesteps = vtk.vtkIntArray()
timesteps.SetNumberOfComponents(1)
timesteps.SetName("timesteps")
i = 0
for step in timeIndexList:
if (self.Pattern % step).replace(' ', '0') in fileList:
timesteps.InsertTuple1(i, step)
i += 1
fileName = (self.Pattern % step).replace(' ', '0')
timeIndex = step
reader.InputFileName = os.path.abspath(
os.path.join(self.InputDirectoryName, fileName))
reader.Execute()
mesh = reader.Mesh
if step == self.FirstTimeStep:
mesh.CopyStructure(mesh)
self.Mesh = mesh
if self.Pressure:
p = mesh.GetPointData().GetArray(self.PressureArrayName)
p.SetName(self.PressureArrayName + str(step))
self.Mesh.GetPointData().AddArray(p)
if self.Wsr:
taux = mesh.GetPointData().GetArray(taux_name)
taux.SetName(taux_name + str(step))
self.Mesh.GetPointData().AddArray(taux)
tauy = mesh.GetPointData().GetArray(tauy_name)
tauy.SetName(tauy_name + str(step))
self.Mesh.GetPointData().AddArray(tauy)
tauz = mesh.GetPointData().GetArray(tauz_name)
tauz.SetName(tauz_name + str(step))
self.Mesh.GetPointData().AddArray(tauz)
if self.VelocityVector:
j = 0
u_component = vtk.vtkDoubleArray()
u_component.SetNumberOfComponents(1)
v_component = vtk.vtkDoubleArray()
v_component.SetNumberOfComponents(1)
w_component = vtk.vtkDoubleArray()
w_component.SetNumberOfComponents(1)
while j < mesh.GetPointData().GetArray(
self.VelocityVectorArrayName).GetNumberOfTuples():
u_val = mesh.GetPointData().GetArray(
self.VelocityVectorArrayName).GetComponent(j, 0)
v_val = mesh.GetPointData().GetArray(
self.VelocityVectorArrayName).GetComponent(j, 1)
w_val = mesh.GetPointData().GetArray(
self.VelocityVectorArrayName).GetComponent(j, 2)
u_component.InsertTuple1(j, u_val)
v_component.InsertTuple1(j, v_val)
w_component.InsertTuple1(j, w_val)
j += 1
u_component.SetName(u_name + "_" + str(step))
self.Mesh.GetPointData().AddArray(u_component)
v_component.SetName(v_name + "_" + str(step))
self.Mesh.GetPointData().AddArray(v_component)
w_component.SetName(w_name + "_" + str(step))
self.Mesh.GetPointData().AddArray(w_component)
else:
u = mesh.GetPointData().GetArray(u_name)
u.SetName(u_name + "_" + str(step))
self.Mesh.GetPointData().AddArray(u)
v = mesh.GetPointData().GetArray(v_name)
v.SetName(v_name + "_" + str(step))
self.Mesh.GetPointData().AddArray(v)
w = mesh.GetPointData().GetArray(w_name)
w.SetName(w_name + "_" + str(step))
self.Mesh.GetPointData().AddArray(w)
if self.WsrVector:
vectorFromComponents.Mesh = self.Mesh
vectorFromComponents.VectorArrayName = "Wsr_" + str(step)
vectorFromComponents.ComponentsArrayNames = [
taux.GetName(),
tauy.GetName(),
tauz.GetName()
]
vectorFromComponents.RemoveComponentArrays = True
vectorFromComponents.Execute()
field.AddArray(timesteps)
self.Mesh.SetFieldData(field)
def Execute(self):
if (self.InputDirectoryName == None):
self.PrintError('Error: no directory.')
if (self.Pattern == None):
self.PrintError('Error: no pattern.')
if (self.FirstTimeStep == None):
self.PrintError('Error: no first timestep.')
if (self.LastTimeStep == None):
self.PrintError('Error: no last timestep.')
if (self.VelocityComponentsArrayNames == None):
self.PrintError('Error: no VelocityComponentsArrayNames.')
for root, dirs, files in os.walk(self.InputDirectoryName):
if root == self.InputDirectoryName:
fileList = [x for x in files if not (x.startswith('.'))]
timeIndexList = range(self.FirstTimeStep,self.LastTimeStep+1,self.IntervalTimeStep)
reader = vmtkmeshreader.vmtkMeshReader()
#if self.VelocityVector or self.WsrVector:
if self.WsrVector:
vectorFromComponents = vmtkmeshvectorfromcomponents.vmtkMeshVectorFromComponents()
u_name = self.VelocityComponentsArrayNames.split(' ')[0]
v_name = self.VelocityComponentsArrayNames.split(' ')[1]
w_name = self.VelocityComponentsArrayNames.split(' ')[2]
if self.Wsr:
taux_name = self.WsrComponentsArrayNames.split(' ')[0]
tauy_name = self.WsrComponentsArrayNames.split(' ')[1]
tauz_name = self.WsrComponentsArrayNames.split(' ')[2]
field = vtk.vtkFieldData()
field.AllocateArrays(1)
timesteps = vtk.vtkIntArray()
timesteps.SetNumberOfComponents(1)
timesteps.SetName("timesteps")
i = 0
for step in timeIndexList:
if (self.Pattern%step).replace(' ','0') in fileList:
timesteps.InsertTuple1(i, step)
i+=1
fileName = (self.Pattern%step).replace(' ','0')
timeIndex = step
reader.InputFileName = os.path.abspath(os.path.join(self.InputDirectoryName,fileName))
reader.Execute()
mesh = reader.Mesh
if step == self.FirstTimeStep:
mesh.CopyStructure(mesh)
self.Mesh = mesh
if self.Pressure:
p = mesh.GetPointData().GetArray(self.PressureArrayName)
p.SetName(self.PressureArrayName+str(step))
self.Mesh.GetPointData().AddArray(p)
if self.Wsr:
taux = mesh.GetPointData().GetArray(taux_name)
taux.SetName(taux_name+str(step))
self.Mesh.GetPointData().AddArray(taux)
tauy = mesh.GetPointData().GetArray(tauy_name)
tauy.SetName(tauy_name+str(step))
self.Mesh.GetPointData().AddArray(tauy)
tauz = mesh.GetPointData().GetArray(tauz_name)
tauz.SetName(tauz_name+str(step))
self.Mesh.GetPointData().AddArray(tauz)
if self.VelocityVector:
j = 0
u_component = vtk.vtkDoubleArray()
u_component.SetNumberOfComponents(1)
v_component = vtk.vtkDoubleArray()
v_component.SetNumberOfComponents(1)
w_component = vtk.vtkDoubleArray()
w_component.SetNumberOfComponents(1)
while j < mesh.GetPointData().GetArray(self.VelocityVectorArrayName).GetNumberOfTuples():
u_val = mesh.GetPointData().GetArray(self.VelocityVectorArrayName).GetComponent(j,0)
v_val = mesh.GetPointData().GetArray(self.VelocityVectorArrayName).GetComponent(j,1)
w_val = mesh.GetPointData().GetArray(self.VelocityVectorArrayName).GetComponent(j,2)
u_component.InsertTuple1(j, u_val)
v_component.InsertTuple1(j, v_val)
w_component.InsertTuple1(j, w_val)
j+=1
u_component.SetName(u_name+"_"+str(step))
self.Mesh.GetPointData().AddArray(u_component)
v_component.SetName(v_name+"_"+str(step))
self.Mesh.GetPointData().AddArray(v_component)
w_component.SetName(w_name+"_"+str(step))
self.Mesh.GetPointData().AddArray(w_component)
else:
u = mesh.GetPointData().GetArray(u_name)
u.SetName(u_name+"_"+str(step))
self.Mesh.GetPointData().AddArray(u)
v = mesh.GetPointData().GetArray(v_name)
v.SetName(v_name+"_"+str(step))
self.Mesh.GetPointData().AddArray(v)
w = mesh.GetPointData().GetArray(w_name)
w.SetName(w_name+"_"+str(step))
self.Mesh.GetPointData().AddArray(w)
if self.WsrVector:
vectorFromComponents.Mesh = self.Mesh
vectorFromComponents.VectorArrayName = "Wsr_"+str(step)
vectorFromComponents.ComponentsArrayNames = [taux.GetName(),tauy.GetName(),tauz.GetName()]
vectorFromComponents.RemoveComponentArrays = True
vectorFromComponents.Execute()
field.AddArray(timesteps)
self.Mesh.SetFieldData(field)
