edit_image.py

import gtk
import color_circle



class Edit_image(object):
    def __init__(self,filename):
        self.filename=filename
        # Backing pixmap for drawing area
        self.pixmap = None
        self.temp_height = 0
        self.temp_width = 0
        self.x1 = None
        self.y1 = None
        # Create a new backing pixmap of the appropriate size
        self.color_selected_object=color_circle.ColorSelectionExample()

        #creating the structure
        self.load_image()
        self.drawing_area()
        self.window()


   
    def window(self):
        self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
        self.window.set_resizable(False)
        self.window.connect("delete_event", self.delete_event)
        self.window.set_size_request(self.pixbuf.get_width(), self.pixbuf.get_height())
        self.window.set_keep_above(True)
        self.window.add(self.drawing_area)
        self.window.show_all()  

    def load_image(self):
        self.image = gtk.Image()
        self.image.set_from_file(self.filename)
        self.pixbuf = self.image.get_pixbuf()
        self.image.connect('expose-event', self.on_image_resize)  

    def drawing_area(self):
        # Create the drawing areadrawing_area
        self.drawing_area = gtk.DrawingArea()
        self.drawing_area.set_size_request(self.pixbuf.get_width(), self.pixbuf.get_height())
        self.drawing_area.show()

        # Signals used to handle backing pixmap
        self.drawing_area.connect("expose_event", self.expose_event)
        self.drawing_area.connect("configure_event", self.configure_event)

        # Event signals
        self.drawing_area.connect("motion_notify_event", self.motion_notify_event)
        self.drawing_area.connect("button_press_event", self.button_press_event) 
        self.drawing_area.set_events(gtk.gdk.EXPOSURE_MASK
                                | gtk.gdk.LEAVE_NOTIFY_MASK
                                | gtk.gdk.BUTTON_PRESS_MASK
                                | gtk.gdk.POINTER_MOTION_MASK
                                | gtk.gdk.POINTER_MOTION_HINT_MASK)
 
      

    def on_image_resize(self, widget, event):
        allocation = widget.get_allocation()
        if self.temp_height != allocation.height or self.temp_width != allocation.width:
            self.temp_height = allocation.height
            self.temp_width = allocation.width
            pixbuf = self.pixbuf.scale_simple(allocation.width, allocation.height, gtk.gdk.INTERP_BILINEAR)
            widget.set_from_pixbuf(pixbuf)

    def close_application(self, widget, event, data=None):
        gtk.main_quit()
        return False

    def configure_event(self,widget, event):
        global pixmap

        x, y, width, height = widget.get_allocation()
        self.pixmap = gtk.gdk.Pixmap(widget.window, width, height)
        self.pixmap.draw_rectangle(widget.get_style().white_gc,
                              True, 0, 0, width, height)
        pixbuf = gtk.gdk.pixbuf_new_from_file(self.filename) #one way to load a pixbuf
        self.pixmap.draw_pixbuf(None, pixbuf, 0, 0, x, y, -1, -1, gtk.gdk.RGB_DITHER_NONE, 0, 0)
       
        return True

    # Redraw the screen from the backing pixmap
    def expose_event(self, widget, event):
        # try:
        x , y, width, height = event.area
        widget.window.draw_drawable(widget.get_style().fg_gc[gtk.STATE_NORMAL], self.pixmap, x, y, x, y, width, height)
        return False
        # except:
        #     pass

    #Draw a rectangle on the screen
    def draw_brush(self, widget, x, y):
        x = int(x)
        y = int(y)
        gc = widget.get_style().black_gc
        # color=gtk.gdk.Color(red=65535, green=0, blue=0)
        gc.set_rgb_fg_color(self.selected_color)

        if self.x1 != None and self.y1 != None:
            self.pixmap.draw_lines(gc,((self.x1, self.y1), (x, y)))
            gc.set_line_attributes(10,gtk.gdk.LINE_SOLID,gtk.gdk.CAP_ROUND,gtk.gdk.JOIN_ROUND)

            widget.queue_draw_area(0, 0, widget.allocation.width, widget.allocation.height)
        self.x1 = x
        self.y1 = y

    def button_press_event(self, widget, event):
        
        self.selected_color=self.color_selected_object.color
        # if event.button == 3:
        #     self.save_image(widget)
        if event.button == 1 and self.pixmap != None:
            self.draw_brush(widget, event.x, event.y)
        return True

    def motion_notify_event(self, widget, event):
        if event.is_hint:
            x, y, state = event.window.get_pointer()
        else:
            x = event.x
            y = event.y
            state = event.state
        
        if state & gtk.gdk.BUTTON1_MASK and self.pixmap != None:
            self.selected_color=self.color_selected_object.color
            self.draw_brush(widget, x, y)
                    
        return True

    def save_image(self, widget):
        print("Save Image")
        pixbuf1 = gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, has_alpha=False, bits_per_sample=8, width=widget.allocation.width, height=widget.allocation.height)
        pixbuf1.get_from_drawable(self.pixmap, self.pixmap.get_colormap(), 0, 0, 0, 0, widget.allocation.width, widget.allocation.height)
        pixbuf1.save(self.filename, "png", {"quality":"100"})

    def delete_event(self, widget, event, data=None):
        self.save_image(widget)
        self.color_selected_object.delete_event(widget, event)
        gtk.main_quit()
        return False



def edit(filename):
    Edit_image(filename)
    gtk.main()

if __name__ == "__main__":
    Edit_image('screenshot.png')
    gtk.main()