Skip to content

Commit

Permalink
refactored loop quote
Browse files Browse the repository at this point in the history
  • Loading branch information
@chakravala
chakravala committed Jun 16, 2024
1 parent bd4e2bd commit 1416e12
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 92 deletions.
10 changes: 10 additions & 0 deletions src/algebra.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1721,6 +1721,16 @@ function leftrightsym(com)
return (left...,right...,br)
end

function product_loop(V,type,loop,VEC)
if mdims(V)<cache_limit/2
return :($type{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],VEC)...)
$(loop[2])
return $type{V,t}(out)
end end
end

for com ∈ (:spinor,:s_m,:m_s,:anti,:a_m,:m_a,:multivector,:s_a,:a_s)
outspin = com ∈ (:spinor,:anti,:s_a,:a_s)
left,leftspin,right,rightspin,br = leftrightsym(com)
Expand Down
105 changes: 13 additions & 92 deletions src/products.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1223,62 +1223,31 @@ for (mop,product!,field) ∈ ((:∧,:exteraddmulti!,false),(:⟑,:geomaddmulti!,
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_multivector(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(Multivector{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],VEC)...)
$(loop[2])
return Multivector{V,t}(out)
end end
product_loop(V,:Multivector,loop,VEC)
end
@generated function $op(a::Spinor{V,T},b::Multivector{V,S},$(args...)) where {V,T,S}
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_s_m(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(Multivector{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],VEC)...)
$(loop[2])
return Multivector{V,t}(out)
end end
product_loop(V,:Multivector,loop,VEC)
end
@generated function $op(a::Multivector{V,T},b::Spinor{V,S},$(args...)) where {V,T,S}
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_m_s(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(Multivector{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],VEC)...)
$(loop[2])
return Multivector{V,t}(out)
end end

product_loop(V,:Multivector,loop,VEC)
end
@generated function $op(a::AntiSpinor{V,T},b::Multivector{V,S},$(args...)) where {V,T,S}
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_a_m(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(Multivector{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],VEC)...)
$(loop[2])
return Multivector{V,t}(out)
end end
product_loop(V,:Multivector,loop,VEC)
end
@generated function $op(a::Multivector{V,T},b::AntiSpinor{V,S},$(args...)) where {V,T,S}
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_m_a(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(Multivector{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],VEC)...)
$(loop[2])
return Multivector{V,t}(out)
end end
product_loop(V,:Multivector,loop,VEC)
end
end
end
Expand Down Expand Up @@ -1320,25 +1289,13 @@ for (mop,product!,field) ∈ ((:∧,:exteraddspin!,false),(:⟑,:geomaddspin!,fa
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_spinor(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(Spinor{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],Symbol(string(VEC)*"s"))...)
$(loop[2])
return Spinor{V,t}(out)
end end
product_loop(V,:Spinor,loop,Symbol(string(VEC)*"s"))
end
@generated function $op(a::AntiSpinor{V,T},b::AntiSpinor{V,S},$(args...)) where {V,T,S}
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_anti(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(Spinor{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],Symbol(string(VEC)*"s"))...)
$(loop[2])
return Spinor{V,t}(out)
end end
product_loop(V,:Spinor,loop,Symbol(string(VEC)*"s"))
end
end
end
Expand All @@ -1353,71 +1310,35 @@ for (mop,product!,field) ∈ ((:∧,:exteraddanti!,false),(:⟑,:geomaddanti!,fa
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_s_a(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(AntiSpinor{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],Symbol(string(VEC)*"s"))...)
$(loop[2])
return AntiSpinor{V,t}(out)
end end
product_loop(V,:AntiSpinor,loop,Symbol(string(VEC)*"s"))
end
@generated function $op(a::AntiSpinor{V,T},b::Spinor{V,S},$(args...)) where {V,T,S}
$indu
MUL,VEC = mulvec(a,b)
loop = generate_loop_a_s(V,:(a.v),:(b.v),promote_type(T,S),MUL,$product!,$preproduct!,$field)
if mdims(V)<cache_limit/2
return :(AntiSpinor{V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],Symbol(string(VEC)*"s"))...)
$(loop[2])
return AntiSpinor{V,t}(out)
end end
product_loop(V,:AntiSpinor,loop,Symbol(string(VEC)*"s"))
end
end
end
@generated function ∨(a::Spinor{V,T},b::Spinor{V,S}) where {V,T,S}
MUL,VEC = mulvec(a,b)
loop = generate_loop_spinor(V,:(a.v),:(b.v),promote_type(T,S),MUL,isodd(mdims(V)) ? meetaddanti! : meetaddspin!,isodd(mdims(V)) ? meetaddanti!_pre : meetaddspin!_pre)
if mdims(V)<cache_limit/2
return :($(isodd(mdims(V)) ? :AntiSpinor : :Spinor){V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],Symbol(string(VEC)*"s"))...)
$(loop[2])
return $(isodd(mdims(V)) ? :AntiSpinor : :Spinor){V,t}(out)
end end
product_loop(V,isodd(mdims(V)) ? :AntiSpinor : :Spinor,loop,Symbol(string(VEC)*"s"))
end
@generated function ∨(a::AntiSpinor{V,T},b::AntiSpinor{V,S}) where {V,T,S}
MUL,VEC = mulvec(a,b)
loop = generate_loop_anti(V,:(a.v),:(b.v),promote_type(T,S),MUL,isodd(mdims(V)) ? meetaddanti! : meetaddspin!,isodd(mdims(V)) ? meetaddanti!_pre : meetaddspin!_pre)
if mdims(V)<cache_limit/2
return :($(isodd(mdims(V)) ? :AntiSpinor : :Spinor){V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],Symbol(string(VEC)*"s"))...)
$(loop[2])
return $(isodd(mdims(V)) ? :AntiSpinor : :Spinor){V,t}(out)
end end
product_loop(V,isodd(mdims(V)) ? :AntiSpinor : :Spinor,loop,Symbol(string(VEC)*"s"))
end
@generated function ∨(a::Spinor{V,T},b::AntiSpinor{V,S}) where {V,T,S}
MUL,VEC = mulvec(a,b)
loop = generate_loop_s_a(V,:(a.v),:(b.v),promote_type(T,S),MUL,isodd(mdims(V)) ? meetaddspin! : meetaddanti!,isodd(mdims(V)) ? meetaddspin!_pre : meetaddanti!_pre)
if mdims(V)<cache_limit/2
return :($(isodd(mdims(V)) ? :Spinor : :AntiSpinor){V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],Symbol(string(VEC)*"s"))...)
$(loop[2])
return $(isodd(mdims(V)) ? :Spinor : :AntiSpinor){V,t}(out)
end end
product_loop(V,isodd(mdims(V)) ? :Spinor : :AntiSpinor,loop,Symbol(string(VEC)*"s"))
end
@generated function ∨(a::AntiSpinor{V,T},b::Spinor{V,S}) where {V,T,S}
MUL,VEC = mulvec(a,b)
loop = generate_loop_a_s(V,:(a.v),:(b.v),promote_type(T,S),MUL,isodd(mdims(V)) ? meetaddspin! : meetaddanti!,isodd(mdims(V)) ? meetaddspin!_pre : meetaddanti!_pre)
if mdims(V)<cache_limit/2
return :($(isodd(mdims(V)) ? :Spinor : :AntiSpinor){V}($(loop[2].args[2])))
else return quote
$(insert_expr(loop[1],Symbol(string(VEC)*"s"))...)
$(loop[2])
return $(isodd(mdims(V)) ? :Spinor : :AntiSpinor){V,t}(out)
end end
product_loop(V,isodd(mdims(V)) ? :Spinor : :AntiSpinor,loop,Symbol(string(VEC)*"s"))
end

for side ∈ (:left,:right)
Expand Down

0 comments on commit 1416e12

Please sign in to comment.