Add staircase fan curve

Author: 0xIO32

README.md

@@ -584,6 +584,25 @@ curves:
         - 80: 255
 ```
 
+#### Staircase
+
+To create a simple, staircase speed curve, use a curve of type `staircase`.
+
+It maintains a static fan speed and avoids constantly changing fan speeds.
+
+```yaml
+curves:
+  - id: staircase_curve
+    staircase:
+      sensor: cpu_package
+      threshold: 6
+      steps:
+        # Sensor value (in degress Celsius) -> Speed (0-255)
+        - 40: 1
+        - 50: 50
+        - 80: 255
+```
+
 #### PID
 
 If you want to get your hands dirty and use a PID based curve, you can use `pid`:

internal/configuration/config.go

@@ -189,40 +189,51 @@ func applyTransformations() {
 	// convert steps in linear curves from strings (with plain numbers or percent values) to floats between 0 and 255
 	for _, curve := range CurrentConfig.Curves {
 		if curve.Linear != nil && len(curve.Linear.InSteps) > 0 {
-			curve.Linear.Steps = make(map[int]float64)
-
-			for temp, origstr := range curve.Linear.InSteps {
-				str := strings.TrimSpace(origstr)
-				l := len(str)
-				isPercent := false
-				if l > 1 && str[l-1] == '%' {
-					isPercent = true
-					str = str[:l-1] // cut off '%' because ParseFloat() wouldn't like it
+			transformCurveSteps(&curve.ID, &curve.Linear.Steps, &curve.Linear.InSteps)
+		}
+		if curve.Staircase != nil && len(curve.Staircase.InSteps) > 0 {
+			if len(curve.Staircase.InSteps) > 0 {
+				transformCurveSteps(&curve.ID, &curve.Staircase.Steps, &curve.Staircase.InSteps)
+			} else {
+				ui.Fatal("Missing steps in curve %s", curve.ID)
+			}
+		}
+	}
+}
+
+func transformCurveSteps(ID *string, Steps *map[int]float64, InSteps *map[int]string) {
+	*Steps = make(map[int]float64)
+
+	for temp, origstr := range *InSteps {
+		str := strings.TrimSpace(origstr)
+		l := len(str)
+		isPercent := false
+		if l > 1 && str[l-1] == '%' {
+			isPercent = true
+			str = str[:l-1] // cut off '%' because ParseFloat() wouldn't like it
+		}
+		speed, err := strconv.ParseFloat(str, 64)
+		if err != nil {
+			ui.Fatal("Invalid curve step value '%s' in %s - must be either just a number or a number followed by '%%'", origstr, *ID)
+		} else {
+			if isPercent {
+				if speed < 0 || speed > 100 {
+					ui.Fatal("invalid curve step value '%s' (=> %f) in %s - must be between 0%% and 100%%", origstr, speed, *ID)
 				}
-				speed, err := strconv.ParseFloat(str, 64)
-				if err != nil {
-					ui.Fatal("Invalid curve step value '%s' in %s - must be either just a number or a number followed by '%%'", origstr, curve.ID)
+				// convert 0-100% into [0..255]
+				if speed < 1 {
+					// less than 1% always turns into 0
+					speed = 0
 				} else {
-					if isPercent {
-						if speed < 0 || speed > 100 {
-							ui.Fatal("invalid curve step value '%s' (=> %f) in %s - must be between 0%% and 100%%", origstr, speed, curve.ID)
-						}
-						// convert 0-100% into [0..255]
-						if speed < 1 {
-							// less than 1% always turns into 0
-							speed = 0
-						} else {
-							// 1% turns into 1, 100% turns into 255
-							// => convert 1..100% to 1..255
-							// => 0..99 to 0..254 and then add 1
-							speed = (speed-1)*(254.0/99.0) + 1
-						}
-					} else if speed < 0 || speed > 255 {
-						ui.Fatal("invalid curve step value '%s' in %s - must be between 0 and 255", origstr, curve.ID)
-					}
-					curve.Linear.Steps[temp] = speed
+					// 1% turns into 1, 100% turns into 255
+					// => convert 1..100% to 1..255
+					// => 0..99 to 0..254 and then add 1
+					speed = (speed-1)*(254.0/99.0) + 1
 				}
+			} else if speed < 0 || speed > 255 {
+				ui.Fatal("invalid curve step value '%s' in %s - must be between 0 and 255", origstr, *ID)
 			}
+			(*Steps)[temp] = speed
 		}
 	}
 }

internal/configuration/curves.go

@@ -5,9 +5,10 @@ type CurveConfig struct {
 	ID string `json:"id"`
 
 	// can be any of the following:
-	Linear   *LinearCurveConfig   `json:"linear,omitempty"`
-	PID      *PidCurveConfig      `json:"pid,omitempty"`
-	Function *FunctionCurveConfig `json:"function,omitempty"`
+	Linear    *LinearCurveConfig    `json:"linear,omitempty"`
+	Staircase *StaircaseCurveConfig `json:"staircase,omitempty"`
+	PID       *PidCurveConfig       `json:"pid,omitempty"`
+	Function  *FunctionCurveConfig  `json:"function,omitempty"`
 }
 
 type LinearCurveConfig struct {
@@ -26,6 +27,20 @@ type LinearCurveConfig struct {
 	Steps map[int]float64 `json:"steps" mapstructure:"-"`
 }
 
+type StaircaseCurveConfig struct {
+	// Sensor is the id of the sensor to use for this curve
+	Sensor string `json:"sensor"`
+	// Threshold is the temperature threshold in degrees
+	Threshold int `json:"threshold"`
+	// Steps is a map of temperature to relative speed value (in range of 0..255 or alternatively 0%..100%)
+	// InSteps contains the speed values as strings (like "42" or "11%"), as read from fan2go.yaml
+	InSteps map[int]string `mapstructure:"steps" json:"-"`
+	// Steps is created from InSteps on load (LoadConfig()), the strings are converted to floats
+	// between 0 and 255 (0% is 0, 1% is 1; from there on it's interpolated linearly so 100% is 255).
+	// If a string only contains a number (without "%"), it's just converted to float
+	Steps map[int]float64 `json:"steps" mapstructure:"-"`
+}
+
 type PidCurveConfig struct {
 	// Sensor is the id of the sensor to use for this curve
 	Sensor string `json:"sensor"`

internal/configuration/validation.go

@@ -124,6 +124,9 @@ func isSensorConfigInUse(config SensorConfig, curves []CurveConfig) bool {
 		if curveConfig.Linear != nil && curveConfig.Linear.Sensor == config.ID {
 			return true
 		}
+		if curveConfig.Staircase != nil && curveConfig.Staircase.Sensor == config.ID {
+			return true
+		}
 		if curveConfig.PID != nil && curveConfig.PID.Sensor == config.ID {
 			return true
 		}
@@ -146,6 +149,9 @@ func validateCurves(config *Configuration) error {
 		if curveConfig.Linear != nil {
 			subConfigs++
 		}
+		if curveConfig.Staircase != nil {
+			subConfigs++
+		}
 		if curveConfig.PID != nil {
 			subConfigs++
 		}
@@ -196,6 +202,20 @@ func validateCurves(config *Configuration) error {
 			}
 		}
 
+		if curveConfig.Staircase != nil {
+			if len(curveConfig.Staircase.Sensor) <= 0 {
+				return fmt.Errorf("curve %s: missing sensorId", curveConfig.ID)
+			}
+
+			if !sensorIdExists(curveConfig.Staircase.Sensor, config) {
+				return fmt.Errorf("curve %s: no sensor definition with id '%s' found", curveConfig.ID, curveConfig.Staircase.Sensor)
+			}
+
+			if len(curveConfig.Staircase.InSteps) <= 0 {
+				return fmt.Errorf("curve %s: missing steps", curveConfig.ID)
+			}
+		}
+
 		if curveConfig.PID != nil {
 			if len(curveConfig.PID.Sensor) <= 0 {
 				return fmt.Errorf("curve %s: missing sensorId", curveConfig.ID)

internal/curves/curve.go

@@ -2,6 +2,7 @@ package curves
 
 import (
 	"fmt"
+	"math"
 
 	"github.com/markusressel/fan2go/internal/configuration"
 	"github.com/markusressel/fan2go/internal/util"
@@ -30,6 +31,14 @@ func NewSpeedCurve(config configuration.CurveConfig) (SpeedCurve, error) {
 		return ret, nil
 	}
 
+	if config.Staircase != nil {
+		ret := &StaircaseSpeedCurve{
+			Config:   config,
+			LastTemp: math.MinInt,
+		}
+		return ret, nil
+	}
+
 	if config.PID != nil {
 		pidLoop := util.NewPidLoop(
 			config.PID.P,

internal/curves/linear_test.go

@@ -1,10 +1,11 @@
 package curves
 
 import (
+	"testing"
+
 	"github.com/markusressel/fan2go/internal/configuration"
 	"github.com/markusressel/fan2go/internal/sensors"
 	"github.com/stretchr/testify/assert"
-	"testing"
 )
 
 // helper function to create a linear curve configuration

internal/curves/staircase.go

@@ -0,0 +1,60 @@
+package curves
+
+import (
+	"github.com/markusressel/fan2go/internal/ui"
+
+	"github.com/markusressel/fan2go/internal/configuration"
+	"github.com/markusressel/fan2go/internal/sensors"
+)
+
+type StaircaseSpeedCurve struct {
+	Config configuration.CurveConfig `json:"config"`
+	Value  float64                   `json:"value"`
+
+	LastTemp int
+}
+
+func (c *StaircaseSpeedCurve) GetId() string {
+	return c.Config.ID
+}
+
+func (c *StaircaseSpeedCurve) Evaluate() (value float64, err error) {
+	sensor, _ := sensors.GetSensor(c.Config.Staircase.Sensor)
+	var measured float64
+	measured, err = sensor.GetValue()
+	if err != nil {
+                ui.Warning("Curve %s: Error getting sensor value: %v", c.Config.ID, err)
+                return c.Value, err
+        }
+
+	steps := c.Config.Staircase.Steps
+
+	var targetTemp int
+	for temp := range steps {
+		if measured >= float64(temp)*1000 {
+			targetTemp = max(targetTemp, temp)
+		}
+	}
+	if targetTemp < c.LastTemp && (c.LastTemp-int(measured/1000)) < c.Config.Staircase.Threshold {
+		targetTemp = c.LastTemp
+	}
+
+	c.LastTemp = targetTemp
+	value = steps[targetTemp]
+
+	ui.Debug("Evaluating curve '%s'. Sensor '%s' temp '%.0f°'. Desired speed: %.2f", c.Config.ID, sensor.GetId(), measured/1000, value)
+	c.SetValue(value)
+	return value, nil
+}
+
+func (c *StaircaseSpeedCurve) SetValue(value float64) {
+	valueMu.Lock()
+	defer valueMu.Unlock()
+	c.Value = value
+}
+
+func (c *StaircaseSpeedCurve) CurrentValue() float64 {
+	valueMu.Lock()
+	defer valueMu.Unlock()
+	return c.Value
+}