PyPost.py

import sys
from sys import platform as _platform

from numpy import *
from collections import deque

from PyQt5 import QtCore, QtGui
from PyQt5.QtCore import Qt, pyqtSignal
from PyQt5.QtWidgets import (QApplication, QDialog, QLineEdit, QTextBrowser,
                          QVBoxLayout, QHBoxLayout, QSizePolicy,
                          QToolBar, QMenuBar, QMenu, QAction, QMainWindow, QWidget)
from PyQt5.QtGui import QKeySequence

import matplotlib

"""
if _platform == "linux" or _platform == "linux2":
  matplotlib.rcParams['backend.qt4']='PySide'
  matplotlib.rcParams['backend'] = 'Qt4Agg'
elif _platform == "darwin":
  # MAC OS X
  matplotlib.rcParams['backend.qt4']='PySide'
elif _platform == "win32":
  # Windows
  pass
  """

# from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure

# from twisted.application import reactors

import CommandInterp
import LineEditHist
import EngMplCanvas
import MplCanvasDict
import ImgMplCanvas

# Running in the virtual environment at 
# /Users/toma/anaconda/envs/pypostenv/bin/

# Previous things I've used:
# Custom python: /Users/toma/Library/Enthought/Canopy_64bit/User/bin/python
# Or: /Users/toma/python278i/bin/python
# Or: /Users/toma/anaconda/bin/python

# Using factories and other Java-esque patterns:
#   https://python-3-patterns-idioms-test.readthedocs.org/en/latest/Factory.html

# BUG:
#   ci t/ngtest
#   gr v(1)
# This crashes the program. Should prevent this with try/catch somewhere.
# Looks like this is actually due to t/ngtest.raw does not exist.

# TODO:
#   When clicking a marker, have the marker save the XY coordinates in a variable.
#   Show the XY coordinates in the transcript.

# TODO:
#   Add a 2-D plot function. Look at code from therm.py, incorporate this in an
#     interactive way, both with png and hdf5 data.
#
#   Add markers. Markers need to support both 1D and 2D.
#
#   Add variables and functions for constants and vectors
#   hdf5 data import
#   compiled scripts
#   HTTP/IPC port, web server
#   Display controls
#   Save/Recall entire state in database
#   Program launch from file that contains state
#   Graphical SVG output, esp useful over HTTP as web page

#   TODO Graphing commands to implement
#   gr somevector
#     should plot the vector versus its index if the x-axis is otherwise undefined.
#   Need a command to set the x-axis variable.
#   gr somevector histo
#     should draw a histogram
#   gr scalar
#     should draw a flat line horizontally across the screen at y-value of scalar
#   ma scalar
#     should draw a straight vertical line at x-value of scalar
#   Could implement these with a base command that draws a straight line segment.

# Make an installer for Windows with https://pythonhosted.org/PyInstaller/
#
# Might need https://winpython.github.io in addition or instead.
# Or https://www.py2exe.org

# Plotting
# https://www.jeffreytratner.com/example-pandas-docs/series-str-2013-10-6/visualization.html
# https://www.physics.ucdavis.edu/~dwittman/Matplotlib-examples/

# Need gr x .vs y
# Need eval on Qtplot command in case it fails
# Need to check Qtplot for x and y different dimensions in plot
# Create a transcript file with command line history in it.
# Create a transcript file with evaluated Python in it.
#   This will be the basis for script generation.

# Consider:
#  In [143]: import numexpr as ne
#  In [146]: ne.evaluate('2*a*(b/c)**2', local_dict=var)
#  Out[146]: array([ 0.88888889,  0.44444444,  4.        ])

# TRY ne
# instead


class Post(QMainWindow):
  def __init__(self, parent=None):
    super(Post, self).__init__(parent)

    self.graphs= MplCanvasDict.MplCanvasDict()

    self.table = MainWin(self)
    self.setCentralWidget(self.table)

    self.createActions()
    self.createMenus()

    self.setWindowTitle('PyPost')

    self.resize(1200, 420)
    self.statusBar().showMessage('Ready')

  def newFile(self):
    self.statusBar().showMessage("Invoked File|New")
    print("Invoked File|New")

  def open_(self):
    self.statusBar().showMessage("Invoked File|Open")
    print("Invoked File|Open")

  def save(self):
    self.statusBar().showMessage("Invoked File|Save")
    print("Invoked File|Save")

  def print_(self):
    self.statusBar().showMessage("Invoked File|Print")
    print("Invoked File|Print")

  def about(self):
    #self.statusBar().showMessage("Invoked Help|About")
    #print("Invoked Help|About")
    QtGui.QMessageBox.about(self, "About PyPost",
                                  "<b>PyPost</b> is an interactive dataset visualizer " +
                                  "for electronic and thermal engineering.")

  def aboutQt(self):
    print("Invoked <b>Help|About Qt</b>")

  def prev(self):
    self.table.lineedit.historyUp()

  def next_(self):
    self.table.lineedit.historyDown()

  def createGraph(self):
    canvasName= self.graphs.create()
    self.graphs.setActive(canvasName)
    graph= self.graphs.getActiveCanvas()
    graph.setWindowTitle(canvasName)
    graph.show()
    graph.activateWindow()
    graph.raise_()
    graph.setCommandDelegate(self.table.commandInterp)
    self.table.commandInterp.setGraphicsActiveDelegate(graph)
    self.statusBar().showMessage("Created graph " + canvasName)

  def createMenus(self):
    menubar= self.menuBar()

    self.fileMenu = menubar.addMenu("File")
    self.fileMenu.addAction(self.newAct)
    self.fileMenu.addAction(self.openAct)
    self.fileMenu.addAction(self.saveAct)
    self.fileMenu.addSeparator()
    self.fileMenu.addAction(self.printAct)

    # FIXME: TODO: Looks like a bug in PySide:
    # One of the next two lines needs to be present in order for the
    # Help menu to show up.
    # Also, there is an unexpected search box in the help.
    # It looks like there is some interference from OSX.

    self.fileMenu.addAction(self.aboutAct)
    # self.fileMenu.addAction(self.aboutQtAct)

    self.cmdMenu = menubar.addMenu("Command")
    self.cmdMenu.addAction(self.previousAct)
    self.cmdMenu.addAction(self.nextAct)

    self.cmdGraph = menubar.addMenu("Graph")
    self.cmdGraph.addAction(self.createGraphAct)

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

  def createActions(self):
    self.newAct = QAction("&New", self,
                                shortcut=QtGui.QKeySequence.New,
                                statusTip="Create a new file", triggered=self.newFile)

    self.openAct = QAction("&Open...", self,
                                 shortcut=QtGui.QKeySequence.Open,
                                 statusTip="Open an existing file", triggered=self.open_)

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

    self.printAct = QAction("&Print...", self,
                                  shortcut=QtGui.QKeySequence.Print,
                                  statusTip="Print the document", triggered=self.print_)

    self.aboutAct = QAction("About", self,
                                  statusTip="About PyPost", triggered=self.about)

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

    self.previousAct = QAction("Previous", self,
                                     shortcut=QtGui.QKeySequence.NextChild,
                                     statusTip="Previous Command", triggered=self.prev)

    self.nextAct = QAction("Next", self,
                                     shortcut=QtGui.QKeySequence.PreviousChild,
                                     statusTip="Next Command", triggered=self.next_)

    self.createGraphAct = QAction("Create", self,
                                 shortcut=QtGui.QKeySequence.New,
                                 statusTip="Create Graph", triggered=self.createGraph)

class MainWin(QWidget):
  """
  Show a transcript, a command line, and a graphing canvas in one window.
  """
  def __init__(self, mainWindow):
    super(MainWin, self).__init__()

    self.resize(720, 320)
    self.browser = QTextBrowser()

    canvasName= mainWindow.graphs.create()
    mainWindow.graphs.setActive(canvasName)
    self.sc= mainWindow.graphs.getActiveCanvas()

    self.lineedit = LineEditHist.lineEditHist("")
    self.lineedit.selectAll()

    self.topLayout = QHBoxLayout()
    self.topLayout.addWidget(self.browser)
    self.topLayout.addWidget(self.sc)

    layout = QVBoxLayout()
    layout.addLayout(self.topLayout)
    layout.addWidget(self.lineedit)

    self.setLayout(layout)
    self.lineedit.setFocus()
    self.lineedit.returnPressed.connect(self.updateUi)

    self.commandInterp= CommandInterp.CommandInterp()
    self.commandInterp.setGraphicsActiveDelegate(self.sc)
    self.commandInterp.setGraphicsWindowsDelegate(mainWindow.graphs)
    self.sc.setCommandDelegate(self.commandInterp)

    self.setWindowTitle("PyPost")

    self.show()
    self.activateWindow()
    self.raise_()

  def updateUi(self):
    cmdText = unicode(self.lineedit.text())
    self.lineedit.history.append('')

    message= self.commandInterp.executeCmd(cmdText)
    pyCode= self.commandInterp.pyCode;
    print("cmdText: " + cmdText + "\nPython code: " + str(pyCode))
    self.lineedit.addCommandToDB(cmdText, pyCode)

    self.browser.append(message)
    self.lineedit.clear()
    self.lineedit.resetHistoryPosition()

    # self.mainWindow.statusBar().showMessage('Next')

if __name__ == "__main__":

  app = QApplication(sys.argv)

  main = Post()
  main.show()

  sys.exit(app.exec_())