# This is only needed for Python v2 but is harmless for Python v3.
import sip
sip.setapi('QVariant', 2)

from PyQt4 import QtCore, QtGui

import viz
import vizact

class MainWindow(QtGui.QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.resize(800,600)
        self.setWindowTitle('Vizard PyQt Example')

        # Create a blank widget to be used as Vizard graphics window
        self.widget = QtGui.QWidget()
        self.setCentralWidget(self.widget)

        # TODO: Initialize Vizard using the existing QtWidget window ID
        viz.go(viz.EMBEDDED,window=int(self.widget.winId()))
        
        # Prevent default escape key from closing Vizard window
        viz.setOption('viz.default_key.quit',0)

        # Add a spinning logo
        self.model = viz.add('logo.ive',pos=(0,1,4))
        self.model.addAction(vizact.spin(0,1,0,90))

        # TODO: Setup a 60 Hz timer to update the Vizard graphics loop
        self.vizTimer = QtCore.QTimer()
        self.vizTimer.start(16)
        QtCore.QObject.connect(self.vizTimer,
                QtCore.SIGNAL("timeout()"),
                self.updateVizard)
        
        # Initialize menu, toolbar, etc..
        self.createActions()
        self.createMenus()
        self.createStatusBar()
        self.setUnifiedTitleAndToolBarOnMac(True)

    def updateVizard(self):
        # TODO: Manually update the Vizard graphics frame
        viz.frame()

    def open(self):
        fileName = QtGui.QFileDialog.getOpenFileName(self)

    def save(self):
        fileName = QtGui.QFileDialog.getSaveFileName(self)

    def about(self):
        QtGui.QMessageBox.about(self, "About",
                "This application demonsrates how to "
                "integrate Vizard within a PyQt application")

    def createActions(self):
        self.openAct = QtGui.QAction("&Open...", self,
                shortcut=QtGui.QKeySequence.Open,
                statusTip="Open an existing file",
                triggered=self.open)

        self.saveAct = QtGui.QAction("&Save", self,
                shortcut=QtGui.QKeySequence.Save,
                statusTip="Save the document to disk",
                triggered=self.save)

        self.exitAct = QtGui.QAction("E&xit", self,
                shortcut="Ctrl+Q",
                statusTip="Exit the application",
                triggered=self.close)

        self.aboutAct = QtGui.QAction("&About", self,
                statusTip="Show the application's About box",
                triggered=self.about)

        self.aboutQtAct = QtGui.QAction("About &Qt", self,
                statusTip="Show the Qt library's About box",
                triggered=QtGui.qApp.aboutQt)

    def createMenus(self):
        self.fileMenu = self.menuBar().addMenu("&File")
        self.fileMenu.addAction(self.openAct)
        self.fileMenu.addAction(self.saveAct)
        self.fileMenu.addSeparator();
        self.fileMenu.addAction(self.exitAct)

        self.helpMenu = self.menuBar().addMenu("&Help")
        self.helpMenu.addAction(self.aboutAct)
        self.helpMenu.addAction(self.aboutQtAct)

    def createStatusBar(self):
        self.statusBar().showMessage("Ready")


if __name__ == '__main__':

    import sys

    app = QtGui.QApplication(sys.argv)
    mainWin = MainWindow()
    mainWin.show()
    app.exec_()