edu.ie3.netpad.grid.info.GridInfoController Maven / Gradle / Ivy

Go to download
/*
 * © 2020. TU Dortmund University,
 * Institute of Energy Systems, Energy Efficiency and Energy Economics,
 * Research group Distribution grid planning and operation
*/
package edu.ie3.netpad.grid.info;

import static javafx.scene.control.CheckBoxTreeItem.checkBoxSelectionChangedEvent;

import edu.ie3.datamodel.models.voltagelevels.VoltageLevel;
import edu.ie3.netpad.grid.event.GridEvent;
import edu.ie3.netpad.grid.event.GridEventListener;
import edu.ie3.netpad.grid.event.ReplaceGridEvent;
import edu.ie3.netpad.util.ListenerUtil;
import java.util.*;
import javafx.beans.property.ObjectProperty;
import javafx.beans.property.SimpleObjectProperty;
import javafx.beans.value.ChangeListener;
import javafx.event.EventHandler;
import javafx.fxml.FXML;
import javafx.scene.control.*;
import javafx.scene.layout.VBox;
import org.controlsfx.control.CheckTreeView;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * //ToDo: Class Description
 *
 * @version 0.1
 * @since 24.05.20
 */
public class GridInfoController implements GridEventListener {

  private static final Logger log = LoggerFactory.getLogger(GridInfoController.class);

  @FXML private VBox upperVBox;

  @FXML private CheckTreeView selectedGridCheckTreeView;

  private final ChangeListener gridEventListener =
      ListenerUtil.createGridEventListener(this);
  private final HashMap, UUID> subGridCheckBoxes = new HashMap<>();

  private final ObjectProperty gridInfoEventProperty = new SimpleObjectProperty<>();

  @Override
  public ChangeListener gridEventListener() {
    return gridEventListener;
  }

  @Override
  public void handleGridEvent(GridEvent gridEvent) {

    if (gridEvent instanceof ReplaceGridEvent) {
      handleReplaceGridEvent((ReplaceGridEvent) gridEvent);
    } else {
      log.warn(
          "The provided GridEvent {} is not supported by the GridInfoController!",
          gridEvent.getClass().getSimpleName());
    }
  }

  private void handleReplaceGridEvent(ReplaceGridEvent gridEvent) {

    CheckBoxTreeItem root = new CheckBoxTreeItem<>(gridEvent.getGridName());
    root.setExpanded(false);
    root.setSelected(true);

    selectedGridCheckTreeView.setRoot(root);

    Map> voltageLvls = new HashMap<>();

    gridEvent
        .getSubGrids()
        .forEach(
            (uuid, subGrid) -> {
              CheckBoxTreeItem voltageLvlChkBox =
                  Optional.ofNullable(voltageLvls.get(subGrid.getPredominantVoltageLevel()))
                      .orElseGet(
                          () ->
                              new CheckBoxTreeItem<>(subGrid.getPredominantVoltageLevel().getId()));

              voltageLvlChkBox.setSelected(true);

              CheckBoxTreeItem checkBoxTreeItem =
                  new CheckBoxTreeItem<>(Integer.toString(subGrid.getSubnet()));

              checkBoxTreeItem.setSelected(true);
              checkBoxTreeItem.addEventHandler(
                  checkBoxSelectionChangedEvent(),
                  (EventHandler>)
                      event -> {
                        CheckBoxTreeItem chk = event.getTreeItem();
                        UUID subGridUUID = subGridCheckBoxes.get(chk);
                        notifyListener(new GridInfoEvent(subGridUUID, chk.isSelected()));
                      });

              voltageLvlChkBox.getChildren().add(checkBoxTreeItem);

              voltageLvls.put(subGrid.getPredominantVoltageLevel(), voltageLvlChkBox);
              subGridCheckBoxes.put(checkBoxTreeItem, uuid);
            });

    voltageLvls.values().forEach(voltageLvlChkBox -> root.getChildren().add(voltageLvlChkBox));
  }

  public ObjectProperty gridInfoEvents() {
    return gridInfoEventProperty;
  }

  private void notifyListener(GridInfoEvent gridInfoEvent) {
    gridInfoEventProperty.set(gridInfoEvent);
  }
}