From a21aaa41faa341a1a34f3e1f0e969b5009b66337 Mon Sep 17 00:00:00 2001
From: hovsep <avakyan.hovsep@yandex.ru>
Date: Tue, 21 Jan 2025 02:25:24 +0200
Subject: [PATCH] [WIP] Simple electric circuit simulation

---
 examples/battery_and_lightbulb/main.go | 94 ++++++++++++++++++++++++++
 fmesh.go                               |  3 +
 2 files changed, 97 insertions(+)
 create mode 100644 examples/battery_and_lightbulb/main.go

diff --git a/examples/battery_and_lightbulb/main.go b/examples/battery_and_lightbulb/main.go
new file mode 100644
index 0000000..451c9ec
--- /dev/null
+++ b/examples/battery_and_lightbulb/main.go
@@ -0,0 +1,94 @@
+package main
+
+import (
+	"fmt"
+	"github.com/hovsep/fmesh"
+	"github.com/hovsep/fmesh/component"
+	"github.com/hovsep/fmesh/port"
+	"github.com/hovsep/fmesh/signal"
+)
+
+const (
+	// lightBulbPowerConsumption represents how much energy the lightbulb consumes per cycle
+	lightBulbPowerConsumption = 37
+
+	// lightBulbLuminousFlux represents how much light is generated during one activation cycle
+	lightBulbLuminousFlux = 1
+)
+
+func main() {
+	//Init battery level (state stored outside the component)
+	batteryLevel := 100
+
+	battery := component.New("battery").
+		WithDescription("electric battery with initial charge level").
+		WithInputs("power_demand").
+		WithOutputs("power_supply").
+		WithActivationFunc(func(inputs *port.Collection, outputs *port.Collection) error {
+			//@TODO add component level loggers
+			fmt.Println("battery:activated, level= ", batteryLevel)
+			//Power demand/supply cycle
+			if inputs.ByName("power_demand").HasSignals() {
+				demandedCurrent := inputs.ByName("power_demand").FirstSignalPayloadOrDefault(0).(int)
+				fmt.Println("battery:consumption = ", demandedCurrent)
+
+				//Emit current represented as a number
+				suppliedCurrent := min(batteryLevel, demandedCurrent)
+				if suppliedCurrent > 0 {
+					outputs.ByName("power_supply").PutSignals(signal.New(suppliedCurrent))
+					fmt.Println("battery:emiting power ", suppliedCurrent)
+
+					//Discharge
+					batteryLevel = max(0, batteryLevel-suppliedCurrent)
+					fmt.Println("battery:discharged to", batteryLevel)
+				} else {
+					fmt.Println("battery:LOW POWER")
+				}
+			}
+
+			return nil
+		})
+
+	lightbulb := component.New("lightbulb").
+		WithDescription("electric lightbulb").
+		WithInputs("power_supply", "start_power_demand").
+		WithOutputs("light_supply", "power_demand").
+		WithActivationFunc(func(inputs *port.Collection, outputs *port.Collection) error {
+			fmt.Println("bulb:activated")
+			//Power consumption cycle (at constant rate)
+			inputPower := inputs.ByName("power_supply").FirstSignalPayloadOrDefault(0).(int)
+			fmt.Println("bulb:got power: ", inputPower)
+
+			if inputPower >= lightBulbPowerConsumption {
+				//Emit light
+				outputs.ByName("light_supply").PutSignals(signal.New(lightBulbLuminousFlux))
+				fmt.Println("bulb:emited light: ", lightBulbLuminousFlux)
+			} else {
+				fmt.Println("bulb:LOW POWER")
+			}
+
+			//Always continue demanding power
+			outputs.ByName("power_demand").PutSignals(signal.New(lightBulbPowerConsumption))
+			fmt.Println("bulb:demanded power: ", lightBulbPowerConsumption)
+			return nil
+		})
+
+	battery.OutputByName("power_supply").PipeTo(lightbulb.InputByName("power_supply"))
+	lightbulb.OutputByName("power_demand").PipeTo(battery.InputByName("power_demand"))
+
+	fm := fmesh.New("battery_and_lightbulb").
+		WithDescription("simple electric simulation").
+		WithComponents(battery, lightbulb)
+
+	// Turn on the lightbulb (yes you can init an output port)
+	lightbulb.InputByName("start_power_demand").PutSignals(signal.New("start"))
+
+	cycles, err := fm.Run()
+
+	if err != nil {
+		fmt.Println("Simulation failed with error: ", err)
+		return
+	}
+
+	fmt.Println(fmt.Sprintf("Simulation finished with %d cycles", len(cycles)))
+}
diff --git a/fmesh.go b/fmesh.go
index 02345e0..16a9308 100644
--- a/fmesh.go
+++ b/fmesh.go
@@ -94,6 +94,9 @@ func (fm *FMesh) WithConfig(config Config) *FMesh {
 func (fm *FMesh) runCycle() {
 	newCycle := cycle.New().WithNumber(fm.cycles.Len() + 1)
 
+	//@TODO:Add debug mode
+	fmt.Println("CYCLE #", newCycle.Number())
+
 	if fm.HasErr() {
 		newCycle.SetErr(fm.Err())
 	}