def DumpPipeline(export_rendering, simulation_input_map, screenshot_info):
global write_frequencies, cp_views, cp_writers
# Initialize the write frequency map
for key in simulation_input_map.values():
write_frequencies[key] = []
# Start trace
smtrace.start_trace(CaptureAllProperties=True, UseGuiName=True)
# update trace globals.
smtrace.trace_globals.proxy_ctor_hook = staticmethod(cp_hook)
smtrace.trace_globals.trace_output = []
# Get list of proxy lists
proxy_lists = smstate.get_proxy_lists_ordered_by_group(WithRendering=export_rendering)
# Now register the proxies with the smtrace module
for proxy_list in proxy_lists:
smstate.register_proxies_by_dependency(proxy_list)
# Calling append_trace causes the smtrace module to sort out all the
# registered proxies and their properties and write them as executable
# python.
smtrace.append_trace()
# Stop trace and print it to the console
smtrace.stop_trace()
for view_proxy in view_proxies:
# Locate which simulation input this write is connected to, if any. If so,
# we update the write_frequencies datastructure accordingly.
sim_inputs = cp_locate_simulation_inputs_for_view(view_proxy)
proxyName = servermanager.ProxyManager().GetProxyName("views", view_proxy)
image_write_frequency = screenshot_info[proxyName][1]
for sim_input_name in sim_inputs:
if not image_write_frequency in write_frequencies[sim_input_name]:
write_frequencies[sim_input_name].append(image_write_frequency)
write_frequencies[sim_input_name].sort()
# Create global fields values
pipelineClassDef = "write_frequencies = " + str(write_frequencies) + "\n"
pipelineClassDef += "simulation_input_map = " + str(simulation_input_map) + "\n"
pipelineClassDef += "\n"
pipelineClassDef += "# ----------------------- Pipeline definition -----------------------\n\n"
# Create the resulting string that will contains the pipeline definition
pipelineClassDef += "def CreatePipeline(datadescription):\n"
pipelineClassDef += " class Pipeline:\n"
pipelineClassDef += " global cp_views, cp_writers\n"
for original_line in smtrace.trace_globals.trace_output:
for line in original_line.split("\n"):
pipelineClassDef += " " + line + "\n";
smtrace.clear_trace()
pipelineClassDef += " return Pipeline()\n";
return pipelineClassDef
except:
raise RuntimeError('could not import paraview.smstate')
# Start trace
smtrace.start_trace(CaptureAllProperties=True, UseGuiName=True)
# update trace globals.
smtrace.trace_globals.proxy_ctor_hook = staticmethod(cp_hook)
smtrace.trace_globals.trace_output = []
# Get list of proxy lists
proxy_lists = smstate.get_proxy_lists_ordered_by_group(WithRendering=export_rendering)
# Now register the proxies with the smtrace module
for proxy_list in proxy_lists:
smstate.register_proxies_by_dependency(proxy_list)
# Calling append_trace causes the smtrace module to sort out all the
# registered proxies and their properties and write them as executable
# python.
smtrace.append_trace()
# Stop trace and print it to the console
smtrace.stop_trace()
for view_proxy in view_proxies:
# Locate which simulation input this write is connected to, if any. If so,
# we update the write_frequencies datastructure accordingly.
sim_inputs = cp_locate_simulation_inputs_for_view(view_proxy)
for sim_input_name in sim_inputs:
def DumpPipeline(export_rendering, simulation_input_map, screenshot_info):
"""
Method that will dump the current pipeline and return it as a string trace
- export_rendering : boolean telling if we want to export rendering
- simulation_input_map: string->string map with key being the proxyname
while value being the simulation input name.
- screenshot_info : map with information about screenshots
key -> view proxy name
value -> [filename, writefreq, fitToScreen,
magnification, width, height]
"""
# reset the global variables.
reset_cpstate_globals()
cpstate_globals.export_rendering = export_rendering
cpstate_globals.simulation_input_map = simulation_input_map
cpstate_globals.screenshot_info = screenshot_info
# Initialize the write frequency map
for key in cpstate_globals.simulation_input_map.values():
cpstate_globals.write_frequencies[key] = []
# Start trace
capture_modified_properties = not smstate._save_full_state
smtrace.start_trace(CaptureAllProperties=True,
CaptureModifiedProperties=capture_modified_properties,
UseGuiName=True)
# Disconnect the smtrace module's observer. It should not be
# active while tracing the state.
smtrace.reset_trace_observer()
# update trace globals.
smtrace.trace_globals.proxy_ctor_hook = staticmethod(cp_hook)
smtrace.trace_globals.trace_output.clear()
# Get list of proxy lists
proxy_lists = smstate.get_proxy_lists_ordered_by_group(WithRendering=cpstate_globals.export_rendering)
# Now register the proxies with the smtrace module
for proxy_list in proxy_lists:
smstate.register_proxies_by_dependency(proxy_list)
# Calling append_trace causes the smtrace module to sort out all the
# registered proxies and their properties and write them as executable
# python.
smtrace.append_trace()
# Stop trace and print it to the console
smtrace.stop_trace()
# During tracing, cp_hook() will fill up the cpstate_globals.view_proxies
# list with view proxies, if rendering was enabled.
for view_proxy in cpstate_globals.view_proxies:
# Locate which simulation input this write is connected to, if any. If so,
# we update the write_frequencies datastructure accordingly.
sim_inputs = locate_simulation_inputs_for_view(view_proxy)
proxyName = servermanager.ProxyManager().GetProxyName("views", view_proxy)
image_write_frequency = cpstate_globals.screenshot_info[proxyName][1]
for sim_input_name in sim_inputs:
if not image_write_frequency in cpstate_globals.write_frequencies[sim_input_name]:
cpstate_globals.write_frequencies[sim_input_name].append(image_write_frequency)
cpstate_globals.write_frequencies[sim_input_name].sort()
# Create global fields values
pipelineClassDef = "\n"
pipelineClassDef += "# ----------------------- CoProcessor definition -----------------------\n\n"
# Create the resulting string that will contains the pipeline definition
pipelineClassDef += "def CreateCoProcessor():\n"
pipelineClassDef += " def _CreatePipeline(coprocessor, datadescription):\n"
pipelineClassDef += " class Pipeline:\n";
# add the traced code.
for original_line in smtrace.trace_globals.trace_output.raw_data():
for line in original_line.split("\n"):
pipelineClassDef += " " + line + "\n";
smtrace.clear_trace()
pipelineClassDef += " return Pipeline()\n";
pipelineClassDef += "\n"
pipelineClassDef += " class CoProcessor(coprocessing.CoProcessor):\n"
pipelineClassDef += " def CreatePipeline(self, datadescription):\n"
pipelineClassDef += " self.Pipeline = _CreatePipeline(self, datadescription)\n"
pipelineClassDef += "\n"
pipelineClassDef += " coprocessor = CoProcessor()\n";
pipelineClassDef += " freqs = " + str(cpstate_globals.write_frequencies) + "\n"
pipelineClassDef += " coprocessor.SetUpdateFrequencies(freqs)\n"
pipelineClassDef += " return coprocessor\n"
return pipelineClassDef
# views refer to representations and scalar_bars
if export_rendering:
proxy_groups = ["implicit_functions", "piecewise_functions",
"lookup_tables", "scalar_bars", "sources",
"representations", "views"]
else:
proxy_groups = ["implicit_functions", "sources"]
# Collect the proxies using a list comprehension
get_func = servermanager.ProxyManager().GetProxiesInGroup
proxy_lists = [get_func(proxy_group).values() for proxy_group in proxy_groups]
# Now register the proxies with the smtrace module
for proxy_list in proxy_lists:
smstate.register_proxies_by_dependency(proxy_list)
# Calling append_trace causes the smtrace module to sort out all the
# registered proxies and their properties and write them as executable
# python.
smtrace.append_trace()
# Stop trace and print it to the console
smtrace.stop_trace()
output_contents = """
try: paraview.simple
except: from paraview.simple import *
cp_writers = []
def DumpPipeline(export_rendering, simulation_input_map, screenshot_info):
"""
Method that will dump the current pipeline and return it as a string trace
- export_rendering : boolean telling if we want to export rendering
- simulation_input_map: string->string map with key being the proxyname
while value being the simulation input name.
- screenshot_info : map with information about screenshots
key -> view proxy name
value -> [filename, writefreq, fitToScreen,
magnification, width, height]
"""
# reset the global variables.
reset_cpstate_globals()
cpstate_globals.export_rendering = export_rendering
cpstate_globals.simulation_input_map = simulation_input_map
cpstate_globals.screenshot_info = screenshot_info
# Initialize the write frequency map
for key in cpstate_globals.simulation_input_map.values():
cpstate_globals.write_frequencies[key] = []
# Start trace
smtrace.start_trace(CaptureAllProperties=True, UseGuiName=True)
# Disconnect the smtrace module's observer. It should not be
# active while tracing the state.
smtrace.reset_trace_observer()
# update trace globals.
smtrace.trace_globals.proxy_ctor_hook = staticmethod(cp_hook)
smtrace.trace_globals.trace_output = []
# Get list of proxy lists
proxy_lists = smstate.get_proxy_lists_ordered_by_group(
WithRendering=cpstate_globals.export_rendering)
# Now register the proxies with the smtrace module
for proxy_list in proxy_lists:
smstate.register_proxies_by_dependency(proxy_list)
# Calling append_trace causes the smtrace module to sort out all the
# registered proxies and their properties and write them as executable
# python.
smtrace.append_trace()
# Stop trace and print it to the console
smtrace.stop_trace()
# During tracing, cp_hook() will fill up the cpstate_globals.view_proxies
# list with view proxies, if rendering was enabled.
for view_proxy in cpstate_globals.view_proxies:
# Locate which simulation input this write is connected to, if any. If so,
# we update the write_frequencies datastructure accordingly.
sim_inputs = locate_simulation_inputs_for_view(view_proxy)
proxyName = servermanager.ProxyManager().GetProxyName(
"views", view_proxy)
image_write_frequency = cpstate_globals.screenshot_info[proxyName][1]
for sim_input_name in sim_inputs:
if not image_write_frequency in cpstate_globals.write_frequencies[
sim_input_name]:
cpstate_globals.write_frequencies[sim_input_name].append(
image_write_frequency)
cpstate_globals.write_frequencies[sim_input_name].sort()
# Create global fields values
pipelineClassDef = "\n"
pipelineClassDef += "# ----------------------- CoProcessor definition -----------------------\n\n"
# Create the resulting string that will contains the pipeline definition
pipelineClassDef += "def CreateCoProcessor():\n"
pipelineClassDef += " def _CreatePipeline(coprocessor, datadescription):\n"
pipelineClassDef += " class Pipeline:\n"
# add the traced code.
for original_line in smtrace.trace_globals.trace_output:
for line in original_line.split("\n"):
pipelineClassDef += " " + line + "\n"
smtrace.clear_trace()
pipelineClassDef += " return Pipeline()\n"
pipelineClassDef += "\n"
pipelineClassDef += " class CoProcessor(coprocessing.CoProcessor):\n"
pipelineClassDef += " def CreatePipeline(self, datadescription):\n"
pipelineClassDef += " self.Pipeline = _CreatePipeline(self, datadescription)\n"
pipelineClassDef += "\n"
pipelineClassDef += " coprocessor = CoProcessor()\n"
pipelineClassDef += " freqs = " + str(
cpstate_globals.write_frequencies) + "\n"
pipelineClassDef += " coprocessor.SetUpdateFrequencies(freqs)\n"
pipelineClassDef += " return coprocessor\n"
return pipelineClassDef
