WIP: entrybox for plot

Signed-off-by: Henry Schreiner <[email protected]>

Author: henryiii

src/uproot_browser/tree.py

@@ -84,7 +84,7 @@ def children(self) -> list[UprootEntry]:
         else:
             items = {obj.name.split(";")[0] for obj in self.item.branches}
         return [
-            UprootEntry(f"{self.path}/{key}", self.item[key]) for key in sorted(items)
+            UprootEntry(key if self.path == "/" else f"{self.path}/{key}", self.item[key]) for key in sorted(items)
         ]
 
 

src/uproot_browser/tui/browser.css

@@ -46,6 +46,14 @@ Browser.-show-tree #tree-view {
     content-align: center middle;
 }
 
+#plot-input-container {
+    height: 3;
+}
+
+#plot-button {
+    dock: right;
+}
+
 
 Footer > .footer--highlight {
     background: $secondary-darken-2;
@@ -82,3 +90,4 @@ HelpScreen {
     height: 3;
     background: $surface-lighten-1;
 }
+

src/uproot_browser/tui/browser.py

@@ -86,6 +86,10 @@ def compose(self) -> textual.app.ComposeResult:
                 id="main-view",
                 initial="logo",
             )
+            with textual.containers.Container(id="plot-input-container"):
+                yield textual.widgets.Input(id="plot-input")
+                yield textual.widgets.Button("Plot", id="plot-button")
+
         yield textual.widgets.Footer()
 
     def on_mount(self, _event: textual.events.Mount) -> None:
@@ -147,6 +151,7 @@ def on_uproot_selected(self, message: UprootSelected) -> None:
             make_plot(message.upfile[message.path], theme, 20)
             self.plot_widget.item = Plotext(message.upfile, message.path, theme)
             content_switcher.current = "plot"
+            self.query_one("#plot-input", textual.widgets.Input).value = f"f[{message.path!r}]"
 
         except EmptyTreeError:
             content_switcher.current = "empty"

src/uproot_browser/tui/right_panel.py

@@ -13,6 +13,7 @@
 import textual.widgets
 
 import uproot_browser.plot
+import uproot_browser.dirs
 
 LOGO = """\
 Scikit-HEP