138 lines
4.3 KiB
Plaintext
138 lines
4.3 KiB
Plaintext
open FunctionRegistry_Core
|
|
open FunctionRegistry_Helpers
|
|
|
|
let twoArgs = E.Tuple2.toFnCall
|
|
|
|
module Declaration = {
|
|
let frType = FRTypeRecord([
|
|
("fn", FRTypeLambda),
|
|
("inputs", FRTypeArray(FRTypeRecord([("min", FRTypeNumber), ("max", FRTypeNumber)]))),
|
|
])
|
|
|
|
let fromExpressionValue = (e: frValue): result<expressionValue, string> => {
|
|
switch FunctionRegistry_Helpers.Prepare.ToValueArray.Record.twoArgs([e]) {
|
|
| Ok([FRValueLambda(lambda), FRValueArray(inputs)]) => {
|
|
open FunctionRegistry_Helpers.Prepare
|
|
let getMinMax = arg =>
|
|
ToValueArray.Record.toArgs([arg])
|
|
->E.R.bind(ToValueTuple.twoNumbers)
|
|
->E.R2.fmap(((min, max)) => Declaration.ContinuousFloatArg.make(min, max))
|
|
inputs
|
|
->E.A2.fmap(getMinMax)
|
|
->E.A.R.firstErrorOrOpen
|
|
->E.R2.fmap(args => ReducerInterface_ExpressionValue.EvDeclaration(
|
|
Declaration.make(lambda, args),
|
|
))
|
|
}
|
|
| Error(r) => Error(r)
|
|
| Ok(_) => Error(FunctionRegistry_Helpers.impossibleError)
|
|
}
|
|
}
|
|
}
|
|
|
|
let inputsTodist = (inputs: array<FunctionRegistry_Core.frValue>, makeDist) => {
|
|
let array = inputs->E.A.unsafe_get(0)->Prepare.ToValueArray.Array.openA
|
|
let xyCoords =
|
|
array->E.R.bind(xyCoords =>
|
|
xyCoords
|
|
->E.A2.fmap(xyCoord =>
|
|
[xyCoord]->Prepare.ToValueArray.Record.twoArgs->E.R.bind(Prepare.ToValueTuple.twoNumbers)
|
|
)
|
|
->E.A.R.firstErrorOrOpen
|
|
)
|
|
let expressionValue =
|
|
xyCoords
|
|
->E.R.bind(r => r->XYShape.T.makeFromZipped->E.R2.errMap(XYShape.Error.toString))
|
|
->E.R2.fmap(r => ReducerInterface_ExpressionValue.EvDistribution(PointSet(makeDist(r))))
|
|
expressionValue
|
|
}
|
|
|
|
let registry = [
|
|
Function.make(
|
|
~name="toContinuousPointSet",
|
|
~definitions=[
|
|
FnDefinition.make(
|
|
~name="toContinuousPointSet",
|
|
~inputs=[FRTypeArray(FRTypeRecord([("x", FRTypeNumeric), ("y", FRTypeNumeric)]))],
|
|
~run=(inputs, _) => inputsTodist(inputs, r => Continuous(Continuous.make(r))),
|
|
),
|
|
],
|
|
),
|
|
Function.make(
|
|
~name="toDiscretePointSet",
|
|
~definitions=[
|
|
FnDefinition.make(
|
|
~name="toDiscretePointSet",
|
|
~inputs=[FRTypeArray(FRTypeRecord([("x", FRTypeNumeric), ("y", FRTypeNumeric)]))],
|
|
~run=(inputs, _) => inputsTodist(inputs, r => Discrete(Discrete.make(r))),
|
|
),
|
|
],
|
|
),
|
|
Function.make(
|
|
~name="Declaration",
|
|
~definitions=[
|
|
FnDefinition.make(~name="declareFn", ~inputs=[Declaration.frType], ~run=(inputs, _) => {
|
|
inputs->E.A.unsafe_get(0)->Declaration.fromExpressionValue
|
|
}),
|
|
],
|
|
),
|
|
Function.make(
|
|
~name="Normal",
|
|
~definitions=[
|
|
TwoArgDist.make("normal", twoArgs(SymbolicDist.Normal.make)),
|
|
TwoArgDist.makeRecordP5P95("normal", r =>
|
|
twoArgs(SymbolicDist.Normal.from90PercentCI, r)->Ok
|
|
),
|
|
TwoArgDist.makeRecordMeanStdev("normal", twoArgs(SymbolicDist.Normal.make)),
|
|
],
|
|
),
|
|
Function.make(
|
|
~name="Lognormal",
|
|
~definitions=[
|
|
TwoArgDist.make("lognormal", twoArgs(SymbolicDist.Lognormal.make)),
|
|
TwoArgDist.makeRecordP5P95("lognormal", r =>
|
|
twoArgs(SymbolicDist.Lognormal.from90PercentCI, r)->Ok
|
|
),
|
|
TwoArgDist.makeRecordMeanStdev("lognormal", twoArgs(SymbolicDist.Lognormal.fromMeanAndStdev)),
|
|
],
|
|
),
|
|
Function.make(
|
|
~name="Uniform",
|
|
~definitions=[TwoArgDist.make("uniform", twoArgs(SymbolicDist.Uniform.make))],
|
|
),
|
|
Function.make(
|
|
~name="Beta",
|
|
~definitions=[TwoArgDist.make("beta", twoArgs(SymbolicDist.Beta.make))],
|
|
),
|
|
Function.make(
|
|
~name="Cauchy",
|
|
~definitions=[TwoArgDist.make("cauchy", twoArgs(SymbolicDist.Cauchy.make))],
|
|
),
|
|
Function.make(
|
|
~name="Gamma",
|
|
~definitions=[TwoArgDist.make("gamma", twoArgs(SymbolicDist.Gamma.make))],
|
|
),
|
|
Function.make(
|
|
~name="Logistic",
|
|
~definitions=[TwoArgDist.make("logistic", twoArgs(SymbolicDist.Logistic.make))],
|
|
),
|
|
Function.make(
|
|
~name="To",
|
|
~definitions=[
|
|
TwoArgDist.make("to", twoArgs(SymbolicDist.From90thPercentile.make)),
|
|
TwoArgDist.make(
|
|
"credibleIntervalToDistribution",
|
|
twoArgs(SymbolicDist.From90thPercentile.make),
|
|
),
|
|
],
|
|
),
|
|
Function.make(
|
|
~name="Exponential",
|
|
~definitions=[OneArgDist.make("exponential", SymbolicDist.Exponential.make)],
|
|
),
|
|
Function.make(
|
|
~name="Bernoulli",
|
|
~definitions=[OneArgDist.make("bernoulli", SymbolicDist.Bernoulli.make)],
|
|
),
|
|
]
|