def __init__(self):
self.move = [0,0,0]
self.dirskeys = ( ( ['w'] , ['s'] ) , ( ['a'] , ['d'] ) , ( ['e'] , ['q'] ) )
for d in self.dirskeys :
for e in d :
for i in range(len(e)) : e[i] = ( gtk.gdk.unicode_to_keyval(ord(e[i])) , False )
self.near = 1
self.far = 1000
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON3_MOTION_MASK )
self.drawing_area.set_size_request(320,240)
builder.get_object("vbox1").pack_start(self.drawing_area)
win_main = builder.get_object("win_main")
win_main.set_events( gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK )
win_main.connect('key-press-event' , self._on_key_pressed )
win_main.connect('key-release-event', self._on_key_released )
self.scene = Scene( self.fov , .01 , self.near , self.far , meshes )
self.drawing_area.add( self.scene )
print 'Scene added'
win_main.show_all()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene.set_ratio( ratio )
builder.connect_signals(self)
self.statbar = builder.get_object('statbar')
self.drawing_area.connect('motion_notify_event',self._on_mouse_motion)
self.drawing_area.connect('button_press_event',self._on_button_pressed)
self.drawing_area.connect('configure_event',self._on_reshape)
self.drawing_area.connect_after('expose_event',self._after_draw)
gtk.timeout_add( 1 , self._refresh )
def __init__(self):
self.button = {}
self.move = [0,0,0]
self.dirskeys = ( ( ['w'] , ['s'] ) , ( ['a'] , ['d'] ) , ( ['e'] , ['q'] ) )
for d in self.dirskeys :
for e in d :
for i in range(len(e)) : e[i] = ( gtk.gdk.unicode_to_keyval(ord(e[i])) , False )
self.near = 1
self.far = 1000
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON1_MOTION_MASK | gtk.gdk.BUTTON2_MOTION_MASK |gtk.gdk.BUTTON3_MOTION_MASK )
self.drawing_area.set_size_request(320,240)
builder.get_object("vbox1").pack_start(self.drawing_area)
win_main = builder.get_object("win_main")
win_main.set_events( gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK )
win_main.connect('key-press-event' , self._on_key_pressed )
win_main.connect('key-release-event', self._on_key_released )
self.scene = Scene( self.fov , .01 , self.near , self.far )
self.drawing_area.add( self.scene , ( 0,0,.5,1) )
self.drawing_area.add( self.scene , (.5,0,.5,1) )
print 'Scene added'
win_main.show_all()
width = self.drawing_area.allocation.width / 2.0
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene.set_ratio( ratio )
builder.connect_signals(self)
# self.statbar = builder.get_object('statbar')
self.drawing_area.connect('motion_notify_event',self._on_mouse_motion)
self.drawing_area.connect('button_press_event',self._on_button_pressed)
self.drawing_area.connect('button_release_event',self._on_button_released)
self.drawing_area.connect('configure_event',self._on_reshape)
self.drawing_area.connect_after('expose_event',self._after_draw)
gtk.timeout_add( 1 , self._refresh )
def __init__(self):
self._init_keyboard()
self.near = .01
self.far = 10
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON3_MOTION_MASK )
self.drawing_area.set_size_request(800, 800)
builder.get_object("vbox1").pack_start(self.drawing_area)
self.scene = Scene( self.fov , 1 , self.near , self.far , self._load_cube_img(skybox_files) , self._load_img(duck_file) )
self.drawing_area.add( self.scene )
win_main = builder.get_object("win_main")
win_main.connect('key-press-event' , self._on_key_pressed )
win_main.connect('key-release-event', self._on_key_released )
win_main.show_all()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene.set_ratio( ratio )
builder.connect_signals(self)
self.statbar = builder.get_object('statbar')
self.drawing_area.connect('motion_notify_event',self._on_mouse_motion)
self.drawing_area.connect('button_press_event',self._on_button_pressed)
self.drawing_area.connect('configure_event',self._on_reshape)
self.drawing_area.connect_after('expose_event',self._after_draw)
gtk.timeout_add( 1 , self._refresh )
def __init__(self):
self._init_keyboard()
self.near = 1
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events(
gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON1_MOTION_MASK
)
self.drawing_area.set_size_request(800, 800)
self.box3d = builder.get_object("hbox_3d")
builder.get_object("vbox4").pack_start(self.drawing_area)
win_main = builder.get_object("win_main")
win_main.set_events(gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK)
win_main.connect("key-press-event", self._on_key_pressed)
win_main.connect("key-release-event", self._on_key_released)
win_main.show_all()
self.box3d.hide()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width) / float(height)
self.scene = Scene(self.fov, ratio, self.near)
self.drawing_area.add(self.scene)
builder.connect_signals(self)
self.statbar = builder.get_object("statbar")
self.drawing_area.connect("motion_notify_event", self._on_mouse_motion)
self.drawing_area.connect("button_press_event", self._on_button_pressed)
self.drawing_area.connect("configure_event", self._on_reshape)
self.drawing_area.connect_after("expose_event", self._after_draw)
self.rbut_trans = builder.get_object("rbut_trans")
self.rbut_isoscale = builder.get_object("rbut_isoscale")
self.rbut_scale = builder.get_object("rbut_scale")
self.rbut_rotate = builder.get_object("rbut_rotate")
self.rbut_xy = builder.get_object("rbut_xy")
self.rbut_xz = builder.get_object("rbut_xz")
self.rbut_yz = builder.get_object("rbut_yz")
self.sp_fov = builder.get_object("sp_fov")
self.sp_fov.set_value(self.fov)
self.sp_near = builder.get_object("sp_near")
self.sp_near.set_value(self.near)
self.sp_pos_x = builder.get_object("sp_pos_x")
self.sp_pos_y = builder.get_object("sp_pos_y")
self.sp_pos_z = builder.get_object("sp_pos_z")
self.sp_look_x = builder.get_object("sp_look_x")
self.sp_look_y = builder.get_object("sp_look_y")
self.sp_look_z = builder.get_object("sp_look_z")
self.on_but_pos_appyly_clicked(None)
self.tbut_add_c0 = builder.get_object("tbut_add_c0")
self.tbut_add_c2 = builder.get_object("tbut_add_c2")
self.tbut_add_inter = builder.get_object("tbut_add_interpolation")
self.tbut_del_curve = builder.get_object("tbut_del_curve")
self.tbut_sel_curve = builder.get_object("tbut_sel_curve")
self.tbut_add_surf_c0 = builder.get_object("tbut_add_surf_c0")
self.tbut_add_surf_c2 = builder.get_object("tbut_add_surf_c2")
self.tbut_add_pipe = builder.get_object("tbut_add_pipe")
self.tbut_add_gregory = builder.get_object("tbut_add_gregory")
self.tbut_cut = builder.get_object("tbut_cut_choose")
self.cbox_first = builder.get_object("cbox_first")
self.cbox_second = builder.get_object("cbox_second")
self.sp_delta = builder.get_object("sp_cut_delta")
self.sp_cut_u1 = builder.get_object("sp_cut_u1")
self.sp_cut_v1 = builder.get_object("sp_cut_v1")
self.sp_cut_u2 = builder.get_object("sp_cut_u2")
self.sp_cut_v2 = builder.get_object("sp_cut_v2")
cell = gtk.CellRendererText()
self.cbox_first.pack_start(cell, True)
self.cbox_first.add_attribute(cell, "text", 0)
cell = gtk.CellRendererText()
self.cbox_second.pack_start(cell, True)
self.cbox_second.add_attribute(cell, "text", 0)
self.tbuts = [
self.tbut_add_c0,
self.tbut_add_c2,
self.tbut_add_inter,
self.tbut_del_curve,
self.tbut_sel_curve,
self.tbut_add_surf_c0,
self.tbut_add_surf_c2,
self.tbut_add_pipe,
self.tbut_add_gregory,
self.tbut_cut,
]
self.sp_surf_x = builder.get_object("sp_surf_x")
self.sp_surf_y = builder.get_object("sp_surf_y")
self.sp_draw_surf_x = builder.get_object("sp_draw_surf_x")
self.sp_draw_surf_y = builder.get_object("sp_draw_surf_y")
self.win_dia_load = builder.get_object("win_dia_load")
self.win_dia_save = builder.get_object("win_dia_save")
if os.path.isdir("../data/"):
self.win_dia_load.set_current_folder("../data/")
self.win_dia_save.set_current_folder("../data/")
self.win_dia_load.set_transient_for(win_main)
self.win_dia_load.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
self.win_dia_load.add_button(gtk.STOCK_OPEN, gtk.RESPONSE_OK)
self.win_dia_save.set_transient_for(win_main)
self.win_dia_save.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
self.win_dia_save.add_button(gtk.STOCK_SAVE, gtk.RESPONSE_OK)
self.save_file = None
class App(object):
"""Application main class"""
def __init__(self):
self._init_keyboard()
self.near = 1
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events(
gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON1_MOTION_MASK
)
self.drawing_area.set_size_request(800, 800)
self.box3d = builder.get_object("hbox_3d")
builder.get_object("vbox4").pack_start(self.drawing_area)
win_main = builder.get_object("win_main")
win_main.set_events(gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK)
win_main.connect("key-press-event", self._on_key_pressed)
win_main.connect("key-release-event", self._on_key_released)
win_main.show_all()
self.box3d.hide()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width) / float(height)
self.scene = Scene(self.fov, ratio, self.near)
self.drawing_area.add(self.scene)
builder.connect_signals(self)
self.statbar = builder.get_object("statbar")
self.drawing_area.connect("motion_notify_event", self._on_mouse_motion)
self.drawing_area.connect("button_press_event", self._on_button_pressed)
self.drawing_area.connect("configure_event", self._on_reshape)
self.drawing_area.connect_after("expose_event", self._after_draw)
self.rbut_trans = builder.get_object("rbut_trans")
self.rbut_isoscale = builder.get_object("rbut_isoscale")
self.rbut_scale = builder.get_object("rbut_scale")
self.rbut_rotate = builder.get_object("rbut_rotate")
self.rbut_xy = builder.get_object("rbut_xy")
self.rbut_xz = builder.get_object("rbut_xz")
self.rbut_yz = builder.get_object("rbut_yz")
self.sp_fov = builder.get_object("sp_fov")
self.sp_fov.set_value(self.fov)
self.sp_near = builder.get_object("sp_near")
self.sp_near.set_value(self.near)
self.sp_pos_x = builder.get_object("sp_pos_x")
self.sp_pos_y = builder.get_object("sp_pos_y")
self.sp_pos_z = builder.get_object("sp_pos_z")
self.sp_look_x = builder.get_object("sp_look_x")
self.sp_look_y = builder.get_object("sp_look_y")
self.sp_look_z = builder.get_object("sp_look_z")
self.on_but_pos_appyly_clicked(None)
self.tbut_add_c0 = builder.get_object("tbut_add_c0")
self.tbut_add_c2 = builder.get_object("tbut_add_c2")
self.tbut_add_inter = builder.get_object("tbut_add_interpolation")
self.tbut_del_curve = builder.get_object("tbut_del_curve")
self.tbut_sel_curve = builder.get_object("tbut_sel_curve")
self.tbut_add_surf_c0 = builder.get_object("tbut_add_surf_c0")
self.tbut_add_surf_c2 = builder.get_object("tbut_add_surf_c2")
self.tbut_add_pipe = builder.get_object("tbut_add_pipe")
self.tbut_add_gregory = builder.get_object("tbut_add_gregory")
self.tbut_cut = builder.get_object("tbut_cut_choose")
self.cbox_first = builder.get_object("cbox_first")
self.cbox_second = builder.get_object("cbox_second")
self.sp_delta = builder.get_object("sp_cut_delta")
self.sp_cut_u1 = builder.get_object("sp_cut_u1")
self.sp_cut_v1 = builder.get_object("sp_cut_v1")
self.sp_cut_u2 = builder.get_object("sp_cut_u2")
self.sp_cut_v2 = builder.get_object("sp_cut_v2")
cell = gtk.CellRendererText()
self.cbox_first.pack_start(cell, True)
self.cbox_first.add_attribute(cell, "text", 0)
cell = gtk.CellRendererText()
self.cbox_second.pack_start(cell, True)
self.cbox_second.add_attribute(cell, "text", 0)
self.tbuts = [
self.tbut_add_c0,
self.tbut_add_c2,
self.tbut_add_inter,
self.tbut_del_curve,
self.tbut_sel_curve,
self.tbut_add_surf_c0,
self.tbut_add_surf_c2,
self.tbut_add_pipe,
self.tbut_add_gregory,
self.tbut_cut,
]
self.sp_surf_x = builder.get_object("sp_surf_x")
self.sp_surf_y = builder.get_object("sp_surf_y")
self.sp_draw_surf_x = builder.get_object("sp_draw_surf_x")
self.sp_draw_surf_y = builder.get_object("sp_draw_surf_y")
self.win_dia_load = builder.get_object("win_dia_load")
self.win_dia_save = builder.get_object("win_dia_save")
if os.path.isdir("../data/"):
self.win_dia_load.set_current_folder("../data/")
self.win_dia_save.set_current_folder("../data/")
self.win_dia_load.set_transient_for(win_main)
self.win_dia_load.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
self.win_dia_load.add_button(gtk.STOCK_OPEN, gtk.RESPONSE_OK)
self.win_dia_save.set_transient_for(win_main)
self.win_dia_save.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
self.win_dia_save.add_button(gtk.STOCK_SAVE, gtk.RESPONSE_OK)
self.save_file = None
def _init_keyboard(self):
self.move = [0, 0, 0]
self.dirskeys = ((["w"], ["s"]), (["a"], ["d"]), (["e"], ["q"]))
for d in self.dirskeys:
for e in d:
for i in range(len(e)):
e[i] = (gtk.gdk.unicode_to_keyval(ord(e[i])), False)
def _on_key_pressed(self, widget, data=None):
if not any(self.move):
gtk.timeout_add(20, self._move_callback)
for i in range(len(self.dirskeys)):
if (data.keyval, False) in self.dirskeys[i][0]:
self.dirskeys[i][0][self.dirskeys[i][0].index((data.keyval, False))] = (data.keyval, True)
self.move[i] += 1
elif (data.keyval, False) in self.dirskeys[i][1]:
self.dirskeys[i][1][self.dirskeys[i][1].index((data.keyval, False))] = (data.keyval, True)
self.move[i] -= 1
def _on_key_released(self, widget, data=None):
for i in range(len(self.dirskeys)):
if (data.keyval, True) in self.dirskeys[i][0]:
self.dirskeys[i][0][self.dirskeys[i][0].index((data.keyval, True))] = (data.keyval, False)
self.move[i] -= 1
elif (data.keyval, True) in self.dirskeys[i][1]:
self.dirskeys[i][1][self.dirskeys[i][1].index((data.keyval, True))] = (data.keyval, False)
self.move[i] += 1
def _move_callback(self):
self.scene.key_pressed(self.move)
self.drawing_area.queue_draw()
return any(self.move)
def _after_draw(self, widget, data=None):
self.update_statusbar()
def update_statusbar(self):
cid = self.statbar.get_context_id("cursor")
self.statbar.pop(cid)
self.statbar.push(cid, str(self.scene.get_cursor_pos()) + " " + str(self.scene.get_cursor_screen_pos()))
def toggle_tbuts(self, omit=None):
for t in self.tbuts:
if t == omit:
continue
t.set_active(False)
def on_tbut_add_toggled(self, widget, data=None):
if widget.get_active():
self.toggle_tbuts(widget)
def on_sp_draw_surf_value_changed(self, widget, data=None):
self.scene.set_surf_density((self.sp_draw_surf_x.get_value_as_int(), self.sp_draw_surf_y.get_value_as_int()))
self.drawing_area.queue_draw()
def _on_reshape(self, widget, data=None):
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width) / float(height)
self.scene.set_screen_size(width, height)
self.scene.set_ratio(ratio)
def _on_button_pressed(self, widget, data=None):
surfdata = (
(self.sp_surf_x.get_value_as_int(), self.sp_surf_y.get_value_as_int()),
(self.sp_draw_surf_x.get_value_as_int(), self.sp_draw_surf_y.get_value_as_int()),
)
if data.button == 1:
self.mouse_pos = -data.x, data.y
elif data.button == 3:
if self.tbut_add_c0.get_active():
self.scene.new_curve_c0()
self.tbut_add_c0.set_active(False)
elif self.tbut_add_c2.get_active():
self.scene.new_curve_c2()
self.tbut_add_c2.set_active(False)
elif self.tbut_add_surf_c0.get_active():
self.scene.new_surface_c0(surfdata)
self.tbut_add_surf_c0.set_active(False)
elif self.tbut_add_surf_c2.get_active():
self.scene.new_surface_c2(surfdata)
self.tbut_add_surf_c2.set_active(False)
elif self.tbut_add_pipe.get_active():
self.scene.new_pipe(surfdata)
self.tbut_add_pipe.set_active(False)
elif self.tbut_add_gregory.get_active():
self.scene.new_gregory(surfdata)
self.tbut_add_gregory.set_active(False)
elif self.tbut_add_inter.get_active():
self.scene.new_curve_interpolation()
self.tbut_add_inter.set_active(False)
elif self.tbut_del_curve.get_active():
self.scene.delete_curve()
self.tbut_del_curve.set_active(False)
elif self.tbut_sel_curve.get_active():
self.scene.select_curve()
self.tbut_sel_curve.set_active(False)
elif self.tbut_cut.get_active():
self.scene.select_to_cut()
self.tbut_cut.set_active(False)
else:
self.scene.activate_cursor()
self.drawing_area.queue_draw()
def _on_mouse_motion(self, widget, data=None):
diff = map(op.sub, self.mouse_pos, (-data.x, data.y))
rowdiff = copy(diff)
if self.rbut_xy.get_active():
diff[2:2] = [0]
axis1 = (0, 1, 0)
axis2 = (1, 0, 0)
elif self.rbut_xz.get_active():
diff[1:1] = [0]
axis1 = (0, 0, 1)
axis2 = (1, 0, 0)
elif self.rbut_yz.get_active():
diff.reverse()
diff[0:0] = [0]
axis1 = (0, 1, 0)
axis2 = (0, 0, 1)
self.scene.mouse_move(rowdiff, diff, axis1, axis2)
self.mouse_pos = -data.x, data.y
self.drawing_area.queue_draw()
def init_glext(self):
# Query the OpenGL extension version.
# print "OpenGL extension version - %d.%d\n" % gtk.gdkgl.query_version()
# Configure OpenGL framebuffer.
# Try to get a double-buffered framebuffer configuration,
# if not successful then try to get a single-buffered one.
display_mode = gtk.gdkgl.MODE_RGB | gtk.gdkgl.MODE_DEPTH | gtk.gdkgl.MODE_DOUBLE
try:
glconfig = gtk.gdkgl.Config(mode=display_mode)
except gtk.gdkgl.NoMatches:
display_mode &= ~gtk.gdkgl.MODE_DOUBLE
glconfig = gtk.gdkgl.Config(mode=display_mode)
# print "is RGBA:", glconfig.is_rgba()
# print "is double-buffered:", glconfig.is_double_buffered()
# print "is stereo:", glconfig.is_stereo()
# print "has alpha:", glconfig.has_alpha()
# print "has depth buffer:", glconfig.has_depth_buffer()
# print "has stencil buffer:", glconfig.has_stencil_buffer()
# print "has accumulation buffer:", glconfig.has_accum_buffer()
# print
return glconfig
def on_win_main_destroy(self, widget, data=None):
gtk.main_quit()
def on_but_quit_clicked(self, widget, data=None):
gtk.main_quit()
def on_cbut_draw_bezier_pts_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BEZIER, Curve.POINTS)
self.drawing_area.queue_draw()
def on_cbut_draw_bezier_curves_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BEZIER, Curve.CURVE)
self.drawing_area.queue_draw()
def on_cbut_draw_bezier_polygons_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BEZIER, Curve.POLYGON)
self.drawing_area.queue_draw()
def on_cbut_draw_bspline_pts_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BSPLINE, Curve.POINTS)
self.drawing_area.queue_draw()
def on_cbut_draw_bspline_curves_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BSPLINE, Curve.CURVE)
self.drawing_area.queue_draw()
def on_cbut_draw_bspline_polygons_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BSPLINE, Curve.POLYGON)
self.drawing_area.queue_draw()
def on_sp_R_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.R = widget.get_value()
self.scene.torus.refresh()
self.drawing_area.queue_draw()
def on_sp_r_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.r = widget.get_value()
self.scene.torus.refresh()
self.drawing_area.queue_draw()
def on_sp_N_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.N = widget.get_value()
self.scene.torus.refresh()
self.drawing_area.queue_draw()
def on_sp_n_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.n = widget.get_value()
self.scene.torus.refresh()
self.drawing_area.queue_draw()
def on_sp_near_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.P0 = -widget.get_value()
self.scene.set_near(widget.get_value())
self.drawing_area.queue_draw()
def on_sp_fov_value_changed(self, widget, data=None):
self.scene.set_fov(widget.get_value())
self.drawing_area.queue_draw()
def on_chbut_3d_toggled(self, widget, data=None):
if self.box3d.get_property("visible"):
self.box3d.hide()
self.scene.set_drawmode(Scene.DRAW2D)
else:
self.box3d.show()
self.scene.set_drawmode(Scene.DRAW3D)
self.drawing_area.queue_draw()
def on_colbut_right_color_set(self, widget, data=None):
CMAX = 65535.0
c = widget.get_color()
c = (c.red / CMAX, c.green / CMAX, c.blue / CMAX)
self.scene.set_right_color(c)
self.drawing_area.queue_draw()
def on_colbut_left_color_set(self, widget, data=None):
CMAX = 65535.0
c = widget.get_color()
c = (c.red / CMAX, c.green / CMAX, c.blue / CMAX)
self.scene.set_left_color(c)
self.drawing_area.queue_draw()
def on_hs_3d_value_changed(self, widget, data=None):
v = widget.get_value() / 2.0
self.scene.set_eyes_split(v)
self.drawing_area.queue_draw()
def on_rbut_none_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.NONE)
def on_rbut_cursor_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.CURSOR)
def on_rbut_scale_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.SCALE)
def on_rbut_isoscale_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.ISOSCALE)
def on_rbut_trans_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.TRANSLATE)
def on_rbut_rotate_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.ROTATE)
def on_rbut_camera_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.CAMERA)
def on_rbut_pnt_add_bezier_pressed(self, widget, data=None):
self.scene.set_cursormode(Scene.PNTBZADD)
def on_rbut_pnt_add_bspline_pressed(self, widget, data=None):
self.scene.set_cursormode(Scene.PNTBSADD)
def on_rbut_pnt_del_pressed(self, widget, data=None):
self.scene.set_cursormode(Scene.PNTDEL)
def on_rbut_pnt_edit_pressed(self, widget, data=None):
self.scene.set_cursormode(Scene.PNTEDIT)
def on_rbut_edit_point_toggled(self, widget, data=None):
self.scene.set_editmode(Points.PNT)
def on_rbut_edit_row_toggled(self, widget, data=None):
self.scene.set_editmode(Points.ROW)
def on_rbut_edit_column_toggled(self, widget, data=None):
self.scene.set_editmode(Points.COL)
def on_rbut_edit_symetric_toggled(self, widget, data=None):
self.scene.set_editmode(Points.SYM)
def on_rbut_gap_toggled(self, widget, data=None):
self.scene.fill_gap(None)
self.drawing_area.queue_draw()
def on_rbut_gap_c0_toggled(self, widget, data=None):
self.scene.fill_gap(Scene.C0)
self.drawing_area.queue_draw()
def on_rbut_gap_c1_toggled(self, widget, data=None):
self.scene.fill_gap(Scene.C1)
self.drawing_area.queue_draw()
def on_rbut_gap_c2_toggled(self, widget, data=None):
self.scene.fill_gap(Scene.C2)
self.drawing_area.queue_draw()
def on_but_pos_appyly_clicked(self, widget, data=None):
self.scene.set_lookat(
(self.sp_pos_x.get_value(), self.sp_pos_y.get_value(), self.sp_pos_z.get_value()),
(self.sp_look_x.get_value(), self.sp_look_y.get_value(), self.sp_look_z.get_value()),
)
self.drawing_area.queue_draw()
def on_but_cut_clicked(self, widget, data=None):
A, B = self.scene.cut_current(
(
self.sp_cut_u1.get_value(),
self.sp_cut_v2.get_value(),
self.sp_cut_u2.get_value(),
self.sp_cut_v2.get_value(),
),
self.sp_delta.get_value(),
)
fm = self.cbox_first.get_model()
sm = self.cbox_second.get_model()
fm.clear()
for a in range(A):
fm.append((str(a),))
sm.clear()
for b in range(B):
sm.append((str(b),))
self.drawing_area.queue_draw()
def on_but_cut_clear_clicked(self, widget, data=None):
self.scene.clear_cut()
self.drawing_area.queue_draw()
def on_cbox_first_changed(self, widget, data=None):
txt = widget.get_active_text()
if txt == None:
i = None
else:
i = int(txt)
self.scene.cut_select(0, i)
self.drawing_area.queue_draw()
def on_cbox_second_changed(self, widget, data=None):
txt = widget.get_active_text()
if txt == None:
i = None
else:
i = int(txt)
self.scene.cut_select(1, i)
self.drawing_area.queue_draw()
def on_mitem_load_activate(self, widget, data=None):
if self.win_dia_load.run() == gtk.RESPONSE_OK:
self.save_file = self.win_dia_load.get_filename()
self.scene.load_from_file(self.save_file)
self.drawing_area.queue_draw()
self.win_dia_load.hide()
def on_mitem_new_activate(self, widget, data=None):
self.scene.clear()
self.drawing_area.queue_draw()
def on_mitem_save_activate(self, widget, data=None):
if self.save_file != None:
self.scene.dump_to_file(self.save_file)
else:
self.on_mitem_saveas_activate(widget, data)
def on_mitem_saveas_activate(self, widget, data=None):
if self.win_dia_save.run() == gtk.RESPONSE_OK:
self.save_file = self.win_dia_save.get_filename()
self.scene.dump_to_file(self.save_file)
self.win_dia_save.hide()
def on_but_gen_clicked(self, widge, data=None):
self.scene.gen_paths()
def on_but_dump_clicked(self, widge, data=None):
self.scene.dump_sign()
def on_show(self, widget, data=None):
widget.show_all()
def on_hide(self, widget, data=None):
widget.hide()
return True
class App(object):
"""Application main class"""
def __init__(self):
self.move = [0,0,0]
self.dirskeys = ( ( ['w'] , ['s'] ) , ( ['a'] , ['d'] ) , ( ['e'] , ['q'] ) )
for d in self.dirskeys :
for e in d :
for i in range(len(e)) : e[i] = ( gtk.gdk.unicode_to_keyval(ord(e[i])) , False )
self.near = 1
self.far = 1000
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.MOTION_NOTIFY | gtk.gdk.BUTTON1_MOTION_MASK | gtk.gdk.BUTTON3_MOTION_MASK )
self.drawing_area.set_size_request(320,240)
builder.get_object("vbox1").pack_start(self.drawing_area)
win_main = builder.get_object("win_main")
self.sp_l1 = builder.get_object("sp_l1")
self.sp_l2 = builder.get_object("sp_l2")
win_main.set_events( gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK )
win_main.connect('key-press-event' , self._on_key_pressed )
win_main.connect('key-release-event', self._on_key_released )
self.scene = Scene( )
self.scene.set_robot_params( self.sp_l1.get_value() , self.sp_l2.get_value() )
self.drawing_area.add( self.scene , ( 0,0,1,1) )
# self.drawing_area.add( self.scene , (.5,0,.5,1) )
print 'Scene added'
win_main.show_all()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
self.scene.set_screen_size( width , height )
builder.connect_signals(self)
# self.statbar = builder.get_object('statbar')
self.drawing_area.connect('motion_notify_event',self._on_mouse_motion)
self.drawing_area.connect('button_press_event',self._on_button_pressed)
self.drawing_area.connect('button_release_event',self._on_button_released)
self.drawing_area.connect('configure_event',self._on_reshape)
self.drawing_area.connect_after('expose_event',self._after_draw)
gtk.timeout_add( 1 , self._refresh )
def _refresh( self ) :
self.drawing_area.queue_draw()
return True
def _after_draw( self , widget , data=None ) :
self.update_statusbar()
def update_statusbar( self ) :
pass
def _on_reshape( self , widget , data=None ) :
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene.set_screen_size( width , height )
def _on_button_pressed( self , widget , data=None ) :
if data.button in (1,3) :
self.mouse_pos = data.x , data.y
self.scene.mouse_but_pressed( data.button , (data.x,data.y) )
self.drawing_area.queue_draw()
def _on_button_released( self , widget , data=None ) :
self.scene.mouse_but_released( data.button , (data.x,data.y) )
self.drawing_area.queue_draw()
def _on_mouse_motion( self , widget , data=None ) :
diff = map( op.sub , self.mouse_pos , (data.x , data.y) )
self.scene.mouse_move( diff , (data.x,data.y) )
self.mouse_pos = data.x , data.y
self.drawing_area.queue_draw()
def _on_key_pressed( self , widget , data=None ) :
if not any(self.move) :
gtk.timeout_add( 20 , self._move_callback )
self.scene.key_pressed( gtk.gdk.keyval_to_unicode( data.keyval ) )
for i in range(len(self.dirskeys)) :
if (data.keyval,False) in self.dirskeys[i][0] :
self.dirskeys[i][0][ self.dirskeys[i][0].index( (data.keyval,False) ) ] = (data.keyval,True)
self.move[i]+= 1
elif (data.keyval,False) in self.dirskeys[i][1] :
self.dirskeys[i][1][ self.dirskeys[i][1].index( (data.keyval,False) ) ] = (data.keyval,True)
self.move[i]-= 1
def _on_key_released( self , widget , data=None ) :
for i in range(len(self.dirskeys)) :
if (data.keyval,True) in self.dirskeys[i][0] :
self.dirskeys[i][0][ self.dirskeys[i][0].index( (data.keyval,True) ) ] = (data.keyval,False)
self.move[i]-= 1
elif (data.keyval,True) in self.dirskeys[i][1] :
self.dirskeys[i][1][ self.dirskeys[i][1].index( (data.keyval,True) ) ] = (data.keyval,False)
self.move[i]+= 1
def _move_callback( self ) :
self.drawing_area.queue_draw()
return any(self.move)
def on_start_set( self , widget , data=None ) :
self.scene.set_start_point()
def on_finish_set( self , widget , data=None ) :
self.scene.set_finish_point()
def on_robot_changed( self , widget , data=None ) :
self.scene.set_robot_params( self.sp_l1.get_value() , self.sp_l2.get_value() )
def init_glext(self):
# Query the OpenGL extension version.
# print "OpenGL extension version - %d.%d\n" % gtk.gdkgl.query_version()
# Configure OpenGL framebuffer.
# Try to get a double-buffered framebuffer configuration,
# if not successful then try to get a single-buffered one.
display_mode = (
gtk.gdkgl.MODE_RGB |
gtk.gdkgl.MODE_DEPTH |
gtk.gdkgl.MODE_STENCIL|
gtk.gdkgl.MODE_DOUBLE )
try:
glconfig = gtk.gdkgl.Config(mode=display_mode)
except gtk.gdkgl.NoMatches:
display_mode &= ~gtk.gdkgl.MODE_DOUBLE
glconfig = gtk.gdkgl.Config(mode=display_mode)
# print "is RGBA:", glconfig.is_rgba()
# print "is double-buffered:", glconfig.is_double_buffered()
# print "is stereo:", glconfig.is_stereo()
# print "has alpha:", glconfig.has_alpha()
# print "has depth buffer:", glconfig.has_depth_buffer()
# print "has stencil buffer:", glconfig.has_stencil_buffer()
# print "has accumulation buffer:", glconfig.has_accum_buffer()
# print
return glconfig
def on_quit(self,widget,data=None):
gtk.main_quit()
def on_anim_toggle(self,widget,data=None) :
self.scene.anim_toggle()
def on_anim_recalculate(self,widget,data=None) :
self.scene.anim_recalc()
def on_anim_reset(self,widget,data=None) :
self.scene.anim_reset()
class App(object):
"""Application main class"""
def __init__(self):
self.button = {}
self.move = [0,0,0]
self.dirskeys = ( ( ['w'] , ['s'] ) , ( ['a'] , ['d'] ) , ( ['e'] , ['q'] ) )
for d in self.dirskeys :
for e in d :
for i in range(len(e)) : e[i] = ( gtk.gdk.unicode_to_keyval(ord(e[i])) , False )
self.near = 1
self.far = 1000
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON1_MOTION_MASK | gtk.gdk.BUTTON2_MOTION_MASK |gtk.gdk.BUTTON3_MOTION_MASK )
self.drawing_area.set_size_request(320,240)
builder.get_object("vbox1").pack_start(self.drawing_area)
win_main = builder.get_object("win_main")
self.save_diag = builder.get_object("save_dialog")
self.open_diag = builder.get_object("open_dialog")
self.leuler = builder.get_object("leuler")
self.lquaternion = builder.get_object("lquaternion")
win_main.set_events( gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK )
win_main.connect('key-press-event' , self._on_key_pressed )
win_main.connect('key-release-event', self._on_key_released )
self.qscene = QuaternionsScene( 10.0 , self.fov , .01 , self.near , self.far )
self.escene = EulersScene( 10.0 , self.fov , .01 , self.near , self.far )
self.drawing_area.add( self.escene , (0, 0,1,.5) )
self.drawing_area.add( self.qscene , (0,.5,1,.5) )
self.qscene.set_matrix( tr.identity_matrix() , tr.identity_matrix() )
self.escene.set_matrix( tr.identity_matrix() , tr.identity_matrix() )
print 'Scene added'
win_main.show_all()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height / 2.0
ratio = float(width)/float(height)
self.qscene.set_ratio( ratio )
self.escene.set_ratio( ratio )
builder.connect_signals(self)
# self.statbar = builder.get_object('statbar')
self.drawing_area.connect('motion_notify_event',self._on_mouse_motion)
self.drawing_area.connect('button_press_event',self._on_button_pressed)
self.drawing_area.connect('button_release_event',self._on_button_released)
self.drawing_area.connect('configure_event',self._on_reshape)
self.drawing_area.connect_after('expose_event',self._after_draw)
self.anim = False
gtk.timeout_add( 1 , self._refresh )
def on_anim_toggle( self , widget , data=None ) :
if self.anim :
self.on_stop()
else :
self.on_start()
def on_start( self ) :
self.anim = True
def on_stop( self ) :
self.anim = False
def _refresh( self ) :
self.leuler.set_text( str(self.escene) )
self.lquaternion.set_text( str(self.qscene) )
self.qscene.step( self.anim )
self.escene.step( self.anim )
self.drawing_area.queue_draw()
return True
def _after_draw( self , widget , data=None ) :
self.update_statusbar()
def update_statusbar( self ) :
pass
def _on_reshape( self , widget , data=None ) :
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height / 2.0
ratio = float(width)/float(height)
self.qscene.set_screen_size( width , height )
self.escene.set_screen_size( width , height )
def _on_button_pressed( self , widget , data=None ) :
if data.button == 1 or data.button == 2 or data.button == 3 :
self.mouse_pos = data.x , data.y
self.button[data.button] = True
self._refresh()
def _on_button_released( self , widget , data=None ) :
self.button[data.button] = False
def _on_mouse_motion( self , widget , data=None ) :
diff = map( op.sub , self.mouse_pos , (data.x , data.y) )
self.qscene.mouse_move( diff , self.button )
self.escene.mouse_move( diff , self.button )
self.mouse_pos = data.x , data.y
self._refresh()
# gtk.gdk.Keymap
def _on_key_pressed( self , widget , data=None ) :
if not any(self.move) :
gtk.timeout_add( 20 , self._move_callback )
for i in range(len(self.dirskeys)) :
if (data.keyval,False) in self.dirskeys[i][0] :
self.dirskeys[i][0][ self.dirskeys[i][0].index( (data.keyval,False) ) ] = (data.keyval,True)
self.move[i]+= 1
elif (data.keyval,False) in self.dirskeys[i][1] :
self.dirskeys[i][1][ self.dirskeys[i][1].index( (data.keyval,False) ) ] = (data.keyval,True)
self.move[i]-= 1
def _on_key_released( self , widget , data=None ) :
for i in range(len(self.dirskeys)) :
if (data.keyval,True) in self.dirskeys[i][0] :
self.dirskeys[i][0][ self.dirskeys[i][0].index( (data.keyval,True) ) ] = (data.keyval,False)
self.move[i]-= 1
elif (data.keyval,True) in self.dirskeys[i][1] :
self.dirskeys[i][1][ self.dirskeys[i][1].index( (data.keyval,True) ) ] = (data.keyval,False)
self.move[i]+= 1
def _move_callback( self ) :
self.qscene.key_pressed( self.move )
self.escene.key_pressed( self.move )
self._refresh()
return any(self.move)
def init_glext(self):
# Query the OpenGL extension version.
# print "OpenGL extension version - %d.%d\n" % gtk.gdkgl.query_version()
# Configure OpenGL framebuffer.
# Try to get a double-buffered framebuffer configuration,
# if not successful then try to get a single-buffered one.
display_mode = (
gtk.gdkgl.MODE_RGB |
gtk.gdkgl.MODE_DEPTH |
gtk.gdkgl.MODE_STENCIL|
gtk.gdkgl.MODE_DOUBLE )
try:
glconfig = gtk.gdkgl.Config(mode=display_mode)
except gtk.gdkgl.NoMatches:
display_mode &= ~gtk.gdkgl.MODE_DOUBLE
glconfig = gtk.gdkgl.Config(mode=display_mode)
# print "is RGBA:", glconfig.is_rgba()
# print "is double-buffered:", glconfig.is_double_buffered()
# print "is stereo:", glconfig.is_stereo()
# print "has alpha:", glconfig.has_alpha()
# print "has depth buffer:", glconfig.has_depth_buffer()
# print "has stencil buffer:", glconfig.has_stencil_buffer()
# print "has accumulation buffer:", glconfig.has_accum_buffer()
# print
return glconfig
def on_reload( self , widget , data=None ):
self.qscene.reset()
self.escene.reset()
def on_quit(self,widget,data=None):
gtk.main_quit()
def on_open( self, widget , data=None ):
chooser = gtk.FileChooserDialog(
title=None,
action=gtk.FILE_CHOOSER_ACTION_SAVE,
buttons=(gtk.STOCK_CANCEL,gtk.RESPONSE_CANCEL,gtk.STOCK_OPEN,gtk.RESPONSE_OK)
)
chooser.set_current_folder( os.path.dirname(sys.argv[0]) )
filter = gtk.FileFilter()
filter.set_name('Numpy dumps')
filter.add_pattern('*.npy')
filter.add_pattern('*.npz')
chooser.add_filter(filter)
if chooser.run() == gtk.RESPONSE_OK :
arr = np.load(chooser.get_filename())
self.qscene.set_matrix( arr['begin'] , arr['end'] )
self.escene.set_matrix( arr['begin'] , arr['end'] )
chooser.destroy()
def on_save( self, widget , data=None ):
chooser = gtk.FileChooserDialog(
title=None,
action=gtk.FILE_CHOOSER_ACTION_SAVE,
buttons=(gtk.STOCK_CANCEL,gtk.RESPONSE_CANCEL,gtk.STOCK_SAVE,gtk.RESPONSE_OK)
)
chooser.set_current_folder( os.path.dirname(sys.argv[0]) )
chooser.set_do_overwrite_confirmation(True)
if chooser.run() == gtk.RESPONSE_OK :
np.savez(chooser.get_filename(),
begin = self.qscene.get_beg_matrix(),
end = self.qscene.get_end_matrix() )
chooser.destroy()
def __init__(self):
self.move = [0,0,0]
self.dirskeys = ( ( ['w'] , ['s'] ) , ( ['a'] , ['d'] ) , ( ['e'] , ['q'] ) )
for d in self.dirskeys :
for e in d :
for i in range(len(e)) : e[i] = ( gtk.gdk.unicode_to_keyval(ord(e[i])) , False )
self.near = 1
self.far = 1000
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON3_MOTION_MASK )
self.drawing_area.set_size_request(640,480)
builder.get_object("vbox1").pack_start(self.drawing_area)
self.tb_run = builder.get_object("tb_run")
self.sp_xb = builder.get_object("sp_xb")
self.sp_xe = builder.get_object("sp_xe")
self.sp_yb = builder.get_object("sp_yb")
self.sp_ye = builder.get_object("sp_ye")
self.sp_zb = builder.get_object("sp_zb")
self.sp_ze = builder.get_object("sp_ze")
self.sp_ms = builder.get_object("sp_miller_size")
self.cb_m = builder.get_object("cb_miller")
self.pbar = builder.get_object("pbar")
win_main = builder.get_object("win_main")
win_main.set_events( gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK )
self.nosetdrill = False
win_main.connect('key-press-event' , self._on_key_pressed )
win_main.connect('key-release-event', self._on_key_released )
win_main.show_all()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene = Scene( self.fov , ratio , self.near , self.far , meshes )
self.scene.set_speed( builder.get_object("sp_spd").get_value() )
self.drawing_area.add( self.scene )
builder.connect_signals(self)
self.drawing_area.connect('motion_notify_event',self._on_mouse_motion)
self.drawing_area.connect('button_press_event',self._on_button_pressed)
self.drawing_area.connect('configure_event',self._on_reshape)
self.drawing_area.connect_after('expose_event',self._after_draw)
gtk.timeout_add( 1 , self._refresh )
self.pbar.set_visible(False)
class App(object):
"""Application main class"""
def __init__(self):
self.move = [0,0,0]
self.dirskeys = ( ( ['w'] , ['s'] ) , ( ['a'] , ['d'] ) , ( ['e'] , ['q'] ) )
for d in self.dirskeys :
for e in d :
for i in range(len(e)) : e[i] = ( gtk.gdk.unicode_to_keyval(ord(e[i])) , False )
self.near = 1
self.far = 1000
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON3_MOTION_MASK )
self.drawing_area.set_size_request(640,480)
builder.get_object("vbox1").pack_start(self.drawing_area)
self.tb_run = builder.get_object("tb_run")
self.sp_xb = builder.get_object("sp_xb")
self.sp_xe = builder.get_object("sp_xe")
self.sp_yb = builder.get_object("sp_yb")
self.sp_ye = builder.get_object("sp_ye")
self.sp_zb = builder.get_object("sp_zb")
self.sp_ze = builder.get_object("sp_ze")
self.sp_ms = builder.get_object("sp_miller_size")
self.cb_m = builder.get_object("cb_miller")
self.pbar = builder.get_object("pbar")
win_main = builder.get_object("win_main")
win_main.set_events( gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK )
self.nosetdrill = False
win_main.connect('key-press-event' , self._on_key_pressed )
win_main.connect('key-release-event', self._on_key_released )
win_main.show_all()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene = Scene( self.fov , ratio , self.near , self.far , meshes )
self.scene.set_speed( builder.get_object("sp_spd").get_value() )
self.drawing_area.add( self.scene )
builder.connect_signals(self)
self.drawing_area.connect('motion_notify_event',self._on_mouse_motion)
self.drawing_area.connect('button_press_event',self._on_button_pressed)
self.drawing_area.connect('configure_event',self._on_reshape)
self.drawing_area.connect_after('expose_event',self._after_draw)
gtk.timeout_add( 1 , self._refresh )
self.pbar.set_visible(False)
def _refresh( self ) :
self.drawing_area.queue_draw()
return True
def _after_draw( self , widget , data=None ) :
self.update_statusbar()
def update_statusbar( self ) :
pass
def _on_reshape( self , widget , data=None ) :
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene.set_screen_size( width , height )
def _on_button_pressed( self , widget , data=None ) :
if data.button == 3 :
self.mouse_pos = data.x , data.y
self.drawing_area.queue_draw()
def _on_mouse_motion( self , widget , data=None ) :
diff = map( op.sub , self.mouse_pos , (data.x , data.y) )
self.scene.mouse_move( diff )
self.mouse_pos = data.x , data.y
self.drawing_area.queue_draw()
# gtk.gdk.Keymap
def _on_key_pressed( self , widget , data=None ) :
if not any(self.move) :
gtk.timeout_add( 20 , self._move_callback )
for i in range(len(self.dirskeys)) :
if (data.keyval,False) in self.dirskeys[i][0] :
self.dirskeys[i][0][ self.dirskeys[i][0].index( (data.keyval,False) ) ] = (data.keyval,True)
self.move[i]+= 1
elif (data.keyval,False) in self.dirskeys[i][1] :
self.dirskeys[i][1][ self.dirskeys[i][1].index( (data.keyval,False) ) ] = (data.keyval,True)
self.move[i]-= 1
def _on_key_released( self , widget , data=None ) :
for i in range(len(self.dirskeys)) :
if (data.keyval,True) in self.dirskeys[i][0] :
self.dirskeys[i][0][ self.dirskeys[i][0].index( (data.keyval,True) ) ] = (data.keyval,False)
self.move[i]-= 1
elif (data.keyval,True) in self.dirskeys[i][1] :
self.dirskeys[i][1][ self.dirskeys[i][1].index( (data.keyval,True) ) ] = (data.keyval,False)
self.move[i]+= 1
def _move_callback( self ) :
self.scene.key_pressed( self.move )
self.drawing_area.queue_draw()
return any(self.move)
def on_run_pause( self , wdg , data =None ) :
if self.scene.running :
self.scene.sim_stop()
else :
self.scene.sim_run()
def on_reset( self , wdg , data=None ) :
drill = self.scene.reset()
self.tb_run.set_active(False)
self.set_drill( drill )
def on_load( self , wdg , data=None ) :
self.set_drill( self.scene.load_path( wdg.get_filename() ) )
def set_drill( self , drill ) :
self.nosetdrill = True
print drill
self.cb_m.set_active( drill[0] )
self.sp_ms.set_value( drill[1] )
self.nosetdrill = False
def on_speed_changed( self , wdg , data=None ) :
self.scene.set_speed( wdg.get_value() )
def on_prec_changed( self , wdg , data=None ) :
self.scene.set_precision( wdg.get_value_as_int() )
def on_size_changed( self , wdg , data=None ) :
self.scene.set_size(
( self.sp_xb.get_value()
, self.sp_yb.get_value()
, self.sp_zb.get_value() ) ,
( self.sp_xe.get_value()
, self.sp_ye.get_value()
, self.sp_ze.get_value() ) )
def on_miller_changed( self , wdg , data=None ) :
if not self.nosetdrill :
self.scene.reset_drill( (Parser.FLAT if self.cb_m.get_active() == 0 else Parser.ROUND , self.sp_ms.get_value_as_int() ) )
def on_miller_length_changed( self , wdg , data=None ) :
self.scene.set_drill_len( wdg.get_value() )
def on_fast_cut( self , wdg , data=None ) :
self.scene.fast_cut( self.pbar )
def init_glext(self):
display_mode = (
gtk.gdkgl.MODE_RGB |
gtk.gdkgl.MODE_DEPTH |
gtk.gdkgl.MODE_STENCIL|
gtk.gdkgl.MODE_DOUBLE )
try:
glconfig = gtk.gdkgl.Config(mode=display_mode)
except gtk.gdkgl.NoMatches:
display_mode &= ~gtk.gdkgl.MODE_DOUBLE
glconfig = gtk.gdkgl.Config(mode=display_mode)
return glconfig
def on_win_main_destroy(self,widget,data=None):
gtk.main_quit()
def on_but_quit_clicked(self,widget,data=None):
gtk.main_quit()
class App(object):
"""Application main class"""
def __init__(self):
self.button = {}
self.move = [0, 0, 0]
self.dirskeys = ((['w'], ['s']), (['a'], ['d']), (['e'], ['q']))
for d in self.dirskeys:
for e in d:
for i in range(len(e)):
e[i] = (gtk.gdk.unicode_to_keyval(ord(e[i])), False)
self.near = 1
self.far = 1000
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events(gtk.gdk.BUTTON_PRESS_MASK
| gtk.gdk.BUTTON_RELEASE_MASK
| gtk.gdk.BUTTON1_MOTION_MASK
| gtk.gdk.BUTTON2_MOTION_MASK
| gtk.gdk.BUTTON3_MOTION_MASK)
self.drawing_area.set_size_request(320, 240)
builder.get_object("vbox1").pack_end(self.drawing_area)
win_main = builder.get_object("win_main")
win_main.set_events(gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK)
win_main.connect('key-press-event', self._on_key_pressed)
win_main.connect('key-release-event', self._on_key_released)
self.scene = Scene(self.fov, .01, self.near, self.far)
self.drawing_area.add(self.scene, (0, 0, 1, 1))
print 'Scene added'
win_main.show_all()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width) / float(height)
self.scene.set_ratio(ratio)
builder.connect_signals(self)
self.drawing_area.connect('motion_notify_event', self._on_mouse_motion)
self.drawing_area.connect('button_press_event',
self._on_button_pressed)
self.drawing_area.connect('button_release_event',
self._on_button_released)
self.drawing_area.connect('configure_event', self._on_reshape)
self.drawing_area.connect_after('expose_event', self._after_draw)
gobject.timeout_add(10, self._refresh)
def _refresh(self):
self.drawing_area.queue_draw()
return True
def _after_draw(self, widget, data=None):
self.update_statusbar()
def update_statusbar(self):
pass
def _on_reshape(self, widget, data=None):
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width) / float(height)
self.scene.set_screen_size(width, height)
def _on_button_pressed(self, widget, data=None):
if data.button == 1 or data.button == 2 or data.button == 3:
self.mouse_pos = data.x, data.y
self.button[data.button] = True
self.drawing_area.queue_draw()
def _on_button_released(self, widget, data=None):
self.button[data.button] = False
def _on_mouse_motion(self, widget, data=None):
diff = map(op.sub, self.mouse_pos, (data.x, data.y))
self.scene.mouse_move(diff, self.button)
self.mouse_pos = data.x, data.y
self.drawing_area.queue_draw()
def _on_key_pressed(self, widget, data=None):
if not any(self.move):
gtk.timeout_add(20, self._move_callback)
for i in range(len(self.dirskeys)):
if (data.keyval, False) in self.dirskeys[i][0]:
self.dirskeys[i][0][self.dirskeys[i][0].index(
(data.keyval, False))] = (data.keyval, True)
self.move[i] += 1
elif (data.keyval, False) in self.dirskeys[i][1]:
self.dirskeys[i][1][self.dirskeys[i][1].index(
(data.keyval, False))] = (data.keyval, True)
self.move[i] -= 1
def _on_key_released(self, widget, data=None):
for i in range(len(self.dirskeys)):
if (data.keyval, True) in self.dirskeys[i][0]:
self.dirskeys[i][0][self.dirskeys[i][0].index(
(data.keyval, True))] = (data.keyval, False)
self.move[i] -= 1
elif (data.keyval, True) in self.dirskeys[i][1]:
self.dirskeys[i][1][self.dirskeys[i][1].index(
(data.keyval, True))] = (data.keyval, False)
self.move[i] += 1
def _move_callback(self):
self.scene.key_pressed(self.move)
self.drawing_area.queue_draw()
return any(self.move)
def init_glext(self):
# Query the OpenGL extension version.
print "OpenGL extension version - %d.%d\n" % gtk.gdkgl.query_version()
# Configure OpenGL framebuffer.
# Try to get a double-buffered framebuffer configuration,
# if not successful then try to get a single-buffered one.
display_mode = (gtk.gdkgl.MODE_RGB | gtk.gdkgl.MODE_DEPTH
| gtk.gdkgl.MODE_DOUBLE)
try:
glconfig = gtk.gdkgl.Config(mode=display_mode)
except gtk.gdkgl.NoMatches:
display_mode &= ~gtk.gdkgl.MODE_DOUBLE
glconfig = gtk.gdkgl.Config(mode=display_mode)
print "is RGBA:", glconfig.is_rgba()
print "is double-buffered:", glconfig.is_double_buffered()
print "is stereo:", glconfig.is_stereo()
print "has alpha:", glconfig.has_alpha()
print "has depth buffer:", glconfig.has_depth_buffer()
print "has stencil buffer:", glconfig.has_stencil_buffer()
print "has accumulation buffer:", glconfig.has_accum_buffer()
print
return glconfig
def on_win_main_destroy(self, widget, data=None):
gtk.main_quit()
def on_but_quit_clicked(self, widget, data=None):
gtk.main_quit()
def on_cb_wireframe_toggled(self, widget, data=None):
self.scene.toggle_wireframe()
def on_cb_solid_toggled(self, widget, data=None):
self.scene.toggle_solid()
def on_cb_gravity_toggled(self, widget, data=None):
self.scene.toggle_gravity()
def on_trace_value_changed(self, widget, data=None):
self.scene.set_trace_len(widget.get_value())
def on_set_dens(self, widget, data=None):
self.scene.set_dens(widget.get_value())
def on_set_x(self, widget, data=None):
self.scene.set_x(widget.get_value())
def on_set_y(self, widget, data=None):
self.scene.set_y(widget.get_value())
def on_set_z(self, widget, data=None):
self.scene.set_z(widget.get_value())
def on_set_a(self, widget, data=None):
self.scene.set_a(widget.get_value())
def on_set_w(self, widget, data=None):
self.scene.set_w(widget.get_value())
def on_reset(self, widget, data=None):
self.scene.reset()
def __init__(self):
self._init_keyboard()
self.near = 1
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events(gtk.gdk.BUTTON_PRESS_MASK
| gtk.gdk.BUTTON_RELEASE_MASK
| gtk.gdk.BUTTON1_MOTION_MASK)
self.drawing_area.set_size_request(800, 800)
self.box3d = builder.get_object("hbox_3d")
builder.get_object("vbox4").pack_start(self.drawing_area)
win_main = builder.get_object("win_main")
win_main.set_events(gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK)
win_main.connect('key-press-event', self._on_key_pressed)
win_main.connect('key-release-event', self._on_key_released)
win_main.show_all()
self.box3d.hide()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width) / float(height)
self.scene = Scene(self.fov, ratio, self.near)
self.drawing_area.add(self.scene)
builder.connect_signals(self)
self.statbar = builder.get_object('statbar')
self.drawing_area.connect('motion_notify_event', self._on_mouse_motion)
self.drawing_area.connect('button_press_event',
self._on_button_pressed)
self.drawing_area.connect('configure_event', self._on_reshape)
self.drawing_area.connect_after('expose_event', self._after_draw)
self.rbut_trans = builder.get_object('rbut_trans')
self.rbut_isoscale = builder.get_object('rbut_isoscale')
self.rbut_scale = builder.get_object('rbut_scale')
self.rbut_rotate = builder.get_object('rbut_rotate')
self.rbut_xy = builder.get_object('rbut_xy')
self.rbut_xz = builder.get_object('rbut_xz')
self.rbut_yz = builder.get_object('rbut_yz')
self.sp_fov = builder.get_object('sp_fov')
self.sp_fov.set_value(self.fov)
self.sp_near = builder.get_object('sp_near')
self.sp_near.set_value(self.near)
self.sp_pos_x = builder.get_object('sp_pos_x')
self.sp_pos_y = builder.get_object('sp_pos_y')
self.sp_pos_z = builder.get_object('sp_pos_z')
self.sp_look_x = builder.get_object('sp_look_x')
self.sp_look_y = builder.get_object('sp_look_y')
self.sp_look_z = builder.get_object('sp_look_z')
self.on_but_pos_appyly_clicked(None)
self.tbut_add_c0 = builder.get_object('tbut_add_c0')
self.tbut_add_c2 = builder.get_object('tbut_add_c2')
self.tbut_add_inter = builder.get_object('tbut_add_interpolation')
self.tbut_del_curve = builder.get_object('tbut_del_curve')
self.tbut_sel_curve = builder.get_object('tbut_sel_curve')
self.tbut_add_surf_c0 = builder.get_object('tbut_add_surf_c0')
self.tbut_add_surf_c2 = builder.get_object('tbut_add_surf_c2')
self.tbut_add_pipe = builder.get_object('tbut_add_pipe')
self.tbut_add_gregory = builder.get_object('tbut_add_gregory')
self.tbut_cut = builder.get_object('tbut_cut_choose')
self.cbox_first = builder.get_object('cbox_first')
self.cbox_second = builder.get_object('cbox_second')
self.sp_delta = builder.get_object('sp_cut_delta')
self.sp_cut_u1 = builder.get_object('sp_cut_u1')
self.sp_cut_v1 = builder.get_object('sp_cut_v1')
self.sp_cut_u2 = builder.get_object('sp_cut_u2')
self.sp_cut_v2 = builder.get_object('sp_cut_v2')
cell = gtk.CellRendererText()
self.cbox_first.pack_start(cell, True)
self.cbox_first.add_attribute(cell, 'text', 0)
cell = gtk.CellRendererText()
self.cbox_second.pack_start(cell, True)
self.cbox_second.add_attribute(cell, 'text', 0)
self.tbuts = [
self.tbut_add_c0, self.tbut_add_c2, self.tbut_add_inter,
self.tbut_del_curve, self.tbut_sel_curve, self.tbut_add_surf_c0,
self.tbut_add_surf_c2, self.tbut_add_pipe, self.tbut_add_gregory,
self.tbut_cut
]
self.sp_surf_x = builder.get_object('sp_surf_x')
self.sp_surf_y = builder.get_object('sp_surf_y')
self.sp_draw_surf_x = builder.get_object('sp_draw_surf_x')
self.sp_draw_surf_y = builder.get_object('sp_draw_surf_y')
self.win_dia_load = builder.get_object('win_dia_load')
self.win_dia_save = builder.get_object('win_dia_save')
if os.path.isdir('../data/'):
self.win_dia_load.set_current_folder('../data/')
self.win_dia_save.set_current_folder('../data/')
self.win_dia_load.set_transient_for(win_main)
self.win_dia_load.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
self.win_dia_load.add_button(gtk.STOCK_OPEN, gtk.RESPONSE_OK)
self.win_dia_save.set_transient_for(win_main)
self.win_dia_save.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
self.win_dia_save.add_button(gtk.STOCK_SAVE, gtk.RESPONSE_OK)
self.save_file = None
class App(object):
"""Application main class"""
def __init__(self):
self._init_keyboard()
self.near = 1
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events(gtk.gdk.BUTTON_PRESS_MASK
| gtk.gdk.BUTTON_RELEASE_MASK
| gtk.gdk.BUTTON1_MOTION_MASK)
self.drawing_area.set_size_request(800, 800)
self.box3d = builder.get_object("hbox_3d")
builder.get_object("vbox4").pack_start(self.drawing_area)
win_main = builder.get_object("win_main")
win_main.set_events(gtk.gdk.KEY_PRESS_MASK | gtk.gdk.KEY_RELEASE_MASK)
win_main.connect('key-press-event', self._on_key_pressed)
win_main.connect('key-release-event', self._on_key_released)
win_main.show_all()
self.box3d.hide()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width) / float(height)
self.scene = Scene(self.fov, ratio, self.near)
self.drawing_area.add(self.scene)
builder.connect_signals(self)
self.statbar = builder.get_object('statbar')
self.drawing_area.connect('motion_notify_event', self._on_mouse_motion)
self.drawing_area.connect('button_press_event',
self._on_button_pressed)
self.drawing_area.connect('configure_event', self._on_reshape)
self.drawing_area.connect_after('expose_event', self._after_draw)
self.rbut_trans = builder.get_object('rbut_trans')
self.rbut_isoscale = builder.get_object('rbut_isoscale')
self.rbut_scale = builder.get_object('rbut_scale')
self.rbut_rotate = builder.get_object('rbut_rotate')
self.rbut_xy = builder.get_object('rbut_xy')
self.rbut_xz = builder.get_object('rbut_xz')
self.rbut_yz = builder.get_object('rbut_yz')
self.sp_fov = builder.get_object('sp_fov')
self.sp_fov.set_value(self.fov)
self.sp_near = builder.get_object('sp_near')
self.sp_near.set_value(self.near)
self.sp_pos_x = builder.get_object('sp_pos_x')
self.sp_pos_y = builder.get_object('sp_pos_y')
self.sp_pos_z = builder.get_object('sp_pos_z')
self.sp_look_x = builder.get_object('sp_look_x')
self.sp_look_y = builder.get_object('sp_look_y')
self.sp_look_z = builder.get_object('sp_look_z')
self.on_but_pos_appyly_clicked(None)
self.tbut_add_c0 = builder.get_object('tbut_add_c0')
self.tbut_add_c2 = builder.get_object('tbut_add_c2')
self.tbut_add_inter = builder.get_object('tbut_add_interpolation')
self.tbut_del_curve = builder.get_object('tbut_del_curve')
self.tbut_sel_curve = builder.get_object('tbut_sel_curve')
self.tbut_add_surf_c0 = builder.get_object('tbut_add_surf_c0')
self.tbut_add_surf_c2 = builder.get_object('tbut_add_surf_c2')
self.tbut_add_pipe = builder.get_object('tbut_add_pipe')
self.tbut_add_gregory = builder.get_object('tbut_add_gregory')
self.tbut_cut = builder.get_object('tbut_cut_choose')
self.cbox_first = builder.get_object('cbox_first')
self.cbox_second = builder.get_object('cbox_second')
self.sp_delta = builder.get_object('sp_cut_delta')
self.sp_cut_u1 = builder.get_object('sp_cut_u1')
self.sp_cut_v1 = builder.get_object('sp_cut_v1')
self.sp_cut_u2 = builder.get_object('sp_cut_u2')
self.sp_cut_v2 = builder.get_object('sp_cut_v2')
cell = gtk.CellRendererText()
self.cbox_first.pack_start(cell, True)
self.cbox_first.add_attribute(cell, 'text', 0)
cell = gtk.CellRendererText()
self.cbox_second.pack_start(cell, True)
self.cbox_second.add_attribute(cell, 'text', 0)
self.tbuts = [
self.tbut_add_c0, self.tbut_add_c2, self.tbut_add_inter,
self.tbut_del_curve, self.tbut_sel_curve, self.tbut_add_surf_c0,
self.tbut_add_surf_c2, self.tbut_add_pipe, self.tbut_add_gregory,
self.tbut_cut
]
self.sp_surf_x = builder.get_object('sp_surf_x')
self.sp_surf_y = builder.get_object('sp_surf_y')
self.sp_draw_surf_x = builder.get_object('sp_draw_surf_x')
self.sp_draw_surf_y = builder.get_object('sp_draw_surf_y')
self.win_dia_load = builder.get_object('win_dia_load')
self.win_dia_save = builder.get_object('win_dia_save')
if os.path.isdir('../data/'):
self.win_dia_load.set_current_folder('../data/')
self.win_dia_save.set_current_folder('../data/')
self.win_dia_load.set_transient_for(win_main)
self.win_dia_load.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
self.win_dia_load.add_button(gtk.STOCK_OPEN, gtk.RESPONSE_OK)
self.win_dia_save.set_transient_for(win_main)
self.win_dia_save.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
self.win_dia_save.add_button(gtk.STOCK_SAVE, gtk.RESPONSE_OK)
self.save_file = None
def _init_keyboard(self):
self.move = [0, 0, 0]
self.dirskeys = ((['w'], ['s']), (['a'], ['d']), (['e'], ['q']))
for d in self.dirskeys:
for e in d:
for i in range(len(e)):
e[i] = (gtk.gdk.unicode_to_keyval(ord(e[i])), False)
def _on_key_pressed(self, widget, data=None):
if not any(self.move):
gtk.timeout_add(20, self._move_callback)
for i in range(len(self.dirskeys)):
if (data.keyval, False) in self.dirskeys[i][0]:
self.dirskeys[i][0][self.dirskeys[i][0].index(
(data.keyval, False))] = (data.keyval, True)
self.move[i] += 1
elif (data.keyval, False) in self.dirskeys[i][1]:
self.dirskeys[i][1][self.dirskeys[i][1].index(
(data.keyval, False))] = (data.keyval, True)
self.move[i] -= 1
def _on_key_released(self, widget, data=None):
for i in range(len(self.dirskeys)):
if (data.keyval, True) in self.dirskeys[i][0]:
self.dirskeys[i][0][self.dirskeys[i][0].index(
(data.keyval, True))] = (data.keyval, False)
self.move[i] -= 1
elif (data.keyval, True) in self.dirskeys[i][1]:
self.dirskeys[i][1][self.dirskeys[i][1].index(
(data.keyval, True))] = (data.keyval, False)
self.move[i] += 1
def _move_callback(self):
self.scene.key_pressed(self.move)
self.drawing_area.queue_draw()
return any(self.move)
def _after_draw(self, widget, data=None):
self.update_statusbar()
def update_statusbar(self):
cid = self.statbar.get_context_id('cursor')
self.statbar.pop(cid)
self.statbar.push(
cid,
str(self.scene.get_cursor_pos()) + " " +
str(self.scene.get_cursor_screen_pos()))
def toggle_tbuts(self, omit=None):
for t in self.tbuts:
if t == omit: continue
t.set_active(False)
def on_tbut_add_toggled(self, widget, data=None):
if widget.get_active(): self.toggle_tbuts(widget)
def on_sp_draw_surf_value_changed(self, widget, data=None):
self.scene.set_surf_density((self.sp_draw_surf_x.get_value_as_int(),
self.sp_draw_surf_y.get_value_as_int()))
self.drawing_area.queue_draw()
def _on_reshape(self, widget, data=None):
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width) / float(height)
self.scene.set_screen_size(width, height)
self.scene.set_ratio(ratio)
def _on_button_pressed(self, widget, data=None):
surfdata = ((self.sp_surf_x.get_value_as_int(),
self.sp_surf_y.get_value_as_int()),
(self.sp_draw_surf_x.get_value_as_int(),
self.sp_draw_surf_y.get_value_as_int()))
if data.button == 1:
self.mouse_pos = -data.x, data.y
elif data.button == 3:
if self.tbut_add_c0.get_active():
self.scene.new_curve_c0()
self.tbut_add_c0.set_active(False)
elif self.tbut_add_c2.get_active():
self.scene.new_curve_c2()
self.tbut_add_c2.set_active(False)
elif self.tbut_add_surf_c0.get_active():
self.scene.new_surface_c0(surfdata)
self.tbut_add_surf_c0.set_active(False)
elif self.tbut_add_surf_c2.get_active():
self.scene.new_surface_c2(surfdata)
self.tbut_add_surf_c2.set_active(False)
elif self.tbut_add_pipe.get_active():
self.scene.new_pipe(surfdata)
self.tbut_add_pipe.set_active(False)
elif self.tbut_add_gregory.get_active():
self.scene.new_gregory(surfdata)
self.tbut_add_gregory.set_active(False)
elif self.tbut_add_inter.get_active():
self.scene.new_curve_interpolation()
self.tbut_add_inter.set_active(False)
elif self.tbut_del_curve.get_active():
self.scene.delete_curve()
self.tbut_del_curve.set_active(False)
elif self.tbut_sel_curve.get_active():
self.scene.select_curve()
self.tbut_sel_curve.set_active(False)
elif self.tbut_cut.get_active():
self.scene.select_to_cut()
self.tbut_cut.set_active(False)
else:
self.scene.activate_cursor()
self.drawing_area.queue_draw()
def _on_mouse_motion(self, widget, data=None):
diff = map(op.sub, self.mouse_pos, (-data.x, data.y))
rowdiff = copy(diff)
if self.rbut_xy.get_active():
diff[2:2] = [0]
axis1 = (0, 1, 0)
axis2 = (1, 0, 0)
elif self.rbut_xz.get_active():
diff[1:1] = [0]
axis1 = (0, 0, 1)
axis2 = (1, 0, 0)
elif self.rbut_yz.get_active():
diff.reverse()
diff[0:0] = [0]
axis1 = (0, 1, 0)
axis2 = (0, 0, 1)
self.scene.mouse_move(rowdiff, diff, axis1, axis2)
self.mouse_pos = -data.x, data.y
self.drawing_area.queue_draw()
def init_glext(self):
# Query the OpenGL extension version.
# print "OpenGL extension version - %d.%d\n" % gtk.gdkgl.query_version()
# Configure OpenGL framebuffer.
# Try to get a double-buffered framebuffer configuration,
# if not successful then try to get a single-buffered one.
display_mode = (gtk.gdkgl.MODE_RGB | gtk.gdkgl.MODE_DEPTH
| gtk.gdkgl.MODE_DOUBLE)
try:
glconfig = gtk.gdkgl.Config(mode=display_mode)
except gtk.gdkgl.NoMatches:
display_mode &= ~gtk.gdkgl.MODE_DOUBLE
glconfig = gtk.gdkgl.Config(mode=display_mode)
# print "is RGBA:", glconfig.is_rgba()
# print "is double-buffered:", glconfig.is_double_buffered()
# print "is stereo:", glconfig.is_stereo()
# print "has alpha:", glconfig.has_alpha()
# print "has depth buffer:", glconfig.has_depth_buffer()
# print "has stencil buffer:", glconfig.has_stencil_buffer()
# print "has accumulation buffer:", glconfig.has_accum_buffer()
# print
return glconfig
def on_win_main_destroy(self, widget, data=None):
gtk.main_quit()
def on_but_quit_clicked(self, widget, data=None):
gtk.main_quit()
def on_cbut_draw_bezier_pts_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BEZIER, Curve.POINTS)
self.drawing_area.queue_draw()
def on_cbut_draw_bezier_curves_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BEZIER, Curve.CURVE)
self.drawing_area.queue_draw()
def on_cbut_draw_bezier_polygons_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BEZIER, Curve.POLYGON)
self.drawing_area.queue_draw()
def on_cbut_draw_bspline_pts_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BSPLINE, Curve.POINTS)
self.drawing_area.queue_draw()
def on_cbut_draw_bspline_curves_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BSPLINE, Curve.CURVE)
self.drawing_area.queue_draw()
def on_cbut_draw_bspline_polygons_toggled(self, widget, data=None):
self.scene.toggle_curve(Curve.BSPLINE, Curve.POLYGON)
self.drawing_area.queue_draw()
def on_sp_R_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.R = widget.get_value()
self.scene.torus.refresh()
self.drawing_area.queue_draw()
def on_sp_r_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.r = widget.get_value()
self.scene.torus.refresh()
self.drawing_area.queue_draw()
def on_sp_N_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.N = widget.get_value()
self.scene.torus.refresh()
self.drawing_area.queue_draw()
def on_sp_n_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.n = widget.get_value()
self.scene.torus.refresh()
self.drawing_area.queue_draw()
def on_sp_near_value_changed(self, widget, data=None):
# FIXME: hardcoded torus
self.scene.torus.P0 = -widget.get_value()
self.scene.set_near(widget.get_value())
self.drawing_area.queue_draw()
def on_sp_fov_value_changed(self, widget, data=None):
self.scene.set_fov(widget.get_value())
self.drawing_area.queue_draw()
def on_chbut_3d_toggled(self, widget, data=None):
if self.box3d.get_property("visible"):
self.box3d.hide()
self.scene.set_drawmode(Scene.DRAW2D)
else:
self.box3d.show()
self.scene.set_drawmode(Scene.DRAW3D)
self.drawing_area.queue_draw()
def on_colbut_right_color_set(self, widget, data=None):
CMAX = 65535.0
c = widget.get_color()
c = (c.red / CMAX, c.green / CMAX, c.blue / CMAX)
self.scene.set_right_color(c)
self.drawing_area.queue_draw()
def on_colbut_left_color_set(self, widget, data=None):
CMAX = 65535.0
c = widget.get_color()
c = (c.red / CMAX, c.green / CMAX, c.blue / CMAX)
self.scene.set_left_color(c)
self.drawing_area.queue_draw()
def on_hs_3d_value_changed(self, widget, data=None):
v = widget.get_value() / 2.0
self.scene.set_eyes_split(v)
self.drawing_area.queue_draw()
def on_rbut_none_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.NONE)
def on_rbut_cursor_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.CURSOR)
def on_rbut_scale_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.SCALE)
def on_rbut_isoscale_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.ISOSCALE)
def on_rbut_trans_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.TRANSLATE)
def on_rbut_rotate_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.ROTATE)
def on_rbut_camera_pressed(self, widget, data=None):
self.scene.set_mousemode(Scene.CAMERA)
def on_rbut_pnt_add_bezier_pressed(self, widget, data=None):
self.scene.set_cursormode(Scene.PNTBZADD)
def on_rbut_pnt_add_bspline_pressed(self, widget, data=None):
self.scene.set_cursormode(Scene.PNTBSADD)
def on_rbut_pnt_del_pressed(self, widget, data=None):
self.scene.set_cursormode(Scene.PNTDEL)
def on_rbut_pnt_edit_pressed(self, widget, data=None):
self.scene.set_cursormode(Scene.PNTEDIT)
def on_rbut_edit_point_toggled(self, widget, data=None):
self.scene.set_editmode(Points.PNT)
def on_rbut_edit_row_toggled(self, widget, data=None):
self.scene.set_editmode(Points.ROW)
def on_rbut_edit_column_toggled(self, widget, data=None):
self.scene.set_editmode(Points.COL)
def on_rbut_edit_symetric_toggled(self, widget, data=None):
self.scene.set_editmode(Points.SYM)
def on_rbut_gap_toggled(self, widget, data=None):
self.scene.fill_gap(None)
self.drawing_area.queue_draw()
def on_rbut_gap_c0_toggled(self, widget, data=None):
self.scene.fill_gap(Scene.C0)
self.drawing_area.queue_draw()
def on_rbut_gap_c1_toggled(self, widget, data=None):
self.scene.fill_gap(Scene.C1)
self.drawing_area.queue_draw()
def on_rbut_gap_c2_toggled(self, widget, data=None):
self.scene.fill_gap(Scene.C2)
self.drawing_area.queue_draw()
def on_but_pos_appyly_clicked(self, widget, data=None):
self.scene.set_lookat(
(self.sp_pos_x.get_value(), self.sp_pos_y.get_value(),
self.sp_pos_z.get_value()),
(self.sp_look_x.get_value(), self.sp_look_y.get_value(),
self.sp_look_z.get_value()))
self.drawing_area.queue_draw()
def on_but_cut_clicked(self, widget, data=None):
A, B = self.scene.cut_current(
(self.sp_cut_u1.get_value(), self.sp_cut_v2.get_value(),
self.sp_cut_u2.get_value(), self.sp_cut_v2.get_value()),
self.sp_delta.get_value())
fm = self.cbox_first.get_model()
sm = self.cbox_second.get_model()
fm.clear()
for a in range(A):
fm.append((str(a), ))
sm.clear()
for b in range(B):
sm.append((str(b), ))
self.drawing_area.queue_draw()
def on_but_cut_clear_clicked(self, widget, data=None):
self.scene.clear_cut()
self.drawing_area.queue_draw()
def on_cbox_first_changed(self, widget, data=None):
txt = widget.get_active_text()
if txt == None: i = None
else: i = int(txt)
self.scene.cut_select(0, i)
self.drawing_area.queue_draw()
def on_cbox_second_changed(self, widget, data=None):
txt = widget.get_active_text()
if txt == None: i = None
else: i = int(txt)
self.scene.cut_select(1, i)
self.drawing_area.queue_draw()
def on_mitem_load_activate(self, widget, data=None):
if self.win_dia_load.run() == gtk.RESPONSE_OK:
self.save_file = self.win_dia_load.get_filename()
self.scene.load_from_file(self.save_file)
self.drawing_area.queue_draw()
self.win_dia_load.hide()
def on_mitem_new_activate(self, widget, data=None):
self.scene.clear()
self.drawing_area.queue_draw()
def on_mitem_save_activate(self, widget, data=None):
if self.save_file != None:
self.scene.dump_to_file(self.save_file)
else:
self.on_mitem_saveas_activate(widget, data)
def on_mitem_saveas_activate(self, widget, data=None):
if self.win_dia_save.run() == gtk.RESPONSE_OK:
self.save_file = self.win_dia_save.get_filename()
self.scene.dump_to_file(self.save_file)
self.win_dia_save.hide()
def on_but_gen_clicked(self, widge, data=None):
self.scene.gen_paths()
def on_but_dump_clicked(self, widge, data=None):
self.scene.dump_sign()
def on_show(self, widget, data=None):
widget.show_all()
def on_hide(self, widget, data=None):
widget.hide()
return True
class App(object):
"""Application main class"""
def __init__(self):
self._init_keyboard()
self.near = .01
self.far = 10
self.fov = 60
builder = gtk.Builder()
builder.add_from_file(ui_file)
glconfig = self.init_glext()
self.drawing_area = GLDrawingArea(glconfig)
self.drawing_area.set_events( gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON3_MOTION_MASK )
self.drawing_area.set_size_request(800, 800)
builder.get_object("vbox1").pack_start(self.drawing_area)
self.scene = Scene( self.fov , 1 , self.near , self.far , self._load_cube_img(skybox_files) , self._load_img(duck_file) )
self.drawing_area.add( self.scene )
win_main = builder.get_object("win_main")
win_main.connect('key-press-event' , self._on_key_pressed )
win_main.connect('key-release-event', self._on_key_released )
win_main.show_all()
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene.set_ratio( ratio )
builder.connect_signals(self)
self.statbar = builder.get_object('statbar')
self.drawing_area.connect('motion_notify_event',self._on_mouse_motion)
self.drawing_area.connect('button_press_event',self._on_button_pressed)
self.drawing_area.connect('configure_event',self._on_reshape)
self.drawing_area.connect_after('expose_event',self._after_draw)
gtk.timeout_add( 1 , self._refresh )
def _init_keyboard( self ) :
self.move = [0,0,0]
self.dirskeys = ( ( ['w'] , ['s'] ) , ( ['a'] , ['d'] ) , ( ['e'] , ['q'] ) )
for d in self.dirskeys :
for e in d :
for i in range(len(e)) : e[i] = ( gtk.gdk.unicode_to_keyval(ord(e[i])) , False )
def _on_key_pressed( self , widget , data=None ) :
if not any(self.move) :
gtk.timeout_add( 20 , self._move_callback )
for i in range(len(self.dirskeys)) :
if (data.keyval,False) in self.dirskeys[i][0] :
self.dirskeys[i][0][ self.dirskeys[i][0].index( (data.keyval,False) ) ] = (data.keyval,True)
self.move[i]+= 1
elif (data.keyval,False) in self.dirskeys[i][1] :
self.dirskeys[i][1][ self.dirskeys[i][1].index( (data.keyval,False) ) ] = (data.keyval,True)
self.move[i]-= 1
def _on_key_released( self , widget , data=None ) :
for i in range(len(self.dirskeys)) :
if (data.keyval,True) in self.dirskeys[i][0] :
self.dirskeys[i][0][ self.dirskeys[i][0].index( (data.keyval,True) ) ] = (data.keyval,False)
self.move[i]-= 1
elif (data.keyval,True) in self.dirskeys[i][1] :
self.dirskeys[i][1][ self.dirskeys[i][1].index( (data.keyval,True) ) ] = (data.keyval,False)
self.move[i]+= 1
def _move_callback( self ) :
self.scene.key_pressed( self.move )
self.drawing_area.queue_draw()
return any(self.move)
def _refresh( self ) :
self.drawing_area.queue_draw()
return True
def _after_draw( self , widget , data=None ) :
self.update_statusbar()
def update_statusbar( self ) :
pass
def _on_reshape( self , widget , data=None ) :
width = self.drawing_area.allocation.width
height = self.drawing_area.allocation.height
ratio = float(width)/float(height)
self.scene.set_screen_size( width , height )
self.scene.set_ratio( ratio )
def _on_button_pressed( self , widget , data=None ) :
if data.button == 3 :
self.mouse_pos = data.x , data.y
self.drawing_area.queue_draw()
def _on_mouse_motion( self , widget , data=None ) :
diff = map( op.sub , self.mouse_pos , (data.x , data.y) )
self.scene.mouse_move( diff )
self.mouse_pos = data.x , data.y
self.drawing_area.queue_draw()
def init_glext(self):
# Query the OpenGL extension version.
# print "OpenGL extension version - %d.%d\n" % gtk.gdkgl.query_version()
# Configure OpenGL framebuffer.
# Try to get a double-buffered framebuffer configuration,
# if not successful then try to get a single-buffered one.
display_mode = (gtk.gdkgl.MODE_RGB |
gtk.gdkgl.MODE_DEPTH |
gtk.gdkgl.MODE_DOUBLE)
try:
glconfig = gtk.gdkgl.Config(mode=display_mode)
except gtk.gdkgl.NoMatches:
display_mode &= ~gtk.gdkgl.MODE_DOUBLE
glconfig = gtk.gdkgl.Config(mode=display_mode)
return glconfig
def _load_img( self , filename ) :
gimg = gtk.image_new_from_file( filename )
gpb = gimg.get_pixbuf()
return Image( gpb.get_width() , gpb.get_height() , gpb.get_pixels() )
def _load_cube_img( self , filenames ) :
pbs = [ gtk.image_new_from_file(filename).get_pixbuf() for filename in filenames ]
w = pbs[0].get_width()
h = pbs[0].get_height()
for p in pbs : assert( w == p.get_width () )
for p in pbs : assert( h == p.get_height() )
return CubeImage( w , h , [ p.get_pixels() for p in pbs ] )
def on_win_main_destroy(self,widget,data=None):
gtk.main_quit()
def on_but_quit_clicked(self,widget,data=None):
gtk.main_quit()