Size changing

Author: halcy

AnsiImage.py

@@ -37,7 +37,13 @@ def __init__(self, min_line_len = None):
             for x in [0, 1, 6, 7]:
                 self.cursor_shape.append((y, x))
 
-    def change_size(self, new_height, new_width):
+    def get_size(self):
+        """
+        Return the current canvas size in characters
+        """
+        return (self.width, self.height)
+    
+    def change_size(self, new_width, new_height):
         """
         Make the image larger or smaller, retaining contents
         """

MainWindow.py

@@ -54,6 +54,7 @@ def createMenuBar(self):
         self.actionExit = QtWidgets.QAction("Exit", self)
 
         menuEdit = self.menuBar().addMenu("Edit")
+        self.actionSize = QtWidgets.QAction("Set size", self)
         self.actionUndo = QtWidgets.QAction("Undo", self)
         self.actionRedo = QtWidgets.QAction("Redo", self)
         self.actionCopy = QtWidgets.QAction("Copy", self)
@@ -68,6 +69,8 @@ def createMenuBar(self):
         menuFile.addSeparator()
         menuFile.addAction(self.actionExit)
 
+        menuEdit.addAction(self.actionSize)
+        menuEdit.addSeparator()
         menuEdit.addAction(self.actionUndo)
         menuEdit.addAction(self.actionRedo)
         menuEdit.addSeparator()
@@ -151,6 +154,8 @@ def connectEvents(self):
         self.actionExit.triggered.connect(self.exit)
         self.actionExit.setShortcut(QtGui.QKeySequence.Quit)
 
+        self.actionSize.triggered.connect(self.resizeCanvas)
+        
         self.actionUndo.triggered.connect(self.undo)
         self.actionUndo.setShortcut(QtGui.QKeySequence.Undo)
 
@@ -414,6 +419,18 @@ def clipboardPaste(self):
             self.addUndo(self.ansiImage.paste(self.pasteBuffer))
             self.redisplayAnsi()
 
+    def resizeCanvas(self):
+        """
+        Get size via dialog and resize
+        """
+        cur_width, cur_height =  self.ansiImage.get_size()
+        sizeDialog = SizeDialog(cur_width, cur_height)
+        if sizeDialog.exec() == 1:
+            new_width = sizeDialog.spinBoxWidth.value()
+            new_height = sizeDialog.spinBoxHeight.value()
+            self.ansiImage.change_size(new_width, new_height)
+            self.redisplayAnsi()
+            
     def addUndo(self, operation):
         """
         Add an undo step (and clean out the redo stack)

SizeDialog.py

@@ -0,0 +1,51 @@
+from PyQt5 import QtCore, QtGui, QtWidgets
+
+class SizeDialog(QtWidgets.QDialog):
+    def __init__(self, width, height):
+        super(SizeDialog, self).__init__()
+        self.setModal(True)
+        self.setWindowTitle("Enter new size...")
+        self.setSizeGripEnabled(False)
+        
+        self.labelWidth = QtWidgets.QLabel("Width")
+        self.labelWidth.setMinimumSize(50, 1)
+        
+        self.spinBoxWidth = QtWidgets.QSpinBox()
+        self.spinBoxWidth.setMinimum(1)
+        self.spinBoxWidth.setMaximum(9999)
+        self.spinBoxWidth.setValue(width)
+        
+        self.labelHeight = QtWidgets.QLabel("Height")
+        self.labelHeight.setMinimumSize(50, 1)
+        
+        self.spinBoxHeight = QtWidgets.QSpinBox()
+        self.spinBoxHeight.setMinimum(1)
+        self.spinBoxHeight.setMaximum(99999)
+        self.spinBoxHeight.setValue(height)
+        
+        self.cancelButton = QtWidgets.QPushButton("Cancel")
+        self.cancelButton.clicked.connect(self.reject)
+        
+        self.acceptButton = QtWidgets.QPushButton("Ok!")
+        self.acceptButton.clicked.connect(self.accept)
+        
+        widthLayout = QtWidgets.QHBoxLayout()
+        widthLayout.addWidget(self.labelWidth)
+        widthLayout.addWidget(self.spinBoxWidth)             
+             
+        heightLayout = QtWidgets.QHBoxLayout()
+        heightLayout.addWidget(self.labelHeight)
+        heightLayout.addWidget(self.spinBoxHeight)
+        
+        buttonLayout = QtWidgets.QHBoxLayout()
+        buttonLayout.addWidget(self.cancelButton)
+        buttonLayout.addWidget(self.acceptButton)
+        buttonLayout.setAlignment(QtCore.Qt.AlignRight)
+        
+        mainLayout = QtWidgets.QVBoxLayout()
+        mainLayout.addLayout(widthLayout)
+        mainLayout.addLayout(heightLayout)
+        mainLayout.addLayout(buttonLayout)
+        
+        self.setLayout(mainLayout)
+