Minor program refactors

This commit is contained in:
Ozzie Gooen 2022-02-18 17:14:35 -05:00
parent 406a31ce53
commit dd064be7f5
3 changed files with 11 additions and 12 deletions

View File

@ -91,7 +91,7 @@ module Internals = {
} }
let inputsToLeaf = (inputs: Inputs.inputs) => let inputsToLeaf = (inputs: Inputs.inputs) =>
Parser.fromString(inputs.squiggleString) |> E.R.bind(_, g => runProgram(inputs, g)) Parser.fromString(inputs.squiggleString) -> E.R.bind(g => runProgram(inputs, g))
let outputToDistPlus = (inputs: Inputs.inputs, pointSetDist: PointSetTypes.pointSetDist) => let outputToDistPlus = (inputs: Inputs.inputs, pointSetDist: PointSetTypes.pointSetDist) =>
DistPlus.make(~pointSetDist, ~squiggleString=Some(inputs.squiggleString), ()) DistPlus.make(~pointSetDist, ~squiggleString=Some(inputs.squiggleString), ())

View File

@ -1,6 +1,5 @@
open ASTTypes open ASTTypes
type t = node
type tResult = node => result<node, string> type tResult = node => result<node, string>
/* Given two random variables A and B, this returns the distribution /* Given two random variables A and B, this returns the distribution
@ -8,7 +7,7 @@ type tResult = node => result<node, string>
For instance, normal(0, 1) + normal(1, 1) -> normal(1, 2). For instance, normal(0, 1) + normal(1, 1) -> normal(1, 2).
In general, this is implemented via convolution. */ In general, this is implemented via convolution. */
module AlgebraicCombination = { module AlgebraicCombination = {
let tryAnalyticalSimplification = (operation, t1: t, t2: t) => let tryAnalyticalSimplification = (operation, t1: node, t2: node) =>
switch (operation, t1, t2) { switch (operation, t1, t2) {
| (operation, #SymbolicDist(d1), #SymbolicDist(d2)) => | (operation, #SymbolicDist(d1), #SymbolicDist(d2)) =>
switch SymbolicDist.T.tryAnalyticalSimplification(d1, d2, operation) { switch SymbolicDist.T.tryAnalyticalSimplification(d1, d2, operation) {
@ -61,8 +60,8 @@ module AlgebraicCombination = {
let operationToLeaf = ( let operationToLeaf = (
evaluationParams: evaluationParams, evaluationParams: evaluationParams,
algebraicOp: Operation.algebraicOperation, algebraicOp: Operation.algebraicOperation,
t1: t, t1: node,
t2: t, t2: node,
): result<node, string> => ): result<node, string> =>
algebraicOp algebraicOp
|> tryAnalyticalSimplification(_, t1, t2) |> tryAnalyticalSimplification(_, t1, t2)
@ -77,7 +76,7 @@ module AlgebraicCombination = {
module PointwiseCombination = { module PointwiseCombination = {
//TODO: This is crude and slow. It forces everything to be pointSetDist, even though much //TODO: This is crude and slow. It forces everything to be pointSetDist, even though much
//of the process could happen on symbolic distributions without a conversion to be a pointSetDist. //of the process could happen on symbolic distributions without a conversion to be a pointSetDist.
let pointwiseAdd = (evaluationParams: evaluationParams, t1: t, t2: t) => let pointwiseAdd = (evaluationParams: evaluationParams, t1: node, t2: node) =>
switch (Node.render(evaluationParams, t1), Node.render(evaluationParams, t2)) { switch (Node.render(evaluationParams, t1), Node.render(evaluationParams, t2)) {
| (Ok(#RenderedDist(rs1)), Ok(#RenderedDist(rs2))) => | (Ok(#RenderedDist(rs1)), Ok(#RenderedDist(rs2))) =>
Ok( Ok(
@ -98,7 +97,7 @@ module PointwiseCombination = {
| _ => Error("Pointwise combination: rendering failed.") | _ => Error("Pointwise combination: rendering failed.")
} }
let pointwiseCombine = (fn, evaluationParams: evaluationParams, t1: t, t2: t) => let pointwiseCombine = (fn, evaluationParams: evaluationParams, t1: node, t2: node) =>
switch // TODO: construct a function that we can easily sample from, to construct switch // TODO: construct a function that we can easily sample from, to construct
// a RenderedDist. Use the xMin and xMax of the rendered pointSetDists to tell the sampling function where to look. // a RenderedDist. Use the xMin and xMax of the rendered pointSetDists to tell the sampling function where to look.
// TODO: This should work for symbolic distributions too! // TODO: This should work for symbolic distributions too!
@ -113,8 +112,8 @@ module PointwiseCombination = {
let operationToLeaf = ( let operationToLeaf = (
evaluationParams: evaluationParams, evaluationParams: evaluationParams,
pointwiseOp: Operation.pointwiseOperation, pointwiseOp: Operation.pointwiseOperation,
t1: t, t1: node,
t2: t, t2: node,
) => ) =>
switch pointwiseOp { switch pointwiseOp {
| #Add => pointwiseAdd(evaluationParams, t1, t2) | #Add => pointwiseAdd(evaluationParams, t1, t2)
@ -204,7 +203,7 @@ module FunctionCall = {
} }
module Render = { module Render = {
let rec operationToLeaf = (evaluationParams: evaluationParams, t: node): result<t, string> => let rec operationToLeaf = (evaluationParams: evaluationParams, t: node): result<node, string> =>
switch t { switch t {
| #Function(_) => Error("Cannot render a function") | #Function(_) => Error("Cannot render a function")
| #SymbolicDist(d) => | #SymbolicDist(d) =>
@ -224,7 +223,7 @@ module Render = {
but most often it will produce a RenderedDist. but most often it will produce a RenderedDist.
This function is used mainly to turn a parse tree into a single RenderedDist This function is used mainly to turn a parse tree into a single RenderedDist
that can then be displayed to the user. */ that can then be displayed to the user. */
let rec toLeaf = (evaluationParams: ASTTypes.evaluationParams, node: t): result<t, string> => let rec toLeaf = (evaluationParams: ASTTypes.evaluationParams, node: node): result<node, string> =>
switch node { switch node {
// Leaf nodes just stay leaf nodes // Leaf nodes just stay leaf nodes
| #SymbolicDist(_) | #SymbolicDist(_)