bias.go

package model

import (
	"fmt"
	"github.com/Azbesciak/RealDecisionMaker/lib/utils"
)

//go:generate easytags $GOFILE json:camel

type BiasesParams = []interface{}
type BiasProps = interface{}
type BiasMap = map[string]Bias
type Biases = []Bias
type BiasesWithProps = []BiasWithProps

type BiasParams struct {
	Name             string    `json:"name"`
	Disabled         bool      `json:"disabled"`
	ApplyProbability float64   `json:"applyProbability"`
	Props            BiasProps `json:"props"`
}

type BiasWithProps struct {
	Bias  *Bias       `json:"bias"`
	Props *BiasParams `json:"props"`
}

type Bias interface {
	utils.Identifiable
	Apply(
		original, current *DecisionMakingParams,
		props *BiasProps,
		listener *BiasListener,
	) *BiasedResult
}
type BiasedResult struct {
	DMP   *DecisionMakingParams `json:"dm"`
	Props BiasProps             `json:"props"`
}

func AsBiasesMap(h *Biases) *BiasMap {
	result := make(BiasMap, len(*h))
	for _, bias := range *h {
		result[bias.Identifier()] = bias
	}
	return &result
}

func ChooseBiases(available *BiasMap, choose *BiasesParams) *BiasesWithProps {
	var result BiasesWithProps
	for _, props := range *choose {
		biasParams := BiasParams{ApplyProbability: 1}
		utils.DecodeToStruct(props, &biasParams)
		if biasParams.Disabled {
			continue
		}
		bias, ok := (*available)[biasParams.Name]
		if !ok {
			var keys []string
			for k := range *available {
				keys = append(keys, k)
			}
			panic(fmt.Errorf("bias '%s' not found, available are '%s'", biasParams.Name, keys))
		}
		result = append(result, BiasWithProps{Bias: &bias, Props: &biasParams})
	}
	return &result
}

func UpdateBiasesProps(oldProps *BiasParams, update BiasProps) *BiasParams {
	return &BiasParams{
		Name:             oldProps.Name,
		ApplyProbability: oldProps.ApplyProbability,
		Props:            update,
	}
}