1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
# -*- coding: utf-8 -*-
#
# PartMgr - Image
#
# Copyright 2014-2022 Michael Buesch <m@bues.ch>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
from partmgr.core.util import *
class Image:
def __init__(self, data=None):
self.convertFrom(data)
def isNull(self):
return self.pixmap.isNull()
def convertFrom(self, data, dataFormat="PNG"):
self.pixmap = QPixmap()
if not data:
return
if isinstance(data, str):
data = fromBase64(data, toBytes=True)
if not data:
return
if isinstance(data, bytes):
data = QByteArray(data)
if isinstance(data, QByteArray):
pix = QPixmap()
pix.loadFromData(data, dataFormat)
data = pix
if not isinstance(data, QPixmap):
raise PartMgrError("Unsupported Image object type")
self.pixmap = data
def fromFile(self, filename, fileFormat):
try:
fd = open(filename, "rb")
data = fd.read()
fd.close()
except IOError as e:
raise PartMgrError("Failed to read %s: %s" %\
(filename, str(e)))
self.convertFrom(data, fileFormat)
if self.isNull():
raise PartMgrError("%s: Unknown image format" % filename)
def toString(self):
return toBase64(self.toBytes())
def toBytes(self):
return self.toQByteArray().data()
def toQByteArray(self):
ba = QByteArray()
buf = QBuffer(ba)
buf.open(QIODeviceBase.OpenModeFlag.WriteOnly)
self.toPixmap().save(buf, "PNG")
return ba
def toPixmap(self):
return self.pixmap
def size(self):
return self.pixmap.size()
def scaleToMaxSize(self, maxSize):
if self.size().height() <= maxSize.height() and\
self.size().width() <= maxSize.width():
return self
return Image(self.pixmap.scaled(
maxSize,
Qt.AspectRatioMode.KeepAspectRatio,
Qt.TransformationMode.SmoothTransformation))
|
