Example #1
0
File: puma.py Project: jkotur/Puma
	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 )
Example #2
0
	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 )
Example #3
0
File: duck.py Project: jkotur/duck
	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 )
Example #4
0
    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
Example #5
0
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
Example #6
0
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()
Example #7
0
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()
Example #8
0
	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)
Example #9
0
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()
Example #10
0
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()
Example #11
0
    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
Example #12
0
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
Example #13
0
File: duck.py Project: jkotur/duck
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()