def export(self, filename, w=1920, h=1080):
"""
Save the grid to a png file
"""
name, extension = os.path.splitext(filename.lower())
if extension == '.png':
image = QImage(w, h, QImage.Format_ARGB32_Premultiplied)
image.fill(Qt.transparent)
painter = QPainter(image)
painter.setRenderHint(QPainter.Antialiasing)
self.diagramScene.render(painter)
image.save(filename)
painter.end()
elif extension == '.svg':
svg_gen = QSvgGenerator()
svg_gen.setFileName(filename)
svg_gen.setSize(QSize(w, h))
svg_gen.setViewBox(QRect(0, 0, w, h))
svg_gen.setTitle("Electrical grid schematic")
svg_gen.setDescription("An SVG drawing created by GridCal")
painter = QPainter(svg_gen)
self.diagramScene.render(painter)
painter.end()
else:
raise Exception('Extension ' + str(extension) + ' not supported :(')
def export_to_svg(self, filename: Path) -> bool:
generator = QSvgGenerator()
generator.setFileName(str(filename))
generator.setSize(self.size())
generator.setViewBox(self.rect())
self.render(generator)
return True