71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
|
// CookieJar - A contestant's algorithm toolbox
|
||
|
|
// Copyright (c) 2014 Peter Szilagyi. All rights reserved.
|
||
|
|
//
|
||
|
|
// CookieJar is dual licensed: use of this source code is governed by a BSD
|
||
|
|
// license that can be found in the LICENSE file. Alternatively, the CookieJar
|
||
|
|
// toolbox may be used in accordance with the terms and conditions contained
|
||
|
|
// in a signed written agreement between you and the author(s).
|
||
|
|
|
||
|
|
package utility
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math"
|
||
|
|
|
||
|
|
"gopkg.in/karalabe/cookiejar.v2/collections/bag"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Combination utility derived from two other utilities.
|
||
|
|
type comboUtility struct {
|
||
|
|
combinator Combinator // Curve transformation combinator
|
||
|
|
srcA, srcB singleUtility // Base utilities from which to derive this one
|
||
|
|
|
||
|
|
children *bag.Bag // Derived utilities based on the current one
|
||
|
|
|
||
|
|
reset bool // Flag whether the output is not yet calculated
|
||
|
|
output float64 // Cached output utility value
|
||
|
|
}
|
||
|
|
|
||
|
|
// Creates a new derived utility based on two existing ones.
|
||
|
|
func newComboUtility(combinator Combinator) *comboUtility {
|
||
|
|
return &comboUtility{
|
||
|
|
combinator: combinator,
|
||
|
|
children: bag.New(),
|
||
|
|
reset: true,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Finishes initialization with the two source components.
|
||
|
|
func (u *comboUtility) Init(srcA, srcB utility) {
|
||
|
|
u.srcA = srcA.(singleUtility)
|
||
|
|
u.srcB = srcB.(singleUtility)
|
||
|
|
|
||
|
|
srcA.Dependency(u)
|
||
|
|
srcB.Dependency(u)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Resets the utility value, forcing it to recalculate when needed again.
|
||
|
|
func (u *comboUtility) Reset() {
|
||
|
|
if !u.reset {
|
||
|
|
u.reset = true
|
||
|
|
u.children.Do(func(util interface{}) {
|
||
|
|
util.(*comboUtility).Reset()
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Adds a new dependency to the utility hierarchy.
|
||
|
|
func (u *comboUtility) Dependency(util utility) {
|
||
|
|
u.children.Insert(util)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Evaluates the utility chain and returns the current value.
|
||
|
|
func (u *comboUtility) Evaluate() float64 {
|
||
|
|
// If the utility was reset, reevaluate it
|
||
|
|
if u.reset {
|
||
|
|
u.output = math.Min(1, math.Max(0, u.combinator(u.srcA.Evaluate(), u.srcB.Evaluate())))
|
||
|
|
u.reset = false
|
||
|
|
}
|
||
|
|
// Return the currently set value
|
||
|
|
return u.output
|
||
|
|
}
|