Skip to content

Commit 0e9c903

Browse files
committed
Restoring scenario LCA calculations
1 parent 5557d92 commit 0e9c903

1 file changed

Lines changed: 12 additions & 3 deletions

File tree

activity_browser/app/pages/calculation_setup/scenario_section.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
from activity_browser import app
1212
from activity_browser.ui import icons, widgets
13+
from activity_browser.ui import core
1314

1415

1516

@@ -264,8 +265,8 @@ def __init__(self, index: int, parent=None):
264265
self.load_btn.setToolTip("Load (new) data for this scenario table")
265266
self.remove_btn = QtWidgets.QPushButton(icons.qicons.delete, "Delete")
266267
self.remove_btn.setToolTip("Remove this scenario table")
267-
self.view = widgets.ABTreeView(self)
268-
self.model = widgets.ABItemModel(self)
268+
self.view = ScenarioImportView(self)
269+
self.model = ScenarioImportModel(parent=self)
269270
self.view.setModel(self.model)
270271
self.scenario_df = pd.DataFrame(columns=ss.SUPERSTRUCTURE)
271272

@@ -373,7 +374,7 @@ def sync_superstructure(self, df: pd.DataFrame) -> None:
373374
return
374375
self.scenario_df = df
375376
cols = ss.scenario_names_from_df(self.scenario_df)
376-
self.model.setDataFrame(pd.DataFrame(cols, columns=["Scenarios"]))
377+
self.model.set_dataframe(pd.DataFrame(cols, columns=["Scenarios"]))
377378
self._parent.combined_dataframe()
378379

379380
def scenario_db_check(self, df: pd.DataFrame) -> pd.DataFrame:
@@ -399,6 +400,14 @@ def dataframe(self) -> pd.DataFrame:
399400
return self.scenario_df
400401

401402

403+
class ScenarioImportView(widgets.ABTreeView):
404+
"""Tree view for scenario imports."""
405+
406+
407+
class ScenarioImportModel(core.ABTreeModel):
408+
"""Model for displaying imported scenario names."""
409+
410+
402411
class ExcelReadDialog(QtWidgets.QDialog):
403412
SUFFIXES = {
404413
".xls",

0 commit comments

Comments
 (0)