KCWI_Offset.sin

#! @KPYTHON3@


import sys,os
import subprocess
from PyQt5.QtWidgets import QCheckBox,QFrame,QLabel,QHBoxLayout,QLineEdit,QPushButton,QVBoxLayout,QApplication,QWidget, QTextEdit, QGridLayout



def main():
    app = QApplication(sys.argv)
    w = MyWindow()
    w.show()
    sys.exit(app.exec_())


class separator(QFrame):
    def __init__(self):
        super().__init__()
        self.setFrameShape(QFrame.HLine)
        self.setFrameShadow(QFrame.Sunken)
        self.setLineWidth(3)


class MyWindow(QWidget):
    def __init__(self, *args):
        super().__init__()
        #self.runMode = 'debug'
        self.objectMoveColor = 'Gold'
        self.telescopeMoveColor = 'SkyBlue'
        self.slicerMoveColor = 'Plum'
        self.takeGuiderImageColor = 'LawnGreen'

        self.runMode = ''
        self.init_ui()
        self.setWindowTitle("KCWI Target Alignment")

    def init_ui(self):
        # create objects
        #self.lbl = QLabel('Offsetig mode')
        #self.en = QRadioButton('East-North (arcseconds')
        #self.xy = QRadioButton('Magiq coordinates (arcseconds)')
        #self.xyp = QRadioButton('Magiq coordinates (pixels)')
        #self.sl = QRadioButton('Slicer (slices)')
        # layout for radio buttons
        #v1_layout = QVBoxLayout()
        #v1_layout.addWidget(self.en)
        #v1_layout.addWidget(self.xy)
        #v1_layout.addWidget(self.xyp)
        #v1_layout.addWidget(self.sl)
        separator1 = separator()
        separator2 = separator()
        separator3 = separator()
        separator4 = separator()

        # value of offset
        self.lbl1 = QLabel('Value')
        self.value = QLineEdit()
        self.value.setText('1')
        self.lbl5 = QLabel('positive value, arcseconds or slices')
        h1_layout = QHBoxLayout()
        h1_layout.addWidget(self.lbl1)
        h1_layout.addWidget(self.value)
        h1_layout.addWidget(self.lbl5)

        # direction of offset
        self.lbl2 = QLabel('Move object in guider coordinates')
        self.up = QPushButton('Up')
        self.down = QPushButton('Down')
        self.left = QPushButton('Left')
        self.right = QPushButton('Right')
        list = [self.up, self.down, self.left, self.right]
        for button in list:
            button.setStyleSheet("background-color : %s" % self.objectMoveColor)

        g2_layout = QGridLayout()
        g2_layout.addWidget(self.lbl2,0,0,1,3)
        g2_layout.addWidget(self.up,1,1)
        g2_layout.addWidget(self.down,3,1)
        g2_layout.addWidget(self.left,2,0)
        g2_layout.addWidget(self.right,2,2)

        # direction of offset
        self.lbl3 = QLabel('Move telescope in absolute coordinates')
        self.e = QPushButton('E')
        self.w = QPushButton('W')
        self.n = QPushButton('N')
        self.s = QPushButton('S')
        list = [self.e, self.w, self.n, self.s]
        for button in list:
            button.setStyleSheet("background-color : %s" % self.telescopeMoveColor)
        

        g3_layout = QGridLayout()
        g3_layout.addWidget(self.lbl3, 0, 0, 1, 3)
        g3_layout.addWidget(self.e, 2, 0)
        g3_layout.addWidget(self.w, 2, 2)
        g3_layout.addWidget(self.n, 1, 1)
        g3_layout.addWidget(self.s, 3, 1)

        # direction of offset
        self.lbl4 = QLabel('Move object across slicer')
        self.sliceleft = QPushButton('Slice Left')
        self.sliceright = QPushButton('Slice Right')
        list = [self.sliceleft, self.sliceright]
        for button in list:
            button.setStyleSheet("background-color : %s" % self.slicerMoveColor)
        h4_layout = QHBoxLayout()
        h4_layout.addWidget(self.lbl4)
        h4_layout.addWidget(self.sliceleft)
        h4_layout.addWidget(self.sliceright)


        self.te = QTextEdit()
        self.take_guider = QCheckBox('Take guider image after every move')
        self.test_mode = QCheckBox('Test mode (show command, no moves)')
        # layout
        v_layout = QVBoxLayout(self)
        #v_layout.addLayout(v1_layout)
        v_layout.addLayout(h1_layout)
        v_layout.addWidget(separator1)
        v_layout.addLayout(g2_layout)
        v_layout.addWidget(separator2)
        v_layout.addLayout(g3_layout)
        v_layout.addWidget(separator3)
        
        v_layout.addLayout(h4_layout)
        v_layout.addWidget(separator4)
        self.take_guider_button = QPushButton('Take Guider Image')
        self.take_guider_button.setStyleSheet("background-color: %s" % self.takeGuiderImageColor)

        v_layout.addWidget(self.take_guider_button)

        v_layout.addWidget(self.te)
        v_layout.addWidget(self.take_guider)
        v_layout.addWidget(self.test_mode)

        self.setLayout(v_layout)

        # associate action
        buttons = [self.up, self.down, self.right, self.left, self.e, self.w, self.s, self.n, self.sliceleft, self.sliceright, self.take_guider_button]
        for button in buttons:
                button.clicked.connect(self.btn_click)


    def run_command(self, command):
        try:
            kroot = os.environ['KROOT']
        except:
            kroot = ''
        cmdline = os.path.join(kroot,'rel','default','bin',command)
        if self.test_mode.isChecked():
            self.runMode = 'debug'
        else:
            self.runMode = 'normal'
            
        if self.runMode is not 'debug':
            p = subprocess.Popen(cmdline, stdout = subprocess.PIPE,stderr = subprocess.PIPE, shell=True)
            output, errors = p.communicate()
            if len(errors)>0:
                    output = output + errors
            self.te.setText(str(output.decode()))
        else:
            self.te.setText(str(cmdline))

    def btn_click(self):
        sender = self.sender()
        value = self.value.text()

        if sender.text() == 'Up':
            command = '%s 0 -%f' % ('gxy', float(value))
        elif sender.text() == 'Down':
            command = '%s 0 %f' % ('gxy', float(value))
        elif sender.text() == 'Left':
            command = '%s %f 0' % ('gxy', float(value))
        elif sender.text() == 'Right':
            command = '%s -%f 0' % ('gxy', float(value))
        elif sender.text() == 'E':
            command = '%s %f 0' % ('en', float(value))
        elif sender.text() == 'W':
            command = '%s -%f 0' % ('en', float(value))
        elif sender.text() == 'N':
            command = '%s 0 %f' % ('en', float(value))
        elif sender.text() == 'S':
            command = '%s 0 -%f' % ('en', float(value))
        elif sender.text() == 'Slice Left':
            command = '%s left %f' % ('moveSlicer',float(value))
        elif sender.text() == 'Slice Right':
            command = '%s right %f' % ('moveSlicer',float(value))
        elif sender.text() == 'Take Guider Image':
            command = 'saveGuiderImageLocal'
        else:
            command = None

        if command is not None:
            self.run_command(command)

        if self.take_guider.isChecked():
            self.run_command('saveGuiderImageLocal')




if __name__ == "__main__":
    main()