brightway-basic-explorer 0.9.dev2__tar.gz → 0.9.dev4__tar.gz

{brightway_basic_explorer-0.9.dev2 → brightway_basic_explorer-0.9.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brightway-basic-explorer
-Version: 0.9.dev2
+Version: 0.9.dev4
 Summary: Implement basic GUI to explorer brightway ativities
 Author-email: Benoît GSCHWIND <benoit.gschwind@minesparis.psl.eu>, OIE - Mines Paris PSL <benoit.gschwind@minesparis.psl.eu>
 Maintainer-email: Benoît GSCHWIND <benoit.gschwind@minesparis.psl.eu>

{brightway_basic_explorer-0.9.dev2 → brightway_basic_explorer-0.9.dev4}/brightway_basic_explorer/__init__.py

@@ -1,7 +1,7 @@
 # coding=utf-8
 
 from IPython.external.qt_for_kernel import QtGui, QtCore
-from IPython.lib.guisupport import start_event_loop_qt4, get_app_qt4, is_event_loop_running_qt4
+from IPython.lib.guisupport import get_app_qt4, is_event_loop_running_qt4
 
 import os
 import sys
@@ -18,6 +18,39 @@ def activity_to_json(act):
     a["exchanges"] = exs
     return a
 
+def activity_to_json_with_params(act, params):
+    try:
+        from lca_algebraic.params import (
+            all_params,
+            _complete_and_expand_params,
+            _getAmountOrFormula,
+        )
+
+        from sympy import Basic
+    except:
+        raise Exception("lca_algebraic not found, please install it before using show_activity_with_params")
+
+    a = dict(act)
+    exs = list()
+    for e in act.exchanges():
+        de = dict(bw2data.get_activity(e["input"]))
+
+        amount = _getAmountOrFormula(e)
+
+        # Params provided ? Evaluate formulas
+        if isinstance(amount, Basic):
+            new_params = list(_complete_and_expand_params(params, list(all_params().keys())).items())
+            amount = amount.subs(new_params)
+            if amount.is_number:
+                amount = float(amount.evalf())
+            de["computed_amount"] = True
+
+        de["amount"] = amount
+        de["formula"] = e.get("formula", None)
+        exs.append(de)
+    a["exchanges"] = exs
+    return a
+
 class QStandardItemRO(QtGui.QStandardItem):
     def __init__(self, *args, data=None, **kwargs):
         super().__init__(*args, **kwargs)
@@ -46,6 +79,9 @@ class TableModel(QtGui.QStandardItemModel):
                 for k in ["name", "unit", "categories", "location", "amount", "formula"]
             ]
 
+            if "computed_amount" in e:
+               row[-2].setForeground(QtGui.QBrush(QtCore.Qt.GlobalColor.red))
+
             etype = e.get("type", "unknown")
             if etype == "emission":
                 path = os.path.join(os.path.dirname(__file__), "icons", "emission.png")
@@ -73,6 +109,9 @@ class ActionMenu(QtGui.QMenu):
         self.action_explore = self.addAction("Explore")
         self.action_explore.triggered.connect(self.explore_triggered)
 
+        self.action_explore = self.addAction("Explore in new window")
+        self.action_explore.triggered.connect(self.explore_in_new_window_triggered)
+
     def copy_triggered(self):
         QtGui.QGuiApplication.clipboard().setText(self.index.data())
 
@@ -85,31 +124,82 @@ class ActionMenu(QtGui.QMenu):
     def explore_triggered(self):
         self.parentWidget().explore(self.index)
 
+    def explore_in_new_window_triggered(self):
+        self.parentWidget().explore_in_new_window(self.index)
+
+class TabBar(QtGui.QTabBar):
+    def __init__(self):
+        super().__init__()
+
+    def tabSizeHint(self, index):
+        value = super().tabSizeHint(index)
+        return QtCore.QSize(200, value.height())
+
+    def minimumTabSizeHint(self, index):
+        value = super().minimumTabSizeHint(index)
+        return QtCore.QSize(100, value.height())
+
 
 class ActivityWindow(QtGui.QMainWindow):
-    keep = dict()
+    keep = list()
 
-    def __init__(self, act):
+    def __init__(self, activity_json, params=None):
         super().__init__()
-        self.act = act
-
-        data = activity_to_json(act)
+        self.tabs = dict()
+        self.xcount = 0
+        self.params = params
 
         self.setWindowTitle("Activity Viewer")
-        self.setGeometry(100, 100, 800, 600)
+        self.setMinimumSize(QtCore.QSize(800, 600))
+
+        self.tabs_widget = QtGui.QTabWidget()
+        self.tabs_widget.setTabBar(TabBar())
+        self.tabs_widget.setTabsClosable(True)
+        self.tabs_widget.setMovable(True)
+        self.tabs_widget.tabCloseRequested.connect(self.close_tab)
+        self.tabs_widget.setElideMode(QtCore.Qt.TextElideMode.ElideRight)
+        self.setCentralWidget(self.tabs_widget)
+        self.add_activity(activity_json)
+
+    def add_activity(self, activity_json):
+        key = (activity_json["database"], activity_json["code"])
+        if key in self.tabs:
+            activity_widget = self.tabs[key]
+            self.tabs_widget.setCurrentWidget(activity_widget)
+            return
+
+        activity_widget = ActivityTab(self, activity_json)
+        self.tabs[activity_widget.activity_key] = activity_widget
+        self.tabs_widget.addTab(activity_widget, f"#{self.xcount} "+activity_json["name"])
+        self.tabs_widget.setCurrentWidget(activity_widget)
+        activity_widget.tree_view.setColumnWidth(0, 400)
+        self.xcount += 1
+
+    def close_tab(self, index):
+        w = self.tabs_widget.widget(index)
+        del self.tabs[w.activity_key]
+        self.tabs_widget.removeTab(index)
+
+    def closeEvent(self, ev):
+        ActivityWindow.keep.remove(self)
+        super().closeEvent(ev)
+
+class ActivityTab(QtGui.QWidget):
+    def __init__(self, xparent, activity_json):
+        super().__init__()
+
+        self.xparent = xparent
+        self.activity_key = (activity_json["database"], activity_json["code"])
 
-        # Create central widget and layout
-        central_widget = QtGui.QWidget()
-        self.setCentralWidget(central_widget)
         layout = QtGui.QVBoxLayout()
-        central_widget.setLayout(layout)
+        self.setLayout(layout)
 
         grid = QtGui.QGridLayout()
         layout.addLayout(grid)
 
         for i, k in enumerate(["database", "name", "location", "unit", "categories", "type"]):
             grid.addWidget(QtGui.QLabel(f"{k}:"), i, 0)
-            x = QtGui.QLabel(f"{str(data.get(k, '-'))}")
+            x = QtGui.QLabel(f"{str(activity_json.get(k, '-'))}")
             x.setTextInteractionFlags(QtCore.Qt.TextInteractionFlag.TextSelectableByMouse)
             grid.addWidget(x, i, 1)
         next_row = grid.rowCount()
@@ -134,7 +224,7 @@ class ActivityWindow(QtGui.QMainWindow):
         self.tree_view.setAlternatingRowColors(True)
 
         self.model = TableModel()
-        self.model.load(data["exchanges"])
+        self.model.load(activity_json["exchanges"])
         self.tree_view.setModel(self.model)
         self.tree_view.doubleClicked.connect(self.doubleCliked)
         self.tree_view.setExpandsOnDoubleClick(False)
@@ -148,11 +238,6 @@ class ActivityWindow(QtGui.QMainWindow):
         self.action_menu = ActionMenu(self, index)
         self.action_menu.exec(self.tree_view.viewport().mapToGlobal(point))
 
-    def closeEvent(self, ev):
-        if self.act.key in ActivityWindow.keep:
-            del ActivityWindow.keep[self.act.key]
-        super().closeEvent(ev)
-
     def update_filter(self, *args):
         text = self.filter_edit.text()
         if self.ignore_case.isChecked():
@@ -173,11 +258,19 @@ class ActivityWindow(QtGui.QMainWindow):
     def explore(self, index):
         r = index.row()
         v = self.model.root.child(r, 0).data()
-        show_activity(bw2data.get_activity((v["database"], v["code"])))
+        act = bw2data.get_activity((v["database"], v["code"]))
+        if self.xparent.params is None:
+            activity_json = activity_to_json(act)
+        else:
+            activity_json = activity_to_json_with_params(act, self.xparent.params)
+        self.xparent.add_activity(activity_json)
+
+    def explore_in_new_window(self, index):
+        r = index.row()
+        v = self.model.root.child(r, 0).data()
+        act = bw2data.get_activity((v["database"], v["code"]))
+        show_activity(act, self.xparent.params)
 
-    def show(self, *args, **kwargs):
-        super().show(*args, **kwargs)
-        self.tree_view.setColumnWidth(0, 400)
 
 # Replace IPython version to one compatible with Qt6
 def start_event_loop_qt4(app=None):
@@ -194,7 +287,7 @@ def start_event_loop_qt4(app=None):
     else:
         app._in_event_loop = True
 
-def show_activity(act):
+def show_activity(act, params=None):
 
     if 'matplotlib' in sys.modules:
         import matplotlib
@@ -204,9 +297,13 @@ def show_activity(act):
 
     app = get_app_qt4()
 
-    if (window := ActivityWindow.keep.get(act.key, None)) is None:
-        window = ActivityWindow(act)
-        ActivityWindow.keep[act.key] = window
+    if params is None:
+        activity_json = activity_to_json(act)
+    else:
+        activity_json = activity_to_json_with_params(act, params)
+
+    window = ActivityWindow(activity_json, params)
+    ActivityWindow.keep.append(window)
     window.show()
 
     if not is_event_loop_running_qt4(app):
@@ -215,5 +312,5 @@ def show_activity(act):
     window.activateWindow()
 
 def close_all():
-    for w in list(ActivityWindow.keep.values()):
+    for w in list(ActivityWindow.keep):
         w.close()

{brightway_basic_explorer-0.9.dev2 → brightway_basic_explorer-0.9.dev4}/brightway_basic_explorer.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: brightway-basic-explorer
-Version: 0.9.dev2
+Version: 0.9.dev4
 Summary: Implement basic GUI to explorer brightway ativities
 Author-email: Benoît GSCHWIND <benoit.gschwind@minesparis.psl.eu>, OIE - Mines Paris PSL <benoit.gschwind@minesparis.psl.eu>
 Maintainer-email: Benoît GSCHWIND <benoit.gschwind@minesparis.psl.eu>

{brightway_basic_explorer-0.9.dev2 → brightway_basic_explorer-0.9.dev4}/pyproject.toml

@@ -12,7 +12,7 @@ name = "brightway-basic-explorer"
 description = "Implement basic GUI to explorer brightway ativities"
 readme = "README.md"
 license-files = ["COPYING"]
-version = "0.9.dev2"
+version = "0.9.dev4"
 dependencies = [ "ipython" ]
 requires-python = ">= 3.10"
 keywords = ["lca", "ecoinvent", "brightway"]