From 7774da51c532006da82617ce307d136897693280 Mon Sep 17 00:00:00 2001
From: Tom Smeding <tom@tomsmeding.com>
Date: Sat, 23 Nov 2024 12:12:03 +0100
Subject: Prepare for introducing top-level args in accum mod

---
 src/CHAD/Heuristics.hs | 14 ++++++++++++++
 src/CHAD/Types.hs      | 14 ++++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 src/CHAD/Heuristics.hs

(limited to 'src/CHAD')

diff --git a/src/CHAD/Heuristics.hs b/src/CHAD/Heuristics.hs
new file mode 100644
index 0000000..6ab8222
--- /dev/null
+++ b/src/CHAD/Heuristics.hs
@@ -0,0 +1,14 @@
+{-# LANGUAGE GADTs #-}
+module CHAD.Heuristics where
+
+import AST
+
+
+hasArrays :: STy t' -> Bool
+hasArrays STNil = False
+hasArrays (STPair a b) = hasArrays a || hasArrays b
+hasArrays (STEither a b) = hasArrays a || hasArrays b
+hasArrays (STMaybe t) = hasArrays t
+hasArrays STArr{} = True
+hasArrays STScal{} = False
+hasArrays STAccum{} = error "Accumulators not allowed in source program"
diff --git a/src/CHAD/Types.hs b/src/CHAD/Types.hs
index 0b73a3a..130493d 100644
--- a/src/CHAD/Types.hs
+++ b/src/CHAD/Types.hs
@@ -68,3 +68,17 @@ d2 STAccum{} = error "Accumulators not allowed in input program"
 d2e :: SList STy env -> SList STy (D2E env)
 d2e SNil = SNil
 d2e (SCons t ts) = SCons (d2 t) (d2e ts)
+
+
+data CHADConfig = CHADConfig
+  { -- | D[let] will bind variables containing arrays in accumulator mode.
+    chcLetArrayAccum :: Bool
+  , -- | D[case] will bind variables containing arrays in accumulator mode.
+    chcCaseArrayAccum :: Bool
+  }
+
+defaultConfig :: CHADConfig
+defaultConfig = CHADConfig
+  { chcLetArrayAccum = False
+  , chcCaseArrayAccum = False
+  }
-- 
cgit v1.2.3-70-g09d2