from krita import *
from PyQt5.QtWidgets import QDialog, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QComboBox, QCheckBox, QPushButton
from PyQt5.QtCore import *

class BoxGenerator(Extension):

    def __init__(self, parent):
        super().__init__(parent)

    # Krita.instance() exists, so do any setup work
    def setup(self):
        pass

    # called after setup(self)
    def createActions(self, window):
        # Create menu item in Tools > Scripts.
        action = window.createAction("boxgen", "Box Generator")
        action.triggered.connect(self.box_generator)

    def box_generator(self):
        # Get the current selected layer, called a 'node'
        self.doc = Krita.instance().activeDocument()

        if not self.doc:
            # Create and show a new document (A4 300ppi).
            self.doc = Krita.instance().createDocument(3508, 2480, "", "RGBA", "U8", "", 300.0)
            Krita.instance().activeWindow().addView(self.doc)

        # Dialog creation.
        newDialog = QDialog()
        newDialog.setWindowTitle("Box Generator")

        # Description row.
        desc = QLabel('Creates outlines for a foldable box with lid.')
        desc.setAlignment(Qt.AlignCenter)
        row0 = QHBoxLayout()
        row0.addWidget(desc)

        # Dimension row.
        self.lengthInput = QLineEdit('67')
        self.widthInput = QLineEdit('93')
        self.heightInput = QLineEdit('20')
        self.unitInput = QComboBox()
        self.unitInput.addItems( ['mm', 'inch', 'px'] )
        row1 = QHBoxLayout()
        row1.addWidget(QLabel('Dimensions: L:'))
        row1.addWidget(self.lengthInput)
        row1.addWidget(QLabel(' x W:'))
        row1.addWidget(self.widthInput)
        row1.addWidget(QLabel(' x H:'))
        row1.addWidget(self.heightInput)
        row1.addWidget(self.unitInput)

        # Decision row.
        self.boxTypeInput = QComboBox()
        self.boxTypeInput.addItems( ['Tuckbox', 'Telescopic'] )
        row2 = QHBoxLayout()
        row2.addWidget(QLabel('Box Type:'))
        row2.addWidget(self.boxTypeInput)

        # Do It row.
        goButton = QPushButton("Create Box")
        goButton.setIcon( Krita.instance().icon('animation_play') )
        row3 = QHBoxLayout()
        row3.addWidget(goButton)

        layoutMain = QVBoxLayout()
        layoutMain.addLayout(row0)
        layoutMain.addLayout(row1)
        layoutMain.addLayout(row2)
        layoutMain.addLayout(row3)
        newDialog.setLayout(layoutMain)

        # hook up the actions.
        goButton.clicked.connect( self.generateBox )
        # self.unitInput.currentIndexChanged.connect( self.convertCount )

        # show the dialog.
        newDialog.exec_()


    ##########
    # Slots
    ##########

    # Actually generates the box!
    def generateBox(self, e):
        length_text = self.lengthInput.text()
        width_text = self.widthInput.text()
        height_text = self.heightInput.text()

        # Box dimensions in pixels.
        doc_ppi = self.doc.resolution()
        mm = doc_ppi/25.4
        multiplier = 1
        if self.unitInput.currentText() == "inch":
            multiplier = doc_ppi
        if self.unitInput.currentText() == "mm":
            multiplier = mm
        length = float(length_text) * multiplier
        height = float(height_text) * multiplier
        width = float(width_text) * multiplier

        layer = self.doc.createVectorLayer(f"""Box {length_text}x{width_text}x{height_text}""")
        root = self.doc.rootNode()
        root.addChildNode(layer, None)

        # Create outline using SVG (starts from top left corner).
        if self.boxTypeInput.currentIndex() == 0: # Tuckbox

            stringy = f"""<svg style="stroke:black; fill:none; stroke-width:1">
                <path d="
                m {1*mm+height*1.66}
                h {length-height*2*0.66}
                a {height*0.7} {height*0.7} 0 0 1 {height*0.66 - 0.7*mm} {height*0.66 - 0.25*mm}
                h -{4*mm}
                v {0.5*mm}
                h {4.7*mm}
                v {height - 0.25*mm}
                l {height*0.1} -{height}
                h {height*0.4}
                a {height*0.5} {height} 0 0 1 {height*0.5-0.25*mm} {height-3.5*mm}
                h {0.6*mm}
                v {3.5*mm}
                h {length*0.5-0.35*mm - 7*mm}
                a {7.3*mm} {7.3*mm} 0 0 0 {14*mm} 0
                h {length*0.5 - 7*mm}
                h {height-0.25*mm}
                v {width-0.25*mm}
                h -{height-0.25*mm}
                v {height-0.25*mm}
                h -{length}
                v -{height-0.5*mm}
                l -{height*0.1} {height}
                h -{height*0.8}
                l -{height*0.1} -{height}
                v {height}
                h -{length}
                v -{height}
                l -{height*0.1} {height}
                h -{height*0.8}
                l -{height*0.1} -{height}
                v -{width}
                h -{0.25*mm}
                v -{3.5*mm}
                h {0.5*mm}
                a {height*0.5} {height} 0 0 1 {height*0.5-0.25*mm} -{height-3.5*mm}
                h {height*0.4}
                l {height*0.1} {height}
                v -{height - 0.25*mm}
                h {4.7*mm}
                v -{0.5*mm}
                h -{4*mm}
                a {height*0.7} {height*0.7} 0 0 1 {height*0.66 - 0.7*mm} -{height*0.66 - 0.25*mm}
                z
                "></path>
                <path d="
                M {1*mm+height} {height*0.66 + 0.25*mm}
                h -{1*mm}
                M {1*mm+height+length} {height*0.66 + 0.25*mm}
                h {1*mm}
                M 0 {height*1.66}
                h {1*mm}
                M 0 {height*1.66+width}
                h {1*mm}
                M {1*mm+height*2+length*2} {height*1.66}
                v -{1*mm}
                M {1*mm+height*2+length*2} {height*1.66+width}
                h {1*mm}
                M {1*mm} {height*1.66 - 3.5*mm}
                v -{1*mm}
                M {1*mm+height*2+length} {height*1.66 - 3.5*mm}
                v -{1*mm}
                M {1*mm+height*2+length} {height*1.66}
                h {0.35*mm}
                "></path>
                <rect x="{1*mm+height*2+length+2.5*mm}"
                y="{height*1.66+width+2.5*mm}" width="{length-5*mm}" height="{height-0.5*mm - 3*mm}"></rect>
                <rect x="{1*mm+height*2+length*2+2.5*mm}" y="{height*1.66+2.5*mm}" width="{height-0.25*mm - 3*mm}" height="{width-5*mm}"></rect>
                </svg>"""

        else: # Telescopic
            total_length = length + 5 * height
            total_width = width + 5 * height
            stringy = f"""<svg width="{total_length}" height="{total_width}" style="stroke:black; fill:none; stroke-width:1">
                <path d="
                m {height*3}
                h {length-height}
                l {height/2} {height/2}
                v {height}
                a {height} {height} 0 0 1 {height} {height}
                h {height}
                l {height/2} {height/2}
                v {width-height}
                l -{height/2} {height/2}
                h -{height}
                a {height} {height} 0 0 1 -{height} {height}
                v {height}
                l -{height/2} {height/2}
                h -{length-height}
                l -{height/2} -{height/2}
                v -{height}
                a {height} {height} 0 0 1 -{height} -{height}
                h -{height}
                l -{height/2} -{height/2}
                v -{width-height}
                l {height/2} -{height/2}
                h {height}
                a {height} {height} 0 0 1 {height} -{height}
                v -{height}
                Z
                "></path>
                </svg>"""

        layer.addShapesFromSvg(stringy)