Order variables by coordinate and then element
In other words, order coordinates like (rₛ₁, rₛ₂, sₛ₁, sₛ₂, xₛ₁, xₛ₂, xₚ₃, yₛ₁, yₛ₂, yₚ₃, zₛ₁, zₛ₂, zₚ₃) instead of like (rₛ₁, sₛ₁, xₛ₁, yₛ₁, zₛ₁, rₛ₂, sₛ₂, xₛ₂, yₛ₂, zₛ₂, xₚ₃, yₚ₃, zₚ₃). In the test cases, this really cuts down the size of the Gröbner basis.
This commit is contained in:
parent
463a3b21e1
commit
86dbd9ea45
@ -12,46 +12,68 @@ using Groebner
|
|||||||
abstract type Element{T} end
|
abstract type Element{T} end
|
||||||
|
|
||||||
mutable struct Point{T} <: Element{T}
|
mutable struct Point{T} <: Element{T}
|
||||||
coords::Union{Vector{MPolyRingElem{T}}, Nothing}
|
coords::Vector{MPolyRingElem{T}}
|
||||||
vec::Union{Vector{MPolyRingElem{T}}, Nothing}
|
vec::Union{Vector{MPolyRingElem{T}}, Nothing}
|
||||||
rel::Nothing
|
rel::Nothing
|
||||||
|
|
||||||
## [to do] constructor argument never needed?
|
## [to do] constructor argument never needed?
|
||||||
Point{T}(
|
Point{T}(
|
||||||
coords::Union{Vector{MPolyRingElem{T}}, Nothing} = nothing,
|
coords::Vector{MPolyRingElem{T}} = MPolyRingElem{T}[],
|
||||||
vec::Union{Vector{MPolyRingElem{T}}, Nothing} = nothing
|
vec::Union{Vector{MPolyRingElem{T}}, Nothing} = nothing
|
||||||
) where T = new(coords, vec, nothing)
|
) where T = new(coords, vec, nothing)
|
||||||
end
|
end
|
||||||
|
|
||||||
coordnames(_::Point) = [:xₚ, :yₚ, :zₚ]
|
##coordnames(_::Point) = [:xₚ, :yₚ, :zₚ]
|
||||||
|
|
||||||
function buildvec(pt::Point, coordqueue)
|
function buildvec!(pt::Point)
|
||||||
coordring = parent(coordqueue[1])
|
coordring = parent(pt.coords[1])
|
||||||
pt.coords = splice!(coordqueue, 1:3)
|
|
||||||
pt.vec = [one(coordring), dot(pt.coords, pt.coords), pt.coords...]
|
pt.vec = [one(coordring), dot(pt.coords, pt.coords), pt.coords...]
|
||||||
end
|
end
|
||||||
|
|
||||||
mutable struct Sphere{T} <: Element{T}
|
mutable struct Sphere{T} <: Element{T}
|
||||||
coords::Union{Vector{MPolyRingElem{T}}, Nothing}
|
coords::Vector{MPolyRingElem{T}}
|
||||||
vec::Union{Vector{MPolyRingElem{T}}, Nothing}
|
vec::Union{Vector{MPolyRingElem{T}}, Nothing}
|
||||||
rel::Union{MPolyRingElem{T}, Nothing}
|
rel::Union{MPolyRingElem{T}, Nothing}
|
||||||
|
|
||||||
|
## [to do] constructor argument never needed?
|
||||||
Sphere{T}(
|
Sphere{T}(
|
||||||
coords::Union{Vector{MPolyRingElem{T}}, Nothing} = nothing,
|
coords::Vector{MPolyRingElem{T}} = MPolyRingElem{T}[],
|
||||||
vec::Union{Vector{MPolyRingElem{T}}, Nothing} = nothing,
|
vec::Union{Vector{MPolyRingElem{T}}, Nothing} = nothing,
|
||||||
rel::Union{MPolyRingElem{T}, Nothing} = nothing
|
rel::Union{MPolyRingElem{T}, Nothing} = nothing
|
||||||
) where T = new(coords, vec, rel)
|
) where T = new(coords, vec, rel)
|
||||||
end
|
end
|
||||||
|
|
||||||
coordnames(_::Sphere) = [:rₛ, :sₛ, :xₛ, :yₛ, :zₛ]
|
##coordnames(_::Sphere) = [:rₛ, :sₛ, :xₛ, :yₛ, :zₛ]
|
||||||
|
|
||||||
function buildvec(sph::Sphere, coordqueue)
|
function buildvec!(sph::Sphere)
|
||||||
coordring = parent(coordqueue[1])
|
coordring = parent(sph.coords[1])
|
||||||
sph.coords = splice!(coordqueue, 1:5)
|
|
||||||
sph.vec = sph.coords
|
sph.vec = sph.coords
|
||||||
sph.rel = mprod(sph.coords, sph.coords) + one(coordring)
|
sph.rel = mprod(sph.coords, sph.coords) + one(coordring)
|
||||||
end
|
end
|
||||||
|
|
||||||
|
const coordnames = IdDict{Symbol, Vector{Union{Symbol, Nothing}}}(
|
||||||
|
nameof(Point) => [nothing, nothing, :xₚ, :yₚ, :zₚ],
|
||||||
|
nameof(Sphere) => [:rₛ, :sₛ, :xₛ, :yₛ, :zₛ]
|
||||||
|
)
|
||||||
|
|
||||||
|
coordname(elem::Element, index) = coordnames[nameof(typeof(elem))][index]
|
||||||
|
|
||||||
|
function pushcoordname!(coordnamelist, indexed_elem::Tuple{Any, Element}, coordindex)
|
||||||
|
elemindex, elem = indexed_elem
|
||||||
|
name = coordname(elem, coordindex)
|
||||||
|
if !isnothing(name)
|
||||||
|
subscript = Subscripts.sub(string(elemindex))
|
||||||
|
push!(coordnamelist, Symbol(name, subscript))
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
function takecoord!(coordlist, indexed_elem::Tuple{Any, Element}, coordindex)
|
||||||
|
elem = indexed_elem[2]
|
||||||
|
if !isnothing(coordname(elem, coordindex))
|
||||||
|
push!(elem.coords, popfirst!(coordlist))
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
# --- primitive relations ---
|
# --- primitive relations ---
|
||||||
|
|
||||||
abstract type Relation{T} end
|
abstract type Relation{T} end
|
||||||
@ -99,22 +121,38 @@ function Base.push!(ctx::Construction{T}, rel::Relation{T}) where T
|
|||||||
end
|
end
|
||||||
|
|
||||||
function realize(ctx::Construction{T}) where T
|
function realize(ctx::Construction{T}) where T
|
||||||
# collect variable names
|
# collect coordinate names
|
||||||
coordnamelist = Symbol[]
|
coordnamelist = Symbol[]
|
||||||
elemenum = enumerate(ctx.elements)
|
elemenum = enumerate(ctx.elements)
|
||||||
for (index, elem) in elemenum
|
for coordindex in 1:5
|
||||||
subscript = Subscripts.sub(string(index))
|
for indexed_elem in elemenum
|
||||||
append!(coordnamelist,
|
pushcoordname!(coordnamelist, indexed_elem, coordindex)
|
||||||
[Symbol(name, subscript) for name in coordnames(elem)]
|
end
|
||||||
)
|
|
||||||
end
|
end
|
||||||
|
|
||||||
|
display(collect(elemenum))
|
||||||
|
display(coordnamelist)
|
||||||
|
println()
|
||||||
|
|
||||||
# construct coordinate ring
|
# construct coordinate ring
|
||||||
coordring, coordqueue = polynomial_ring(parent_type(T)(), coordnamelist, ordering = :degrevlex)
|
coordring, coordqueue = polynomial_ring(parent_type(T)(), coordnamelist, ordering = :degrevlex)
|
||||||
|
|
||||||
|
# retrieve coordinates
|
||||||
|
for (_, elem) in elemenum
|
||||||
|
empty!(elem.coords)
|
||||||
|
end
|
||||||
|
for coordindex in 1:5
|
||||||
|
for indexed_elem in elemenum
|
||||||
|
takecoord!(coordqueue, indexed_elem, coordindex)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
# construct coordinate vectors
|
# construct coordinate vectors
|
||||||
for (_, elem) in elemenum
|
for (_, elem) in elemenum
|
||||||
buildvec(elem, coordqueue)
|
buildvec!(elem)
|
||||||
|
display(elem.coords)
|
||||||
|
display(elem.vec)
|
||||||
|
println()
|
||||||
end
|
end
|
||||||
|
|
||||||
# turn relations into equations
|
# turn relations into equations
|
||||||
|
Loading…
Reference in New Issue
Block a user