#!/usr/bin/env python

import sys
from PyQt4 import QtGui, QtCore


class PlotWidget(QtGui.QWidget):
    
    def __init__(self):
        super(PlotWidget, self).__init__()
        
        self.initUI()
        
    def initUI(self):
        
        self.setGeometry(300, 300, 300, 150)
        self.setWindowTitle('Select plotting method')
        self.gridLayout = QtGui.QGridLayout()
        self.setLayout(self.gridLayout)
        self.rbutton1 = QtGui.QRadioButton('Plot intensities', self)
        self.rbutton2 = QtGui.QRadioButton('Plot average', self)
        self.rbutton3 = QtGui.QRadioButton('Plot intensities & average', self)
        self.verticalLayout = QtGui.QVBoxLayout()
        self.verticalLayout.addWidget(self.rbutton1)
        self.verticalLayout.addWidget(self.rbutton2)
        self.verticalLayout.addWidget(self.rbutton3)
        self.gridLayout.addLayout(self.verticalLayout, 0, 0, 1, 1)
        self.label = QtGui.QLabel("", self)	
        self.verticalLayout.addWidget(self.label)
        self.hLayout = QtGui.QHBoxLayout()
        self.pbutton1 = QtGui.QPushButton('Accept', self)
        self.pbutton2 = QtGui.QPushButton('Cancel', self)
        self.hLayout.addWidget(self.pbutton1)
        self.hLayout.addWidget(self.pbutton2)
        self.gridLayout.addLayout(self.hLayout, 1, 0, 1, 1)

        QtCore.QObject.connect(self.pbutton1, QtCore.SIGNAL("clicked()"), self.accept)
        QtCore.QObject.connect(self.pbutton2, QtCore.SIGNAL("clicked()"), self.close)

    def accept(self):
        if self.rbutton1.isChecked():
            print "moo"
            self.close()
        elif self.rbutton2.isChecked():
            print "poo"
            self.close()
        elif self.rbutton3.isChecked():
            print "foo"
            self.close()
        else:
            self.label.setText('Check a mode')

        
def main():
    
    app = QtGui.QApplication(sys.argv)
    ex = PlotWidget()
    ex.show()
    sys.exit(app.exec_())


if __name__ == '__main__':
    main()