Merge pull request #24 from eliascarv/gencolumns

Add `gencolumns` function
eliascarv · Feb 4, 2023 · d499a70 · d499a70
2 parents f909fe3 + bf03ac1
commit d499a70
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/macro.jl b/src/macro.jl
@@ -111,9 +111,7 @@ function _truthtable(expr::Expr, full::Bool)
   preprocess!(expr)
 
   colnames = propnames(expr)
-  n = length(colnames)
-  rows = Iterators.product(fill([true, false], n)...)
-  columns = Vector{Bool}[vec([row[i] for row in rows]) for i in n:-1:1]
+  columns  = gencolumns(length(colnames))
   colexprs = Expr[]
 
   if full

diff --git a/src/macroutils.jl b/src/macroutils.jl
@@ -1,3 +1,10 @@
+function gencolumns(n::Integer)
+  bools = [true, false]
+  outers = [2^x for x in 0:n-1]
+  inners = reverse(outers)
+  [repeat(bools; inner, outer) for (inner, outer) in zip(inners, outers)]
+end
+
 const OPRS = (
   :&, :∧,
   :|, :∨,