preserves/implementations/javascript/packages/core/src/float.ts

103 lines
2.8 KiB
TypeScript

import { Tag } from "./constants";
import { stringify } from "./text";
import { Value } from "./values";
import type { GenericEmbedded } from "./embedded";
import type { Encoder, Preservable } from "./encoder";
import type { Writer, PreserveWritable } from "./writer";
import * as IO from "./iolist";
export type FloatType = 'Single' | 'Double';
export const FloatType = Symbol.for('FloatType');
export abstract class Float {
readonly value: number;
constructor(value: number | Float) {
this.value = typeof value === 'number' ? value : value.value;
}
toString() {
return stringify(this);
}
equals(other: any): boolean {
return Object.is(other.constructor, this.constructor) && (other.value === this.value);
}
hashCode(): number {
return (this.value | 0); // TODO: something better?
}
abstract get [FloatType](): FloatType;
static isFloat = (x: any): x is Float => x?.[FloatType] !== void 0;
static isSingle = (x: any): x is SingleFloat => x?.[FloatType] === 'Single';
static isDouble = (x: any): x is DoubleFloat => x?.[FloatType] === 'Double';
}
export function floatValue(f: any): number {
if (typeof f === 'number') {
return f;
} else if (Float.isFloat(f)) {
return f.value;
} else {
return NaN;
}
}
export class SingleFloat extends Float implements Preservable<any>, PreserveWritable<any> {
__as_preserve__<T = GenericEmbedded>(): Value<T> {
return this;
}
static __from_preserve__<T>(v: Value<T>): undefined | SingleFloat {
return Float.isSingle(v) ? v : void 0;
}
__preserve_on__<T>(_encoder: Encoder<T>): IO.IOList {
const bs = new Uint8Array(4);
new DataView(bs.buffer).setFloat32(0, this.value, false);
return [Tag.Float, bs];
}
__preserve_text_on__(w: Writer<any>) {
w.state.pieces.push('' + this.value + 'f');
}
get [FloatType](): 'Single' {
return 'Single';
}
}
export function Single(value: number | Float): SingleFloat {
return new SingleFloat(value);
}
export class DoubleFloat extends Float implements Preservable<any>, PreserveWritable<any> {
__as_preserve__<T = GenericEmbedded>(): Value<T> {
return this;
}
static __from_preserve__<T>(v: Value<T>): undefined | DoubleFloat {
return Float.isDouble(v) ? v : void 0;
}
__preserve_on__<T>(_encoder: Encoder<T>): IO.IOList {
const bs = new Uint8Array(8);
new DataView(bs.buffer).setFloat64(0, this.value, false);
return [Tag.Float, bs];
}
__preserve_text_on__(w: Writer<any>) {
w.state.pieces.push('' + this.value);
}
get [FloatType](): 'Double' {
return 'Double';
}
}
export function Double(value: number | Float): DoubleFloat {
return new DoubleFloat(value);
}