Save images with same size as defined sensor resolution

fishui/widgets/FisheyeWidget.py

@@ -46,17 +46,18 @@ class FisheyeWidget(QWidget):
 
     @Slot()
     def saveImage(self):
-        filename = QFileDialog.getSaveFileName(self, "Save file", ".", ".png")[0]
+        filename = QFileDialog.getSaveFileName(self, "Save file", ".", "Images (*.png)")[0]
         if len(filename) == 0:
             return
         if not filename.endswith(".png"):
             filename = filename + ".png"
-        height = 1080
-        scale = 1080 / self.size().height()
-        width = scale * self.size().width()
+        width = self.viewModel.sensor_size_px[0]
+        height = self.viewModel.sensor_size_px[1]
+        scaleW = width / self.size().width()
+        scaleH = height / self.size().height()
         image = QImage(width, height, QImage.Format_RGB32)
         painter = QPainter(image)
-        painter.scale(scale, scale)
+        painter.scale(scaleW, scaleH)
         self.render(painter, QPoint())
         painter.end()
         image.save(filename)

fishui/widgets/PerspectiveWidget.py

@@ -10,6 +10,7 @@ class PerspectiveWidget(QWidget):
 
     def __init__(self, parent, viewModel: MainViewModel):
         super().__init__(parent)
+        self.viewModel = viewModel
         self.perspectiveBackgroundWidget = PerspectiveBackgroundWidget(self, viewModel)
         self.perspectiveBlockWidget = PerspectiveBlockWidget(self, viewModel)
         self.show()
@@ -33,17 +34,18 @@ class PerspectiveWidget(QWidget):
 
     @Slot()
     def saveImage(self):
-        filename = QFileDialog.getSaveFileName(self, "Save file", ".", ".png")[0]
+        filename = QFileDialog.getSaveFileName(self, "Save file", ".", "Images (*.png)")[0]
         if len(filename) == 0:
             return
         if not filename.endswith(".png"):
             filename = filename + ".png"
-        height = 1080
-        scale = 1080 / self.size().height()
-        width = scale * self.size().width()
+        width = self.viewModel.sensor_size_px[0]
+        height = self.viewModel.sensor_size_px[1]
+        scaleW = width / self.size().width()
+        scaleH = height / self.size().height()
         image = QImage(width, height, QImage.Format_RGB32)
         painter = QPainter(image)
-        painter.scale(scale, scale)
+        painter.scale(scaleW, scaleH)
         self.render(painter, QPoint())
         painter.end()
         image.save(filename)