class EMImageRotorModule(EMGUIModule):
def get_desktop_hint(self):
return "rotor"
def get_gl_widget(self,qt_context_parent,gl_context_parent):
from emfloatingwidgets import EM3DGLWindowOverride
if self.gl_widget == None:
self.gl_context_parent = gl_context_parent
self.qt_context_parent = qt_context_parent
self.gl_widget = EM3DGLWindowOverride(self,self.rotor)
self.rotor.target_zoom_events_allowed(False)
self.rotor.target_translations_allowed(False)
return self.gl_widget
def __del__(self):
for widget in self.rotor.get_widgets():
self.application.deregister_qt_emitter(widget.get_drawable().get_drawable())
def get_qt_widget(self):
if self.qt_context_parent == None:
from emimageutil import EMParentWin
self.gl_context_parent = EMImageRotorWidget(self)
self.qt_context_parent = EMParentWin(self.gl_context_parent)
self.gl_widget = self.gl_context_parent
self.qt_context_parent.setAcceptDrops(True)
return self.qt_context_parent
#def get_qt_widget(self):
#if self.parent == None:
#from emimageutil import EMParentWin
#self.gl_parent = EMImageRotorWidget(self)
#self.parent = EMParentWin(self.gl_parent)
#self.set_gl_parent(self.gl_parent)
#return self.parent
def __init__(self, data=None,application=None):
self.rotor = EMGLRotorWidget(self,15,10,45,EMGLRotorWidget.LEFT_ROTARY)
EMGUIModule.__init__(self,ensure_gl_context=True)
self.data=None
self.initsizeflag = True
if data:
self.set_data(data)
self.widget = EM3DGLWindow(self,self.rotor)
self.widget.set_draw_frame(False)
self.z_near = 0
self.z_far = 0
self.hud_data = [] # a list of strings to be rendered to the heads up display (hud)
self.load_font_renderer()
def optimally_resize(self):
if isinstance(self.gl_context_parent,EMImageRotorWidget):
self.qt_context_parent.resize(*self.get_optimal_size())
else:
self.gl_widget.resize(*self.get_optimal_size())
def load_font_renderer(self):
try:
self.font_renderer = get_3d_font_renderer()
self.font_renderer.set_face_size(20)
self.font_renderer.set_depth(4)
self.font_renderer.set_font_mode(FTGLFontMode.EXTRUDE)
# self.font_renderer.set_font_file_name("/usr/share/fonts/dejavu/DejaVuSerif.ttf")
self.font_render_mode = EMGUIModule.FTGL
except:
self.font_render_mode = EMGUIModule.GLUT
def set_extra_hud_data(self,hud_data):
self.hud_data = hud_data
def get_optimal_size(self):
lr = self.rotor.get_lr_bt_nf()
width = lr[1] - lr[0]
height = lr[3] - lr[2]
return [width+80,height+20]
#self.rotor.set_shapes([],1.01)
def context(self):
# asking for the OpenGL context from the parent
return self.gl_context_parent.context()
def emit(self,*args,**kargs):
#print "emitting",signal,event
self.application.get_qt_emitter(self).emit(*args,**kargs)
#if b != None:
#elif a != None:
#self.application.get_qt_emitter(self).emit(signal,event,a)
#else:
#self.application.get_qt_emitter(self).emit(signal,event)
def set_mouse_mode(self,mode):
self.mmode = mode
self.rotor.set_mouse_mode(mode)
def set_frozen(self,frozen,idx=0):
self.rotor.set_frozen(frozen,idx)
def set_shapes(self,shapes,shrink,idx=0):
self.rotor.set_shapes(shapes,shrink,idx)
def register_animatable(self,animatable):
self.get_qt_context_parent().register_animatable(animatable)
def get_inspector(self):
return None
def width(self):
try: return self.get_parent().width()
except: return 0
def height(self):
try: return self.get_parent().height()
except: return 0
def getImageFileName(self):
''' warning - could return none in some circumstances'''
try: return self.gl_widget.getImageFileName()
except: return None
def set_data(self,data):
if data == None or not isinstance(data,list) or len(data)==0:
self.data = []
return
self.data = data
self.rotor.clear_widgets()
for d in self.data:
w = EM2DGLView(self,d)
#w.get_drawable().set_app(self.application)
# self.application.register_qt_emitter(w.get_drawable(),self.application.get_qt_emitter(self))
x = EM2DGLWindow(self,w)
x.set_width(w.width())
x.set_height(w.height())
self.rotor.add_widget(x)
def updateGL(self):
try: self.gl_widget.updateGL()
except: pass
def render(self):
if not self.data : return
#glLoadIdentity()
GL.glEnable(GL.GL_DEPTH_TEST)
GL.glEnable(GL.GL_LIGHTING)
lr = self.rotor.get_lr_bt_nf()
lrt = lr
#lr = self.widget.get_lr_bt_nf()
z = self.gl_context_parent.get_depth_for_height(abs(lr[3]-lr[2]))
z_near = z-lrt[4]
z_trans = 0
z_far = z-lrt[5]
if z_near < 0:
z_trans = z_near
z_near = 0.1
z_far -= z_trans
if z_far < 0: z_far = 0.1 # hacking alert
if self.z_near != z_near or self.z_far != z_far:
self.z_near = z_near
self.z_far = z_far
if isinstance(self.gl_context_parent,EMImageRotorWidget):
self.gl_context_parent.set_near_far(self.z_near,self.z_far)
else: print "bug"
#FTGL.print_message("hello world",36);
#print self.z_near,self.z_far,-self.parent.get_depth_for_height(abs(lr[3]-lr[2]))+z_trans
glPushMatrix()
glTranslate(0,0,-z)
#glTranslate(0,-75,0) # This number is a FIXME issue
#FTGL.print_message("hello hello",36);
self.widget.draw()
glPopMatrix()
self.draw_hud()
def dragEnterEvent(self,event):
pass
def dropEvent(self,event):
pass
def mousePressEvent(self, event):
if event.button()==Qt.MidButton or (event.button()==Qt.LeftButton and event.modifiers()&Qt.AltModifier):
self.show_inspector(True)
else:
if self.widget.isinwin(event.x(),self.gl_context_parent.viewport_height()-event.y()):
self.widget.mousePressEvent(event)
self.updateGL()
def mouseMoveEvent(self, event):
if self.widget.isinwin(event.x(),self.gl_context_parent.viewport_height()-event.y()):
self.widget.mouseMoveEvent(event)
self.updateGL()
def mouseReleaseEvent(self, event):
if self.widget.isinwin(event.x(),self.gl_context_parent.viewport_height()-event.y()):
self.widget.mouseReleaseEvent(event)
self.updateGL()
def wheelEvent(self, event):
if self.widget.isinwin(event.x(),self.gl_context_parent.viewport_height()-event.y()):
self.widget.wheelEvent(event)
self.updateGL()
def leaveEvent(self):
pass
def keyPressEvent(self,event):
#if self.widget.isinwin(event.x(),self.gl_context_parent.viewport_height()-event.y()):
self.widget.keyPressEvent(event)
self.updateGL()
def resize_event(self, width, height):
self.rotor.resize_event(width,height)
def draw_hud(self):
width = self.gl_widget.width()
height = self.gl_widget.height()
glMatrixMode(GL_PROJECTION)
glPushMatrix()
glLoadIdentity()
glOrtho(0,width,0,height,-100,100)
glMatrixMode(GL_MODELVIEW)
glLoadIdentity()
glEnable(GL_LIGHTING)
glEnable(GL_NORMALIZE)
glMaterial(GL_FRONT,GL_AMBIENT,(0.2, 1.0, 0.2,1.0))
glMaterial(GL_FRONT,GL_DIFFUSE,(0.2, 1.0, 0.9,1.0))
glMaterial(GL_FRONT,GL_SPECULAR,(1.0 , 0.5, 0.2,1.0))
glMaterial(GL_FRONT,GL_SHININESS,20.0)
enable_depth = glIsEnabled(GL_DEPTH_TEST)
glDisable(GL_DEPTH_TEST)
glColor(1.0,1.0,1.0)
if self.font_render_mode == EMImageMXModule.FTGL:
panels = len(self.rotor)
idx = self.rotor.current_index()
string = str(idx+1) + ' / ' + str(panels)
bbox = self.font_renderer.bounding_box(string)
x_offset = width-(bbox[3]-bbox[0]) - 10
y_offset = 10
glPushMatrix()
glTranslate(x_offset,y_offset,0)
glRotate(20,0,1,0)
self.font_renderer.render_string(string)
glPopMatrix()
y_offset += bbox[4]-bbox[1]
for s in self.hud_data:
string = str(s)
bbox = self.font_renderer.bounding_box(string)
x_offset = width-(bbox[3]-bbox[0]) - 10
y_offset += 10
glPushMatrix()
glTranslate(x_offset,y_offset,0)
glRotate(20,0,1,0)
self.font_renderer.render_string(string)
glPopMatrix()
y_offset += bbox[4]-bbox[1]
else:
pass
if enable_depth: glEnable(GL_DEPTH_TEST)
glMatrixMode(GL_PROJECTION)
glPopMatrix()
glMatrixMode(GL_MODELVIEW)
