1
0
mirror of https://github.com/kalmarek/Groups.jl.git synced 2024-12-25 18:15:29 +01:00

rework DirectProdIter to the new iteration protocol

This commit is contained in:
kalmarek 2018-09-21 18:14:48 +02:00
parent e28c296bce
commit 0750a47853
2 changed files with 13 additions and 23 deletions

View File

@ -118,7 +118,7 @@ parent(g::DirectProductGroupElem) =
#
###############################################################################
Base.size(g::DirectProductGroupElem) = size(g.elts)
size(g::DirectProductGroupElem) = size(g.elts)
Base.IndexStyle(::Type{DirectProductGroupElem}) = Base.LinearFast()
Base.getindex(g::DirectProductGroupElem, i::Int) = g.elts[i]
@ -276,10 +276,7 @@ end
#
###############################################################################
import Base: size, length, start, next, done, eltype
struct DirectPowerIter{Gr<:AbstractAlgebra.Group, GrEl<:AbstractAlgebra.GroupElem}
G::Gr
struct DirectPowerIter{GrEl<:AbstractAlgebra.GroupElem}
N::Int
elts::Vector{GrEl}
totalorder::Int
@ -287,27 +284,20 @@ struct DirectPowerIter{Gr<:AbstractAlgebra.Group, GrEl<:AbstractAlgebra.GroupEle
end
function DirectPowerIter(G::Gr, N::Integer) where {Gr<:AbstractAlgebra.Group}
return DirectPowerIter{Gr, elem_type(G)}(
G,
N,
vec(collect(elements(G))),
Int(order(G))^N,
Int(order(G))
)
return DirectPowerIter{elem_type(G)}(N, collect(G), order(G)^N, order(G))
end
Base.size(DPIter::DirectPowerIter) = ntuple(i -> DPIter.orderG, DPIter.N)
Base.length(DPIter::DirectPowerIter) = DPIter.totalorder
Base.start(::DirectPowerIter) = 0
length(DPIter::DirectPowerIter) = DPIter.totalorder
function Base.next(DPIter::DirectPowerIter, state)
idx = ind2sub(size(DPIter), state+1)
function iterate(DPIter::DirectPowerIter, state=0)
if state >= DPIter.totalorder
return nothing
end
idx = Tuple(CartesianIndices(ntuple(i -> DPIter.orderG, DPIter.N))[state+1])
return DirectProductGroupElem([DPIter.elts[i] for i in idx]), state+1
end
Base.done(DPIter::DirectPowerIter, state) = DPIter.totalorder <= state
Base.eltype(::Type{DirectPowerIter{Gr, GrEl}}) where {Gr, GrEl} = DirectProductGroupElem{GrEl}
eltype(::Type{DirectPowerIter{GrEl}}) where {GrEl} = DirectProductGroupElem{GrEl}
@doc doc"""
elements(G::DirectProductGroup)

View File

@ -4,9 +4,9 @@ module Groups
using AbstractAlgebra
import AbstractAlgebra: Group, GroupElem, Ring
import AbstractAlgebra: parent, parent_type, elem_type
import AbstractAlgebra: elements, order, gens, matrix_repr
import AbstractAlgebra: order, gens, matrix_repr
import Base: length, ==, hash, show, convert
import Base: length, ==, hash, show, convert, eltype, iterate
import Base: inv, reduce, *, ^, power_by_squaring
import Base: findfirst, findnext
import Base: deepcopy_internal