Skip to content

Add a render quality setting to VolumeRenderer. #11

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Apr 15, 2014
Merged

Add a render quality setting to VolumeRenderer. #11

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
43 changes: 40 additions & 3 deletions ensemble/volren/volume_renderer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from mayavi.core.ui.api import MlabSceneModel
from mayavi.sources.vtk_data_source import VTKDataSource
from mayavi.tools.tools import add_dataset
from traits.api import CInt, Float, HasTraits, Instance, List, \
from traits.api import CInt, Enum, Float, HasTraits, Instance, List, \
on_trait_change
from tvtk.api import tvtk

Expand All @@ -13,6 +13,36 @@
from ensemble.volren.volume_data import VolumeData

CLIP_MAX = 100
QUALITY_SETTINGS = {
'best': {
'mapper': {
'sample_distance': 0.05,
},
'property': {
'shade': False,
'ambient': 0.5,
'diffuse': 0.4,
'specular': 0.1,
'specular_power': 100,
}
},
'default': {
'mapper': {
'sample_distance': 0.2,
},
'property': {
'shade': False
}
},
'performance': {
'mapper': {
'sample_distance': 1.0,
},
'property': {
'shade': False
}
},
}


class VolumeRenderer(HasTraits):
Expand All @@ -35,6 +65,9 @@ class VolumeRenderer(HasTraits):
# Clip plane positions
clip_bounds = List(Float)

# Render quality setting
render_quality = Enum('default', QUALITY_SETTINGS.keys())

# The transfer function editor
ctf_editor = Instance(CtfEditor)

Expand Down Expand Up @@ -69,6 +102,9 @@ def _volume_data_changed(self):
def _clip_bounds_items_changed(self):
self._set_volume_clip_planes()

def _render_quality_changed(self):
self._setup_volume()

@on_trait_change('ctf_editor.function_updated')
def ctf_updated(self):
ctf = tvtk.ColorTransferFunction()
Expand Down Expand Up @@ -113,8 +149,9 @@ def display_model(self):
#--------------------------------------------------------------------------

def _setup_volume(self):
self.volume.volume_mapper.trait_set(sample_distance=0.2)
self.volume.volume_property.trait_set(shade=False)
render_settings = QUALITY_SETTINGS[self.render_quality]
self.volume.volume_mapper.trait_set(**render_settings['mapper'])
self.volume.volume_property.trait_set(**render_settings['property'])
self.ctf_updated()

def _set_volume_clip_planes(self):
Expand Down