def test_area(self):
r = Rect(Point(0, 0), Size(10, 10))
self.assertEqual(r.area(), 100)
r = Rect(Point(10, 10), Size(20, 20))
self.assertEqual(r.area(), 400)
r = Rect(Point(20, 20), Size(50, 50))
self.assertEqual(r.area(), 2500)
r = Rect(Point(100, 100), Size(200, 300))
self.assertEqual(r.area(), 60000)
class Image(Rect):
"""
self.origin - показывает положение в атласе
self.size - оригинальный размер изображения
self.crop - область изображения взятого из оригинала после обрезания прозрачных краёв
"""
BorderPadding = 1
CropThreshold = 10
def __init__(self, directory, filename):
self.directory = directory
self.filename = filename
image = QImage(self.directory + QDir.separator() + self.filename)
Rect.__init__(self, Point(), Size(image.width(), image.height()))
cropped_image, x, y, width, height = self.__getCrop(image, False)
self.crop = Rect(Point(x, y), Size(width, height))
self.rotated = False
def area(self):
return self.crop.area()
def draw(self, painter, offset=Point()):
cropped_image, x, y, width, height = self.__getCrop()
origin_offset = QPoint(self.origin.x + offset.x, self.origin.y + offset.y)
if self.rotated:
old_transform = QTransform(painter.transform())
painter.setTransform(painter.transform().translate(origin_offset.x() + self.crop.size.height,
origin_offset.y()).rotate(90, Qt.ZAxis))
painter.drawImage(0, 0, cropped_image)
painter.setTransform(old_transform)
else:
painter.drawImage(origin_offset, cropped_image)
# Rect(self.origin, self.crop.size).draw(painter, offset)
def toJson(self):
return {
'offset': {
'x': self.origin.x,
'y': self.origin.y
},
'original_size': {
'width': self.size.width,
'height': self.size.height
},
'crop': {
'origin': {
'x': self.crop.origin.x,
'y': self.crop.origin.y
},
'size': {
'width': self.crop.size.width,
'height': self.crop.size.height
}
},
'rotated': self.rotated,
'filename': self.filename
}
# noinspection PyUnreachableCode
def __str__(self):
return '{klass}({file}, {origin}, {size}, {crop}, {rotated})'.format(klass=self.__class__.__name__,
file=self.filename,
crop=self.crop,
origin=self.origin,
size=self.size,
rotated=self.rotated)
def __repr__(self):
return self.__str__()
def __getCrop(self, image=None, loadCrop=True):
if self.__hasCrop():
cropped_image, crop_info = self.__loadCrop(loadCrop)
x, y, width, height = crop_info['x'], crop_info['y'], crop_info['width'], crop_info['height']
else:
if not image:
raise RuntimeError('Image is none')
cropped_image, x, y, width, height = crop_image(image, Image.CropThreshold)
self.__saveCrop(cropped_image, x, y, width, height)
return cropped_image, x, y, width, height
def __fileNameWoExt(self):
image = QFileInfo(self.directory + QDir.separator() + self.filename)
return image.absoluteDir().absolutePath() + QDir.separator() + image.baseName()
def __hasCrop(self):
name_wo_ext = self.__fileNameWoExt()
return QFileInfo(name_wo_ext + '.json').exists() and QFileInfo(name_wo_ext + '.crop').exists()
def __saveCrop(self, image, x, y, width, height):
image.save(self.__fileNameWoExt() + '.crop', 'PNG')
with open(self.__fileNameWoExt() + '.json', 'w') as f:
json.dump({
'x': x,
'y': y,
'width': width,
'height': height
}, fp=f)
def __loadCrop(self, loadCrop=True):
name_wo_ext = self.__fileNameWoExt()
json_data = open(name_wo_ext + '.json').read()
image = QImage()
if loadCrop:
image.load(name_wo_ext + '.crop', 'PNG')
return image, json.loads(json_data)
