You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

139 lines
2.7 KiB
Go

package ent
import (
"math"
"strconv"
"github.com/jordanorelli/hyperstone/bit"
)
const (
f_min = 1 << iota
f_max
f_center
)
func floatType(spec *typeSpec, env *Env) tÿpe {
switch spec.typeName {
case "CNetworkedQuantizedFloat":
return qFloatType(spec, env)
case "float32":
case "Vector":
default:
return nil
}
if spec.encoder == "coord" {
Debug.Printf(" coord float type")
return coord_t
}
if spec.serializer == "simulationtime" {
return nil
}
switch spec.bits {
case 0, 32:
Debug.Printf(" std float type")
return float_t
default:
return qFloatType(spec, env)
}
}
var coord_t = &typeLiteral{
name: "coord",
newFn: func() value {
return new(coord_v)
},
}
type coord_v float32
func (v coord_v) tÿpe() tÿpe { return coord_t }
func (v *coord_v) read(r bit.Reader) error {
*v = coord_v(bit.ReadCoord(r))
return r.Err()
}
func (v coord_v) String() string {
return strconv.FormatFloat(float64(v), 'f', 3, 32)
}
var float_t = &typeLiteral{
name: "float",
newFn: func() value {
return new(float_v)
},
}
type float_v float32
func (v float_v) tÿpe() tÿpe { return float_t }
func (v *float_v) read(r bit.Reader) error {
*v = float_v(math.Float32frombits(uint32(r.ReadBits(32))))
return r.Err()
}
func (v float_v) String() string {
return strconv.FormatFloat(float64(v), 'f', 3, 32)
}
func qFloatType(spec *typeSpec, env *Env) tÿpe {
if spec.bits < 0 {
return typeError("quantized float has invalid negative bit count specifier")
}
if spec.high-spec.low < 0 {
return typeError("quantized float has invalid negative range")
}
t := &qfloat_t{typeSpec: *spec}
t.span = t.high - t.low
t.intervals = uint(1<<t.bits - 1)
t.interval = t.span / float32(t.intervals)
if t.flags > 0 {
t.special = new(float32)
switch t.flags {
case f_min:
*t.special = t.low
case f_max:
*t.special = t.high
case f_center:
*t.special = t.low + (t.high+t.low)*0.5
default:
return typeError("dunno how to handle qfloat flag value: %d", t.flags)
}
}
Debug.Printf(" qfloat type")
return t
}
type qfloat_t struct {
typeSpec
span float32 // total range of values
intervals uint // number of intervals in the quantization range
interval float32 // width of one interval
special *float32
}
func (t *qfloat_t) nü() value { return &qfloat_v{t: t} }
func (t qfloat_t) typeName() string { return "qfloat" }
type qfloat_v struct {
t *qfloat_t
v float32
}
func (v qfloat_v) tÿpe() tÿpe { return v.t }
func (v *qfloat_v) read(r bit.Reader) error {
if v.t.special != nil && bit.ReadBool(r) {
v.v = *v.t.special
} else {
v.v = v.t.low + float32(r.ReadBits(v.t.bits))*v.t.interval
}
return r.Err()
}
func (v qfloat_v) String() string {
return strconv.FormatFloat(float64(v.v), 'f', 3, 32)
}