def test_RenameSource(self): source = simple.Sphere(guiName='oldName') simple.SetActiveSource(source) simple.RenameSource('newName') # changing the source name should unregister the old name self.assertEqual(None, self.pxm.GetProxy('sources', 'oldName')) self.assertEqual(source, self.pxm.GetProxy('sources', 'newName')) # renaming as the current name should not unregister the source simple.RenameSource('newName') self.assertEqual(source, self.pxm.GetProxy('sources', 'newName'))
def __init__(self, center=(0, 0, 0), radius=1): self.center = center self.radius = radius self.method = 'vtk.data.add_sphere' self.kwargs = { 'Center': self.center, 'Radius': self.radius, } simple.Show(simple.Sphere(**self.kwargs)) self.id = f"sphere_{self._id}" Sphere._id += 1 super().__init__(type='sphere')
def showSphere(self, params): if self.sphere is not None: simple.Delete(self.sphere) maxDim = max(self.bounds[1] - self.bounds[0], self.bounds[3] - self.bounds[2], self.bounds[5] - self.bounds[4]) self.sphere = simple.Sphere() self.sphere.Radius = maxDim / 100.0 self.sphere.Center = params['point'] rep = simple.Show() rep.Representation = 'Surface' rep.DiffuseColor = params['color'] simple.SetActiveSource(self.srcObj)
def create_sphere(self, r, center, color_RGB=[1., 1., 1.], phiRes=30, thetaRes=30): "Cria uma esfera." sphere = pv.Sphere(Radius=r, PhiResolution=phiRes, ThetaResolution=thetaRes) sphere.Center = center disp = pv.Show() disp.DiffuseColor = color_RGB self.id_ += 1 self.objects_[self.id_] = sphere return self.id_
def runTest(): options = servermanager.vtkProcessModule.GetProcessModule().GetOptions() url = options.GetServerURL() smp.Connect(getHost(url), getPort(url)) sphere = smp.Sphere() f = smp.ProgrammableFilter(sphere) # test that vtk is imported automatically and contains the name vtkPolyData script = 'assert vtk.vtkPolyData' assert testScript(f, script) # test that variables can be passed using the Parameters property script = 'assert foo == "bar"' f.SetPropertyWithName('Parameters', ['foo', '"bar"']) assert testScript(f, script) smp.Disconnect()
def initialize(self): # Bring used components self.registerVtkWebProtocol(pv_protocols.ParaViewWebMouseHandler()) self.registerVtkWebProtocol( pv_protocols.ParaViewWebViewPort( MultiViewServer.viewportScale, MultiViewServer.viewportMaxWidth, MultiViewServer.viewportMaxHeight)) self.registerVtkWebProtocol( pv_protocols.ParaViewWebPublishImageDelivery(decode=False)) # Update authentication key to use self.updateSecret(MultiViewServer.authKey) # tell the C++ web app to use no encoding. ParaViewWebPublishImageDelivery must be set to decode=False to match. self.getApplication().SetImageEncoding(0) # Create of 2 views with different content self.registerVtkWebProtocol( UserProtocol( [self.addView(simple.Cone()), self.addView(simple.Sphere())]))
def runTest(): options = servermanager.vtkRemotingCoreConfiguration.GetInstance() url = options.GetServerURL() smp.Connect(getHost(url), getPort(url)) r = smp.CreateRenderView() r.RemoteRenderThreshold = 20 s = smp.Sphere() s.PhiResolution = 80 s.ThetaResolution = 80 d = smp.Show() d.Representation = "Wireframe" smp.Render() r.RemoteRenderThreshold = 0 smp.Render() s.PhiResolution = 8 s.ThetaResolution = 8 smp.Render() smtesting.ProcessCommandLineArguments() if not smtesting.DoRegressionTesting(r.SMProxy): raise smtesting.TestError ("Test failed!!!") print ("Test Passed")
from paraview import simple import tonic from tonic.paraview.dataset_builder import * dataset_destination_path = '/Users/seb/Desktop/mpi-sphere' sphere = simple.Sphere() rep = simple.Show() simple.ColorBy(rep, ('POINTS', 'vtkProcessId')) view = simple.Render() view.ResetCamera() rep.RescaleTransferFunctionToDataRange(True) phi = range(0, 360, 10) theta = range(-60, 61, 10) dh = ImageDataSetBuilder(dataset_destination_path, 'image/jpg', { 'type': 'spherical', 'phi': phi, 'theta': theta }) dh.start(view) dh.writeImages() dh.stop()
rvs4.AxesGrid.Visibility = 1 #rvs4.AxesGrid.AxesToLabel = 5 rvs4.AxesGrid.GridColor = [0.8, 0.8, 0.8] rvs4.AxesGrid.ShowGrid = 1 rvs4.AxesGrid.XLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.XTitleColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.ZLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.ZTitleColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.YLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.YTitleColor = [0.2, 0.2, 0.2] rvs4.OrientationAxesVisibility = 0 rvs4.ViewSize = [1920*5, 1280*5] rvs4.Background = [1.0, 1.0, 1.0] # create a sphere to represent Earth earth = pv.Sphere() earth.Radius = RE_el earth.ThetaResolution = 128 earth.PhiResolution = 128 earth_disp4 = pv.Show(earth, rvs4) earth_disp4.DiffuseColor = [0.51, 0.51, 0.51] # display check lines for key in t96_K0_ds2.field_lines: if t96_K0_ds2.field_lines[key] is None: continue forward = pv.Show(t96_K0_ds2.field_lines[key].fieldLineObj_f, rvs4) forward.DiffuseColor = [1, 0.0, 0.0] forward.LineWidth = 2.0
# This is how you can run this script: # mpiexec -n NUM_PROCS pvbatch Test_Parallel.py import paraview.simple as pv import paraview.servermanager as pvserver num_partitions = pvserver.ActiveConnection.GetNumberOfDataPartitions() print("Rendering {} data partitions".format(num_partitions)) sphere = pv.Sphere() pids = pv.ProcessIdScalars() rep = pv.Show() lut = pv.GetLookupTableForArray( "ProcessId", 1, RGBPoints=[0.0, 0.23, 0.3, 0.754, num_partitions, 0.706, 0.016, 0.15], ColorSpace='Diverging') rep.LookupTable = lut rep.ColorArrayName = 'ProcessId' pv.Render() pv.WriteImage('Parallel.png') print( "Rendering done. You should find an image file named `Parallel.png` that shows a sphere. The sphere should be divided into wedges that are color-coded by processor." )
rv.AxesGrid.XTitleColor = [0.2, 0.2, 0.2] rv.AxesGrid.ZLabelColor = [0.2, 0.2, 0.2] rv.AxesGrid.ZTitleColor = [0.2, 0.2, 0.2] rv.OrientationAxesVisibility = 0 # rv.ViewSize = [1280*3, 600*3] rv.ViewSize = [1280, 600] td = pv.Show(t96_128_dp, view=rv) td.DiffuseColor = [0.0, 0.0, 0.0] td.EdgeColor = [0.0, 0.0, 0.0] td.AmbientColor = [0.0, 0.0, 0.0] # create a sphere to represent Earth earth = pv.Sphere() earth.Radius = 1.0 earth.ThetaResolution = 128 earth.PhiResolution = 128 earth_disp = pv.Show(earth, rv) earth_disp.DiffuseColor = [0.0, 0.3333333333333333, 0.0] # create test field line fline = fl.fieldLine(t96_128_dp, start=[-10.0,0.0,0.0]) forward = pv.Show(fline.fieldLineObj_f,rv) backward = pv.Show(fline.fieldLineObj_b,rv) forward.DiffuseColor = [0.5, 0.0, 0.0] backward.DiffuseColor = [0., 0.0, 0.5] # Mark New Center (Should be B_min) new_center = fline.fieldLinePoints_f[0]
def createPipeline(self): self.renderView = pvsimple.CreateRenderView() pvsimple.Sphere() pvsimple.Show()
rvs4.CameraParallelScale = 50 rvs4.AxesGrid.Visibility = 1 rvs4.AxesGrid.GridColor = [0.8, 0.8, 0.8] rvs4.AxesGrid.ShowGrid = 1 rvs4.AxesGrid.XLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.XTitleColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.ZLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.ZTitleColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.YLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.YTitleColor = [0.2, 0.2, 0.2] rvs4.OrientationAxesVisibility = 0 rvs4.ViewSize = [1920, 1280] rvs4.Background = [1.0, 1.0, 1.0] # create a sphere to represent Earth earth = pv.Sphere() earth.Radius = 1.0 earth.ThetaResolution = 128 earth.PhiResolution = 128 earth_disp4 = pv.Show(earth, rvs4) earth_disp4.DiffuseColor = [0.51, 0.51, 0.51] L_Text = pv.Text() L_Text.Text = "$L^*=$ " + str(L_100) Ldisp = pv.Show(L_Text, rvs4) Ldisp.WindowLocation = 'LowerCenter' for key in t96_K0_ds.field_lines: if t96_K0_ds.field_lines[key] is None: continue
rvs.CameraViewUp = [0, 0, 1] rvs.CameraParallelScale = 11 rvs.AxesGrid.Visibility = 1 rvs.AxesGrid.AxesToLabel = 5 rvs.AxesGrid.GridColor = [0.8, 0.8, 0.8] rvs.AxesGrid.ShowGrid = 1 rvs.AxesGrid.XLabelColor = [0.2, 0.2, 0.2] rvs.AxesGrid.XTitleColor = [0.2, 0.2, 0.2] rvs.AxesGrid.ZLabelColor = [0.2, 0.2, 0.2] rvs.AxesGrid.ZTitleColor = [0.2, 0.2, 0.2] rvs.OrientationAxesVisibility = 0 rvs.ViewSize = [1920, 1280] rvs.Background = [1.0, 1.0, 1.0] # create a sphere to represent Earth earth = pv.Sphere() earth.Radius = 1.0 earth.ThetaResolution = 128 earth.PhiResolution = 128 earth_disp = pv.Show(earth, rvs) earth_disp.DiffuseColor = [0.0, 0.3333333333333333, 0.0] for start_p in start_positions: print "Starting Postion: {}".format(start_p) # get field lines and integrals flines[start_p] = fl.fieldLine(t96_128, start=start_p) i_integral[start_p] = flines[start_p].get_i_integrals() # make field line plots dsp[start_p] = pv.Show(t96_128, view=rvs) dsp[start_p].CubeAxesVisibility = 0
# 1. Create a virtual environment with ParaView's Python: # ``` # path/to/paraview/install/python3 -m venv ./env # ``` # 2. Install `h5py` in the environment **using ParaView's HDF5**: # ``` # . ./env/bin/activate # HDF5_DIR=path/to/paraview/install/ pip install --no-binary=h5py h5py # ``` # Note that the `HDF5_DIR` should have `include` and `lib` directories with ParaView's HDF5. # 3. Set the path below to `path/to/env/lib/python3.7/site-packages` extra_packages_path = '/data/home/nfischer/test-paraview/env/lib/python3.7/site-packages' # Then, this is how you can run this script: # pvpython Test_H5.py import paraview.simple as pv import sys sys.path.append(extra_packages_path) import h5py pv.Sphere() pv.Show() pv.SaveScreenshot("H5.png") print( "Rendering done. You should find an image file named `H5.png` that shows a sphere." )
# get the proxy manager. smpxm = c.Session.GetSessionProxyManager() #============================================================================== # create a proxy and confirm we can obtain a class for it. sphere = smpxm.NewProxy("sources", "SphereSource") sphere.UnRegister(None) cls = c.ProxiesNS.getClass(sphere) assert cls is not None assert type(getattr(cls, "Radius")) == property # let's create another proxy of the same type using a different route and # confirm the same class type is returned. sphere2 = simple.Sphere() sphere3 = simple.Sphere() assert sphere2.__class__ == cls assert sphere3.__class__ == cls #============================================================================== # now, confirm that is two proxies with same name but different groups are created # we get different classes for each. This avoids issues like #20672. p1 = smpxm.NewProxy("extract_writers", "PNG") p1.UnRegister(None) cls1 = c.ProxiesNS.getClass(p1) p2 = smpxm.NewProxy("animation_writers", "PNG") p2.UnRegister(None) cls2 = c.ProxiesNS.getClass(p2)
arrayName = str(random.random()) programmableFilter.Script = script + textwrap.dedent(''' passedArray = vtk.vtkIntArray() passedArray.SetName('%s') self.GetOutput().GetFieldData().AddArray(passedArray)''' % arrayName) programmableFilter.UpdatePipeline() return programmableFilter.GetClientSideObject().GetOutput().GetFieldData( ).GetArray(arrayName) is not None SMPythonTesting.ProcessCommandLineArguments() tempDir = SMPythonTesting.TempDir stateDir = SMPythonTesting.SMStatesDir sphere = smp.Sphere() f = smp.ProgrammableFilter(sphere) script = ''' assert 1+1 == 2 ''' assert testScript(f, script) script = ''' assert foo == 'bar' ''' f.SetPropertyWithName('Parameters', ['foo', '"bar"'])
rvs4.AxesGrid.Visibility = 1 #rvs4.AxesGrid.AxesToLabel = 5 rvs4.AxesGrid.GridColor = [0.8, 0.8, 0.8] rvs4.AxesGrid.ShowGrid = 1 rvs4.AxesGrid.XLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.XTitleColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.ZLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.ZTitleColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.YLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.YTitleColor = [0.2, 0.2, 0.2] rvs4.OrientationAxesVisibility = 0 rvs4.ViewSize = [1920, 1280] rvs4.Background = [1.0, 1.0, 1.0] # create a sphere to represent Earth earth = pv.Sphere() earth.Radius = 1.0 earth.ThetaResolution = 128 earth.PhiResolution = 128 earth_disp = pv.Show(earth, rvs) earth_disp2 = pv.Show(earth, rvs2) earth_disp3 = pv.Show(earth, rvs3) earth_disp4 = pv.Show(earth, rvs4) earth_disp4.DiffuseColor = [0.81, 0.81, 0.81] earth_disp3.DiffuseColor = [0.333, 0.333, 0.333] earth_disp2.DiffuseColor = [0.333, 0.333, 0.333] earth_disp.DiffuseColor = [0.333, 0.333, 0.333] Bmag_calc = pv.Calculator(Input=t96) Bmag_calc.ResultArrayName = 'Bmag' Bmag_calc.Function = 'mag(B)'
pl.title("Field Line Geometry Integral $I$ with respect to $B_{mirror}$ \n For Field line that passes through (-40, 0, 0)") fig1.savefig("IwrtBm.png") # Create a paraview render view so we can see visual progress. # pv._DisableFirstRenderCameraReset() rv128 = pv.CreateRenderView() rv128.InteractionMode = '2D' rv128.CameraPosition = [0, -30, 0] rv128.CameraViewUp = [0, 0, 1] rv128.CameraParallelScale = 10 rv128.ViewSize = [1280, 1024] dipoleDisplay = pv.GetDisplayProperties(dipole128, view=rv128) # pv.Hide(dipole128, view=rv128) # create a new 'Sphere' sphere1 = pv.Sphere() sphere1.Radius = 1.0 sphere1.ThetaResolution = 64 sphere1.PhiResolution = 64 pv.Show(sphere1, rv128) # get field line forward = pv.Show(fline.fieldLineObj_f,rv128) backward = pv.Show(fline.fieldLineObj_b,rv128) forward.DiffuseColor = [0.6666666666666666, 0.0, 0.0] backward.DiffuseColor = [0., 0., 0.66] # Find B_min Bmin_loc, Bmin_val = fline.get_field_min()
def makeSphere(): smp.Sphere() smp.Show() smp.ResetCamera() smp.Render()
def render_frames( scene, frames_dir=None, frame_window=None, render_missing_frames=False, save_state_to_file=None, no_render=False, show_preview=False, show_progress=False, job_id=None, ): # Validate scene if scene["View"]["ViewSize"][0] % 16 != 0: logger.warning( "The view width should be a multiple of 16 to be compatible with" " QuickTime.") if scene["View"]["ViewSize"][1] % 2 != 0: logger.warning( "The view height should be even to be compatible with QuickTime.") render_start_time = time.time() # Setup layout layout = pv.CreateLayout("Layout") # Setup view if "Background" in scene["View"]: bg_config = scene["View"]["Background"] del scene["View"]["Background"] if isinstance(bg_config, list): if isinstance(bg_config[0], list): assert len(bg_config) == 2, ( "When 'Background' is a list of colors, it must have 2" " entries.") bg_config = dict( BackgroundColorMode="Gradient", Background=parse_as.color(bg_config[0]), Background2=parse_as.color(bg_config[1]), ) else: bg_config = dict( BackgroundColorMode="Single Color", Background=parse_as.color(bg_config), ) bg_config["UseColorPaletteForBackground"] = 0 scene["View"].update(bg_config) bg_config = None else: bg_config = None view = pv.CreateRenderView(**scene["View"]) pv.AssignViewToLayout(view=view, layout=layout, hint=0) # Set spherical background texture if bg_config is not None: bg_config["BackgroundColorMode"] = "Texture" skybox_datasource = bg_config["Datasource"] del bg_config["Datasource"] background_texture = pvserver.rendering.ImageTexture( FileName=parse_as.path(scene["Datasources"][skybox_datasource])) background_sphere = pv.Sphere(Radius=bg_config["Radius"], ThetaResolution=100, PhiResolution=100) background_texture_map = pv.TextureMaptoSphere(Input=background_sphere) pv.Show( background_texture_map, view, Texture=background_texture, BackfaceRepresentation="Cull Frontface", Ambient=1.0, ) # Load the waveform data file waveform_h5file, waveform_subfile = parse_as.file_and_subfile( scene["Datasources"]["Waveform"]) waveform_data = WaveformDataReader(FileName=waveform_h5file, Subfile=waveform_subfile) pv.UpdatePipeline() # Generate volume data from the waveform. Also sets the available time range. # TODO: Pull KeepEveryNthTimestep out of datasource waveform_to_volume_configs = scene["WaveformToVolume"] if isinstance(waveform_to_volume_configs, dict): waveform_to_volume_configs = [{ "Object": waveform_to_volume_configs, }] if "VolumeRepresentation" in scene: waveform_to_volume_configs[0]["VolumeRepresentation"] = scene[ "VolumeRepresentation"] waveform_to_volume_objects = [] for waveform_to_volume_config in waveform_to_volume_configs: volume_data = WaveformToVolume( WaveformData=waveform_data, SwshCacheDirectory=parse_as.path( scene["Datasources"]["SwshCache"]), **waveform_to_volume_config["Object"], ) if "Modes" in waveform_to_volume_config["Object"]: volume_data.Modes = waveform_to_volume_config["Object"]["Modes"] if "Polarizations" in waveform_to_volume_config["Object"]: volume_data.Polarizations = waveform_to_volume_config["Object"][ "Polarizations"] waveform_to_volume_objects.append(volume_data) # Compute timing and frames information time_range_in_M = ( volume_data.TimestepValues[0], volume_data.TimestepValues[-1], ) logger.debug(f"Full available data time range: {time_range_in_M} (in M)") if "FreezeTime" in scene["Animation"]: frozen_time = scene["Animation"]["FreezeTime"] logger.info(f"Freezing time at {frozen_time}.") view.ViewTime = frozen_time animation = None else: if "Crop" in scene["Animation"]: time_range_in_M = scene["Animation"]["Crop"] logger.debug(f"Cropping time range to {time_range_in_M} (in M).") animation_speed = scene["Animation"]["Speed"] frame_rate = scene["Animation"]["FrameRate"] num_frames = animate.num_frames( max_animation_length=time_range_in_M[1] - time_range_in_M[0], animation_speed=animation_speed, frame_rate=frame_rate, ) animation_length_in_seconds = num_frames / frame_rate animation_length_in_M = animation_length_in_seconds * animation_speed time_per_frame_in_M = animation_length_in_M / num_frames logger.info(f"Rendering {animation_length_in_seconds:.2f}s movie with" f" {num_frames} frames ({frame_rate} FPS or" f" {animation_speed:.2e} M/s or" f" {time_per_frame_in_M:.2e} M/frame)...") if frame_window is not None: animation_window_num_frames = frame_window[1] - frame_window[0] animation_window_time_range = ( time_range_in_M[0] + frame_window[0] * time_per_frame_in_M, time_range_in_M[0] + (frame_window[1] - 1) * time_per_frame_in_M, ) logger.info( f"Restricting rendering to {animation_window_num_frames} frames" f" (numbers {frame_window[0]} to {frame_window[1] - 1}).") else: animation_window_num_frames = num_frames animation_window_time_range = time_range_in_M frame_window = (0, num_frames) # Setup animation so that sources can retrieve the `UPDATE_TIME_STEP` animation = pv.GetAnimationScene() # animation.UpdateAnimationUsingDataTimeSteps() # Since the data can be evaluated at arbitrary times we define the time steps # here by setting the number of frames within the full range animation.PlayMode = "Sequence" animation.StartTime = animation_window_time_range[0] animation.EndTime = animation_window_time_range[1] animation.NumberOfFrames = animation_window_num_frames logger.debug( f"Animating from scene time {animation.StartTime} to" f" {animation.EndTime} in {animation.NumberOfFrames} frames.") def scene_time_from_real(real_time): return (real_time / animation_length_in_seconds * animation_length_in_M) # For some reason the keyframe time for animations is expected to be within # (0, 1) so we need to transform back and forth from this "normalized" time def scene_time_from_normalized(normalized_time): return animation.StartTime + normalized_time * ( animation.EndTime - animation.StartTime) def normalized_time_from_scene(scene_time): return (scene_time - animation.StartTime) / (animation.EndTime - animation.StartTime) # Setup progress measuring already here so volume data computing for # initial frame is measured if show_progress and not no_render: logging.getLogger().handlers = [TqdmLoggingHandler()] animation_window_frame_range = tqdm.trange( animation_window_num_frames, desc="Rendering", unit="frame", miniters=1, position=job_id, ) else: animation_window_frame_range = range(animation_window_num_frames) # Set the initial time step animation.GoToFirst() # Display the volume data. This will trigger computing the volume data at the # current time step. for volume_data, waveform_to_volume_config in zip( waveform_to_volume_objects, waveform_to_volume_configs): vol_repr = (waveform_to_volume_config["VolumeRepresentation"] if "VolumeRepresentation" in waveform_to_volume_config else {}) volume_color_by = config_color.extract_color_by(vol_repr) if (vol_repr["VolumeRenderingMode"] == "GPU Based" and len(volume_color_by) > 2): logger.warning( "The 'GPU Based' volume renderer doesn't support multiple" " components.") volume = pv.Show(volume_data, view, **vol_repr) pv.ColorBy(volume, value=volume_color_by) if "Slices" in scene: for slice_config in scene["Slices"]: slice_obj_config = slice_config.get("Object", {}) slice = pv.Slice(Input=volume_data) slice.SliceType = "Plane" slice.SliceOffsetValues = [0.0] slice.SliceType.Origin = slice_obj_config.get( "Origin", [0.0, 0.0, -0.3]) slice.SliceType.Normal = slice_obj_config.get( "Normal", [0.0, 0.0, 1.0]) slice_rep = pv.Show(slice, view, **slice_config.get("Representation", {})) pv.ColorBy(slice_rep, value=volume_color_by) # Display the time if "TimeAnnotation" in scene: time_annotation = pv.AnnotateTimeFilter(volume_data, **scene["TimeAnnotation"]) pv.Show(time_annotation, view, **scene["TimeAnnotationRepresentation"]) # Add spheres if "Spheres" in scene: for sphere_config in scene["Spheres"]: sphere = pv.Sphere(**sphere_config["Object"]) pv.Show(sphere, view, **sphere_config["Representation"]) # Add trajectories and objects that follow them if "Trajectories" in scene: for trajectory_config in scene["Trajectories"]: trajectory_name = trajectory_config["Name"] radial_scale = (trajectory_config["RadialScale"] if "RadialScale" in trajectory_config else 1.0) # Load the trajectory data traj_data_reader = TrajectoryDataReader( RadialScale=radial_scale, **scene["Datasources"]["Trajectories"][trajectory_name], ) # Make sure the data is loaded so we can retrieve timesteps. # TODO: This should be fixed in `TrajectoryDataReader` by # communicating time range info down the pipeline, but we had issues # with that (see also `WaveformDataReader`). traj_data_reader.UpdatePipeline() if "Objects" in trajectory_config: with animate.restore_animation_state(animation): follow_traj = FollowTrajectory( TrajectoryData=traj_data_reader) for traj_obj_config in trajectory_config["Objects"]: for traj_obj_key in traj_obj_config: if traj_obj_key in [ "Representation", "Visibility", "TimeShift", "Glyph", ]: continue traj_obj_type = getattr(pv, traj_obj_key) traj_obj_glyph = traj_obj_type( **traj_obj_config[traj_obj_key]) follow_traj.UpdatePipeline() traj_obj = pv.Glyph(Input=follow_traj, GlyphType=traj_obj_glyph) # Can't set this in the constructor for some reason traj_obj.ScaleFactor = 1.0 for glyph_property in (traj_obj_config["Glyph"] if "Glyph" in traj_obj_config else []): setattr( traj_obj, glyph_property, traj_obj_config["Glyph"][glyph_property], ) traj_obj.UpdatePipeline() if "TimeShift" in traj_obj_config: traj_obj = animate.apply_time_shift( traj_obj, traj_obj_config["TimeShift"]) pv.Show(traj_obj, view, **traj_obj_config["Representation"]) if "Visibility" in traj_obj_config: animate.apply_visibility( traj_obj, traj_obj_config["Visibility"], normalized_time_from_scene, scene_time_from_real, ) if "Tail" in trajectory_config: with animate.restore_animation_state(animation): traj_tail = TrajectoryTail(TrajectoryData=traj_data_reader) if "TimeShift" in trajectory_config: traj_tail = animate.apply_time_shift( traj_tail, trajectory_config["TimeShift"]) tail_config = trajectory_config["Tail"] traj_color_by = config_color.extract_color_by(tail_config) if "Visibility" in tail_config: tail_visibility_config = tail_config["Visibility"] del tail_config["Visibility"] else: tail_visibility_config = None tail_rep = pv.Show(traj_tail, view, **tail_config) pv.ColorBy(tail_rep, value=traj_color_by) if tail_visibility_config is not None: animate.apply_visibility( traj_tail, tail_visibility_config, normalized_time_from_scene=normalized_time_from_scene, scene_time_from_real=scene_time_from_real, ) if "Move" in trajectory_config: move_config = trajectory_config["Move"] logger.debug( f"Animating '{move_config['guiName']}' along trajectory.") with h5py.File(trajectory_file, "r") as traj_data_file: trajectory_data = np.array( traj_data_file[trajectory_subfile]) if radial_scale != 1.0: trajectory_data[:, 1:] *= radial_scale logger.debug(f"Trajectory data shape: {trajectory_data.shape}") animate.follow_path( gui_name=move_config["guiName"], trajectory_data=trajectory_data, num_keyframes=move_config["NumKeyframes"], scene_time_range=time_range_in_M, normalized_time_from_scene=normalized_time_from_scene, ) # Add non-spherical horizon shapes (instead of spherical objects following # trajectories) if "Horizons" in scene: for horizon_config in scene["Horizons"]: with animate.restore_animation_state(animation): horizon = pv.PVDReader(FileName=scene["Datasources"] ["Horizons"][horizon_config["Name"]]) if horizon_config.get("InterpolateTime", False): horizon = pv.TemporalInterpolator( Input=horizon, DiscreteTimeStepInterval=0) if "TimeShift" in horizon_config: horizon = animate.apply_time_shift(horizon, horizon_config["TimeShift"], animation) # Try to make horizon surfaces smooth. At low angular resoluton # they still show artifacts, so perhaps more can be done. horizon = pv.ExtractSurface(Input=horizon) horizon = pv.GenerateSurfaceNormals(Input=horizon) horizon_rep_config = horizon_config.get("Representation", {}) if "Representation" not in horizon_rep_config: horizon_rep_config["Representation"] = "Surface" if "AmbientColor" not in horizon_rep_config: horizon_rep_config["AmbientColor"] = [0.0, 0.0, 0.0] if "DiffuseColor" not in horizon_rep_config: horizon_rep_config["DiffuseColor"] = [0.0, 0.0, 0.0] if "Specular" not in horizon_rep_config: horizon_rep_config["Specular"] = 0.2 if "SpecularPower" not in horizon_rep_config: horizon_rep_config["SpecularPower"] = 10 if "SpecularColor" not in horizon_rep_config: horizon_rep_config["SpecularColor"] = [1.0, 1.0, 1.0] if "ColorBy" in horizon_rep_config: horizon_color_by = config_color.extract_color_by( horizon_rep_config) else: horizon_color_by = None horizon_rep = pv.Show(horizon, view, **horizon_rep_config) if horizon_color_by is not None: pv.ColorBy(horizon_rep, value=horizon_color_by) # Animate visibility if "Visibility" in horizon_config: animate.apply_visibility( horizon, horizon_config["Visibility"], normalized_time_from_scene=normalized_time_from_scene, scene_time_from_real=scene_time_from_real, ) if "Contours" in horizon_config: for contour_config in horizon_config["Contours"]: contour = pv.Contour(Input=horizon, **contour_config["Object"]) contour_rep = pv.Show(contour, view, **contour_config["Representation"]) pv.ColorBy(contour_rep, None) if "Visibility" in horizon_config: animate.apply_visibility( contour, horizon_config["Visibility"], normalized_time_from_scene= normalized_time_from_scene, scene_time_from_real=scene_time_from_real, ) # Configure transfer functions if "TransferFunctions" in scene: for tf_config in scene["TransferFunctions"]: colored_field = tf_config["Field"] transfer_fctn = pv.GetColorTransferFunction(colored_field) opacity_fctn = pv.GetOpacityTransferFunction(colored_field) tf.configure_transfer_function(transfer_fctn, opacity_fctn, tf_config["TransferFunction"]) # Save state file before configuring camera keyframes. # TODO: Make camera keyframes work with statefile if save_state_to_file is not None: pv.SaveState(save_state_to_file + ".pvsm") # Camera shots # TODO: Make this work with freezing time while the camera is swinging if animation is None: for i, shot in enumerate(scene["CameraShots"]): if (i == len(scene["CameraShots"]) - 1 or (shot["Time"] if "Time" in shot else 0.0) >= view.ViewTime): camera_motion.apply(shot) break else: camera_motion.apply_swings( scene["CameraShots"], scene_time_range=time_range_in_M, scene_time_from_real=scene_time_from_real, normalized_time_from_scene=normalized_time_from_scene, ) # Report time if animation is not None: report_time_cue = pv.PythonAnimationCue() report_time_cue.Script = """ def start_cue(self): pass def tick(self): import paraview.simple as pv import logging logger = logging.getLogger('Animation') scene_time = pv.GetActiveView().ViewTime logger.info(f"Scene time: {scene_time}") def end_cue(self): pass """ animation.Cues.append(report_time_cue) if show_preview and animation is not None: animation.PlayMode = "Real Time" animation.Duration = 10 animation.Play() animation.PlayMode = "Sequence" if no_render: logger.info("No rendering requested. Total time:" f" {time.time() - render_start_time:.2f}s") return if frames_dir is None: raise RuntimeError("Trying to render but `frames_dir` is not set.") if os.path.exists(frames_dir): logger.warning( f"Output directory '{frames_dir}' exists, files may be overwritten." ) else: os.makedirs(frames_dir) if animation is None: pv.Render() pv.SaveScreenshot(os.path.join(frames_dir, "frame.png")) else: # Iterate over frames manually to support filling in missing frames. # If `pv.SaveAnimation` would support that, here's how it could be # invoked: # pv.SaveAnimation( # os.path.join(frames_dir, 'frame.png'), # view, # animation, # FrameWindow=frame_window, # SuffixFormat='.%06d') # Note that `FrameWindow` appears to be buggy, so we set up the # `animation` according to the `frame_window` above so the frame files # are numberd correctly. for animation_window_frame_i in animation_window_frame_range: frame_i = frame_window[0] + animation_window_frame_i frame_file = os.path.join(frames_dir, f"frame.{frame_i:06d}.png") if render_missing_frames and os.path.exists(frame_file): continue logger.debug(f"Rendering frame {frame_i}...") animation.AnimationTime = ( animation.StartTime + time_per_frame_in_M * animation_window_frame_i) pv.Render() pv.SaveScreenshot(frame_file) logger.info(f"Rendered frame {frame_i}.") logger.info( f"Rendering done. Total time: {time.time() - render_start_time:.2f}s")
def test_slice(self): pv.Connect() # using a dedicated server state for each test print "\nTEST_SLICE" # set up some processing task view_proxy = pv.CreateRenderView() view_proxy.OrientationAxesVisibility = 0 s = pv.Sphere() sliceFilt = pv.Slice( SliceType="Plane", Input=s, SliceOffsetValues=[0.0]) sliceFilt.SliceType.Normal = [0, 1, 0] sliceRep = pv.Show(sliceFilt) # make or open a cinema data store to put results in fname = "/tmp/test_pv_slice/info.json" cs = file_store.FileStore(fname) cs.add_metadata({'type': 'parametric-image-stack'}) cs.add_metadata({'store_type': 'FS'}) cs.add_metadata({'version': '0.0'}) cs.filename_pattern = "{phi}_{theta}_{offset}_{color}_slice.png" cs.add_parameter( "phi", store.make_parameter('phi', [90, 120, 140])) cs.add_parameter( "theta", store.make_parameter('theta', [-90, -30, 30, 90])) cs.add_parameter( "offset", store.make_parameter('offset', [-.4, -.2, 0, .2, .4])) cs.add_parameter( "color", store.make_parameter( 'color', ['yellow', 'cyan', "purple"], typechoice='list')) colorChoice = pv_explorers.ColorList() colorChoice.AddSolidColor('yellow', [1, 1, 0]) colorChoice.AddSolidColor('cyan', [0, 1, 1]) colorChoice.AddSolidColor('purple', [1, 0, 1]) # associate control points with parameters of the data store cam = pv_explorers.Camera([0, 0, 0], [0, 1, 0], 10.0, view_proxy) filt = pv_explorers.Slice("offset", sliceFilt) col = pv_explorers.Color("color", colorChoice, sliceRep) params = ["phi", "theta", "offset", "color"] e = pv_explorers.ImageExplorer( cs, params, [cam, filt, col], view_proxy) # run through all parameter combinations and put data into the store e.explore() # Reproduce an entry and compare vs. loaded # First set the parameters to reproduce cam.execute(store.Document({'theta': -30, 'phi': 120})) filt.execute(store.Document({'offset': -.4})) col.execute(store.Document({'color': 'cyan'})) imageslice = ch.pvRenderToArray(view_proxy) # Now load the corresponding entry cs2 = file_store.FileStore(fname) cs2.load() docs = [] for doc in cs2.find( {'theta': -30, 'phi': 120, 'offset': -.4, 'color': 'cyan'}): docs.append(doc.data) # print "gen entry: \n", # imageslice, "\n", # imageslice.shape,"\n", # "loaded: \n", # docs[0], "\n", # docs[0].shape # compare the two l2error = ch.compare_l2(imageslice, docs[0]) ncc = ch.compare_ncc(imageslice, docs[0]) self.assertTrue((l2error < 1.0) and (ncc > 0.99)) pv.Disconnect() # using a dedicated server state for each test
rvs4.AxesGrid.Visibility = 1 #rvs4.AxesGrid.AxesToLabel = 5 rvs4.AxesGrid.GridColor = [0.8, 0.8, 0.8] rvs4.AxesGrid.ShowGrid = 1 rvs4.AxesGrid.XLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.XTitleColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.ZLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.ZTitleColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.YLabelColor = [0.2, 0.2, 0.2] rvs4.AxesGrid.YTitleColor = [0.2, 0.2, 0.2] rvs4.OrientationAxesVisibility = 0 rvs4.ViewSize = [1920, 1280] rvs4.Background = [1.0, 1.0, 1.0] # create a sphere to represent Earth earth = pv.Sphere() earth.Radius = 1.0 earth.ThetaResolution = 128 earth.PhiResolution = 128 earth_disp4 = pv.Show(earth, rvs4) earth_disp4.DiffuseColor = [0.51, 0.51, 0.51] # pv.Hide(earth,rvs4) for key in t96_K0_ds2.field_lines: if t96_K0_ds2.field_lines[key] is None: continue forward = pv.Show(t96_K0_ds2.field_lines[key].fieldLineObj_f, rvs4) forward.DiffuseColor = [1, 0.0, 0.0] forward.LineWidth = 2.0