From 533c97c41c5bc972010ec4dd434b3364f3c37005 Mon Sep 17 00:00:00 2001
From: Ozzie Gooen <ozzieagooen@gmail.com>
Date: Tue, 24 May 2022 17:23:37 -0400
Subject: [PATCH] Added very simple lambdaDeclaration to React components

---
 .../src/components/SquiggleChart.tsx          | 19 +++++++++++++++++++
 packages/squiggle-lang/src/js/index.ts        |  2 ++
 .../squiggle-lang/src/js/rescript_interop.ts  |  8 ++++++++
 .../src/rescript/TypescriptInterface.res      |  3 +++
 4 files changed, 32 insertions(+)

diff --git a/packages/components/src/components/SquiggleChart.tsx b/packages/components/src/components/SquiggleChart.tsx
index 50fcebb3..1c5407c9 100644
--- a/packages/components/src/components/SquiggleChart.tsx
+++ b/packages/components/src/components/SquiggleChart.tsx
@@ -202,6 +202,25 @@ const SquiggleItem: React.FC<SquiggleItemProps> = ({
         </VariableBox>
       );
     }
+    case "lambdaDeclaration": {
+      switch (expression.value.tag) {
+        case "Continuous": {
+          return (
+            <FunctionChart
+              fn={expression.value.value.fn}
+              chartSettings={chartSettings}
+              environment={{
+                sampleCount: environment.sampleCount / 10,
+                xyPointLength: environment.xyPointLength / 10,
+              }}
+            />
+          );
+        }
+        case "RelativeComparison": {
+          return <></>;
+        }
+      }
+    }
     case "lambda":
       return (
         <FunctionChart
diff --git a/packages/squiggle-lang/src/js/index.ts b/packages/squiggle-lang/src/js/index.ts
index 8b2a46c0..3da19334 100644
--- a/packages/squiggle-lang/src/js/index.ts
+++ b/packages/squiggle-lang/src/js/index.ts
@@ -185,5 +185,7 @@ function createTsExport(
       return tag("date", x.value);
     case "EvTimeDuration":
       return tag("timeDuration", x.value);
+    case "EvDeclaration":
+      return tag("lambdaDeclaration", x.value);
   }
 }
diff --git a/packages/squiggle-lang/src/js/rescript_interop.ts b/packages/squiggle-lang/src/js/rescript_interop.ts
index 7bb98b5d..9d521927 100644
--- a/packages/squiggle-lang/src/js/rescript_interop.ts
+++ b/packages/squiggle-lang/src/js/rescript_interop.ts
@@ -9,6 +9,7 @@ import {
   discreteShape,
   continuousShape,
   lambdaValue,
+  lambdaDeclaration
 } from "../rescript/TypescriptInterface.gen";
 import { Distribution } from "./distribution";
 import { tagged, tag } from "./types";
@@ -63,6 +64,10 @@ export type rescriptExport =
   | {
       TAG: 11; // EvTimeDuration
       _0: number;
+    }
+  | {
+      TAG: 12; // EvDeclaration
+      _0: lambdaDeclaration;
     };
 
 type rescriptDist =
@@ -96,6 +101,7 @@ export type squiggleExpression =
   | tagged<"number", number>
   | tagged<"date", Date>
   | tagged<"timeDuration", number>
+  | tagged<"lambdaDeclaration", lambdaDeclaration>
   | tagged<"record", { [key: string]: squiggleExpression }>;
 
 export { lambdaValue };
@@ -141,6 +147,8 @@ export function convertRawToTypescript(
       return tag("date", result._0);
     case 11: // EvTimeDuration
       return tag("number", result._0);
+    case 12: // EvDeclaration
+      return tag("lambdaDeclaration", result._0);
   }
 }
 
diff --git a/packages/squiggle-lang/src/rescript/TypescriptInterface.res b/packages/squiggle-lang/src/rescript/TypescriptInterface.res
index 93af9832..44051cc1 100644
--- a/packages/squiggle-lang/src/rescript/TypescriptInterface.res
+++ b/packages/squiggle-lang/src/rescript/TypescriptInterface.res
@@ -76,6 +76,9 @@ let distributionErrorToString = DistributionTypes.Error.toString
 @genType
 type lambdaValue = ReducerInterface_ExpressionValue.lambdaValue
 
+@genType
+type lambdaDeclaration = ReducerInterface_ExpressionValue.lambdaDeclaration
+
 @genType
 let defaultSamplingEnv = DistributionOperation.defaultEnv