display_only(prs, view)
# Show scalar bar
+ global _current_bar
if show_bar and _current_bar:
_current_bar.Visibility = 1
os.makedirs(pic_dir)
# Save picture
+ print "Write image:", file_name
pvs.WriteImage(file_name, view=view, Magnification=1)
def set_visible_lines(xy_prs, lines):
"""Set visible only the given lines for XYChartRepresentation."""
- sv = xy_prs.GetProperty("SeriesVisibilityInfo").GetData()
+ sv = xy_prs.GetProperty("SeriesVisibility").GetData()
visible = '0'
for i in xrange(0, len(sv)):
fields_info = proxy.GetProperty("FieldsTreeInfo")[::2]
arr_name_with_dis=[elt.split("/")[-1] for elt in fields_info]
- proxy.AllArrays = []
- proxy.UpdatePipeline()
-
fields = []
for name in on_gauss:
fields.append(name+_med_field_sep+'GAUSS')
if arr_name_with_dis.count(name) > 0:
index = arr_name_with_dis.index(name)
field_list.append(fields_info[index])
-
- proxy.AllArrays = field_list
- proxy.UpdatePipeline()
+
+ if field_list:
+ proxy.AllArrays = field_list
+ proxy.UpdatePipeline()
return len(field_list) != 0
# TODO: VTN. Looks like this code is out of date.
-
+
#all_cell_types = proxy.CellTypes.Available
all_cell_types = proxy.Entity.Available
all_arrays = list(proxy.CellArrays.GetData())
# Reassign the current bar
_current_bar = scalar_bar
- return scalar_bar
+ return _current_bar
def get_bar():
"""Get current scalar bar."""
- global _current_bar
-
return _current_bar
def get_time(proxy, timestamp_nb):
"""Get time value by timestamp number."""
- proxy.UpdatePipeline()
+ #proxy.UpdatePipeline()
# Check timestamp number
timestamps = []
-
+
if (hasattr(proxy, 'TimestepValues')):
timestamps = proxy.TimestepValues.GetData()
elif (hasattr(proxy.Input, 'TimestepValues')):
if (timestamp_nb > 0 and (timestamp_nb - 1) not in xrange(length) ) or (timestamp_nb < 0 and -timestamp_nb > length):
raise ValueError("Timestamp number is out of range: " + str(timestamp_nb))
+ if not timestamps:
+ return 0.0
+
# Return time value
if timestamp_nb > 0:
return timestamps[timestamp_nb - 1]
lookup_table.LockScalarRange = 1
lookup_table.RGBPoints = [data_range[0], 0, 0, 1, data_range[1], 1, 0, 0]
# Set properties
- scalarmap.ColorAttributeType = EntityType.get_pvtype(entity)
- scalarmap.ColorArrayName = field_name
+ scalarmap.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
scalarmap.LookupTable = lookup_table
# Add scalar bar
lookup_table.RGBPoints = [data_range[0], 0, 0, 1, data_range[1], 1, 0, 0]
# Set properties
- cut_planes.ColorAttributeType = EntityType.get_pvtype(entity)
- cut_planes.ColorArrayName = field_name
+ cut_planes.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
cut_planes.LookupTable = lookup_table
# Add scalar bar
lookup_table.RGBPoints = [data_range[0], 0, 0, 1, data_range[1], 1, 0, 0]
# Set properties
- cut_lines.ColorAttributeType = EntityType.get_pvtype(entity)
- cut_lines.ColorArrayName = field_name
+ cut_lines.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
cut_lines.LookupTable = lookup_table
# Set wireframe represenatation mode
lookup_table.RGBPoints = [data_range[0], 0, 0, 1, data_range[1], 1, 0, 0]
# Set properties
- cut_segment.ColorAttributeType = EntityType.get_pvtype(entity)
- cut_segment.ColorArrayName = field_name
+ cut_segment.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
cut_segment.LookupTable = lookup_table
# Set wireframe represenatation mode
glyph = pvs.Glyph(source)
glyph.Vectors = vector_array
glyph.ScaleMode = 'vector'
- glyph.MaskPoints = 0
+ #glyph.MaskPoints = 0
# Set glyph type
glyph.GlyphType = glyph_type
glyph.GlyphType.Center = [0.0, 0.0, 0.0]
if scale_factor is not None:
- glyph.SetScaleFactor = scale_factor
+ glyph.ScaleFactor = scale_factor
else:
def_scale = get_default_scale(PrsTypeEnum.DEFORMEDSHAPE,
proxy, entity, field_name)
- glyph.SetScaleFactor = def_scale
+ glyph.ScaleFactor = def_scale
glyph.UpdatePipeline()
# Set properties
if (is_colored):
- vectors.ColorArrayName = 'GlyphVector'
+ vectors.ColorArrayName = (EntityType.get_pvtype(entity), 'GlyphVector')
else:
- vectors.ColorArrayName = ''
+ vectors.ColorArrayName = (None, '')
vectors.LookupTable = lookup_table
vectors.LineWidth = 1.0
# Do merge
source = pvs.MergeBlocks(proxy)
+ pvs.UpdatePipeline()
# Cell data to point data
if is_data_on_cells(proxy, field_name):
# Set properties
if is_colored:
- defshape.ColorAttributeType = EntityType.get_pvtype(entity)
- defshape.ColorArrayName = field_name
+ defshape.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
else:
- defshape.ColorArrayName = ''
+ defshape.ColorArrayName = (None, '')
defshape.LookupTable = lookup_table
# Set wireframe represenatation mode
# Do merge
source = pvs.MergeBlocks(proxy)
+ pvs.UpdatePipeline()
# Cell data to point data
if is_data_on_cells(proxy, field_name):
lookup_table.RGBPoints = [data_range[0], 0, 0, 1, data_range[1], 1, 0, 0]
# Set properties
- defshapemap.ColorArrayName = scalar_field
+ defshapemap.ColorArrayName = (EntityType.get_pvtype(scalar_field_entity), scalar_field)
defshapemap.LookupTable = lookup_table
- defshapemap.ColorAttributeType = EntityType.get_pvtype(scalar_field_entity)
# Add scalar bar
add_scalar_bar(field_name, nb_components,
lookup_table.RGBPoints = [data_range[0], 0, 0, 1, data_range[1], 1, 0, 0]
# Set properties
- plot3d.ColorAttributeType = EntityType.get_pvtype(entity)
- plot3d.ColorArrayName = field_name
+ plot3d.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
plot3d.LookupTable = lookup_table
# Add scalar bar
# Do merge
source = pvs.MergeBlocks(proxy)
+ pvs.UpdatePipeline()
# Transform cell data into point data if necessary
if is_data_on_cells(proxy, field_name):
# Set display properties
if (is_colored):
- isosurfaces.ColorAttributeType = EntityType.get_pvtype(entity)
- isosurfaces.ColorArrayName = field_name
+ isosurfaces.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
else:
- isosurfaces.ColorArrayName = ''
+ isosurfaces.ColorArrayName = (None, '')
if color:
isosurfaces.DiffuseColor = color
isosurfaces.LookupTable = lookup_table
# Set timestamp
pvs.GetRenderView().ViewTime = time_value
- proxy.UpdatePipeline(time=time_value)
+ pvs.UpdatePipeline(time_value, proxy)
source = proxy
vector_array = field_name
# If the given vector array has only 2 components, add the third one
if nb_components == 2:
- calc = get_add_component_calc(source,
- EntityType.NODE, field_name)
+ calc = get_add_component_calc(source, EntityType.NODE, field_name)
vector_array = calc.ResultArrayName
source = calc
# Set display properties
if is_colored:
- gausspnt.ColorAttributeType = EntityType.get_pvtype(entity)
- gausspnt.ColorArrayName = field_name
+ gausspnt.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
else:
- gausspnt.ColorArrayName = ''
+ gausspnt.ColorArrayName = (None, '')
if color:
gausspnt.DiffuseColor = color
# Create Gauss Points object
source = pvs.GaussPoints(proxy)
source.UpdatePipeline()
-
+
# Get Gauss Points representation object
gausspnt = pvs.GetRepresentation(source)
# Set display properties
if is_colored:
- gausspnt.ColorAttributeType = EntityType.get_pvtype(entity)
- gausspnt.ColorArrayName = field_name
+ gausspnt.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
else:
- gausspnt.ColorArrayName = ''
+ gausspnt.ColorArrayName = (None, '')
if color:
gausspnt.DiffuseColor = color
# Do merge
source = pvs.MergeBlocks(proxy)
+ pvs.UpdatePipeline()
# Cell data to point data
if is_data_on_cells(proxy, field_name):
cell_to_point = pvs.CellDatatoPointData(source)
cell_to_point.PassCellData = 1
- cell_to_point.UpdatePipeline()
+ pvs.UpdatePipeline()
source = cell_to_point
vector_array = field_name
if nb_components == 2:
calc = get_add_component_calc(source, EntityType.NODE, field_name)
vector_array = calc.ResultArrayName
- calc.UpdatePipeline()
+ pvs.UpdatePipeline()
source = calc
# Stream Tracer
stream = pvs.StreamTracer(source)
stream.SeedType = "Point Source"
stream.Vectors = ['POINTS', vector_array]
- stream.SeedType = "Point Source"
stream.IntegrationDirection = direction
stream.IntegratorType = 'Runge-Kutta 2'
stream.UpdatePipeline()
# Set properties
if is_colored:
- streamlines.ColorAttributeType = EntityType.get_pvtype(entity)
- streamlines.ColorArrayName = field_name
+ streamlines.ColorArrayName = (EntityType.get_pvtype(entity), field_name)
else:
- streamlines.ColorArrayName = ''
+ streamlines.ColorArrayName = (None, '')
if color:
streamlines.DiffuseColor = color
if not mesh_full_name:
raise RuntimeError, "The given mesh name was not found"
# Select only the given mesh
- proxy.AllArrays = []
- proxy.UpdatePipeline()
proxy.AllArrays = [mesh_full_name]
proxy.UpdatePipeline()
prs = None
if (proxy.GetDataInformation().GetNumberOfPoints() or
proxy.GetDataInformation().GetNumberOfCells()):
- prs = pvs.GetRepresentation(proxy)
- prs.ColorArrayName = ''
+ my_view = pvs.GetRenderView()
+ prs = pvs.GetRepresentation(proxy, view=my_view)
+ prs.ColorArrayName = (None, '')
return prs
return prs
-def CreatePrsForFile(paravis_instance, file_name, prs_types,
+def CreatePrsForFile(file_name, prs_types,
picture_dir, picture_ext):
"""Build presentations of the given types for the file.
Build presentations for all fields on all timestamps.
Arguments:
- paravis_instance: ParaVis module instance object
file_name: full path to the MED file
prs_types: the list of presentation types to build
picture_dir: the directory path for saving snapshots
if proxy is None:
print "FAILED"
else:
- proxy.UpdatePipeline()
+ #proxy.UpdatePipeline()
print "OK"
except:
print "FAILED"
else:
# Get view
view = pvs.GetRenderView()
+ time_value = get_time(proxy, 0)
+ view.ViewTime = time_value
+ pvs.UpdatePipeline(time=time_value, proxy=proxy)
# Create required presentations for the proxy
CreatePrsForProxy(proxy, view, prs_types,
print "OK"
# Construct image file name
pic_name = picture_dir + get_field_short_name(mesh_name) + "." + picture_ext
-
+
# Show and dump the presentation into a graphics file
process_prs_for_test(prs, view, pic_name, False)
print "OK"
# Construct image file name
pic_name = picture_dir + get_group_short_name(group) + "." + picture_ext
-
+
# Show and dump the presentation into a graphics file
process_prs_for_test(prs, view, pic_name, False)
if field_name == get_field_mesh_name(field):
continue
field_entity = get_field_entity(field)
- # Clear fields selection state
- proxy.AllArrays = []
- proxy.UpdatePipeline()
# Select only the current field:
# necessary for getting the right timestamps
- proxy.AllArrays = field
+ proxy.AllArrays = [field]
proxy.UpdatePipeline()
# Get timestamps