variable_tab.py

Go to the documentation of this file.

#!/usr/bin/env python2
 
import Tkinter as tk
import ttk
 
import ROOT
ROOT.PyConfig.IgnoreCommandLineOptions = True
 
from .util import PreserveGDir
 
class VariableTab(object):
 
    def __init__(self, main, frame):
        self.main = main
        self._setup_GUI(frame)
        self.variables = {}
        self.OnUpdateAll_Click()
 
    def _setup_GUI(self, frame):
        self.frame = frame
        self._MakeButtons(frame)
        self._MakeTreeView(frame)
 
    def _MakeButtons(self, parent):
        # Text entries
        frame = tk.Frame(parent)
        tk.Label(frame, text='Name').grid(row=0, column=0)
        tk.Label(frame, text='Value').grid(row=0, column=1)
        self.var_name = tk.StringVar()
        tk.Entry(frame, textvariable=self.var_name).grid(row=1, column=0)
        self.var_value = tk.StringVar()
        entry = tk.Entry(frame, textvariable=self.var_value)
        entry.bind('<Return>',self.OnSetReplaceVariable_ClickOnSetReplaceVariable_Click)
        entry.grid(row=1, column=1)
        frame.pack(fill=tk.X,expand=False)
 
        # Buttons
        frame = tk.Frame(parent)
        tk.Button(frame, text='Set/Replace',
                  command=self.OnSetReplaceVariable_ClickOnSetReplaceVariable_Click).pack(side=tk.LEFT)
        tk.Button(frame, text='Delete',
                  command=self.OnDeleteVariable_Click).pack(side=tk.LEFT)
        tk.Button(frame, text='Update',
                  command=self.OnUpdateAll_Click).pack(side=tk.LEFT)
        tk.Button(frame, text='Save',
                  command=self.OnSave_Click).pack(side=tk.LEFT)
        frame.pack(fill=tk.X,expand=False)
 
    def _MakeTreeView(self, parent):
        self.treeview = ttk.Treeview(parent, columns=('value',))
        self.treeview.heading('#0', text='Name')
        self.treeview.heading('value', text='Value')
        self.treeview.pack(fill=tk.BOTH,expand=True)
        self.treeview.bind("<Double-1>",self.VariableSelectionVariableSelection)
 
    def OnUpdateAll_Click(self, *args):
        t = ROOT.GValue.AllValues()
        for val in t:
            self.SetReplaceVariable(val.GetName(), val.GetValue(),
                                    update_cpp = False)
 
    def OnSave_Click(self, *args):
        ROOT.GValue.WriteValFile("myvalues.val")
 
 
    def SetReplaceVariable(self, name, value, update_cpp = True):
        self.variables[name] = value
        if name in self.treeview.get_children():
            self.treeview.item(name, values=(str(value),))
        else:
            self.treeview.insert('','end',name, text=name, values=(str(value),))
 
        if update_cpp:
            ROOT.GValue.SetReplaceValue(name, value)
 
 
    def DeleteVariable(self, name):
        self.variables.pop(name, None)
        self.treeview.delete(name)
 
    def OnSetReplaceVariable_Click(self, *args):
        name = self.var_name.get()
        if not name:
            return
 
        try:
            value = float(self.var_value.get())
        except ValueError:
            print 'Could not convert "{}" to float'.format(self.var_value.get())
            return
 
        self.SetReplaceVariable(name, value)
 
    def OnDeleteVariable_Click(self):
        selection = self.treeview.selection()
        if not selection:
            return
        self.DeleteVariable(selection[0])
 
    def VariableSelection(self, event):
        name = event.widget.selection()[0]
        self.var_name.set(name)
        self.var_value.set(str(self.variables[name]))
 
    def _dump_to_tfile(self):
        #pipeline = ROOT.GetPipeline(0)
        #if not pipeline:
        #    return
 
        tdir = ROOT.gDirectory.mkdir('variables')
        #with PreserveGDir(tdir):
        #    for obj in pipeline.GetVariables():
        #        obj.Write()
 
    def _variable_patterns(self):
        return self.variables
 
    def _load_variable_patterns(self, patterns):
        for name, value in patterns.items():
            self.SetReplaceVariable(name, value)
 
    def AddFile(self, tfile):
        tdir = tfile.GetListOfKeys().FindObject('variables')
        if (tdir and
            issubclass(getattr(ROOT, tdir.GetClassName()), ROOT.TDirectory)):
            tdir = tdir.ReadObj()
            for key in tdir.GetListOfKeys():
                obj = key.ReadObj()
                #print " Name : " + obj.GetName()
                #print " Value : " + str(obj.GetValue())
                self.SetReplaceVariable(obj.GetName(),
                                        obj.GetValue())