def button_released(self, widget, event):
self.active_tool.end(event.x, event.y)
self.swap_buffers()
self.active_tool.commit()
if self.active_tool.name == "ColorPicker":
col = self.active_tool.col
self.picker_col = RGBAColor(col[2], col[1], col[0], col[3])
self.emit("color_pick_event", event)
self.active_tool = self.previous_tool
def motion_event(self, widget, event):
if self.active_tool.mode != self.active_tool.DRAWING:
sp = self.__over_scaling_point(event)
if sp != 0:
if sp == self.CORNER_SCALING_POINT:
self.toolchest["canvas-both-scale"].select()
elif sp == self.RIGHT_SCALING_POINT:
self.toolchest["canvas-hor-scale"].select()
elif sp == self.BOTTOM_SCALING_POINT:
self.toolchest["canvas-ver-scale"].select()
else:
self.active_tool.select()
if event.x > self.width or event.y > self.height:
self.window.set_cursor(gtk.gdk.Cursor(gtk.gdk.ARROW))
else:
self.active_tool.move(event.x, event.y)
if self.active_tool.name == "ColorPicker":
col = self.active_tool.col
self.picker_col = RGBAColor(col[2], col[1], col[0], col[3])
self.emit("color_pick_event", event)
self.swap_buffers()
def __init__(self):
# Initializing gtk.DrawingArea superclass
super(Canvas, self).__init__()
# Resize Square Size
self.RSS = 7
# Margin (to draw shadows and resize squares)
self.margin_size = 20
# Registering events
self.add_events(gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK
| gtk.gdk.BUTTON1_MOTION_MASK | gtk.gdk.DRAG_MOTION
| gtk.gdk.POINTER_MOTION_MASK)
self.connect("button-press-event", self.button_pressed)
self.connect("button-release-event", self.button_released)
self.connect("expose-event", self.expose)
self.connect("motion-notify-event", self.motion_event)
self.undo_buffer = UndoBuffer()
self.modified = False
self.set_size(550, 412)
self.alpha_pattern = cairo.SurfacePattern(
cairo.ImageSurface.create_from_png("GUI/alpha-pattern.png"))
self.alpha_pattern.set_extend(cairo.EXTEND_REPEAT)
self.bg_init = 0
self.primary = RGBAColor(0, 0, 0, 1)
self.secondary = RGBAColor(1, 1, 1, 1)
self.figure_linewidth = 0
self.figure_corner_radius = 0
self.airbrush_width = 0
self.fig_fill_type = 0
self.set_selection(False)
self.select_xp = None
self.select_yp = None
# Surface is the image in the canvas
self.surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width,
self.height)
#overlay is for selection boxes - etc
self.overlay = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width,
self.height)
#clipboard
self.clipboard = gtk.clipboard_get(selection="CLIPBOARD")
# Shadows to distribute around canvas
str = "GUI/bl-corner-shadow.png"
self.BL_CORNER_SHADOW = cairo.ImageSurface.create_from_png(str)
str = "GUI/tr-corner-shadow.png"
self.TR_CORNER_SHADOW = cairo.ImageSurface.create_from_png(str)
self.side_alpha_channels = [
0.4, 0.39, 0.37, 0.32, 0.24, 0.16, 0.08, 0.04, 0.01
]
self.toolchest = {
"draw-rounded-rectangle": tools.RoundedRectangleTool(self),
"draw-rectangle": tools.RectangleTool(self),
"straight-line": tools.StraightLineTool(self),
"pencil": tools.PencilTool(self),
"paintbrush": tools.PaintbrushTool(self),
"bucket-fill": tools.BucketFillTool(self),
"eraser": tools.EraserTool(self),
"draw-ellipse": tools.EllipseTool(self),
"color-picker": tools.ColorPickerTool(self),
"rect-select": tools.RectangleSelectTool(self),
"airbrush": tools.AirBrushTool(self),
"canvas-both-scale": tools.BothScalingTool(self),
"canvas-hor-scale": tools.HorizontalScalingTool(self),
"canvas-ver-scale": tools.VerticalScalingTool(self),
"dummy_tool": tools.Tool(self)
}
self.active_tool = self.toolchest["dummy_tool"]
self.previous_tool = self.active_tool # Previous tool (to recover from a rescale)
class Canvas(gtk.DrawingArea):
CORNER_SCALING_POINT = 1
RIGHT_SCALING_POINT = 2
BOTTOM_SCALING_POINT = 3
active_tool = None
picker_col = None
bg_init = None
undo_buffer = None
select_active = None
modified = None
fig_fill_type = None
margin_size = None
RSS = None
primary = None
secondary = None
def __init__(self):
# Initializing gtk.DrawingArea superclass
super(Canvas, self).__init__()
# Resize Square Size
self.RSS = 7
# Margin (to draw shadows and resize squares)
self.margin_size = 20
# Registering events
self.add_events(gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK
| gtk.gdk.BUTTON1_MOTION_MASK | gtk.gdk.DRAG_MOTION
| gtk.gdk.POINTER_MOTION_MASK)
self.connect("button-press-event", self.button_pressed)
self.connect("button-release-event", self.button_released)
self.connect("expose-event", self.expose)
self.connect("motion-notify-event", self.motion_event)
self.undo_buffer = UndoBuffer()
self.modified = False
self.set_size(550, 412)
self.alpha_pattern = cairo.SurfacePattern(
cairo.ImageSurface.create_from_png("GUI/alpha-pattern.png"))
self.alpha_pattern.set_extend(cairo.EXTEND_REPEAT)
self.bg_init = 0
self.primary = RGBAColor(0, 0, 0, 1)
self.secondary = RGBAColor(1, 1, 1, 1)
self.figure_linewidth = 0
self.figure_corner_radius = 0
self.airbrush_width = 0
self.fig_fill_type = 0
self.set_selection(False)
self.select_xp = None
self.select_yp = None
# Surface is the image in the canvas
self.surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width,
self.height)
#overlay is for selection boxes - etc
self.overlay = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width,
self.height)
#clipboard
self.clipboard = gtk.clipboard_get(selection="CLIPBOARD")
# Shadows to distribute around canvas
str = "GUI/bl-corner-shadow.png"
self.BL_CORNER_SHADOW = cairo.ImageSurface.create_from_png(str)
str = "GUI/tr-corner-shadow.png"
self.TR_CORNER_SHADOW = cairo.ImageSurface.create_from_png(str)
self.side_alpha_channels = [
0.4, 0.39, 0.37, 0.32, 0.24, 0.16, 0.08, 0.04, 0.01
]
self.toolchest = {
"draw-rounded-rectangle": tools.RoundedRectangleTool(self),
"draw-rectangle": tools.RectangleTool(self),
"straight-line": tools.StraightLineTool(self),
"pencil": tools.PencilTool(self),
"paintbrush": tools.PaintbrushTool(self),
"bucket-fill": tools.BucketFillTool(self),
"eraser": tools.EraserTool(self),
"draw-ellipse": tools.EllipseTool(self),
"color-picker": tools.ColorPickerTool(self),
"rect-select": tools.RectangleSelectTool(self),
"airbrush": tools.AirBrushTool(self),
"canvas-both-scale": tools.BothScalingTool(self),
"canvas-hor-scale": tools.HorizontalScalingTool(self),
"canvas-ver-scale": tools.VerticalScalingTool(self),
"dummy_tool": tools.Tool(self)
}
self.active_tool = self.toolchest["dummy_tool"]
self.previous_tool = self.active_tool # Previous tool (to recover from a rescale)
def set_size(self, width, height):
self.width = max(width, 1)
self.height = max(height, 1)
self.set_size_request(self.width + self.margin_size,
self.height + self.margin_size)
def get_width(self):
return self.width
def get_height(self):
return self.height
def set_active_tool(self, toolname):
self.active_tool = self.toolchest[toolname]
def button_pressed(self, widget, event):
self.previous_tool = self.active_tool
# When the click is outside the canvas, a scaling point might have been
# clicked.
if event.x >= self.width or event.y >= self.height:
sp = self.__over_scaling_point(event)
if sp == self.CORNER_SCALING_POINT:
self.active_tool = self.toolchest["canvas-both-scale"]
elif sp == self.RIGHT_SCALING_POINT:
self.active_tool = self.toolchest["canvas-hor-scale"]
elif sp == self.BOTTOM_SCALING_POINT:
self.active_tool = self.toolchest["canvas-ver-scale"]
else:
self.active_tool = self.toolchest["dummy_tool"]
if self.active_tool.name != 'NotSet':
if event.type == gtk.gdk.BUTTON_PRESS:
self.active_tool.begin(event.x, event.y, event.button)
self.swap_buffers()
def button_released(self, widget, event):
self.active_tool.end(event.x, event.y)
self.swap_buffers()
self.active_tool.commit()
if self.active_tool.name == "ColorPicker":
col = self.active_tool.col
self.picker_col = RGBAColor(col[2], col[1], col[0], col[3])
self.emit("color_pick_event", event)
self.active_tool = self.previous_tool
def motion_event(self, widget, event):
if self.active_tool.mode != self.active_tool.DRAWING:
sp = self.__over_scaling_point(event)
if sp != 0:
if sp == self.CORNER_SCALING_POINT:
self.toolchest["canvas-both-scale"].select()
elif sp == self.RIGHT_SCALING_POINT:
self.toolchest["canvas-hor-scale"].select()
elif sp == self.BOTTOM_SCALING_POINT:
self.toolchest["canvas-ver-scale"].select()
else:
self.active_tool.select()
if event.x > self.width or event.y > self.height:
self.window.set_cursor(gtk.gdk.Cursor(gtk.gdk.ARROW))
else:
self.active_tool.move(event.x, event.y)
if self.active_tool.name == "ColorPicker":
col = self.active_tool.col
self.picker_col = RGBAColor(col[2], col[1], col[0], col[3])
self.emit("color_pick_event", event)
self.swap_buffers()
def swap_buffers(self):
rect = gtk.gdk.Rectangle(0, 0, self.width, self.height)
self.window.invalidate_rect(
rect, True) #invalidating the rectangle forces gtk to run expose.
def expose(self, widget,
event): # Run when buffers are swapped: updates screen.
#temporary surface size of canvas
tmp_surf = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width,
self.height)
context = cairo.Context(tmp_surf)
#draw to this temporary surface
self.draw(context)
if self.active_tool.name == "AirBrush" and self.active_tool.mode == self.active_tool.DRAWING:
self.surface = tmp_surf
#get widget window as context
wincontext = widget.window.cairo_create()
#clip to image size
wincontext.rectangle(0, 0, self.width, self.height)
wincontext.clip()
#paint alpha pattern over whole clipped region
wincontext.set_source(self.alpha_pattern)
wincontext.paint()
#paint
wincontext.set_source_surface(tmp_surf)
wincontext.paint()
#overlay
wincontext.set_source_surface(self.overlay)
wincontext.paint()
# Retrieving cairo context
context = widget.window.cairo_create()
# Modify clipping area to draw decorations outside the canvas
# Draw decorations
self.__draw_shadows(context)
self.__draw_scaling_points(context)
def print_tool(self):
self.clear_overlay()
#temporary surface size of canvas
tmp_surf = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width,
self.height)
context = cairo.Context(tmp_surf)
#draw to temporary surface
self.draw(context)
self.surface = tmp_surf #surface now swapped with updated surface
def draw(self, context):
# Drawing the background
self.__draw_background(context)
#Draw the current surface over the background
context.set_source_surface(self.surface)
context.paint()
#Draw any active tool if applicable.
if self.active_tool.Draw2Overlay:
ov_context = cairo.Context(self.overlay)
self.active_tool.draw(ov_context)
else:
self.active_tool.draw(context)
def __draw_background(self, context):
#if the background has never been initialsed (first print) then
#fill whole canvas, else fill new regions
if self.bg_init == 0:
context.rectangle(0, 0, self.width, self.height)
self.bg_init = 1
else:
context.rectangle(self.surface.get_width(), 0,
self.width - self.surface.get_width(),
self.height)
context.rectangle(0, self.surface.get_height(), self.width,
self.height - self.surface.get_height())
context.set_source_rgba(self.secondary.get_red(),
self.secondary.get_green(),
self.secondary.get_blue(),
self.secondary.get_alpha())
context.fill()
def clear_overlay(self):
#temporary surface size of canvase
tmp_surf = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width,
self.height)
context = cairo.Context(tmp_surf)
#paint surface transparent
context.rectangle(0, 0, self.width, self.height)
context.set_source_rgba(0, 0, 0, 0)
context.fill()
#set as overlay
self.overlay = tmp_surf
self.set_selection(False)
def set_selection(self, value):
if self.select_active != value:
self.select_active = value
self.emit("change_sensitivty", senstivity_data('crop', value))
def get_image(self):
return self.surface
def set_image(self, surface):
self.surface = surface
self.set_size(surface.get_width(), surface.get_height())
def get_color(self): #used by color_pick_event callback
return self.picker_col
def undo(self):
if self.undo_buffer.n_buf_full > 0:
self.modified = True
self.update_undo_buffer(0)
buf = self.undo_buffer.prev_buf()
data = self.surface.get_data()
w = self.surface.get_width()
h = self.surface.get_height()
bw = self.undo_buffer.width[buf]
bh = self.undo_buffer.height[buf]
if bh != h | bw != w:
self.set_size(bw, bh)
self.print_tool()
data = self.surface.get_data()
data[:] = self.undo_buffer.Buffer[buf][:]
self.undo_buffer.n_buf_full -= 1
if self.undo_buffer.redos_allowed < self.undo_buffer.n_buf:
self.undo_buffer.redos_allowed += 1
self.undo_buffer.cur_buf = buf
self.swap_buffers()
self.emit("change_sensitivty", senstivity_data('redo', True))
if self.undo_buffer.n_buf_full == 0:
self.emit("change_sensitivty", senstivity_data('undo', False))
def redo(self):
if self.undo_buffer.redos_allowed > 0:
self.modified = True
buf = self.undo_buffer.next_buf()
data = self.surface.get_data()
w = self.surface.get_width()
h = self.surface.get_height()
bw = self.undo_buffer.width[buf]
bh = self.undo_buffer.height[buf]
if bh != h | bw != w:
self.set_size(bw, bh)
self.print_tool()
data = self.surface.get_data()
data[:] = self.undo_buffer.Buffer[buf][:]
self.undo_buffer.redos_allowed -= 1
self.undo_buffer.n_buf_full += 1
self.undo_buffer.cur_buf = buf
self.swap_buffers()
self.emit("change_sensitivty", senstivity_data('undo', True))
if self.undo_buffer.redos_allowed == 0:
self.emit("change_sensitivty", senstivity_data('redo', False))
def update_undo_buffer(self, iterate):
self.modified = True
w = self.surface.get_width()
h = self.surface.get_height()
s = self.surface.get_stride()
data = self.surface.get_data()
buf = self.undo_buffer.cur_buf
self.undo_buffer.Buffer[buf] = create_string_buffer(s * h)
self.undo_buffer.Buffer[buf][:] = data[:]
self.undo_buffer.width[buf] = w
self.undo_buffer.height[buf] = h
if iterate == 1:
self.emit("change_sensitivty", senstivity_data('undo', True))
self.emit("change_sensitivty", senstivity_data('redo', False))
self.undo_buffer.redos_allowed = 0
if self.undo_buffer.n_buf_full < self.undo_buffer.n_buf:
self.undo_buffer.n_buf_full += 1
self.undo_buffer.cur_buf = self.undo_buffer.next_buf()
def clear_undo_buffer(self):
self.emit("change_sensitivty", senstivity_data('undo', False))
self.emit("change_sensitivty", senstivity_data('redo', False))
self.undo_buffer.cur_buf = 0
self.undo_buffer.n_buf_full = 0
self.undo_buffer.redos_allowed = 0
def copy(self, cut):
data = self.surface.get_data()
t_data = list(data)
t_data[::4] = data[2::4]
t_data[2::4] = data[::4]
s = self.surface.get_stride()
w = self.surface.get_width()
h = self.surface.get_height()
if self.select_active:
xp = [min(max(0, x), w) for x in self.select_xp]
yp = [min(max(0, y), h) for y in self.select_yp]
c_w = int(max(xp) - min(xp))
c_h = int(max(yp) - min(yp))
if c_h > 0 and c_w > 0:
c_s = c_w * 4
c_data = [t_data[0]] * (c_h * c_s)
c_y = int(min(yp))
c_x = int(min(xp))
for n in range(c_h):
c_data[n * c_s:(n + 1) *
c_s] = t_data[(n + c_y) * s +
c_x * 4:(n + c_y) * s + c_x * 4 + c_s]
c_data = ''.join(c_data)
PixBuf = gtk.gdk.pixbuf_new_from_data(c_data,
gtk.gdk.COLORSPACE_RGB,
True, 8, c_w, c_h, c_s)
self.clipboard.set_image(PixBuf)
if cut:
self.delete()
else:
t_data = ''.join(t_data)
PixBuf = gtk.gdk.pixbuf_new_from_data(t_data,
gtk.gdk.COLORSPACE_RGB, True,
8, w, h, s)
self.clipboard.set_image(PixBuf)
if cut:
self.delete()
def delete(self):
w = self.surface.get_width()
h = self.surface.get_height()
if self.select_active:
xp = [min(max(0, x), w) for x in self.select_xp]
yp = [min(max(0, y), h) for y in self.select_yp]
c_w = int(max(xp) - min(xp))
c_h = int(max(yp) - min(yp))
if c_h > 0 and c_w > 0:
c_y = int(min(yp))
c_x = int(min(xp))
self.update_undo_buffer(1)
aux = cairo.ImageSurface(cairo.FORMAT_ARGB32, c_w, c_h)
context = cairo.Context(aux)
context.rectangle(0, 0, self.width, self.height)
context.set_source_rgba(self.secondary.get_red(),
self.secondary.get_green(),
self.secondary.get_blue(),
self.secondary.get_alpha())
context.fill()
mask = cairo.ImageSurface(cairo.FORMAT_ARGB32, w, h)
context = cairo.Context(mask)
context.rectangle(0, 0, self.width, self.height)
context.set_source_rgba(1, 1, 1, 0)
context.fill()
context.rectangle(c_x, c_y, c_w, c_h)
context.set_source_rgba(1, 1, 1, 1)
context.fill()
context = cairo.Context(self.surface)
context.set_source_surface(aux, c_x, c_y)
context.set_operator(cairo.OPERATOR_SOURCE)
context.mask(cairo.SurfacePattern(mask))
self.swap_buffers()
else:
data = self.surface.get_data()
self.update_undo_buffer(1)
context = cairo.Context(self.surface)
context.rectangle(0, 0, self.width, self.height)
context.set_source_rgba(self.secondary.get_red(),
self.secondary.get_green(),
self.secondary.get_blue(),
self.secondary.get_alpha())
context.set_operator(cairo.OPERATOR_SOURCE)
context.fill()
#if the context is filled blank the pixels wont edit properly for bucket fill
#The hacky solution is to copy the black pixels to a temporary array
#Paint new pixels, and then copy the blank ones back in.
if self.secondary.get_alpha() == 0:
t_data = data[:]
context = cairo.Context(self.surface)
context.paint()
data[:] = t_data[:]
self.swap_buffers()
def paste(self):
image = self.clipboard.wait_for_image()
if image != None:
self.update_undo_buffer(1)
self.set_size(max(self.width, image.get_width()),
max(self.height, image.get_height()))
self.print_tool()
aux = cairo.ImageSurface(cairo.FORMAT_ARGB32, image.get_width(),
image.get_height())
im_data = image.get_pixels()
# the next two semingly useless lines give the surface real pixels to be edited.
context = cairo.Context(aux)
context.paint()
#directly write to the pixels in aux
data = aux.get_data()
#pasted image has alpha channel
if image.get_rowstride() == aux.get_stride():
data[2::4] = im_data[0::4] #red
data[0::4] = im_data[2::4] #blue
data[1::4] = im_data[1::4] #green
data[3::4] = im_data[3::4] #alpha
else: #pasted data no alpha channel
data[2::4] = im_data[0::3] #red
data[0::4] = im_data[2::3] #blue
data[1::4] = im_data[1::3] #green
#use cairo to add the pasted image to the current image
context = cairo.Context(self.surface)
context.set_source_surface(aux, 0, 0)
context.paint()
self.swap_buffers()
def crop(self):
if self.select_active:
self.update_undo_buffer(1)
w = self.surface.get_width()
h = self.surface.get_height()
xp = [min(max(0, x), w) for x in self.select_xp]
yp = [min(max(0, y), h) for y in self.select_yp]
c_w = int(max(xp)) - int(min(xp))
c_h = int(max(yp)) - int(min(yp))
aux = cairo.ImageSurface(cairo.FORMAT_ARGB32, w, h)
context = cairo.Context(aux)
context.set_source_surface(self.surface, 0, 0)
context.paint()
context = cairo.Context(self.surface)
context.set_source_surface(aux, -int(min(xp)), -int(min(yp)))
context.paint()
self.set_size(c_w, c_h)
self.clear_overlay()
self.print_tool()
def is_modified(self):
return self.modified
def __draw_shadows(self, context):
# Shadow displacements
disp = 2
csw = self.BL_CORNER_SHADOW.get_width()
if self.width > 10:
# Bottom left corner
context.set_source_surface(self.BL_CORNER_SHADOW, disp,
self.height)
context.paint()
# Bottom shadow
context.translate(0, self.height)
for i in range(len(self.side_alpha_channels)):
alpha = self.side_alpha_channels[i]
context.rectangle(disp + csw, i, self.width - disp - csw, 1)
context.set_source_rgba(0, 0, 0, alpha)
context.fill()
context.translate(0, -self.height)
if self.height > 10:
# Top right corner
context.set_source_surface(self.TR_CORNER_SHADOW, self.width, disp)
context.paint()
# Side shadow
context.translate(self.width, 0)
for i in range(len(self.side_alpha_channels)):
alpha = self.side_alpha_channels[i]
context.rectangle(i, disp + csw, 1, self.height - disp - csw)
context.set_source_rgba(0, 0, 0, alpha)
context.fill()
context.translate(-self.width, 0)
def __draw_scaling_points(self, context):
# Right scaling point
if self.height > self.RSS * 4:
self.__draw_scaling_point(context, self.width,
(self.height - self.RSS) / 2)
# Bottom scaling point
if self.width > self.RSS * 4:
self.__draw_scaling_point(context, (self.width - self.RSS) / 2,
self.height)
# Corner scaling point
self.__draw_scaling_point(context, self.width, self.height)
def __draw_scaling_point(self, context, x, y):
# Dark border
context.set_source_rgba(0.156, 0.402, 0546, 1)
context.rectangle(x + 1, y + 1, self.RSS - 1, self.RSS - 1)
context.fill()
# Light border
context.set_source_rgba(.556, .802, .946, 1)
context.rectangle(x, y, self.RSS - 1, self.RSS - 1)
context.fill()
# The point itself
context.set_source_rgba(.26, .67, .91, 1.0)
context.rectangle(x + 1, y + 1, self.RSS - 2, self.RSS - 2)
context.fill()
def __over_scaling_point(self, event):
if self.width < event.x + 1 < self.width + self.RSS:
if self.height < event.y + 1 < self.height + self.RSS:
return self.CORNER_SCALING_POINT
elif (self.height - self.RSS) / 2 < event.y + 1 < (self.height +
self.RSS) / 2:
return self.RIGHT_SCALING_POINT
elif self.height < event.y + 1 < self.height + self.RSS:
if (self.width - self.RSS) / 2 < event.x + 1 < (self.width +
self.RSS) / 2:
return self.BOTTOM_SCALING_POINT
return 0
def __init__(self):
# Initializing gtk.DrawingArea superclass
super(Canvas, self).__init__()
# Resize Square Size
self.RSS = 7
# Margin (to draw shadows and resize squares)
self.margin_size = 20
# Registering events
self.add_events(gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON1_MOTION_MASK | gtk.gdk.DRAG_MOTION | gtk.gdk.POINTER_MOTION_MASK)
self.connect("button-press-event", self.button_pressed)
self.connect("button-release-event", self.button_released)
self.connect("expose-event", self.expose)
self.connect("motion-notify-event", self.motion_event)
self.undo_buffer = UndoBuffer()
self.modified = False
self.set_size(550, 412)
self.alpha_pattern = cairo.SurfacePattern(cairo.ImageSurface.create_from_png("GUI/alpha-pattern.png"))
self.alpha_pattern.set_extend(cairo.EXTEND_REPEAT)
self.bg_init=0
self.primary = RGBAColor(0, 0, 0, 1)
self.secondary = RGBAColor(1, 1, 1, 1)
self.figure_linewidth=0
self.figure_corner_radius=0
self.airbrush_width=0
self.fig_fill_type = 0
self.set_selection(False)
self.select_xp = None
self.select_yp = None
# Surface is the image in the canvas
self.surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width, self.height)
#overlay is for selection boxes - etc
self.overlay = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width, self.height)
#clipboard
self.clipboard = gtk.clipboard_get(selection="CLIPBOARD")
# Shadows to distribute around canvas
str = "GUI/bl-corner-shadow.png"
self.BL_CORNER_SHADOW = cairo.ImageSurface.create_from_png(str)
str = "GUI/tr-corner-shadow.png"
self.TR_CORNER_SHADOW = cairo.ImageSurface.create_from_png(str)
self.side_alpha_channels = [0.4, 0.39, 0.37, 0.32, 0.24, 0.16, 0.08, 0.04, 0.01]
self.toolchest = {
"draw-rounded-rectangle" : tools.RoundedRectangleTool(self),
"draw-rectangle" : tools.RectangleTool(self),
"straight-line" : tools.StraightLineTool(self),
"pencil" : tools.PencilTool(self),
"paintbrush" : tools.PaintbrushTool(self),
"bucket-fill" : tools.BucketFillTool(self),
"eraser" : tools.EraserTool(self),
"draw-ellipse" : tools.EllipseTool(self),
"color-picker" : tools.ColorPickerTool(self),
"rect-select" : tools.RectangleSelectTool(self),
"airbrush" : tools.AirBrushTool(self),
"canvas-both-scale" : tools.BothScalingTool(self),
"canvas-hor-scale" : tools.HorizontalScalingTool(self),
"canvas-ver-scale" : tools.VerticalScalingTool(self),
"dummy_tool" : tools.Tool(self)}
self.active_tool = self.toolchest["dummy_tool"]
self.previous_tool = self.active_tool# Previous tool (to recover from a rescale)
class Canvas(gtk.DrawingArea):
CORNER_SCALING_POINT = 1
RIGHT_SCALING_POINT = 2
BOTTOM_SCALING_POINT = 3
active_tool = None
picker_col = None
bg_init=None
undo_buffer = None
select_active = None
modified = None
fig_fill_type = None
margin_size = None
RSS = None
primary = None
secondary = None
def __init__(self):
# Initializing gtk.DrawingArea superclass
super(Canvas, self).__init__()
# Resize Square Size
self.RSS = 7
# Margin (to draw shadows and resize squares)
self.margin_size = 20
# Registering events
self.add_events(gtk.gdk.BUTTON_PRESS_MASK | gtk.gdk.BUTTON_RELEASE_MASK | gtk.gdk.BUTTON1_MOTION_MASK | gtk.gdk.DRAG_MOTION | gtk.gdk.POINTER_MOTION_MASK)
self.connect("button-press-event", self.button_pressed)
self.connect("button-release-event", self.button_released)
self.connect("expose-event", self.expose)
self.connect("motion-notify-event", self.motion_event)
self.undo_buffer = UndoBuffer()
self.modified = False
self.set_size(550, 412)
self.alpha_pattern = cairo.SurfacePattern(cairo.ImageSurface.create_from_png("GUI/alpha-pattern.png"))
self.alpha_pattern.set_extend(cairo.EXTEND_REPEAT)
self.bg_init=0
self.primary = RGBAColor(0, 0, 0, 1)
self.secondary = RGBAColor(1, 1, 1, 1)
self.figure_linewidth=0
self.figure_corner_radius=0
self.airbrush_width=0
self.fig_fill_type = 0
self.set_selection(False)
self.select_xp = None
self.select_yp = None
# Surface is the image in the canvas
self.surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width, self.height)
#overlay is for selection boxes - etc
self.overlay = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width, self.height)
#clipboard
self.clipboard = gtk.clipboard_get(selection="CLIPBOARD")
# Shadows to distribute around canvas
str = "GUI/bl-corner-shadow.png"
self.BL_CORNER_SHADOW = cairo.ImageSurface.create_from_png(str)
str = "GUI/tr-corner-shadow.png"
self.TR_CORNER_SHADOW = cairo.ImageSurface.create_from_png(str)
self.side_alpha_channels = [0.4, 0.39, 0.37, 0.32, 0.24, 0.16, 0.08, 0.04, 0.01]
self.toolchest = {
"draw-rounded-rectangle" : tools.RoundedRectangleTool(self),
"draw-rectangle" : tools.RectangleTool(self),
"straight-line" : tools.StraightLineTool(self),
"pencil" : tools.PencilTool(self),
"paintbrush" : tools.PaintbrushTool(self),
"bucket-fill" : tools.BucketFillTool(self),
"eraser" : tools.EraserTool(self),
"draw-ellipse" : tools.EllipseTool(self),
"color-picker" : tools.ColorPickerTool(self),
"rect-select" : tools.RectangleSelectTool(self),
"airbrush" : tools.AirBrushTool(self),
"canvas-both-scale" : tools.BothScalingTool(self),
"canvas-hor-scale" : tools.HorizontalScalingTool(self),
"canvas-ver-scale" : tools.VerticalScalingTool(self),
"dummy_tool" : tools.Tool(self)}
self.active_tool = self.toolchest["dummy_tool"]
self.previous_tool = self.active_tool# Previous tool (to recover from a rescale)
def set_size(self, width, height):
self.width = max(width, 1)
self.height = max(height, 1)
self.set_size_request(self.width + self.margin_size, self.height + self.margin_size)
def get_width(self):
return self.width
def get_height(self):
return self.height
def set_active_tool(self, toolname):
self.active_tool = self.toolchest[toolname]
def button_pressed(self, widget, event):
self.previous_tool = self.active_tool
# When the click is outside the canvas, a scaling point might have been
# clicked.
if event.x >= self.width or event.y >= self.height:
sp = self.__over_scaling_point(event)
if sp == self.CORNER_SCALING_POINT:
self.active_tool = self.toolchest["canvas-both-scale"]
elif sp == self.RIGHT_SCALING_POINT:
self.active_tool = self.toolchest["canvas-hor-scale"]
elif sp == self.BOTTOM_SCALING_POINT:
self.active_tool = self.toolchest["canvas-ver-scale"]
else:
self.active_tool = self.toolchest["dummy_tool"]
if self.active_tool.name != 'NotSet':
if event.type == gtk.gdk.BUTTON_PRESS:
self.active_tool.begin(event.x, event.y,event.button)
self.swap_buffers()
def button_released(self, widget, event):
self.active_tool.end(event.x, event.y)
self.swap_buffers()
self.active_tool.commit()
if self.active_tool.name == "ColorPicker":
col = self.active_tool.col
self.picker_col = RGBAColor(col[2], col[1], col[0], col[3])
self.emit("color_pick_event", event)
self.active_tool = self.previous_tool
def motion_event(self, widget, event):
if self.active_tool.mode != self.active_tool.DRAWING:
sp = self.__over_scaling_point(event)
if sp != 0:
if sp == self.CORNER_SCALING_POINT:
self.toolchest["canvas-both-scale"].select()
elif sp == self.RIGHT_SCALING_POINT:
self.toolchest["canvas-hor-scale"].select()
elif sp == self.BOTTOM_SCALING_POINT:
self.toolchest["canvas-ver-scale"].select()
else:
self.active_tool.select()
if event.x > self.width or event.y > self.height:
self.window.set_cursor(gtk.gdk.Cursor(gtk.gdk.ARROW))
else:
self.active_tool.move(event.x, event.y)
if self.active_tool.name == "ColorPicker":
col = self.active_tool.col
self.picker_col = RGBAColor(col[2], col[1], col[0], col[3])
self.emit("color_pick_event", event)
self.swap_buffers()
def swap_buffers(self):
rect = gtk.gdk.Rectangle(0, 0, self.width, self.height)
self.window.invalidate_rect(rect, True) #invalidating the rectangle forces gtk to run expose.
def expose(self, widget, event): # Run when buffers are swapped: updates screen.
#temporary surface size of canvas
tmp_surf = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width, self.height)
context = cairo.Context(tmp_surf)
#draw to this temporary surface
self.draw(context)
if self.active_tool.name == "AirBrush" and self.active_tool.mode == self.active_tool.DRAWING:
self.surface = tmp_surf
#get widget window as context
wincontext = widget.window.cairo_create()
#clip to image size
wincontext.rectangle(0, 0, self.width, self.height)
wincontext.clip()
#paint alpha pattern over whole clipped region
wincontext.set_source(self.alpha_pattern)
wincontext.paint()
#paint
wincontext.set_source_surface(tmp_surf)
wincontext.paint()
#overlay
wincontext.set_source_surface(self.overlay)
wincontext.paint()
# Retrieving cairo context
context = widget.window.cairo_create()
# Modify clipping area to draw decorations outside the canvas
# Draw decorations
self.__draw_shadows(context)
self.__draw_scaling_points(context)
def print_tool(self):
self.clear_overlay()
#temporary surface size of canvas
tmp_surf = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width, self.height)
context = cairo.Context(tmp_surf)
#draw to temporary surface
self.draw(context)
self.surface = tmp_surf #surface now swapped with updated surface
def draw(self, context):
# Drawing the background
self.__draw_background(context)
#Draw the current surface over the background
context.set_source_surface(self.surface)
context.paint()
#Draw any active tool if applicable.
if self.active_tool.Draw2Overlay:
ov_context = cairo.Context(self.overlay)
self.active_tool.draw(ov_context)
else:
self.active_tool.draw(context)
def __draw_background(self, context):
#if the background has never been initialsed (first print) then
#fill whole canvas, else fill new regions
if self.bg_init==0:
context.rectangle(0, 0, self.width, self.height)
self.bg_init=1
else:
context.rectangle(self.surface.get_width(), 0, self.width-self.surface.get_width(), self.height)
context.rectangle(0, self.surface.get_height(), self.width, self.height-self.surface.get_height())
context.set_source_rgba(self.secondary.get_red(),self.secondary.get_green(),self.secondary.get_blue(),self.secondary.get_alpha())
context.fill()
def clear_overlay(self):
#temporary surface size of canvase
tmp_surf = cairo.ImageSurface(cairo.FORMAT_ARGB32, self.width, self.height)
context= cairo.Context(tmp_surf)
#paint surface transparent
context.rectangle(0, 0, self.width, self.height)
context.set_source_rgba(0, 0, 0, 0)
context.fill()
#set as overlay
self.overlay = tmp_surf
self.set_selection(False)
def set_selection(self,value):
if self.select_active != value:
self.select_active = value
self.emit("change_sensitivty", senstivity_data('crop',value))
def get_image(self):
return self.surface
def set_image(self, surface):
self.surface = surface
self.set_size(surface.get_width(), surface.get_height())
def get_color(self):#used by color_pick_event callback
return self.picker_col
def undo(self):
if self.undo_buffer.n_buf_full>0:
self.modified=True
self.update_undo_buffer(0)
buf = self.undo_buffer.prev_buf()
data = self.surface.get_data()
w = self.surface.get_width()
h = self.surface.get_height()
bw = self.undo_buffer.width[buf]
bh = self.undo_buffer.height[buf]
if bh!=h | bw!=w:
self.set_size(bw,bh)
self.print_tool()
data = self.surface.get_data()
data[:] = self.undo_buffer.Buffer[buf][:]
self.undo_buffer.n_buf_full -=1
if self.undo_buffer.redos_allowed<self.undo_buffer.n_buf:
self.undo_buffer.redos_allowed += 1
self.undo_buffer.cur_buf = buf
self.swap_buffers()
self.emit("change_sensitivty", senstivity_data('redo',True))
if self.undo_buffer.n_buf_full==0:
self.emit("change_sensitivty", senstivity_data('undo',False))
def redo(self):
if self.undo_buffer.redos_allowed>0:
self.modified=True
buf = self.undo_buffer.next_buf()
data = self.surface.get_data()
w = self.surface.get_width()
h = self.surface.get_height()
bw = self.undo_buffer.width[buf]
bh = self.undo_buffer.height[buf]
if bh!=h | bw!=w:
self.set_size(bw,bh)
self.print_tool()
data = self.surface.get_data()
data[:] = self.undo_buffer.Buffer[buf][:]
self.undo_buffer.redos_allowed -=1
self.undo_buffer.n_buf_full +=1
self.undo_buffer.cur_buf = buf
self.swap_buffers()
self.emit("change_sensitivty", senstivity_data('undo',True))
if self.undo_buffer.redos_allowed==0:
self.emit("change_sensitivty", senstivity_data('redo',False))
def update_undo_buffer(self,iterate):
self.modified=True
w = self.surface.get_width()
h = self.surface.get_height()
s = self.surface.get_stride()
data = self.surface.get_data()
buf = self.undo_buffer.cur_buf
self.undo_buffer.Buffer[buf] = create_string_buffer(s*h)
self.undo_buffer.Buffer[buf][:] = data[:]
self.undo_buffer.width[buf] = w
self.undo_buffer.height[buf] = h
if iterate==1:
self.emit("change_sensitivty", senstivity_data('undo',True))
self.emit("change_sensitivty", senstivity_data('redo',False))
self.undo_buffer.redos_allowed = 0
if self.undo_buffer.n_buf_full<self.undo_buffer.n_buf:
self.undo_buffer.n_buf_full += 1
self.undo_buffer.cur_buf = self.undo_buffer.next_buf()
def clear_undo_buffer(self):
self.emit("change_sensitivty", senstivity_data('undo',False))
self.emit("change_sensitivty", senstivity_data('redo',False))
self.undo_buffer.cur_buf = 0
self.undo_buffer.n_buf_full = 0
self.undo_buffer.redos_allowed = 0
def copy(self,cut):
data = self.surface.get_data()
t_data=list(data)
t_data[::4] = data[2::4]
t_data[2::4] = data[::4]
s = self.surface.get_stride()
w = self.surface.get_width()
h = self.surface.get_height()
if self.select_active:
xp= [min(max(0,x),w) for x in self.select_xp]
yp= [min(max(0,y),h) for y in self.select_yp]
c_w= int(max(xp)-min(xp))
c_h= int(max(yp)-min(yp))
if c_h>0 and c_w>0:
c_s = c_w*4
c_data=[t_data[0]]*(c_h*c_s)
c_y = int(min(yp))
c_x = int(min(xp))
for n in range(c_h):
c_data[n*c_s:(n+1)*c_s] = t_data[(n+c_y)*s+c_x*4:(n+c_y)*s+c_x*4+c_s]
c_data = ''.join(c_data)
PixBuf = gtk.gdk.pixbuf_new_from_data(c_data,gtk.gdk.COLORSPACE_RGB, True, 8, c_w,c_h,c_s)
self.clipboard.set_image(PixBuf)
if cut:
self.delete()
else:
t_data = ''.join(t_data)
PixBuf = gtk.gdk.pixbuf_new_from_data(t_data,gtk.gdk.COLORSPACE_RGB, True, 8, w,h,s)
self.clipboard.set_image(PixBuf)
if cut:
self.delete()
def delete(self):
w = self.surface.get_width()
h = self.surface.get_height()
if self.select_active:
xp= [min(max(0,x),w) for x in self.select_xp]
yp= [min(max(0,y),h) for y in self.select_yp]
c_w= int(max(xp)-min(xp))
c_h= int(max(yp)-min(yp))
if c_h>0 and c_w>0:
c_y = int(min(yp))
c_x = int(min(xp))
self.update_undo_buffer(1)
aux = cairo.ImageSurface(cairo.FORMAT_ARGB32, c_w, c_h)
context = cairo.Context(aux)
context.rectangle(0, 0, self.width, self.height)
context.set_source_rgba(self.secondary.get_red(),self.secondary.get_green(),self.secondary.get_blue(),self.secondary.get_alpha())
context.fill()
mask = cairo.ImageSurface(cairo.FORMAT_ARGB32, w, h)
context = cairo.Context(mask)
context.rectangle(0, 0, self.width, self.height)
context.set_source_rgba(1,1,1,0)
context.fill()
context.rectangle(c_x, c_y, c_w, c_h)
context.set_source_rgba(1,1,1,1)
context.fill()
context = cairo.Context(self.surface)
context.set_source_surface(aux, c_x, c_y)
context.set_operator(cairo.OPERATOR_SOURCE)
context.mask(cairo.SurfacePattern(mask))
self.swap_buffers()
else:
data = self.surface.get_data()
self.update_undo_buffer(1)
context = cairo.Context(self.surface)
context.rectangle(0, 0, self.width, self.height)
context.set_source_rgba(self.secondary.get_red(),self.secondary.get_green(),self.secondary.get_blue(),self.secondary.get_alpha())
context.set_operator(cairo.OPERATOR_SOURCE)
context.fill()
#if the context is filled blank the pixels wont edit properly for bucket fill
#The hacky solution is to copy the black pixels to a temporary array
#Paint new pixels, and then copy the blank ones back in.
if self.secondary.get_alpha() == 0:
t_data = data[:]
context = cairo.Context(self.surface)
context.paint()
data[:] = t_data[:]
self.swap_buffers()
def paste(self):
image = self.clipboard.wait_for_image()
if image != None:
self.update_undo_buffer(1)
self.set_size(max(self.width,image.get_width()), max(self.height,image.get_height()))
self.print_tool()
aux = cairo.ImageSurface(cairo.FORMAT_ARGB32, image.get_width(), image.get_height())
im_data = image.get_pixels()
# the next two semingly useless lines give the surface real pixels to be edited.
context = cairo.Context(aux)
context.paint()
#directly write to the pixels in aux
data = aux.get_data()
#pasted image has alpha channel
if image.get_rowstride()==aux.get_stride():
data[2::4] = im_data[0::4]#red
data[0::4] = im_data[2::4] #blue
data[1::4] = im_data[1::4] #green
data[3::4] = im_data[3::4] #alpha
else: #pasted data no alpha channel
data[2::4] = im_data[0::3]#red
data[0::4] = im_data[2::3] #blue
data[1::4] = im_data[1::3] #green
#use cairo to add the pasted image to the current image
context = cairo.Context(self.surface)
context.set_source_surface(aux, 0, 0)
context.paint()
self.swap_buffers()
def crop(self):
if self.select_active:
self.update_undo_buffer(1)
w = self.surface.get_width()
h = self.surface.get_height()
xp= [min(max(0,x),w) for x in self.select_xp]
yp= [min(max(0,y),h) for y in self.select_yp]
c_w= int(max(xp))-int(min(xp))
c_h= int(max(yp))-int(min(yp))
aux = cairo.ImageSurface(cairo.FORMAT_ARGB32, w,h)
context = cairo.Context(aux)
context.set_source_surface(self.surface,0,0)
context.paint()
context = cairo.Context(self.surface)
context.set_source_surface(aux,-int(min(xp)),-int(min(yp)))
context.paint()
self.set_size(c_w,c_h)
self.clear_overlay()
self.print_tool()
def is_modified(self):
return self.modified
def __draw_shadows(self, context):
# Shadow displacements
disp = 2
csw = self.BL_CORNER_SHADOW.get_width()
if self.width > 10:
# Bottom left corner
context.set_source_surface(self.BL_CORNER_SHADOW, disp, self.height)
context.paint()
# Bottom shadow
context.translate(0, self.height)
for i in range(len(self.side_alpha_channels)):
alpha = self.side_alpha_channels[i]
context.rectangle(disp+csw, i, self.width-disp-csw, 1)
context.set_source_rgba(0, 0, 0, alpha)
context.fill()
context.translate(0, -self.height)
if self.height > 10:
# Top right corner
context.set_source_surface(self.TR_CORNER_SHADOW, self.width, disp)
context.paint()
# Side shadow
context.translate(self.width, 0)
for i in range(len(self.side_alpha_channels)):
alpha = self.side_alpha_channels[i]
context.rectangle(i, disp+csw, 1, self.height-disp-csw)
context.set_source_rgba(0, 0, 0, alpha)
context.fill()
context.translate(-self.width, 0)
def __draw_scaling_points(self, context):
# Right scaling point
if self.height > self.RSS*4:
self.__draw_scaling_point(context, self.width,(self.height-self.RSS)/2)
# Bottom scaling point
if self.width > self.RSS*4:
self.__draw_scaling_point(context, (self.width-self.RSS)/2,self.height)
# Corner scaling point
self.__draw_scaling_point(context, self.width, self.height)
def __draw_scaling_point(self, context, x, y):
# Dark border
context.set_source_rgba(0.156, 0.402, 0546, 1)
context.rectangle(x+1, y+1, self.RSS-1, self.RSS-1)
context.fill()
# Light border
context.set_source_rgba(.556, .802, .946, 1)
context.rectangle(x, y, self.RSS-1, self.RSS-1)
context.fill()
# The point itself
context.set_source_rgba(.26,.67,.91,1.0)
context.rectangle(x+1, y+1, self.RSS-2, self.RSS-2)
context.fill()
def __over_scaling_point(self, event):
if self.width < event.x+1 < self.width + self.RSS:
if self.height < event.y+1 < self.height + self.RSS:
return self.CORNER_SCALING_POINT
elif (self.height-self.RSS)/2 < event.y+1 < (self.height+self.RSS)/2:
return self.RIGHT_SCALING_POINT
elif self.height < event.y+1 < self.height + self.RSS:
if (self.width-self.RSS)/2 < event.x+1 < (self.width+self.RSS)/2:
return self.BOTTOM_SCALING_POINT
return 0