class ImodViewer(pwviewer.Viewer):
""" Wrapper to visualize different type of objects
with the Imod program 3dmod
"""
_environments = [pwviewer.DESKTOP_TKINTER, Plugin.getEnviron()]
_targets = [
tomo.objects.TiltSeries,
tomo.objects.Tomogram,
tomo.objects.SetOfTomograms,
tomo.objects.SetOfTiltSeries,
tomo.objects.SetOfLandmarkModels,
tomo.objects.LandmarkModel
]
def _visualize(self, obj, **kwargs):
env = Plugin.getEnviron()
cls = type(obj)
if issubclass(cls, tomo.objects.TiltSeries):
view = ImodObjectView(obj.getFirstItem())
elif issubclass(cls, tomo.objects.Tomogram):
view = ImodObjectView(obj)
elif issubclass(cls, tomo.objects.LandmarkModel):
view = ImodObjectView(obj)
else:
view = ImodGenericViewer(self.getTkRoot(), self.protocol, obj)
view._env = env
return [view]
def _visualize(self, obj, **kwargs):
env = Plugin.getEnviron()
cls = type(obj)
if issubclass(cls, tomo.objects.TiltSeries):
view = ImodObjectView(obj.getFirstItem())
elif issubclass(cls, tomo.objects.Tomogram):
view = ImodObjectView(obj)
elif issubclass(cls, tomo.objects.LandmarkModel):
view = ImodObjectView(obj)
else:
view = ImodGenericViewer(self.getTkRoot(), self.protocol, obj)
view._env = env
return [view]
