Add max branches configuration from GUI

davet2001 · Jan 6, 2025 · 1fe4426 · 1fe4426
1 parent 8316699
commit 1fe4426
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/hui-power-flow-card.ts b/src/hui-power-flow-card.ts
@@ -219,6 +219,8 @@ class HuiPowerFlowCard extends LitElement implements LovelaceCard {
       delete (config.generation_entity)
     }
 
+    const maxConsumerBranches = this._config.max_consumer_branches || 0;
+
     let gridInRoute: ElecRoute | null = null;
     if (config.power_from_grid_entity) {
       const stateObj = this.hass.states[config.power_from_grid_entity];
@@ -322,6 +324,7 @@ class HuiPowerFlowCard extends LitElement implements LovelaceCard {
             .gridOutRoute=${gridOutRoute || undefined}
             .generationInRoutes=${generationInRoutes}
             .consumerRoutes=${consumerRoutes}
+            .maxConsumerBranches=${maxConsumerBranches}
           ></ha-elec-sankey>
         </div>
       </ha-card>

diff --git a/src/power-flow-card-editor.ts b/src/power-flow-card-editor.ts
@@ -36,6 +36,16 @@ const schema = [
       }
     }
   },
+  {
+    name: "max_consumer_branches",
+    selector: {
+      number: {
+        min: 0,
+        max: 10,
+        mode: "slider",
+      }
+    }
+  }
   // { name: "group_small", selector: { boolean: {} } },
 ];
 
@@ -61,6 +71,7 @@ export class PowerFlowCardEditor extends LitElement implements LovelaceCardEdito
       case "power_from_grid_entity": return "Power from grid";
       case "group_small": return "Group low values together";
       case "generation_entity": return "Power from generation (optional)";
+      case "max_consumer_branches": return "Limit displayed consumer branches (0 for unlimited)";
     }
     console.error("Error name key missing for '" + schema.name + "'")
     return ""
@@ -145,6 +156,11 @@ export class PowerFlowCardEditor extends LitElement implements LovelaceCardEdito
         configValue = "generation_entity";
         value = value.generation_entity;
       }
+      else if (value.max_consumer_branches
+        !== this._config.max_consumer_branches || -1) {
+        configValue = "max_consumer_branches";
+        value = value.max_consumer_branches;
+      }
       else {
         console.warn("unhandled change in <ha-form>");
       }