mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-06-23 15:30:45 -04:00
Start work on image editor
This commit is contained in:
Kovid Goyal
parent
8a36b10085
commit
d12fa98765
297
src/calibre/gui2/tweak_book/editor/image.py
Normal file
297
src/calibre/gui2/tweak_book/editor/image.py
Normal file
@ -0,0 +1,297 @@
|
||||
#!/usr/bin/env python
|
||||
# vim:fileencoding=utf-8
|
||||
from __future__ import (unicode_literals, division, absolute_import,
|
||||
print_function)
|
||||
|
||||
__license__ = 'GPL v3'
|
||||
__copyright__ = '2013, Kovid Goyal <kovid at kovidgoyal.net>'
|
||||
|
||||
import sys, string
|
||||
from functools import wraps
|
||||
|
||||
from PyQt4.Qt import (
|
||||
QWidget, QImage, QPainter, QColor, QApplication, Qt, QPixmap, QRectF,
|
||||
QPointF, QPen)
|
||||
|
||||
from calibre import fit_image
|
||||
|
||||
def painter(func):
|
||||
@wraps(func)
|
||||
def ans(self, painter):
|
||||
painter.save()
|
||||
try:
|
||||
return func(self, painter)
|
||||
finally:
|
||||
painter.restore()
|
||||
return ans
|
||||
|
||||
ucase_map = {l:string.ascii_uppercase[i] for i, l in enumerate(string.ascii_lowercase)}
|
||||
def capitalize(x):
|
||||
return ucase_map[x[0]] + x[1:]
|
||||
|
||||
class Canvas(QWidget):
|
||||
|
||||
BACKGROUND = QColor(60, 60, 60)
|
||||
SHADE_COLOR = QColor(0, 0, 0, 180)
|
||||
SELECT_PEN = QPen(QColor(Qt.white))
|
||||
|
||||
def __init__(self, parent=None):
|
||||
QWidget.__init__(self, parent)
|
||||
self.setMouseTracking(True)
|
||||
|
||||
self.current_image_data = None
|
||||
self.current_image = None
|
||||
self.current_scaled_pixmap = None
|
||||
self.last_canvas_size = None
|
||||
self.last_press_point = None
|
||||
self.last_move_point = None
|
||||
self.target = QRectF(0, 0, 0, 0)
|
||||
self.current_mode = None
|
||||
self.selection_rect = None
|
||||
self.in_selection = False
|
||||
self.drag_corner = (None, None)
|
||||
self.last_drag_pos = None
|
||||
self.drag_mode = None
|
||||
|
||||
def show_image(self, data):
|
||||
self.current_image_data = data
|
||||
self.current_image = i = QImage()
|
||||
i.loadFromData(data)
|
||||
self.is_valid = not i.isNull()
|
||||
self.update()
|
||||
|
||||
@property
|
||||
def dc_size(self):
|
||||
sr = self.selection_rect
|
||||
dx = min(75, sr.width() / 4)
|
||||
dy = min(75, sr.height() / 4)
|
||||
return dx, dy
|
||||
|
||||
def get_drag_corner(self, pos):
|
||||
dx, dy = self.dc_size
|
||||
sr = self.selection_rect
|
||||
x, y = pos.x(), pos.y()
|
||||
hedge = 'left' if x < sr.x() + dx else 'right' if x > sr.right() - dx else None
|
||||
vedge = 'top' if y < sr.y() + dy else 'bottom' if y > sr.bottom() - dy else None
|
||||
return (hedge, vedge)
|
||||
|
||||
def get_drag_rect(self):
|
||||
sr = self.selection_rect
|
||||
dc = self.drag_corner
|
||||
if sr is None or dc == (None, None):
|
||||
return
|
||||
dx, dy = self.dc_size
|
||||
if None in dc:
|
||||
# An edge
|
||||
if dc[0] is None:
|
||||
top = sr.top() if dc[1] == 'top' else sr.bottom() - dy
|
||||
ans = QRectF(sr.left() + dx, top, sr.width() - 2 * dx, dy)
|
||||
else:
|
||||
left = sr.left() if dc[0] == 'left' else sr.right() - dx
|
||||
ans = QRectF(left, sr.top() + dy, dx, sr.height() - 2 * dy)
|
||||
else:
|
||||
# A corner
|
||||
left = sr.left() if dc[0] == 'left' else sr.right() - dx
|
||||
top = sr.top() if dc[1] == 'top' else sr.bottom() - dy
|
||||
ans = QRectF(left, top, dx, dy)
|
||||
return ans
|
||||
|
||||
def get_cursor(self):
|
||||
dc = self.drag_corner
|
||||
if dc == (None, None):
|
||||
ans = Qt.CrossCursor
|
||||
elif None in dc:
|
||||
ans = Qt.SizeVerCursor if dc[0] is None else Qt.SizeHorCursor
|
||||
else:
|
||||
ans = Qt.SizeBDiagCursor if dc in {('left', 'bottom'), ('right', 'top')} else Qt.SizeFDiagCursor
|
||||
return ans
|
||||
|
||||
def move_edge(self, edge, dp):
|
||||
sr = self.selection_rect
|
||||
horiz = edge in {'left', 'right'}
|
||||
func = getattr(sr, 'set' + capitalize(edge))
|
||||
delta = getattr(dp, 'x' if horiz else 'y')()
|
||||
buf = 50
|
||||
if horiz:
|
||||
minv = self.target.left() if edge == 'left' else sr.left() + buf
|
||||
maxv = sr.right() - buf if edge == 'left' else self.target.right()
|
||||
else:
|
||||
minv = self.target.top() if edge == 'top' else sr.top() + buf
|
||||
maxv = sr.bottom() - buf if edge == 'top' else self.target.bottom()
|
||||
func(max(minv, min(maxv, delta + getattr(sr, edge)())))
|
||||
|
||||
def move_selection(self, dp):
|
||||
sr = self.selection_rect
|
||||
dm = self.drag_mode
|
||||
if dm == (None, None):
|
||||
half_width = sr.width() / 2.0
|
||||
half_height = sr.height() / 2.0
|
||||
c = sr.center()
|
||||
nx = c.x() + dp.x()
|
||||
ny = c.y() + dp.y()
|
||||
minx = self.target.left() + half_width
|
||||
maxx = self.target.right() - half_width
|
||||
miny, maxy = self.target.top() + half_height, self.target.bottom() - half_height
|
||||
nx = max(minx, min(maxx, nx))
|
||||
ny = max(miny, min(maxy, ny))
|
||||
sr.moveCenter(QPointF(nx, ny))
|
||||
else:
|
||||
for edge in dm:
|
||||
if edge is not None:
|
||||
self.move_edge(edge, dp)
|
||||
|
||||
def mousePressEvent(self, ev):
|
||||
if ev.button() == Qt.LeftButton and self.target.contains(ev.pos()):
|
||||
pos = ev.pos()
|
||||
self.last_press_point = pos
|
||||
if self.current_mode is None:
|
||||
self.current_mode = 'select'
|
||||
|
||||
elif self.current_mode == 'selected':
|
||||
if self.selection_rect is not None and self.selection_rect.contains(pos):
|
||||
self.drag_mode = self.get_drag_corner(pos)
|
||||
self.last_drag_pos = pos
|
||||
else:
|
||||
self.current_mode = 'select'
|
||||
self.selection_rect = None
|
||||
|
||||
def mouseMoveEvent(self, ev):
|
||||
changed = False
|
||||
if self.in_selection:
|
||||
changed = True
|
||||
self.in_selection = False
|
||||
self.drag_corner = (None, None)
|
||||
pos = ev.pos()
|
||||
cursor = Qt.ArrowCursor
|
||||
try:
|
||||
if not self.target.contains(pos):
|
||||
return
|
||||
if ev.buttons() & Qt.LeftButton:
|
||||
if self.last_press_point is not None:
|
||||
if self.current_mode == 'select':
|
||||
self.selection_rect = QRectF(self.last_press_point, pos).normalized()
|
||||
changed = True
|
||||
elif self.drag_mode is not None and self.last_drag_pos is not None:
|
||||
self.drag_corner = self.drag_mode
|
||||
self.in_selection = True
|
||||
dp = pos - self.last_drag_pos
|
||||
cursor = self.get_cursor()
|
||||
changed = True
|
||||
self.last_drag_pos = pos
|
||||
self.move_selection(dp)
|
||||
else:
|
||||
if self.selection_rect is None or not self.selection_rect.contains(pos):
|
||||
return
|
||||
if self.current_mode == 'selected':
|
||||
if self.selection_rect is not None and self.selection_rect.contains(pos):
|
||||
self.drag_corner = self.get_drag_corner(pos)
|
||||
self.in_selection = True
|
||||
cursor = self.get_cursor()
|
||||
changed = True
|
||||
finally:
|
||||
if changed:
|
||||
self.update()
|
||||
self.setCursor(cursor)
|
||||
|
||||
def mouseReleaseEvent(self, ev):
|
||||
if ev.button() == Qt.LeftButton:
|
||||
self.last_press_point = None
|
||||
self.drag_mode = None
|
||||
self.last_drag_pos = None
|
||||
if self.current_mode == 'select':
|
||||
self.current_mode = 'selected'
|
||||
self.update()
|
||||
|
||||
@painter
|
||||
def draw_background(self, painter):
|
||||
painter.fillRect(self.rect(), self.BACKGROUND)
|
||||
|
||||
@painter
|
||||
def draw_image_error(self, painter):
|
||||
font = painter.font()
|
||||
font.setPointSize(3 * font.pointSize())
|
||||
font.setBold(True)
|
||||
painter.setFont(font)
|
||||
painter.setPen(QColor(Qt.black))
|
||||
painter.drawText(self.rect(), Qt.AlignCenter, _('Not a valid image'))
|
||||
|
||||
def load_pixmap(self):
|
||||
canvas_size = self.rect().width(), self.rect().height()
|
||||
if self.last_canvas_size != canvas_size:
|
||||
if self.last_canvas_size is not None and self.selection_rect is not None:
|
||||
self.current_mode = None
|
||||
self.selection_rect = None
|
||||
# TODO: Migrate the selection rect
|
||||
self.last_canvas_size = canvas_size
|
||||
self.current_scaled_pixmap = None
|
||||
if self.current_scaled_pixmap is None:
|
||||
pwidth, pheight = self.last_canvas_size
|
||||
i = self.current_image
|
||||
width, height = i.width(), i.height()
|
||||
scaled, width, height = fit_image(width, height, pwidth, pheight)
|
||||
if scaled:
|
||||
i = self.current_image.scaled(width, height, transformMode=Qt.SmoothTransformation)
|
||||
self.current_scaled_pixmap = QPixmap.fromImage(i)
|
||||
|
||||
@painter
|
||||
def draw_pixmap(self, painter):
|
||||
p = self.current_scaled_pixmap
|
||||
width, height = p.width(), p.height()
|
||||
pwidth, pheight = self.last_canvas_size
|
||||
x = int(abs(pwidth - width)/2.)
|
||||
y = int(abs(pheight - height)/2.)
|
||||
self.target = QRectF(x, y, width, height)
|
||||
painter.drawPixmap(self.target, p, QRectF(p.rect()))
|
||||
|
||||
@painter
|
||||
def draw_selection_rect(self, painter):
|
||||
cr, sr = self.target, self.selection_rect
|
||||
painter.setPen(self.SELECT_PEN)
|
||||
if self.current_mode == 'selected':
|
||||
# Shade out areas outside the selection rect
|
||||
for r in (
|
||||
QRectF(cr.topLeft(), QPointF(sr.left(), cr.bottom())), # left
|
||||
QRectF(QPointF(sr.left(), cr.top()), sr.topRight()), # top
|
||||
QRectF(QPointF(sr.right(), cr.top()), cr.bottomRight()), # right
|
||||
QRectF(sr.bottomLeft(), QPointF(sr.right(), cr.bottom())), # bottom
|
||||
):
|
||||
painter.fillRect(r, self.SHADE_COLOR)
|
||||
|
||||
dr = self.get_drag_rect()
|
||||
if self.in_selection and dr is not None:
|
||||
# Draw the resize rectangle
|
||||
painter.save()
|
||||
painter.setCompositionMode(QPainter.RasterOp_SourceAndNotDestination)
|
||||
painter.setClipRect(sr)
|
||||
painter.drawRect(dr)
|
||||
painter.restore()
|
||||
|
||||
# Draw the selection rectangle
|
||||
painter.setCompositionMode(QPainter.RasterOp_SourceAndNotDestination)
|
||||
painter.drawRect(sr)
|
||||
|
||||
def paintEvent(self, event):
|
||||
QWidget.paintEvent(self, event)
|
||||
p = QPainter(self)
|
||||
p.setRenderHints(QPainter.Antialiasing | QPainter.SmoothPixmapTransform)
|
||||
try:
|
||||
self.draw_background(p)
|
||||
if self.current_image_data is None:
|
||||
return
|
||||
if not self.is_valid:
|
||||
return self.draw_image_error(p)
|
||||
self.load_pixmap()
|
||||
self.draw_pixmap(p)
|
||||
if self.selection_rect is not None:
|
||||
self.draw_selection_rect(p)
|
||||
finally:
|
||||
p.end()
|
||||
|
||||
if __name__ == '__main__':
|
||||
app = QApplication([])
|
||||
with open(sys.argv[-1], 'rb') as f:
|
||||
data = f.read()
|
||||
c = Canvas()
|
||||
c.show_image(data)
|
||||
c.show()
|
||||
app.exec_()
|
||||
Loading…
x
Reference in New Issue
Block a user