diff --git a/src/Cxx.md b/src/Cxx.md index 264bddf0d6f9f1346f80bf006b7598d2804e3a51..80df14dd8065327dbffacd44fac0adb1c7133bfc 100644 --- a/src/Cxx.md +++ b/src/Cxx.md @@ -69,7 +69,7 @@ If you are new to VTK then these [tutorials](#tutorial) will help to get you sta [ReadSTL](/Cxx/IO/ReadSTL) | Read stereo lithography STL (.stl) files [ReadTextFile](/Cxx/IO/ReadTextFile) | Read a plain text file into a polydata. [SimplePointsReader](/Cxx/IO/SimplePointsReader) | Read a simple "xyz" file of points. -[TransientHDFReader](/Cxx/IO/TransientHDFReader) | Read transient data writen within a vtkhdf file. +[TransientHDFReader](/Cxx/IO/TransientHDFReader) | Read transient data written inside a vtkhdf file. ###### Importers diff --git a/src/Cxx/IO/TransientHDFReader.cxx b/src/Cxx/IO/TransientHDFReader.cxx index 008fddc2ce96b9ed6d9669bbd7be1abdc2f26524..bd5328eb7015fb114ed0700c699d8865780f4fc9 100644 --- a/src/Cxx/IO/TransientHDFReader.cxx +++ b/src/Cxx/IO/TransientHDFReader.cxx @@ -33,14 +33,14 @@ int main(int ac, char** av) vtkNew<vtkNamedColors> colors; - // Read the dataset + // Read the dataset. vtkNew<vtkHDFReader> reader; reader->SetFileName(av[1]); reader->Update(); std::cout << "number of steps: " << reader->GetNumberOfSteps() << endl; auto polydata = vtkPolyData::SafeDownCast(reader->GetOutput()); - // Render the dataset + // Render the dataset. vtkNew<vtkPolyDataMapper> mapper; mapper->SetInputData(polydata); mapper->SetLookupTable(GetCTF()); @@ -58,14 +58,15 @@ int main(int ac, char** av) vtkNew<vtkRenderWindow> renWin; renWin->AddRenderer(renderer); renWin->SetWindowName("TransientHDFReader"); + renWin->SetSize(1024, 512); renWin->Render(); - // Add the animation callback + // Add the animation callback. vtkNew<vtkCallbackCommand> command; command->SetCallback(&Animate); command->SetClientData(reader); - // Add the interactor + // Add the interactor. vtkNew<vtkRenderWindowInteractor> iren; iren->SetRenderWindow(renWin); iren->AddObserver(vtkCommand::TimerEvent, command); diff --git a/src/Python.md b/src/Python.md index 0830555dcefcde09bd028e177df3fe018e78e959..8d5582312a356e7395e97040b91f83aa480db5f1 100644 --- a/src/Python.md +++ b/src/Python.md @@ -48,6 +48,8 @@ If you are new to VTK then these [tutorials](#tutorial) will help to get you sta [ReadSTL](/Python/IO/ReadSTL) | Read an STL file. [ReadUnstructuredGrid](/Python/IO/ReadUnstructuredGrid) | Using vtkUnstructuredGridReader to read a (legacy) *.vtk file [ReadVTP](/Python/IO/ReadVTP) | Read a VTK XML PolyData file. +[TransientHDFReader](/Python/IO/TransientHDFReader) | Read transient data written inside a vtkhdf file. + ###### Importers diff --git a/src/Python/IO/TransientHDFReader.md b/src/Python/IO/TransientHDFReader.md new file mode 100644 index 0000000000000000000000000000000000000000..e2eb9305da68f9b19e883fb0467a2d09b2fb59b0 --- /dev/null +++ b/src/Python/IO/TransientHDFReader.md @@ -0,0 +1,3 @@ +### Description + +That example uses a feature of vtk_hdf5 that is only available in [vtk/master](https://gitlab.kitware.com/vtk/vtk) and will be released with VTK 9.3. See [this blog post](https://www.kitware.com/how-to-write-time-dependent-data-in-vtkhdf-files/) for more information. diff --git a/src/Python/IO/TransientHDFReader.py b/src/Python/IO/TransientHDFReader.py new file mode 100755 index 0000000000000000000000000000000000000000..e1015b53fbeb99f39d3aebecfd11ef49e612fe38 --- /dev/null +++ b/src/Python/IO/TransientHDFReader.py @@ -0,0 +1,148 @@ +#!/usr/bin/env python3 + +# noinspection PyUnresolvedReferences +import vtkmodules.vtkRenderingContextOpenGL2 +from vtkmodules.vtkCommonColor import vtkNamedColors +from vtkmodules.vtkIOHDF import vtkHDFReader +from vtkmodules.vtkInteractionStyle import vtkInteractorStyleTrackballCamera +from vtkmodules.vtkRenderingCore import ( + vtkActor, + vtkDiscretizableColorTransferFunction, + vtkPolyDataMapper, + vtkRenderWindow, + vtkRenderWindowInteractor, + vtkRenderer +) + + +def get_program_parameters(): + import argparse + description = 'Read transient data writen inside a vtkhdf file.' + epilogue = '''''' + parser = argparse.ArgumentParser(description=description, epilog=epilogue, + formatter_class=argparse.RawDescriptionHelpFormatter) + parser.add_argument('file_name', help='warping_spheres.vtkhdf') + args = parser.parse_args() + return args.file_name + + +def main(): + fn = get_program_parameters() + + colors = vtkNamedColors() + + # Read the dataset + reader = vtkHDFReader() + reader.SetFileName(fn) + reader.Update() + print('number of steps: ', reader.GetNumberOfSteps()) + polydata = reader.GetOutput() + + # Render the dataset + mapper = vtkPolyDataMapper() + mapper.SetInputData(polydata) + mapper.SetLookupTable(get_ctf()) + mapper.SetScalarModeToUsePointFieldData() + mapper.SelectColorArray('SpatioTemporalHarmonics') + + actor = vtkActor() + actor.SetMapper(mapper) + + renderer = vtkRenderer() + renderer.SetBackground(colors.GetColor3d('Wheat')) + renderer.UseHiddenLineRemovalOn() + renderer.AddActor(actor) + + ren_win = vtkRenderWindow() + ren_win.AddRenderer(renderer) + ren_win.SetWindowName('TransientHDFReader') + ren_win.SetSize(1024, 512) + ren_win.Render() + + # Add the interactor + iren = vtkRenderWindowInteractor() + iren.SetRenderWindow(ren_win) + iren.CreateRepeatingTimer(50) + # Add the animation callback + observer = AnimationObserver(iren, reader) + iren.AddObserver('TimerEvent', observer) + + i_style = vtkInteractorStyleTrackballCamera() + iren.SetInteractorStyle(i_style) + + iren.Start() + + +def get_ctf(): + ctf = vtkDiscretizableColorTransferFunction() + ctf.SetColorSpaceToLab() + ctf.SetScaleToLinear() + + ctf.AddRGBPoint(-30.3399130649763, 0.0862745098039216, 0.00392156862745098, 0.298039215686275) + ctf.AddRGBPoint(-29.3502559661865, 0.113725, 0.0235294, 0.45098) + ctf.AddRGBPoint(-28.5283393859863, 0.105882, 0.0509804, 0.509804) + ctf.AddRGBPoint(-27.958028793335, 0.0392157, 0.0392157, 0.560784) + ctf.AddRGBPoint(-27.4044914245605, 0.0313725, 0.0980392, 0.6) + ctf.AddRGBPoint(-26.8677291870117, 0.0431373, 0.164706, 0.639216) + ctf.AddRGBPoint(-26.096134185791, 0.054902, 0.243137, 0.678431) + ctf.AddRGBPoint(-25.0729293823242, 0.054902, 0.317647, 0.709804) + ctf.AddRGBPoint(-23.8148933330084, 0.0509804, 0.396078, 0.741176) + ctf.AddRGBPoint(-22.9992658665124, 0.0392157, 0.466667, 0.768627) + ctf.AddRGBPoint(-22.1836384000164, 0.0313725, 0.537255, 0.788235) + ctf.AddRGBPoint(-21.3323650360107, 0.0313725, 0.615686, 0.811765) + ctf.AddRGBPoint(-20.4601268768311, 0.0235294, 0.709804, 0.831373) + ctf.AddRGBPoint(-19.5878868103027, 0.0509804, 0.8, 0.85098) + ctf.AddRGBPoint(-18.8666133880615, 0.0705882, 0.854902, 0.870588) + ctf.AddRGBPoint(-18.1956596374512, 0.262745, 0.901961, 0.862745) + ctf.AddRGBPoint(-17.6085758209229, 0.423529, 0.941176, 0.87451) + ctf.AddRGBPoint(-16.7027893066406, 0.572549, 0.964706, 0.835294) + ctf.AddRGBPoint(-16.0989303588867, 0.658824, 0.980392, 0.843137) + ctf.AddRGBPoint(-15.6628112792969, 0.764706, 0.980392, 0.866667) + ctf.AddRGBPoint(-15.1931447982788, 0.827451, 0.980392, 0.886275) + ctf.AddRGBPoint(-14.2705841064453, 0.913725, 0.988235, 0.937255) + ctf.AddRGBPoint(-13.9854288101196, 1, 1, 0.972549019607843) + ctf.AddRGBPoint(-13.7002735137939, 0.988235, 0.980392, 0.870588) + ctf.AddRGBPoint(-13.2809276580811, 0.992156862745098, 0.972549019607843, 0.803921568627451) + ctf.AddRGBPoint(-12.9622249603271, 0.992157, 0.964706, 0.713725) + ctf.AddRGBPoint(-12.4254627227783, 0.988235, 0.956863, 0.643137) + ctf.AddRGBPoint(-11.5699977874756, 0.980392, 0.917647, 0.509804) + ctf.AddRGBPoint(-10.8487224578857, 0.968627, 0.87451, 0.407843) + ctf.AddRGBPoint(-10.1106739044189, 0.94902, 0.823529, 0.321569) + ctf.AddRGBPoint(-9.57391166687012, 0.929412, 0.776471, 0.278431) + ctf.AddRGBPoint(-8.78554153442383, 0.909804, 0.717647, 0.235294) + ctf.AddRGBPoint(-8.08104133605957, 0.890196, 0.658824, 0.196078) + ctf.AddRGBPoint(-7.50234400308847, 0.878431, 0.619608, 0.168627) + ctf.AddRGBPoint(-6.68671653659248, 0.870588, 0.54902, 0.156863) + ctf.AddRGBPoint(-5.87108907009648, 0.85098, 0.47451, 0.145098) + ctf.AddRGBPoint(-5.05546160360049, 0.831373, 0.411765, 0.133333) + ctf.AddRGBPoint(-4.23983413710449, 0.811765, 0.345098, 0.113725) + ctf.AddRGBPoint(-3.4242066706085, 0.788235, 0.266667, 0.0941176) + ctf.AddRGBPoint(-2.6085792041125, 0.741176, 0.184314, 0.0745098) + ctf.AddRGBPoint(-1.79295173761651, 0.690196, 0.12549, 0.0627451) + ctf.AddRGBPoint(-0.977324271120517, 0.619608, 0.0627451, 0.0431373) + ctf.AddRGBPoint(-0.214114964008331, 0.54902, 0.027451, 0.0705882) + ctf.AddRGBPoint(0.456838220357895, 0.470588, 0.0156863, 0.0901961) + ctf.AddRGBPoint(1.21166050434113, 0.4, 0.00392157, 0.101961) + ctf.AddRGBPoint(2.28518559486346, 0.188235294117647, 0, 0.0705882352941176) + + ctf.SetNumberOfValues(46) + ctf.DiscretizeOff() + + return ctf + + +class AnimationObserver(object): + def __init__(self, interactor, reader): + self.interactor = interactor + self.reader = reader + + def __call__(self, caller, ev): + x = 0 if (self.reader.GetStep() == self.reader.GetNumberOfSteps() - 1) else self.reader.GetStep() + 1 + self.reader.SetStep(x) + print('Current step: ', self.reader.GetStep()) + self.reader.Update() + self.interactor.Render() + + +if __name__ == '__main__': + main() diff --git a/src/Testing/Baseline/Cxx/IO/TestTransientHDFReader.png b/src/Testing/Baseline/Cxx/IO/TestTransientHDFReader.png index c2c3f6fced1e469f9018af1635188e53b502dbf1..6054c2757dcfe29bfe742561297b99154a381941 100644 --- a/src/Testing/Baseline/Cxx/IO/TestTransientHDFReader.png +++ b/src/Testing/Baseline/Cxx/IO/TestTransientHDFReader.png @@ -1,3 +1,3 @@ version https://git-lfs.github.com/spec/v1 -oid sha256:89fe69f18c3ef305aac25c6dccc1663753071b71ee488f12121ac16acaaa4472 -size 54961 +oid sha256:d1448e1f0a15e01d7371731bcf1b0d72716af59ac62ff4a90626c9f4892956cb +size 53670 diff --git a/src/Testing/Baseline/Python/IO/TestTransientHDFReader.png b/src/Testing/Baseline/Python/IO/TestTransientHDFReader.png new file mode 100644 index 0000000000000000000000000000000000000000..6054c2757dcfe29bfe742561297b99154a381941 --- /dev/null +++ b/src/Testing/Baseline/Python/IO/TestTransientHDFReader.png @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d1448e1f0a15e01d7371731bcf1b0d72716af59ac62ff4a90626c9f4892956cb +size 53670