Exemplo n.º 1
0
    def __init__(self, parent=None, infile=None, inimage=None, center_ra=None, center_dec=None, position_angle=None):
        QtGui.QWidget.__init__(self, parent)
        
        #set up the main UI
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
#        OptimizeTab.__init__(self,self.ui)
        #set up the slitmask
        self.slitmask=SlitMask(center_ra=center_ra, center_dec=center_dec, position_angle=position_angle )
        self.slitlets=self.slitmask.slitlets

        #setup the image interaction
        self.imagedisplay=ImageDisplay(target='pySlitMask:5909')
        self.position_angle=position_angle 
        if inimage:
           self.loadimage(inimage)
        #set up some variables that will be needed later
        self.xmlfile=None
        self.fcfile=None

        #read in the input data if available
        self.infile=infile 
        if infile:
            self.ui.radioButtonInfo_Catalogue.setChecked(True)
            self.setmode2cat()
            self.entercatalog(infile)
            print self.slitmask.center_ra, self.slitmask.center_dec
            if self.slitmask.center_ra is None and self.slitmask.center_dec is None:
                self.slitmask.set_MaskPosition()
                self.displayfootprint()

            #if self.slitmask.center_ra and self.slitmask.center_dec:
                #self.imagedisplay.rssregion(self.slitmask.center_ra, self.slitmask.center_dec)
            
            self.ui.lineEditMain_CenRA.setText(str(self.slitmask.center_ra))
            self.ui.lineEditMain_CenDEC.setText(str(self.slitmask.center_dec))
            self.ui.lineEditMain_PA.setText(str(self.slitmask.position_angle))
            self.ui.lineEditMain_Equinox.setText(str(self.slitmask.equinox))
            
            self.ui.lineEditMain_TargetName.setText(self.slitmask.target_name)
            self.ui.lineEditMain_MaskName.setText(self.slitmask.mask_name)
            self.ui.lineEditInfo_Creator.setText(self.slitmask.creator)
            self.ui.lineEditInfo_Proposer.setText(self.slitmask.proposer)
            self.ui.lineEditInfo_ProposalCode.setText(self.slitmask.proposal_code)
        else:
            self.ui.radioButtonInfo_Catalogue.setChecked(False)
            self.ui.radioButtonInfo_Manual.setChecked(True)
            self.setmode2manual()
            self.ui.toolButtonCat_Load.setEnabled(True)



              #self.displayfootprint()

#        self.opttab = OptimizeTab()
        # setup default values for the optimizer
        self.opt_yspacing = 1.
        self.opt_iter = 10
        self.ui.lineEditOpt_Yspacing.setText(str(self.opt_yspacing))
        self.ui.lineEditOpt_Niter.setText(str(self.opt_iter))

#        self.slitmask.outFoV()
#        print self.slitlets.data['fov_flag']
#        self.updatetabs()
        #Listen to different signals

        #menu items
        QtCore.QObject.connect(self.ui.actionLoad_Catalogue, QtCore.SIGNAL("triggered()"), self.loadcatalog)
        QtCore.QObject.connect(self.ui.actionLoad_Image, QtCore.SIGNAL("triggered()"), self.loadimage)

        #main tabs
        QtCore.QObject.connect(self.ui.lineEditMain_CenRA, QtCore.SIGNAL("editingFinished()"), self.loadCenRA)
        QtCore.QObject.connect(self.ui.lineEditMain_CenDEC, QtCore.SIGNAL("editingFinished()"), self.loadCenDEC)
        QtCore.QObject.connect(self.ui.lineEditMain_PA, QtCore.SIGNAL("editingFinished()"), self.loadpositionangle)
        QtCore.QObject.connect(self.ui.lineEditMain_Equinox, QtCore.SIGNAL("editingFinished()"), self.loadequinox)
        QtCore.QObject.connect(self.ui.lineEditMain_TargetName, QtCore.SIGNAL("editingFinished()"), self.loadtargetname)
        QtCore.QObject.connect(self.ui.lineEditMain_MaskName, QtCore.SIGNAL("editingFinished()"), self.loadmaskname)
        
        #info tabs
        QtCore.QObject.connect(self.ui.lineEditInfo_ProposalCode, QtCore.SIGNAL("editingFinished()"), self.loadproposalcode)
        QtCore.QObject.connect(self.ui.lineEditInfo_Proposer, QtCore.SIGNAL("editingFinished()"), self.loadproposer)
        QtCore.QObject.connect(self.ui.lineEditInfo_Creator, QtCore.SIGNAL("editingFinished()"), self.loadcreator)

#        QtCore.QObject.connect(self.slitmask, SIGNAL('xmlloaded'), self.setcreator)

        QtCore.QObject.connect(self.ui.radioButtonInfo_Catalogue, QtCore.SIGNAL("clicked()"), self.setmode2cat)
        QtCore.QObject.connect(self.ui.radioButtonInfo_Manual, QtCore.SIGNAL("clicked()"), self.setmode2manual)
        QtCore.QObject.connect(self.ui.checkBoxInfo_CentroidOn, QtCore.SIGNAL("clicked()"), self.setmodecentroiding)

        
        
        #catalog tabs
        QtCore.QObject.connect(self.ui.toolButtonCat_Load, QtCore.SIGNAL("clicked(bool)"), self.loadcatalog)
        QtCore.QObject.connect(self.ui.pushButtonCat_AddSlits, QtCore.SIGNAL("clicked(bool)"), self.addslitfromcatalog)
        QtCore.QObject.connect(self.ui.pushButtonCat_Clear, QtCore.SIGNAL("clicked()"), self.clearContents)

        #slit tab
        QtCore.QObject.connect(self.ui.pushButtonSlit_ClearSlits, QtCore.SIGNAL("clicked()"), self.clearslittable)
        QtCore.QObject.connect(self.ui.pushButtonSlit_AddSlitImage, QtCore.SIGNAL("clicked()"), self.addslitletsfromimage)
        QtCore.QObject.connect(self.ui.pushButtonSlit_AddSlitfromCat, QtCore.SIGNAL("clicked()"), self.addslitletsfromcatalogue)
        QtCore.QObject.connect(self.ui.pushButtonSlit_AddSlit, QtCore.SIGNAL("clicked()"), self.addslitmanually)
        QtCore.QObject.connect(self.ui.pushButtonSlit_DeleteSlit, QtCore.SIGNAL("clicked()"), self.deleteslitmanually)
        QtCore.QObject.connect(self.ui.pushButtonSlit_DeleteSlitImage, QtCore.SIGNAL("clicked()"), self.deleteslitfromimage)
        QtCore.QObject.connect(self.ui.tableWidgetSlits, QtCore.SIGNAL("itemSelectionChanged()"), self.setposition)
        QtCore.QObject.connect(self.ui.tableWidgetSlits, QtCore.SIGNAL("cellChanged(int, int)"), self.slitchanged)

        #optimize tab
        QtCore.QObject.connect(self.ui.pushButtonOpt_Optimize, QtCore.SIGNAL("clicked()"), self.optimize)
        QtCore.QObject.connect(self.ui.lineEditOpt_Yspacing, QtCore.SIGNAL("editingFinished()"), self.setoptimizer_yspacing)
        QtCore.QObject.connect(self.ui.lineEditOpt_Niter, QtCore.SIGNAL("editingFinished()"), self.setoptimizer_iter)
        QtCore.QObject.connect(self.ui.checkBoxOpt_IncRefstars, QtCore.SIGNAL("stateChanged(int)"), self.includerefstars)
        QtCore.QObject.connect(self.ui.lineEditOpt_NumRefstars, QtCore.SIGNAL("editingFinished()"), self.setnumrefstars)

        #ref stars
        QtCore.QObject.connect(self.ui.pushButtonRef_ClearRefstars, QtCore.SIGNAL("clicked()"), self.clearrefstartable)
        QtCore.QObject.connect(self.ui.pushButtonRef_AddRefstarImage, QtCore.SIGNAL("clicked()"), self.addslitletsfromimage)
        QtCore.QObject.connect(self.ui.pushButtonRef_AddRefstarsfromCat, QtCore.SIGNAL("clicked()"), self.addrefstarsfromcatalogue)
        QtCore.QObject.connect(self.ui.pushButtonRef_AddRefstar, QtCore.SIGNAL("clicked()"), self.addrefstarmanually)
        QtCore.QObject.connect(self.ui.pushButtonRef_DeleteRefstar, QtCore.SIGNAL("clicked()"), self.deleterefstarmanually)
        QtCore.QObject.connect(self.ui.pushButtonRef_DeleteRefstar_2, QtCore.SIGNAL("clicked()"), self.deleteslitfromimage)
        QtCore.QObject.connect(self.ui.tableWidgetRefstars, QtCore.SIGNAL(" itemSelectionChanged()"), self.setrefposition)
        QtCore.QObject.connect(self.ui.tableWidgetRefstars, QtCore.SIGNAL("cellChanged(int, int)"), self.refchanged)

        # finalize tab
        QtCore.QObject.connect(self.ui.pushButtonFin_Validate, QtCore.SIGNAL("clicked(bool)"),self.validator)
        QtCore.QObject.connect(self.ui.pushButtonFin_WriteXML, QtCore.SIGNAL("clicked(bool)"), self.writexml)
        QtCore.QObject.connect(self.ui.toolButtonFin_WriteRSMT, QtCore.SIGNAL("clicked(bool)"), self.writersmt)
        QtCore.QObject.connect(self.ui.pushButtonFin_CreateFChart_Current, QtCore.SIGNAL("clicked(bool)"), self.writeFC_Current)
        QtCore.QObject.connect(self.ui.pushButtonFin_CreateFChart_DSS, QtCore.SIGNAL("clicked(bool)"), self.writeFC_DSS)

 
        self.ui.tabWidget.setCurrentIndex(0)
Exemplo n.º 2
0
    def __init__(self, parent=None, infile=None, inimage=None, center_ra=None, center_dec=None, position_angle=None):
        QtGui.QWidget.__init__(self, parent)

        # set up the main UI
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        #        OptimizeTab.__init__(self,self.ui)
        # set up the slitmask
        self.slitmask = SlitMask(center_ra=center_ra, center_dec=center_dec, position_angle=position_angle)
        self.slitlets = self.slitmask.slitlets

        # setup the image interaction
        self.imagedisplay = ImageDisplay(target="pySlitMask:5909")
        self.position_angle = position_angle
        if inimage:
            self.loadimage(inimage)
        # set up some variables that will be needed later
        self.xmlfile = None
        self.fcfile = None

        # read in the input data if available
        self.infile = infile
        if infile:
            self.ui.radioButtonInfo_Catalogue.setChecked(True)
            self.setmode2cat()
            self.entercatalog(infile)
            print self.slitmask.center_ra, self.slitmask.center_dec
            if self.slitmask.center_ra is None and self.slitmask.center_dec is None:
                self.slitmask.set_MaskPosition()
                self.displayfootprint()

            # if self.slitmask.center_ra and self.slitmask.center_dec:
            # self.imagedisplay.rssregion(self.slitmask.center_ra, self.slitmask.center_dec)

            self.ui.lineEditMain_CenRA.setText(str(self.slitmask.center_ra))
            self.ui.lineEditMain_CenDEC.setText(str(self.slitmask.center_dec))
            self.ui.lineEditMain_PA.setText(str(self.slitmask.position_angle))
            self.ui.lineEditMain_Equinox.setText(str(self.slitmask.equinox))

            self.ui.lineEditMain_TargetName.setText(self.slitmask.target_name)
            self.ui.lineEditMain_MaskName.setText(self.slitmask.mask_name)
            self.ui.lineEditInfo_Creator.setText(self.slitmask.creator)
            self.ui.lineEditInfo_Proposer.setText(self.slitmask.proposer)
            self.ui.lineEditInfo_ProposalCode.setText(self.slitmask.proposal_code)
        else:
            self.ui.radioButtonInfo_Catalogue.setChecked(False)
            self.ui.radioButtonInfo_Manual.setChecked(True)
            self.setmode2manual()
            self.ui.toolButtonCat_Load.setEnabled(True)

            # self.displayfootprint()

        #        self.opttab = OptimizeTab()
        # setup default values for the optimizer
        self.opt_yspacing = 1.0
        self.opt_iter = 10
        self.ui.lineEditOpt_Yspacing.setText(str(self.opt_yspacing))
        self.ui.lineEditOpt_Niter.setText(str(self.opt_iter))

        #        self.slitmask.outFoV()
        #        print self.slitlets.data['fov_flag']
        #        self.updatetabs()
        # Listen to different signals

        # menu items
        QtCore.QObject.connect(self.ui.actionLoad_Catalogue, QtCore.SIGNAL("triggered()"), self.loadcatalog)
        QtCore.QObject.connect(self.ui.actionLoad_Image, QtCore.SIGNAL("triggered()"), self.loadimage)

        # main tabs
        QtCore.QObject.connect(self.ui.lineEditMain_CenRA, QtCore.SIGNAL("editingFinished()"), self.loadCenRA)
        QtCore.QObject.connect(self.ui.lineEditMain_CenDEC, QtCore.SIGNAL("editingFinished()"), self.loadCenDEC)
        QtCore.QObject.connect(self.ui.lineEditMain_PA, QtCore.SIGNAL("editingFinished()"), self.loadpositionangle)
        QtCore.QObject.connect(self.ui.lineEditMain_Equinox, QtCore.SIGNAL("editingFinished()"), self.loadequinox)
        QtCore.QObject.connect(self.ui.lineEditMain_TargetName, QtCore.SIGNAL("editingFinished()"), self.loadtargetname)
        QtCore.QObject.connect(self.ui.lineEditMain_MaskName, QtCore.SIGNAL("editingFinished()"), self.loadmaskname)

        # info tabs
        QtCore.QObject.connect(
            self.ui.lineEditInfo_ProposalCode, QtCore.SIGNAL("editingFinished()"), self.loadproposalcode
        )
        QtCore.QObject.connect(self.ui.lineEditInfo_Proposer, QtCore.SIGNAL("editingFinished()"), self.loadproposer)
        QtCore.QObject.connect(self.ui.lineEditInfo_Creator, QtCore.SIGNAL("editingFinished()"), self.loadcreator)

        #        QtCore.QObject.connect(self.slitmask, SIGNAL('xmlloaded'), self.setcreator)

        QtCore.QObject.connect(self.ui.radioButtonInfo_Catalogue, QtCore.SIGNAL("clicked()"), self.setmode2cat)
        QtCore.QObject.connect(self.ui.radioButtonInfo_Manual, QtCore.SIGNAL("clicked()"), self.setmode2manual)
        QtCore.QObject.connect(self.ui.checkBoxInfo_CentroidOn, QtCore.SIGNAL("clicked()"), self.setmodecentroiding)

        # catalog tabs
        QtCore.QObject.connect(self.ui.toolButtonCat_Load, QtCore.SIGNAL("clicked(bool)"), self.loadcatalog)
        QtCore.QObject.connect(self.ui.pushButtonCat_AddSlits, QtCore.SIGNAL("clicked(bool)"), self.addslitfromcatalog)
        QtCore.QObject.connect(self.ui.pushButtonCat_Clear, QtCore.SIGNAL("clicked()"), self.clearContents)

        # slit tab
        QtCore.QObject.connect(self.ui.pushButtonSlit_ClearSlits, QtCore.SIGNAL("clicked()"), self.clearslittable)
        QtCore.QObject.connect(
            self.ui.pushButtonSlit_AddSlitImage, QtCore.SIGNAL("clicked()"), self.addslitletsfromimage
        )
        QtCore.QObject.connect(
            self.ui.pushButtonSlit_AddSlitfromCat, QtCore.SIGNAL("clicked()"), self.addslitletsfromcatalogue
        )
        QtCore.QObject.connect(self.ui.pushButtonSlit_AddSlit, QtCore.SIGNAL("clicked()"), self.addslitmanually)
        QtCore.QObject.connect(self.ui.pushButtonSlit_DeleteSlit, QtCore.SIGNAL("clicked()"), self.deleteslitmanually)
        QtCore.QObject.connect(
            self.ui.pushButtonSlit_DeleteSlitImage, QtCore.SIGNAL("clicked()"), self.deleteslitfromimage
        )
        QtCore.QObject.connect(self.ui.tableWidgetSlits, QtCore.SIGNAL("itemSelectionChanged()"), self.setposition)
        QtCore.QObject.connect(self.ui.tableWidgetSlits, QtCore.SIGNAL("cellChanged(int, int)"), self.slitchanged)

        # optimize tab
        QtCore.QObject.connect(self.ui.pushButtonOpt_Optimize, QtCore.SIGNAL("clicked()"), self.optimize)
        QtCore.QObject.connect(
            self.ui.lineEditOpt_Yspacing, QtCore.SIGNAL("editingFinished()"), self.setoptimizer_yspacing
        )
        QtCore.QObject.connect(self.ui.lineEditOpt_Niter, QtCore.SIGNAL("editingFinished()"), self.setoptimizer_iter)
        QtCore.QObject.connect(
            self.ui.checkBoxOpt_IncRefstars, QtCore.SIGNAL("stateChanged(int)"), self.includerefstars
        )
        QtCore.QObject.connect(self.ui.lineEditOpt_NumRefstars, QtCore.SIGNAL("editingFinished()"), self.setnumrefstars)

        # ref stars
        QtCore.QObject.connect(self.ui.pushButtonRef_ClearRefstars, QtCore.SIGNAL("clicked()"), self.clearrefstartable)
        QtCore.QObject.connect(
            self.ui.pushButtonRef_AddRefstarImage, QtCore.SIGNAL("clicked()"), self.addslitletsfromimage
        )
        QtCore.QObject.connect(
            self.ui.pushButtonRef_AddRefstarsfromCat, QtCore.SIGNAL("clicked()"), self.addrefstarsfromcatalogue
        )
        QtCore.QObject.connect(self.ui.pushButtonRef_AddRefstar, QtCore.SIGNAL("clicked()"), self.addrefstarmanually)
        QtCore.QObject.connect(
            self.ui.pushButtonRef_DeleteRefstar, QtCore.SIGNAL("clicked()"), self.deleterefstarmanually
        )
        QtCore.QObject.connect(
            self.ui.pushButtonRef_DeleteRefstar_2, QtCore.SIGNAL("clicked()"), self.deleteslitfromimage
        )
        QtCore.QObject.connect(
            self.ui.tableWidgetRefstars, QtCore.SIGNAL(" itemSelectionChanged()"), self.setrefposition
        )
        QtCore.QObject.connect(self.ui.tableWidgetRefstars, QtCore.SIGNAL("cellChanged(int, int)"), self.refchanged)

        # finalize tab
        QtCore.QObject.connect(self.ui.pushButtonFin_Validate, QtCore.SIGNAL("clicked(bool)"), self.validator)
        QtCore.QObject.connect(self.ui.pushButtonFin_WriteXML, QtCore.SIGNAL("clicked(bool)"), self.writexml)
        QtCore.QObject.connect(self.ui.toolButtonFin_WriteRSMT, QtCore.SIGNAL("clicked(bool)"), self.writersmt)
        QtCore.QObject.connect(
            self.ui.pushButtonFin_CreateFChart_Current, QtCore.SIGNAL("clicked(bool)"), self.writeFC_Current
        )
        QtCore.QObject.connect(self.ui.pushButtonFin_CreateFChart_DSS, QtCore.SIGNAL("clicked(bool)"), self.writeFC_DSS)

        self.ui.tabWidget.setCurrentIndex(0)
Exemplo n.º 3
0
class SlitMaskGui(QtGui.QMainWindow, InfoTab, CatalogTab, OptimizeTab, SlitTab, RefTab, FinalizeTab):
    def __init__(self, parent=None, infile=None, inimage=None, center_ra=None, center_dec=None, position_angle=None):
        QtGui.QWidget.__init__(self, parent)
        
        #set up the main UI
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
#        OptimizeTab.__init__(self,self.ui)
        #set up the slitmask
        self.slitmask=SlitMask(center_ra=center_ra, center_dec=center_dec, position_angle=position_angle )
        self.slitlets=self.slitmask.slitlets

        #setup the image interaction
        self.imagedisplay=ImageDisplay(target='pySlitMask:5909')
        self.position_angle=position_angle 
        if inimage:
           self.loadimage(inimage)
        #set up some variables that will be needed later
        self.xmlfile=None
        self.fcfile=None

        #read in the input data if available
        self.infile=infile 
        if infile:
            self.ui.radioButtonInfo_Catalogue.setChecked(True)
            self.setmode2cat()
            self.entercatalog(infile)
            print self.slitmask.center_ra, self.slitmask.center_dec
            if self.slitmask.center_ra is None and self.slitmask.center_dec is None:
                self.slitmask.set_MaskPosition()
                self.displayfootprint()

            #if self.slitmask.center_ra and self.slitmask.center_dec:
                #self.imagedisplay.rssregion(self.slitmask.center_ra, self.slitmask.center_dec)
            
            self.ui.lineEditMain_CenRA.setText(str(self.slitmask.center_ra))
            self.ui.lineEditMain_CenDEC.setText(str(self.slitmask.center_dec))
            self.ui.lineEditMain_PA.setText(str(self.slitmask.position_angle))
            self.ui.lineEditMain_Equinox.setText(str(self.slitmask.equinox))
            
            self.ui.lineEditMain_TargetName.setText(self.slitmask.target_name)
            self.ui.lineEditMain_MaskName.setText(self.slitmask.mask_name)
            self.ui.lineEditInfo_Creator.setText(self.slitmask.creator)
            self.ui.lineEditInfo_Proposer.setText(self.slitmask.proposer)
            self.ui.lineEditInfo_ProposalCode.setText(self.slitmask.proposal_code)
        else:
            self.ui.radioButtonInfo_Catalogue.setChecked(False)
            self.ui.radioButtonInfo_Manual.setChecked(True)
            self.setmode2manual()
            self.ui.toolButtonCat_Load.setEnabled(True)



              #self.displayfootprint()

#        self.opttab = OptimizeTab()
        # setup default values for the optimizer
        self.opt_yspacing = 1.
        self.opt_iter = 10
        self.ui.lineEditOpt_Yspacing.setText(str(self.opt_yspacing))
        self.ui.lineEditOpt_Niter.setText(str(self.opt_iter))

#        self.slitmask.outFoV()
#        print self.slitlets.data['fov_flag']
#        self.updatetabs()
        #Listen to different signals

        #menu items
        QtCore.QObject.connect(self.ui.actionLoad_Catalogue, QtCore.SIGNAL("triggered()"), self.loadcatalog)
        QtCore.QObject.connect(self.ui.actionLoad_Image, QtCore.SIGNAL("triggered()"), self.loadimage)

        #main tabs
        QtCore.QObject.connect(self.ui.lineEditMain_CenRA, QtCore.SIGNAL("editingFinished()"), self.loadCenRA)
        QtCore.QObject.connect(self.ui.lineEditMain_CenDEC, QtCore.SIGNAL("editingFinished()"), self.loadCenDEC)
        QtCore.QObject.connect(self.ui.lineEditMain_PA, QtCore.SIGNAL("editingFinished()"), self.loadpositionangle)
        QtCore.QObject.connect(self.ui.lineEditMain_Equinox, QtCore.SIGNAL("editingFinished()"), self.loadequinox)
        QtCore.QObject.connect(self.ui.lineEditMain_TargetName, QtCore.SIGNAL("editingFinished()"), self.loadtargetname)
        QtCore.QObject.connect(self.ui.lineEditMain_MaskName, QtCore.SIGNAL("editingFinished()"), self.loadmaskname)
        
        #info tabs
        QtCore.QObject.connect(self.ui.lineEditInfo_ProposalCode, QtCore.SIGNAL("editingFinished()"), self.loadproposalcode)
        QtCore.QObject.connect(self.ui.lineEditInfo_Proposer, QtCore.SIGNAL("editingFinished()"), self.loadproposer)
        QtCore.QObject.connect(self.ui.lineEditInfo_Creator, QtCore.SIGNAL("editingFinished()"), self.loadcreator)

#        QtCore.QObject.connect(self.slitmask, SIGNAL('xmlloaded'), self.setcreator)

        QtCore.QObject.connect(self.ui.radioButtonInfo_Catalogue, QtCore.SIGNAL("clicked()"), self.setmode2cat)
        QtCore.QObject.connect(self.ui.radioButtonInfo_Manual, QtCore.SIGNAL("clicked()"), self.setmode2manual)
        QtCore.QObject.connect(self.ui.checkBoxInfo_CentroidOn, QtCore.SIGNAL("clicked()"), self.setmodecentroiding)

        
        
        #catalog tabs
        QtCore.QObject.connect(self.ui.toolButtonCat_Load, QtCore.SIGNAL("clicked(bool)"), self.loadcatalog)
        QtCore.QObject.connect(self.ui.pushButtonCat_AddSlits, QtCore.SIGNAL("clicked(bool)"), self.addslitfromcatalog)
        QtCore.QObject.connect(self.ui.pushButtonCat_Clear, QtCore.SIGNAL("clicked()"), self.clearContents)

        #slit tab
        QtCore.QObject.connect(self.ui.pushButtonSlit_ClearSlits, QtCore.SIGNAL("clicked()"), self.clearslittable)
        QtCore.QObject.connect(self.ui.pushButtonSlit_AddSlitImage, QtCore.SIGNAL("clicked()"), self.addslitletsfromimage)
        QtCore.QObject.connect(self.ui.pushButtonSlit_AddSlitfromCat, QtCore.SIGNAL("clicked()"), self.addslitletsfromcatalogue)
        QtCore.QObject.connect(self.ui.pushButtonSlit_AddSlit, QtCore.SIGNAL("clicked()"), self.addslitmanually)
        QtCore.QObject.connect(self.ui.pushButtonSlit_DeleteSlit, QtCore.SIGNAL("clicked()"), self.deleteslitmanually)
        QtCore.QObject.connect(self.ui.pushButtonSlit_DeleteSlitImage, QtCore.SIGNAL("clicked()"), self.deleteslitfromimage)
        QtCore.QObject.connect(self.ui.tableWidgetSlits, QtCore.SIGNAL("itemSelectionChanged()"), self.setposition)
        QtCore.QObject.connect(self.ui.tableWidgetSlits, QtCore.SIGNAL("cellChanged(int, int)"), self.slitchanged)

        #optimize tab
        QtCore.QObject.connect(self.ui.pushButtonOpt_Optimize, QtCore.SIGNAL("clicked()"), self.optimize)
        QtCore.QObject.connect(self.ui.lineEditOpt_Yspacing, QtCore.SIGNAL("editingFinished()"), self.setoptimizer_yspacing)
        QtCore.QObject.connect(self.ui.lineEditOpt_Niter, QtCore.SIGNAL("editingFinished()"), self.setoptimizer_iter)
        QtCore.QObject.connect(self.ui.checkBoxOpt_IncRefstars, QtCore.SIGNAL("stateChanged(int)"), self.includerefstars)
        QtCore.QObject.connect(self.ui.lineEditOpt_NumRefstars, QtCore.SIGNAL("editingFinished()"), self.setnumrefstars)

        #ref stars
        QtCore.QObject.connect(self.ui.pushButtonRef_ClearRefstars, QtCore.SIGNAL("clicked()"), self.clearrefstartable)
        QtCore.QObject.connect(self.ui.pushButtonRef_AddRefstarImage, QtCore.SIGNAL("clicked()"), self.addslitletsfromimage)
        QtCore.QObject.connect(self.ui.pushButtonRef_AddRefstarsfromCat, QtCore.SIGNAL("clicked()"), self.addrefstarsfromcatalogue)
        QtCore.QObject.connect(self.ui.pushButtonRef_AddRefstar, QtCore.SIGNAL("clicked()"), self.addrefstarmanually)
        QtCore.QObject.connect(self.ui.pushButtonRef_DeleteRefstar, QtCore.SIGNAL("clicked()"), self.deleterefstarmanually)
        QtCore.QObject.connect(self.ui.pushButtonRef_DeleteRefstar_2, QtCore.SIGNAL("clicked()"), self.deleteslitfromimage)
        QtCore.QObject.connect(self.ui.tableWidgetRefstars, QtCore.SIGNAL(" itemSelectionChanged()"), self.setrefposition)
        QtCore.QObject.connect(self.ui.tableWidgetRefstars, QtCore.SIGNAL("cellChanged(int, int)"), self.refchanged)

        # finalize tab
        QtCore.QObject.connect(self.ui.pushButtonFin_Validate, QtCore.SIGNAL("clicked(bool)"),self.validator)
        QtCore.QObject.connect(self.ui.pushButtonFin_WriteXML, QtCore.SIGNAL("clicked(bool)"), self.writexml)
        QtCore.QObject.connect(self.ui.toolButtonFin_WriteRSMT, QtCore.SIGNAL("clicked(bool)"), self.writersmt)
        QtCore.QObject.connect(self.ui.pushButtonFin_CreateFChart_Current, QtCore.SIGNAL("clicked(bool)"), self.writeFC_Current)
        QtCore.QObject.connect(self.ui.pushButtonFin_CreateFChart_DSS, QtCore.SIGNAL("clicked(bool)"), self.writeFC_DSS)

 
        self.ui.tabWidget.setCurrentIndex(0)

    def clearContents(self):
        self.slitlets.data = None
        self.ui.tableWidgetCat.clearContents()
        self.ui.tableWidgetCat.setRowCount(0)
        #TODO: Set the number of rows to the current data length
        #print 'nope not doing it'

    # loads mask coordinates
    def loadCenRA(self):
        self.slitmask.validated = False
        self.slitmask.add_center_ra(ra_read(self.ui.lineEditMain_CenRA.text()))
        if self.slitmask.center_ra == None:
            palette = self.setPalette('error')
            self.ui.lineEditMain_CenRA.setPalette(palette)
        else:
            palette = self.setPalette('normal')
            self.ui.lineEditMain_CenRA.setPalette(palette)
            self.ui.lineEditMain_CenRA.setText(str(self.slitmask.center_ra))
            self.slitmask.outFoV()
            self.updatetabs()

    def loadCenDEC(self):
        self.slitmask.validated = False
        self.slitmask.add_center_dec(dec_read(self.ui.lineEditMain_CenDEC.text()))
        if self.slitmask.center_dec == None:
            palette = self.setPalette('error')
            self.ui.lineEditMain_CenDEC.setPalette(palette)
        else:
            palette = self.setPalette('normal')
            self.ui.lineEditMain_CenDEC.setPalette(palette)
            self.ui.lineEditMain_CenDEC.setText(str(self.slitmask.center_dec))
            self.slitmask.outFoV()
            self.updatetabs()

    def loadpositionangle(self):
        #print self.ui.lineEditMain_PA.text()
        self.slitmask.validated = False
        self.slitmask.add_position_angle(dec_read(self.ui.lineEditMain_PA.text()))
        if self.slitmask.position_angle == None:
            palette = self.setPalette('error')
            self.ui.lineEditMain_PA.setPalette(palette)
        else:
            palette = self.setPalette('normal')
            self.ui.lineEditMain_PA.setPalette(palette)
            self.ui.lineEditMain_PA.setText(str(self.slitmask.position_angle))
            self.slitmask.outFoV()
            self.imagedisplay.rotate(self.slitmask.position_angle)
            self.updatetabs()

    def loadequinox(self):
        self.slitmask.validated = False
        self.slitmask.add_equinox(dec_read(self.ui.lineEditMain_Equinox.text()))
        if self.slitmask.equinox == None:
            palette = self.setPalette('error')
            self.ui.lineEditMain_Equinox.setPalette(palette)
        else:
            palette = self.setPalette('normal')
            self.ui.lineEditMain_Equinox.setPalette(palette)
            self.ui.lineEditMain_Equinox.setText(str(self.slitmask.equinox))
            self.slitmask.outFoV()
            
     # load info from the main window
    def loadtargetname(self):
        self.slitmask.target_name=str(self.ui.lineEditMain_TargetName.text()).strip()
        if self.slitmask.validated:  
           if len(self.slitmask.target_name)==0: self.slitmask.validated=False

    def loadmaskname(self):
        self.slitmask.mask_name=str(self.ui.lineEditMain_MaskName.text()).strip()
        if self.slitmask.validated:  
           if len(self.slitmask.mask_name)==0: self.slitmask.validated=False
   
    def loadValue(self):
        self.updatetabs()

    def updatetabs(self):
       """Update all of the information after changes to the slitlet class"""
       print 'Updating tabs'
       self.updatecatalogtable()
       self.updateslittable()
       self.updaterefstartable()
       self.imagedisplay.deleteregions()
       self.displayslits()
#       self.displayfootprint()

    def displayslits(self): 
       """Add the slits to the image """
       ids = np.where(self.slitlets.data['inmask_flag']==1)[0]
       fout = open('tmp.reg', 'w')
       fout.write('# Region file format: DS9 version 4.1\n# Filename: sgpR.fits\n')

       fout.write('global color=green dashlist=8 3 width=1 font="helvetica 10 normal roman" select=1 highlite=1 dash=0 fixed=0 edit=1 move=1 delete=1 include=1 source=1\n')
       fout.write('fk5\n')

       for i in ids:
           fout.write(self.slitlets.asregion(i,self.slitmask.position_angle)+'\n')
           # show spectral footprint:
           #fout.write(self.slitlets.asregionspec(i,self.slitmask.position_angle)+'\n')
       fout.close()

       self.imagedisplay.regionfromfile('tmp.reg')


    def displayall(self): 
       """Add the slits to the image """
       ids=np.where(self.slitlets.data['inmask_flag']==1)[0]
       fout=open('oth.reg', 'w')
       fout.write('# Region file format: DS9 version 4.1\n# Filename: sgpR.fits\n')

       fout.write('global color=green dashlist=8 3 width=1 font="helvetica 10 normal roman" select=1 highlite=1 dash=0 fixed=0 edit=1 move=1 delete=1 include=1 source=1\n')
       fout.write('fk5\n')

       for i in ids:
           #fout.write(self.slitlets.asregion(i,self.slitmask.position_angle)+'\n')
           # show spectral footprint:
           fout.write(self.slitlets.asregionspec(i,self.slitmask.position_angle)+'\n')
           # **** also stars, need to be careful about shape
       fout.close()

       self.imagedisplay.regionfromfile('tmp.reg')


    def displayfootprint(self):
        """Add the RSS footprint to the image"""
#       fout=open('tmpfp.reg', 'w')
#       fout.write('# Region file format: DS9 version 4.1\n# Filename: sgpR.fits\n')
#
##       fout.write('global color=yellow dashlist=8 3 width=2 font="helvetica 10 normal roman" select=0 highlite=0 dash=1 fixed=0 edit=0 move=0 delete=0 include=1 source=1\n')
#       fout.write('fk5\n')
#       fout.write('circle(%f,%f,4\') # color=yellow dashlist=8 3 width=2 select=0 highlite=0 dash=1 fixed=0 edit=0 move=1 delete=0 background include=1 source=0\n' % (self.slitmask.center_ra, self.slitmask.center_dec))
#       fout.close()
#       self.imagedisplay.regionfromfile('tmpfp.reg')
        self.imagedisplay.rssregion(self.slitmask.center_ra, self.slitmask.center_dec)

    def loadimage(self, inimage=None):
        if not inimage:
             #launch a file IO dialog
             ldir = os.getcwd()
             inimage = QtGui.QFileDialog.getOpenFileName(caption="Open Catalog", directory=ldir)

        self.inimage=str(inimage)
        self.imagedisplay.display(inimage, pa=self.position_angle)

    def deleteslitfromimage(self):
        #download the slits in the image
        newslits=self.imagedisplay.getregions()
        

        #loop through the list and see which one is missing
        try:
            index=np.where(self.slitlets.data['inmask_flag']==1)[0]
        except:
            return
        
 
        #check to see if it is in the mask
        for i in index:
            sra=self.slitlets.data['targ_ra'][i]
            sdec=self.slitlets.data['targ_dec'][i]
            found=False
            for k in newslits.keys():
              ra = float(newslits[k][0][0])
              dec = float(newslits[k][0][1])
              if abs(sra-ra) < 0.0003 and abs(sdec-dec) < 0.0003:
                 found=True
            if not found:
               self.slitlets.data['inmask_flag'][i]=0

        #update the tabs
        self.updatetabs()      


    def addslitletsfromimage(self):
        """Download the slits from the image and add them to the slitlet or catalog

          If catalog is selected, it will search the catalog for a corresponding object and center the slit on that object

          **TODO**
          If manual centroided is selected, it will centroid around that value in the image and use that position
          If manual uncentroided is selected, it will just use the slit position

        """
        #download the slits in the image
        newslits=self.imagedisplay.getregions()
        #loop through the objects--if they are already in the catalog check to see if they
        #need updating.  If not, then add them to the catalog

        #print "Sorting through regions"

        for i in newslits:
            if newslits[i][1]:
              #if it is tagged we assume it is already in the slitmask
              #print newslits[i]
              pass
            else:
              ra = float(newslits[i][0][0])
              dec = float(newslits[i][0][1])
              #print i,ra,dec
           #width=str(newslits[i][0])
           #height=str(newslits[i][0])
           #tilt=str(newslits[i][0])
              #TODO: This searches that catalog and adds the target that matches the slit drawn
              sid=self.slitlets.findtarget(ra,dec)
              self.slitlets.addtomask(sid)

         
        self.updatetabs()

    def setPalette(self,mode):
        palette = QtGui.QPalette()

        if mode == 'error':
            brush = QtGui.QBrush(QtGui.QColor(255, 148, 148))
            brush.setStyle(QtCore.Qt.SolidPattern)
            palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Base, brush)
            palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Base, brush)

            return palette

        if mode == 'normal':
            brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
            brush.setStyle(QtCore.Qt.SolidPattern)
            palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Base, brush)
            palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Base, brush)

            return palette
Exemplo n.º 4
0
class SlitMaskGui(QtGui.QMainWindow, InfoTab, CatalogTab, OptimizeTab, SlitTab, RefTab, FinalizeTab):
    def __init__(self, parent=None, infile=None, inimage=None, center_ra=None, center_dec=None, position_angle=None):
        QtGui.QWidget.__init__(self, parent)

        # set up the main UI
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        #        OptimizeTab.__init__(self,self.ui)
        # set up the slitmask
        self.slitmask = SlitMask(center_ra=center_ra, center_dec=center_dec, position_angle=position_angle)
        self.slitlets = self.slitmask.slitlets

        # setup the image interaction
        self.imagedisplay = ImageDisplay(target="pySlitMask:5909")
        self.position_angle = position_angle
        if inimage:
            self.loadimage(inimage)
        # set up some variables that will be needed later
        self.xmlfile = None
        self.fcfile = None

        # read in the input data if available
        self.infile = infile
        if infile:
            self.ui.radioButtonInfo_Catalogue.setChecked(True)
            self.setmode2cat()
            self.entercatalog(infile)
            print self.slitmask.center_ra, self.slitmask.center_dec
            if self.slitmask.center_ra is None and self.slitmask.center_dec is None:
                self.slitmask.set_MaskPosition()
                self.displayfootprint()

            # if self.slitmask.center_ra and self.slitmask.center_dec:
            # self.imagedisplay.rssregion(self.slitmask.center_ra, self.slitmask.center_dec)

            self.ui.lineEditMain_CenRA.setText(str(self.slitmask.center_ra))
            self.ui.lineEditMain_CenDEC.setText(str(self.slitmask.center_dec))
            self.ui.lineEditMain_PA.setText(str(self.slitmask.position_angle))
            self.ui.lineEditMain_Equinox.setText(str(self.slitmask.equinox))

            self.ui.lineEditMain_TargetName.setText(self.slitmask.target_name)
            self.ui.lineEditMain_MaskName.setText(self.slitmask.mask_name)
            self.ui.lineEditInfo_Creator.setText(self.slitmask.creator)
            self.ui.lineEditInfo_Proposer.setText(self.slitmask.proposer)
            self.ui.lineEditInfo_ProposalCode.setText(self.slitmask.proposal_code)
        else:
            self.ui.radioButtonInfo_Catalogue.setChecked(False)
            self.ui.radioButtonInfo_Manual.setChecked(True)
            self.setmode2manual()
            self.ui.toolButtonCat_Load.setEnabled(True)

            # self.displayfootprint()

        #        self.opttab = OptimizeTab()
        # setup default values for the optimizer
        self.opt_yspacing = 1.0
        self.opt_iter = 10
        self.ui.lineEditOpt_Yspacing.setText(str(self.opt_yspacing))
        self.ui.lineEditOpt_Niter.setText(str(self.opt_iter))

        #        self.slitmask.outFoV()
        #        print self.slitlets.data['fov_flag']
        #        self.updatetabs()
        # Listen to different signals

        # menu items
        QtCore.QObject.connect(self.ui.actionLoad_Catalogue, QtCore.SIGNAL("triggered()"), self.loadcatalog)
        QtCore.QObject.connect(self.ui.actionLoad_Image, QtCore.SIGNAL("triggered()"), self.loadimage)

        # main tabs
        QtCore.QObject.connect(self.ui.lineEditMain_CenRA, QtCore.SIGNAL("editingFinished()"), self.loadCenRA)
        QtCore.QObject.connect(self.ui.lineEditMain_CenDEC, QtCore.SIGNAL("editingFinished()"), self.loadCenDEC)
        QtCore.QObject.connect(self.ui.lineEditMain_PA, QtCore.SIGNAL("editingFinished()"), self.loadpositionangle)
        QtCore.QObject.connect(self.ui.lineEditMain_Equinox, QtCore.SIGNAL("editingFinished()"), self.loadequinox)
        QtCore.QObject.connect(self.ui.lineEditMain_TargetName, QtCore.SIGNAL("editingFinished()"), self.loadtargetname)
        QtCore.QObject.connect(self.ui.lineEditMain_MaskName, QtCore.SIGNAL("editingFinished()"), self.loadmaskname)

        # info tabs
        QtCore.QObject.connect(
            self.ui.lineEditInfo_ProposalCode, QtCore.SIGNAL("editingFinished()"), self.loadproposalcode
        )
        QtCore.QObject.connect(self.ui.lineEditInfo_Proposer, QtCore.SIGNAL("editingFinished()"), self.loadproposer)
        QtCore.QObject.connect(self.ui.lineEditInfo_Creator, QtCore.SIGNAL("editingFinished()"), self.loadcreator)

        #        QtCore.QObject.connect(self.slitmask, SIGNAL('xmlloaded'), self.setcreator)

        QtCore.QObject.connect(self.ui.radioButtonInfo_Catalogue, QtCore.SIGNAL("clicked()"), self.setmode2cat)
        QtCore.QObject.connect(self.ui.radioButtonInfo_Manual, QtCore.SIGNAL("clicked()"), self.setmode2manual)
        QtCore.QObject.connect(self.ui.checkBoxInfo_CentroidOn, QtCore.SIGNAL("clicked()"), self.setmodecentroiding)

        # catalog tabs
        QtCore.QObject.connect(self.ui.toolButtonCat_Load, QtCore.SIGNAL("clicked(bool)"), self.loadcatalog)
        QtCore.QObject.connect(self.ui.pushButtonCat_AddSlits, QtCore.SIGNAL("clicked(bool)"), self.addslitfromcatalog)
        QtCore.QObject.connect(self.ui.pushButtonCat_Clear, QtCore.SIGNAL("clicked()"), self.clearContents)

        # slit tab
        QtCore.QObject.connect(self.ui.pushButtonSlit_ClearSlits, QtCore.SIGNAL("clicked()"), self.clearslittable)
        QtCore.QObject.connect(
            self.ui.pushButtonSlit_AddSlitImage, QtCore.SIGNAL("clicked()"), self.addslitletsfromimage
        )
        QtCore.QObject.connect(
            self.ui.pushButtonSlit_AddSlitfromCat, QtCore.SIGNAL("clicked()"), self.addslitletsfromcatalogue
        )
        QtCore.QObject.connect(self.ui.pushButtonSlit_AddSlit, QtCore.SIGNAL("clicked()"), self.addslitmanually)
        QtCore.QObject.connect(self.ui.pushButtonSlit_DeleteSlit, QtCore.SIGNAL("clicked()"), self.deleteslitmanually)
        QtCore.QObject.connect(
            self.ui.pushButtonSlit_DeleteSlitImage, QtCore.SIGNAL("clicked()"), self.deleteslitfromimage
        )
        QtCore.QObject.connect(self.ui.tableWidgetSlits, QtCore.SIGNAL("itemSelectionChanged()"), self.setposition)
        QtCore.QObject.connect(self.ui.tableWidgetSlits, QtCore.SIGNAL("cellChanged(int, int)"), self.slitchanged)

        # optimize tab
        QtCore.QObject.connect(self.ui.pushButtonOpt_Optimize, QtCore.SIGNAL("clicked()"), self.optimize)
        QtCore.QObject.connect(
            self.ui.lineEditOpt_Yspacing, QtCore.SIGNAL("editingFinished()"), self.setoptimizer_yspacing
        )
        QtCore.QObject.connect(self.ui.lineEditOpt_Niter, QtCore.SIGNAL("editingFinished()"), self.setoptimizer_iter)
        QtCore.QObject.connect(
            self.ui.checkBoxOpt_IncRefstars, QtCore.SIGNAL("stateChanged(int)"), self.includerefstars
        )
        QtCore.QObject.connect(self.ui.lineEditOpt_NumRefstars, QtCore.SIGNAL("editingFinished()"), self.setnumrefstars)

        # ref stars
        QtCore.QObject.connect(self.ui.pushButtonRef_ClearRefstars, QtCore.SIGNAL("clicked()"), self.clearrefstartable)
        QtCore.QObject.connect(
            self.ui.pushButtonRef_AddRefstarImage, QtCore.SIGNAL("clicked()"), self.addslitletsfromimage
        )
        QtCore.QObject.connect(
            self.ui.pushButtonRef_AddRefstarsfromCat, QtCore.SIGNAL("clicked()"), self.addrefstarsfromcatalogue
        )
        QtCore.QObject.connect(self.ui.pushButtonRef_AddRefstar, QtCore.SIGNAL("clicked()"), self.addrefstarmanually)
        QtCore.QObject.connect(
            self.ui.pushButtonRef_DeleteRefstar, QtCore.SIGNAL("clicked()"), self.deleterefstarmanually
        )
        QtCore.QObject.connect(
            self.ui.pushButtonRef_DeleteRefstar_2, QtCore.SIGNAL("clicked()"), self.deleteslitfromimage
        )
        QtCore.QObject.connect(
            self.ui.tableWidgetRefstars, QtCore.SIGNAL(" itemSelectionChanged()"), self.setrefposition
        )
        QtCore.QObject.connect(self.ui.tableWidgetRefstars, QtCore.SIGNAL("cellChanged(int, int)"), self.refchanged)

        # finalize tab
        QtCore.QObject.connect(self.ui.pushButtonFin_Validate, QtCore.SIGNAL("clicked(bool)"), self.validator)
        QtCore.QObject.connect(self.ui.pushButtonFin_WriteXML, QtCore.SIGNAL("clicked(bool)"), self.writexml)
        QtCore.QObject.connect(self.ui.toolButtonFin_WriteRSMT, QtCore.SIGNAL("clicked(bool)"), self.writersmt)
        QtCore.QObject.connect(
            self.ui.pushButtonFin_CreateFChart_Current, QtCore.SIGNAL("clicked(bool)"), self.writeFC_Current
        )
        QtCore.QObject.connect(self.ui.pushButtonFin_CreateFChart_DSS, QtCore.SIGNAL("clicked(bool)"), self.writeFC_DSS)

        self.ui.tabWidget.setCurrentIndex(0)

    def clearContents(self):
        self.slitlets.data = None
        self.ui.tableWidgetCat.clearContents()
        self.ui.tableWidgetCat.setRowCount(0)
        # TODO: Set the number of rows to the current data length
        # print 'nope not doing it'

    # loads mask coordinates
    def loadCenRA(self):
        self.slitmask.validated = False
        self.slitmask.add_center_ra(ra_read(self.ui.lineEditMain_CenRA.text()))
        if self.slitmask.center_ra == None:
            palette = self.setPalette("error")
            self.ui.lineEditMain_CenRA.setPalette(palette)
        else:
            palette = self.setPalette("normal")
            self.ui.lineEditMain_CenRA.setPalette(palette)
            self.ui.lineEditMain_CenRA.setText(str(self.slitmask.center_ra))
            self.slitmask.outFoV()
            self.updatetabs()

    def loadCenDEC(self):
        self.slitmask.validated = False
        self.slitmask.add_center_dec(dec_read(self.ui.lineEditMain_CenDEC.text()))
        if self.slitmask.center_dec == None:
            palette = self.setPalette("error")
            self.ui.lineEditMain_CenDEC.setPalette(palette)
        else:
            palette = self.setPalette("normal")
            self.ui.lineEditMain_CenDEC.setPalette(palette)
            self.ui.lineEditMain_CenDEC.setText(str(self.slitmask.center_dec))
            self.slitmask.outFoV()
            self.updatetabs()

    def loadpositionangle(self):
        # print self.ui.lineEditMain_PA.text()
        self.slitmask.validated = False
        self.slitmask.add_position_angle(dec_read(self.ui.lineEditMain_PA.text()))
        if self.slitmask.position_angle == None:
            palette = self.setPalette("error")
            self.ui.lineEditMain_PA.setPalette(palette)
        else:
            palette = self.setPalette("normal")
            self.ui.lineEditMain_PA.setPalette(palette)
            self.ui.lineEditMain_PA.setText(str(self.slitmask.position_angle))
            self.slitmask.outFoV()
            self.imagedisplay.rotate(self.slitmask.position_angle)
            self.updatetabs()

    def loadequinox(self):
        self.slitmask.validated = False
        self.slitmask.add_equinox(dec_read(self.ui.lineEditMain_Equinox.text()))
        if self.slitmask.equinox == None:
            palette = self.setPalette("error")
            self.ui.lineEditMain_Equinox.setPalette(palette)
        else:
            palette = self.setPalette("normal")
            self.ui.lineEditMain_Equinox.setPalette(palette)
            self.ui.lineEditMain_Equinox.setText(str(self.slitmask.equinox))
            self.slitmask.outFoV()

    # load info from the main window
    def loadtargetname(self):
        self.slitmask.target_name = str(self.ui.lineEditMain_TargetName.text()).strip()
        if self.slitmask.validated:
            if len(self.slitmask.target_name) == 0:
                self.slitmask.validated = False

    def loadmaskname(self):
        self.slitmask.mask_name = str(self.ui.lineEditMain_MaskName.text()).strip()
        if self.slitmask.validated:
            if len(self.slitmask.mask_name) == 0:
                self.slitmask.validated = False

    def loadValue(self):
        self.updatetabs()

    def updatetabs(self):
        """Update all of the information after changes to the slitlet class"""
        print "Updating tabs"
        self.updatecatalogtable()
        self.updateslittable()
        self.updaterefstartable()
        self.imagedisplay.deleteregions()
        self.displayslits()

    #       self.displayfootprint()

    def displayslits(self):
        """Add the slits to the image """
        ids = np.where(self.slitlets.data["inmask_flag"] == 1)[0]
        fout = open("tmp.reg", "w")
        fout.write("# Region file format: DS9 version 4.1\n# Filename: sgpR.fits\n")

        fout.write(
            'global color=green dashlist=8 3 width=1 font="helvetica 10 normal roman" select=1 highlite=1 dash=0 fixed=0 edit=1 move=1 delete=1 include=1 source=1\n'
        )
        fout.write("fk5\n")

        for i in ids:
            fout.write(self.slitlets.asregion(i, self.slitmask.position_angle) + "\n")
            # show spectral footprint:
            # fout.write(self.slitlets.asregionspec(i,self.slitmask.position_angle)+'\n')
        fout.close()

        self.imagedisplay.regionfromfile("tmp.reg")

    def displayall(self):
        """Add the slits to the image """
        ids = np.where(self.slitlets.data["inmask_flag"] == 1)[0]
        fout = open("oth.reg", "w")
        fout.write("# Region file format: DS9 version 4.1\n# Filename: sgpR.fits\n")

        fout.write(
            'global color=green dashlist=8 3 width=1 font="helvetica 10 normal roman" select=1 highlite=1 dash=0 fixed=0 edit=1 move=1 delete=1 include=1 source=1\n'
        )
        fout.write("fk5\n")

        for i in ids:
            # fout.write(self.slitlets.asregion(i,self.slitmask.position_angle)+'\n')
            # show spectral footprint:
            fout.write(self.slitlets.asregionspec(i, self.slitmask.position_angle) + "\n")
            # **** also stars, need to be careful about shape
        fout.close()

        self.imagedisplay.regionfromfile("tmp.reg")

    def displayfootprint(self):
        """Add the RSS footprint to the image"""
        #       fout=open('tmpfp.reg', 'w')
        #       fout.write('# Region file format: DS9 version 4.1\n# Filename: sgpR.fits\n')
        #
        ##       fout.write('global color=yellow dashlist=8 3 width=2 font="helvetica 10 normal roman" select=0 highlite=0 dash=1 fixed=0 edit=0 move=0 delete=0 include=1 source=1\n')
        #       fout.write('fk5\n')
        #       fout.write('circle(%f,%f,4\') # color=yellow dashlist=8 3 width=2 select=0 highlite=0 dash=1 fixed=0 edit=0 move=1 delete=0 background include=1 source=0\n' % (self.slitmask.center_ra, self.slitmask.center_dec))
        #       fout.close()
        #       self.imagedisplay.regionfromfile('tmpfp.reg')
        self.imagedisplay.rssregion(self.slitmask.center_ra, self.slitmask.center_dec)

    def loadimage(self, inimage=None):
        if not inimage:
            # launch a file IO dialog
            ldir = os.getcwd()
            inimage = QtGui.QFileDialog.getOpenFileName(caption="Open Catalog", directory=ldir)

        self.inimage = str(inimage)
        self.imagedisplay.display(inimage, pa=self.position_angle)

    def deleteslitfromimage(self):
        # download the slits in the image
        newslits = self.imagedisplay.getregions()

        # loop through the list and see which one is missing
        try:
            index = np.where(self.slitlets.data["inmask_flag"] == 1)[0]
        except:
            return

        # check to see if it is in the mask
        for i in index:
            sra = self.slitlets.data["targ_ra"][i]
            sdec = self.slitlets.data["targ_dec"][i]
            found = False
            for k in newslits.keys():
                ra = float(newslits[k][0][0])
                dec = float(newslits[k][0][1])
                if abs(sra - ra) < 0.0003 and abs(sdec - dec) < 0.0003:
                    found = True
            if not found:
                self.slitlets.data["inmask_flag"][i] = 0

        # update the tabs
        self.updatetabs()

    def addslitletsfromimage(self):
        """Download the slits from the image and add them to the slitlet or catalog

          If catalog is selected, it will search the catalog for a corresponding object and center the slit on that object

          **TODO**
          If manual centroided is selected, it will centroid around that value in the image and use that position
          If manual uncentroided is selected, it will just use the slit position

        """
        # download the slits in the image
        newslits = self.imagedisplay.getregions()
        # loop through the objects--if they are already in the catalog check to see if they
        # need updating.  If not, then add them to the catalog

        # print "Sorting through regions"

        for i in newslits:
            if newslits[i][1]:
                # if it is tagged we assume it is already in the slitmask
                # print newslits[i]
                pass
            else:
                ra = float(newslits[i][0][0])
                dec = float(newslits[i][0][1])
                # print i,ra,dec
                # width=str(newslits[i][0])
                # height=str(newslits[i][0])
                # tilt=str(newslits[i][0])
                # TODO: This searches that catalog and adds the target that matches the slit drawn
                sid = self.slitlets.findtarget(ra, dec)
                self.slitlets.addtomask(sid)

        self.updatetabs()

    def setPalette(self, mode):
        palette = QtGui.QPalette()

        if mode == "error":
            brush = QtGui.QBrush(QtGui.QColor(255, 148, 148))
            brush.setStyle(QtCore.Qt.SolidPattern)
            palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Base, brush)
            palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Base, brush)

            return palette

        if mode == "normal":
            brush = QtGui.QBrush(QtGui.QColor(255, 255, 255))
            brush.setStyle(QtCore.Qt.SolidPattern)
            palette.setBrush(QtGui.QPalette.Active, QtGui.QPalette.Base, brush)
            palette.setBrush(QtGui.QPalette.Inactive, QtGui.QPalette.Base, brush)

            return palette