cme/main.py at master · bgriffen/cme · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

from header import Header
from install import InstallFiles
from ics import InitialConditions
from gadgetrun import GadgetRun
from halos import HaloFind
from mergertree import MergerTree
#from analysis import Analysis
from contam import Contamination
from candidates import Candidates

from tvtk.pyface.scene_editor import SceneEditor
from mayavi.tools.mlab_scene_model import MlabSceneModel
from mayavi.core.ui.mayavi_scene import MayaviScene

from Common import *

class ApplicationMain(HasTraits):

    scene = Instance(MlabSceneModel, ())

    headertab = Instance(Header)
    installtab = Instance(InstallFiles)
    #existencetab = Instance(Existence)
    initstab = Instance(InitialConditions)
    gadgetruntab = Instance(GadgetRun)
    halofindtab = Instance(HaloFind)
    mergertreetab = Instance(MergerTree)
    candidatestab = Instance(Candidates)
    contaminationtab = Instance(Contamination)
    #analysistab = Instance(Analysis)

    display = Instance(Figure)
    markercolor = ColorTrait
    markerstyle = Enum(['+',',','*','s','p','d','o'])
    markersize = Range(0,10,2)

    left_panel  = Tabbed(Group(VGroup(Item('display', editor=MPLFigureEditor(),show_label=False, resizable=True)),
                               HGroup(Item(name='markercolor', label="Color", style="custom",springy=True),
                                      Item(name='markerstyle', label="Marker",springy=True),
                                      Item(name='markersize', label="Size",springy=True)), label='Display'),
                               Item(name='scene',label='Mayavi',editor=SceneEditor(scene_class=MayaviScene)),show_labels=False)

#
    right_panel = Tabbed(Item('headertab', style='custom', label='Home',show_label=False),
                         Item('installtab', style='custom', label='Install',show_label=False),
                         Item('candidatestab', style='custom', label='Candidates',show_label=False),
                         Item('initstab', style='custom', label='Initial Conditions',show_label=False),
                         Item('gadgetruntab', style='custom', label='Gadget',show_label=False),
                         Item('halofindtab', style='custom', label='Halo Finder',show_label=False),
                         Item('mergertreetab', style='custom', label='Merger Tree',show_label=False),
                         Item('contaminationtab', style='custom', label='Contam.',show_label=False))
                         #Item('analysistab', style='custom', label='Analysis',show_label=False))

    view = View(HSplit(left_panel,
                 right_panel),
                width = 1280,
                height = 750,
                resizable = True,
                title="Caterpillar Made Easy",
             )

    def _display_default(self):
        """Initialises the display."""
        figure = Figure()
        ax = figure.add_subplot(111)
        ax = figure.axes[0]
        ax.set_xlabel('X')
        ax.set_ylabel('Y')
        ax.set_xlim(0,1)
        ax.set_ylim(0,1)

        # Set matplotlib canvas colour to be white
        rect = figure.patch
        rect.set_facecolor('w')
        return figure

    def _headertab_default(self):
        # Initialize halos the way we want to.
        # Pass a reference of main (e.g. self) downwards
        return Header(self)

    def _installtab_default(self):
        return InstallFiles(self)

    def _initstab_default(self):
        return InitialConditions(self)

    def _gadgetruntab_default(self):
        return GadgetRun(self)

    def _halofindtab_default(self):
        return HaloFind(self)

    def _mergertreetab_default(self):
        return MergerTree(self)

    def _analysistab_default(self):
        return Analysis(self)

    def _candidatestab_default(self):
        return Candidates(self)

    def _contaminationtab_default(self):
        return Contamination(self)

    def _markercolor_changed(self):
        ax = self.display.axes[0]
        if hasattr(self, 'display_points'):
            self.display_points.set_color(self.markercolor)
            self.display_points.set_markeredgecolor(self.markercolor)
            wx.CallAfter(self.display.canvas.draw)

    def _markerstyle_changed(self):
        ax = self.display.axes[0]
        if hasattr(self, 'display_points'):
            self.display_points.set_marker(self.markerstyle)
            wx.CallAfter(self.display.canvas.draw)

    def _markersize_changed(self):
        ax = self.display.axes[0]
        if hasattr(self, 'display_points'):
            self.display_points.set_markersize(self.markersize)
            wx.CallAfter(self.display.canvas.draw)

    def __init__(self, **kwargs):
        self.markercolor = 'blue'
        self.markersize = 2
        self.markerstyle = 'o'

if __name__ == '__main__':
    app = ApplicationMain()
    app.configure_traits()