35 lines
856 B
TypeScript
35 lines
856 B
TypeScript
import { mapTemporal } from './temporal'
|
|
import { convertUnit, serializeUnit, simplifyUnit } from './units'
|
|
import { ASTNode, EvaluatedNode, Unit } from './AST/types'
|
|
|
|
export function simplifyNodeUnit(node) {
|
|
if (!node.unit) {
|
|
return node
|
|
}
|
|
const unit = simplifyUnit(node.unit)
|
|
|
|
return convertNodeToUnit(unit, node)
|
|
}
|
|
|
|
export function convertNodeToUnit<Node extends EvaluatedNode = EvaluatedNode>(
|
|
to: Unit | undefined,
|
|
node: Node
|
|
): Node {
|
|
const temporalValue =
|
|
node.temporalValue && node.unit
|
|
? mapTemporal(
|
|
(value) => convertUnit(node.unit, to, value as number),
|
|
node.temporalValue
|
|
)
|
|
: node.temporalValue
|
|
return {
|
|
...node,
|
|
nodeValue:
|
|
node.unit && typeof node.nodeValue === 'number'
|
|
? convertUnit(node.unit, to, node.nodeValue)
|
|
: node.nodeValue,
|
|
...(temporalValue && { temporalValue }),
|
|
unit: to,
|
|
}
|
|
}
|