First attempt at integrating namespaces

packages/squiggle-lang/src/rescript/FunctionRegistry/FunctionRegistry_Core.res

@@ -39,6 +39,8 @@ and frValueDictParam = (string, frValue)
 and frValueDistOrNumber = FRValueNumber(float) | FRValueDist(DistributionTypes.genericDist)
 
 type fnDefinition = {
+  nameSpace: option<string>,
+  requiresNamespace: bool,
   name: string,
   inputs: array<frType>,
   run: (array<frValue>, DistributionOperation.env) => result<internalExpressionValue, string>,
@@ -327,8 +329,10 @@ module FnDefinition = {
     }
   }
 
-  let make = (~name, ~inputs, ~run): t => {
+  let make = (~nameSpace=None, ~requiresNamespace=true, ~name, ~inputs, ~run, ()): t => {
     name: name,
+    nameSpace: nameSpace,
+    requiresNamespace: requiresNamespace,
     inputs: inputs,
     run: run,
   }

packages/squiggle-lang/src/rescript/FunctionRegistry/FunctionRegistry_Helpers.res

@@ -201,8 +201,11 @@ module TwoArgDist = {
     ->E.R2.fmap(Wrappers.evDistribution)
 
   let make = (name, fn) => {
-    FnDefinition.make(~name, ~inputs=[FRTypeDistOrNumber, FRTypeDistOrNumber], ~run=(inputs, env) =>
+    FnDefinition.make(
-      inputs->Prepare.ToValueTuple.twoDistOrNumber->process(~fn, ~env)
+      ~name,
+      ~inputs=[FRTypeDistOrNumber, FRTypeDistOrNumber],
+      ~run=(inputs, env) => inputs->Prepare.ToValueTuple.twoDistOrNumber->process(~fn, ~env),
+      (),
     )
   }
 
@@ -211,6 +214,7 @@ module TwoArgDist = {
       ~name,
       ~inputs=[FRTypeRecord([("p5", FRTypeDistOrNumber), ("p95", FRTypeDistOrNumber)])],
       ~run=(inputs, env) => inputs->Prepare.ToValueTuple.Record.twoDistOrNumber->process(~fn, ~env),
+      (),
     )
   }
 
@@ -219,6 +223,7 @@ module TwoArgDist = {
       ~name,
       ~inputs=[FRTypeRecord([("mean", FRTypeDistOrNumber), ("stdev", FRTypeDistOrNumber)])],
       ~run=(inputs, env) => inputs->Prepare.ToValueTuple.Record.twoDistOrNumber->process(~fn, ~env),
+      (),
     )
   }
 }
@@ -230,35 +235,51 @@ module OneArgDist = {
     ->E.R2.fmap(Wrappers.evDistribution)
 
   let make = (name, fn) =>
-    FnDefinition.make(~name, ~inputs=[FRTypeDistOrNumber], ~run=(inputs, env) =>
+    FnDefinition.make(
-      inputs->Prepare.ToValueTuple.oneDistOrNumber->process(~fn, ~env)
+      ~name,
+      ~inputs=[FRTypeDistOrNumber],
+      ~run=(inputs, env) => inputs->Prepare.ToValueTuple.oneDistOrNumber->process(~fn, ~env),
+      (),
     )
 }
 
 module ArrayNumberDist = {
   let make = (name, fn) => {
-    FnDefinition.make(~name, ~inputs=[FRTypeArray(FRTypeNumber)], ~run=(inputs, _) =>
+    FnDefinition.make(
+      ~name,
+      ~inputs=[FRTypeArray(FRTypeNumber)],
+      ~run=(inputs, _) =>
         Prepare.ToTypedArray.numbers(inputs)
         ->E.R.bind(r => E.A.length(r) === 0 ? Error("List is empty") : Ok(r))
-      ->E.R.bind(fn)
+        ->E.R.bind(fn),
+      (),
     )
   }
   let make2 = (name, fn) => {
-    FnDefinition.make(~name, ~inputs=[FRTypeArray(FRTypeAny)], ~run=(inputs, _) =>
+    FnDefinition.make(
+      ~name,
+      ~inputs=[FRTypeArray(FRTypeAny)],
+      ~run=(inputs, _) =>
         Prepare.ToTypedArray.numbers(inputs)
         ->E.R.bind(r => E.A.length(r) === 0 ? Error("List is empty") : Ok(r))
-      ->E.R.bind(fn)
+        ->E.R.bind(fn),
+      (),
     )
   }
 }
 
 module NumberToNumber = {
   let make = (name, fn) =>
-    FnDefinition.make(~name, ~inputs=[FRTypeNumber], ~run=(inputs, _) => {
+    FnDefinition.make(
+      ~name,
+      ~inputs=[FRTypeNumber],
+      ~run=(inputs, _) => {
         inputs
         ->getOrError(0)
         ->E.R.bind(Prepare.oneNumber)
         ->E.R2.fmap(fn)
         ->E.R2.fmap(Wrappers.evNumber)
-    })
+      },
+      (),
+    )
 }

packages/squiggle-lang/src/rescript/FunctionRegistry/FunctionRegistry_Library.res

@@ -54,9 +54,12 @@ let registry = [
     ~name="toContinuousPointSet",
     ~definitions=[
       FnDefinition.make(
-        ~name="toContinuousPointSet",
+        ~nameSpace=Some("PointSet"),
+        ~requiresNamespace=true,
+        ~name="makeContinuous",
         ~inputs=[FRTypeArray(FRTypeRecord([("x", FRTypeNumeric), ("y", FRTypeNumeric)]))],
         ~run=(inputs, _) => inputsTodist(inputs, r => Continuous(Continuous.make(r))),
+        (),
       ),
     ],
     (),
@@ -65,9 +68,12 @@ let registry = [
     ~name="toDiscretePointSet",
     ~definitions=[
       FnDefinition.make(
-        ~name="toDiscretePointSet",
+        ~nameSpace=Some("PointSet"),
+        ~requiresNamespace=true,
+        ~name="makeDiscrete",
         ~inputs=[FRTypeArray(FRTypeRecord([("x", FRTypeNumeric), ("y", FRTypeNumeric)]))],
-        ~run=(inputs, _) => inputsTodist(inputs, r => Discrete(Discrete.make(r))),
+        ~run=(inputs, _) => inputsTodist(inputs, r => Continuous(Continuous.make(r))),
+        (),
       ),
     ],
     (),
@@ -75,9 +81,14 @@ let registry = [
   Function.make(
     ~name="Declaration",
     ~definitions=[
-      FnDefinition.make(~name="declareFn", ~inputs=[Declaration.frType], ~run=(inputs, _) => {
+      FnDefinition.make(
+        ~name="declareFn",
+        ~inputs=[Declaration.frType],
+        ~run=(inputs, _) => {
           inputs->getOrError(0)->E.R.bind(Declaration.fromExpressionValue)
-      }),
+        },
+        (),
+      ),
     ],
     (),
   ),
@@ -189,6 +200,7 @@ to(5,10)
         ~name="toContinuousPointSet",
         ~inputs=[FRTypeArray(FRTypeRecord([("x", FRTypeNumeric), ("y", FRTypeNumeric)]))],
         ~run=(inputs, _) => inputsTodist(inputs, r => Continuous(Continuous.make(r))),
+        (),
       ),
     ],
     (),
@@ -207,6 +219,7 @@ to(5,10)
         ~name="toDiscretePointSet",
         ~inputs=[FRTypeArray(FRTypeRecord([("x", FRTypeNumeric), ("y", FRTypeNumeric)]))],
         ~run=(inputs, _) => inputsTodist(inputs, r => Discrete(Discrete.make(r))),
+        (),
       ),
     ],
     (),
@@ -222,9 +235,14 @@ to(5,10)
   ]
 })`,
     ~definitions=[
-      FnDefinition.make(~name="declareFn", ~inputs=[Declaration.frType], ~run=(inputs, _) => {
+      FnDefinition.make(
+        ~name="declareFn",
+        ~inputs=[Declaration.frType],
+        ~run=(inputs, _) => {
           inputs->E.A.unsafe_get(0)->Declaration.fromExpressionValue
-      }),
+        },
+        (),
+      ),
     ],
     ~isExperimental=true,
     (),
@@ -366,6 +384,8 @@ to(5,10)
     ~name="Dict.merge",
     ~definitions=[
       FnDefinition.make(
+        ~nameSpace=Some("Dict"),
+        ~requiresNamespace=true,
         ~name="merge",
         ~inputs=[FRTypeDict(FRTypeAny), FRTypeDict(FRTypeAny)],
         ~run=(inputs, _) => {
@@ -380,6 +400,7 @@ to(5,10)
           | _ => Error(impossibleError)
           }
         },
+        (),
       ),
     ],
     (),
@@ -388,16 +409,19 @@ to(5,10)
   Function.make(
     ~name="Dict.mergeMany",
     ~definitions=[
-      FnDefinition.make(~name="mergeMany", ~inputs=[FRTypeArray(FRTypeDict(FRTypeAny))], ~run=(
+      FnDefinition.make(
-        inputs,
+        ~nameSpace=Some("Dict"),
-        _,
+        ~requiresNamespace=true,
-      ) =>
+        ~name="mergeMany",
+        ~inputs=[FRTypeArray(FRTypeDict(FRTypeAny))],
+        ~run=(inputs, _) =>
           inputs
           ->Prepare.ToTypedArray.dicts
           ->E.R2.fmap(E.Dict.concatMany)
           ->E.R2.fmap(Js.Dict.map((. r) => FunctionRegistry_Core.FRType.matchReverse(r)))
           ->E.R2.fmap(r => r->Js.Dict.entries->Belt.Map.String.fromArray)
-        ->E.R2.fmap(Wrappers.evRecord)
+          ->E.R2.fmap(Wrappers.evRecord),
+        (),
       ),
     ],
     (),
@@ -405,11 +429,18 @@ to(5,10)
   Function.make(
     ~name="Dict.keys",
     ~definitions=[
-      FnDefinition.make(~name="keys", ~inputs=[FRTypeDict(FRTypeAny)], ~run=(inputs, _) =>
+      FnDefinition.make(
+        ~nameSpace=Some("Dict"),
+        ~requiresNamespace=true,
+        ~name="keys",
+        ~inputs=[FRTypeDict(FRTypeAny)],
+        ~run=(inputs, _) =>
           switch inputs {
-        | [FRValueDict(d1)] => Js.Dict.keys(d1)->E.A2.fmap(Wrappers.evString)->Wrappers.evArray->Ok
+          | [FRValueDict(d1)] =>
+            Js.Dict.keys(d1)->E.A2.fmap(Wrappers.evString)->Wrappers.evArray->Ok
           | _ => Error(impossibleError)
-        }
+          },
+        (),
       ),
     ],
     (),
@@ -417,7 +448,12 @@ to(5,10)
   Function.make(
     ~name="Dict.values",
     ~definitions=[
-      FnDefinition.make(~name="values", ~inputs=[FRTypeDict(FRTypeAny)], ~run=(inputs, _) =>
+      FnDefinition.make(
+        ~nameSpace=Some("Dict"),
+        ~requiresNamespace=true,
+        ~name="values",
+        ~inputs=[FRTypeDict(FRTypeAny)],
+        ~run=(inputs, _) =>
           switch inputs {
           | [FRValueDict(d1)] =>
             Js.Dict.values(d1)
@@ -425,7 +461,8 @@ to(5,10)
             ->Wrappers.evArray
             ->Ok
           | _ => Error(impossibleError)
-        }
+          },
+        (),
       ),
     ],
     (),
@@ -433,7 +470,12 @@ to(5,10)
   Function.make(
     ~name="Dict.toList",
     ~definitions=[
-      FnDefinition.make(~name="dictToList", ~inputs=[FRTypeDict(FRTypeAny)], ~run=(inputs, _) =>
+      FnDefinition.make(
+        ~nameSpace=Some("Dict"),
+        ~requiresNamespace=true,
+        ~name="toList",
+        ~inputs=[FRTypeDict(FRTypeAny)],
+        ~run=(inputs, _) =>
           switch inputs {
           | [FRValueDict(dict)] =>
             dict
@@ -447,7 +489,8 @@ to(5,10)
             ->Wrappers.evArray
             ->Ok
           | _ => Error(impossibleError)
-        }
+          },
+        (),
       ),
     ],
     (),
@@ -455,10 +498,12 @@ to(5,10)
   Function.make(
     ~name="Dict.fromList",
     ~definitions=[
-      FnDefinition.make(~name="dictFromList", ~inputs=[FRTypeArray(FRTypeArray(FRTypeAny))], ~run=(
+      FnDefinition.make(
-        inputs,
+        ~nameSpace=Some("Dict"),
-        _,
+        ~requiresNamespace=true,
-      ) => {
+        ~name="fromList",
+        ~inputs=[FRTypeArray(FRTypeArray(FRTypeAny))],
+        ~run=(inputs, _) => {
           let convertInternalItems = items =>
             items
             ->E.A2.fmap(item => {
@@ -473,7 +518,9 @@ to(5,10)
             ->E.R2.fmap(Wrappers.evRecord)
           inputs->getOrError(0)->E.R.bind(Prepare.ToValueArray.Array.arrayOfArrays)
             |> E.R2.bind(convertInternalItems)
-      }),
+        },
+        (),
+      ),
     ],
     (),
   ),
@@ -481,7 +528,12 @@ to(5,10)
     ~name="List.make",
     ~definitions=[
       //Todo: If the second item is a function with no args, it could be nice to run this function and return the result.
-      FnDefinition.make(~name="listMake", ~inputs=[FRTypeNumber, FRTypeAny], ~run=(inputs, _) => {
+      FnDefinition.make(
+        ~nameSpace=Some("List"),
+        ~requiresNamespace=true,
+        ~name="make",
+        ~inputs=[FRTypeNumber, FRTypeAny],
+        ~run=(inputs, _) => {
           switch inputs {
           | [FRValueNumber(number), value] =>
             Belt.Array.make(E.Float.toInt(number), value)
@@ -490,21 +542,29 @@ to(5,10)
             ->Ok
           | _ => Error(impossibleError)
           }
-      }),
+        },
+        (),
+      ),
     ],
     (),
   ),
   Function.make(
     ~name="upTo",
     ~definitions=[
-      FnDefinition.make(~name="upTo", ~inputs=[FRTypeNumber, FRTypeNumber], ~run=(inputs, _) =>
+      FnDefinition.make(
+        ~nameSpace=Some("List"),
+        ~requiresNamespace=true,
+        ~name="upTo",
+        ~inputs=[FRTypeNumber, FRTypeNumber],
+        ~run=(inputs, _) =>
           inputs
           ->Prepare.ToValueTuple.twoNumbers
           ->E.R2.fmap(((low, high)) =>
             E.A.Floats.range(low, high, (high -. low +. 1.0)->E.Float.toInt)
             ->E.A2.fmap(Wrappers.evNumber)
             ->Wrappers.evArray
-        )
+          ),
+        (),
       ),
     ],
     (),