Change logic

Author: tsmbland

src/simulation/investment.rs

@@ -206,20 +206,19 @@ fn update_demand_map(demand: &mut AllDemandMap, flows: &FlowMap, assets: &[Asset
                 time_slice.clone(),
             );
 
-            // Only consider output flows from the primary commodity
-            if flow > &Flow(0.0)
-                && asset
+            // Only consider input flows and output flows from the primary output commodity
+            // (excluding secondary outputs)
+            if (flow < &Flow(0.0))
+                | asset
                     .primary_output()
                     .is_some_and(|p| &p.commodity.id != commodity_id)
             {
-                continue;
+                // Note: we use the negative of the flow as input flows are negative in the flow map.
+                demand
+                    .entry(key)
+                    .and_modify(|value| *value -= *flow)
+                    .or_insert(-*flow);
             }
-
-            // Note: we use the negative of the flow as input flows are negative in the flow map.
-            demand
-                .entry(key)
-                .and_modify(|value| *value -= *flow)
-                .or_insert(-*flow);
         }
     }
 }