Common.py

from matplotlib.pyplot import Figure
from matplotlib.backends.backend_wxagg import FigureCanvasWxAgg as FigureCanvas
from matplotlib.backends.backend_wx import NavigationToolbar2Wx

from enthought.traits.api import *
from enthought.traits.ui.api import View,UItem, Item,Group, Heading, Label, \
        HSplit, Handler, CheckListEditor, EnumEditor, TableEditor, \
        ListEditor, Tabbed, VGroup, HGroup, RangeEditor, Spring, spring
from enthought.traits.ui.menu import NoButtons
from traitsui.ui_editors.array_view_editor import ArrayViewEditor
#from traits.api import Array
#from enthought.traits.api import Any, Instance
from enthought.traits.ui.wx.editor import Editor
from enthought.traits.ui.wx.basic_editor_factory import BasicEditorFactory
import numpy as np
from enthought.enable.api import ColorTrait
from traitsui.api import TabularEditor
from traitsui.tabular_adapter import TabularAdapter
from matplotlib import *

import os
import sys
import wx
import platform
import socket
import random
import subprocess

import pylab as plt

import modules.readsnapshots.readsnapHDF5 as rsHD
import modules.readsnapshots.readsnap as rs
import modules.readsnapshots.readids as readids
import modules.readhalos.readsubf as readsubf
import modules.readhalos.RSDataReaderv2 as RSDataReader
from modules.brendanlib.grifflib import *

from random import randint


class _MPLFigureEditor(Editor):

    scrollable  = True

    def init(self, parent):
        self.control = self._create_canvas(parent)
        self.set_tooltip()

    def update_editor(self):
        pass

    def _create_canvas(self, parent):
        """ Create the MPL canvas. """
        # The panel lets us add additional controls.
        
        panel = wx.Panel(parent, -1, style=wx.CLIP_CHILDREN)
        sizer = wx.BoxSizer(wx.VERTICAL)
        panel.SetSizer(sizer)
        
        # matplotlib commands to create a canvas
        mpl_control = FigureCanvas(panel, -1, self.value)
        sizer.Add(mpl_control, 1, wx.LEFT | wx.TOP | wx.GROW)
        toolbar = NavigationToolbar2Wx(mpl_control)
        sizer.Add(toolbar, 0, wx.EXPAND)
        self.value.canvas.SetMinSize((10,10))
        
        return panel

class MPLFigureEditor(BasicEditorFactory):

    klass = _MPLFigureEditor