$bits conversion handles types with more than 1 dimension

ba7ffdf1 · Zachary Snow · 86195d9e · ba7ffdf1
Commit ba7ffdf1 authored Apr 02, 2019 by Zachary Snow
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 2 deletions

src/Convert/Bits.hs
+9 -2

No files found.
--- a/src/Convert/Bits.hs
+++ b/src/Convert/Bits.hs
@@ -18,6 +18,13 @@ convert =
    convertExpr

 convertExpr :: Expr -> Expr
-convertExpr (Bits (Left (IntegerVector _ _ [r]))) = rangeSize r
-convertExpr (Bits (Left (Implicit        _ [r]))) = rangeSize r
+convertExpr (Bits (Left (IntegerVector _ _ rs))) = size rs
+convertExpr (Bits (Left (Implicit        _ rs))) = size rs
 convertExpr other = other
+
+size :: [Range] -> Expr
+size ranges =
+    simplify $
+    foldl (BinOp Mul) (Number "1") $
+    map rangeSize $
+    ranges