import Base: rationalize

using IntervalArithmetic

IntervalArithmetic.setrounding(Interval, :tight)
IntervalArithmetic.setformat(sigfigs=12)

import IntervalArithmetic.±

function (±){T<:Number}(X::AbstractArray{T}, tol::Real)
    r{T}(x::T) = (x == zero(T)? @interval(0) : x ± tol)
    return r.(X)
end

(±)(X::GroupRingElem, tol::Real) = GroupRingElem(X.coeffs ± tol, parent(X))

EOI{T<:Number}(Δ::GroupRingElem{T}, λ::T) = Δ*Δ - λ*Δ

function groupring_square(vect::AbstractVector, l, pm)
    zzz = zeros(eltype(vect), l)
    zzz[1:length(vect)] .= vect
    return GroupRings.mul!(similar(zzz), zzz, zzz, pm)
end

function compute_SOS(Q::AbstractArray, pm::Array{Int,2}, l::Int)
   n = size(Q,2)

   # result = zeros(eltype(Q), l)
   # for i in 1:n
   #    result .+= groupring_square(view(Q,:,i), l, pm)
   # end

   @everywhere groupring_square = PropertyT.groupring_square

   result = @parallel (+) for i in 1:n
      print(" $i")
      groupring_square(view(Q,:,i), l, pm)
   end

   return result
end

function compute_SOS(Q::AbstractArray, RG::GroupRing, l::Int)
   result = compute_SOS(Q, RG.pm, l)
   return GroupRingElem(result, RG)
end

function distance_to_cone{T<:Interval}(elt::GroupRingElem, Q::AbstractArray{T,2}, wlen::Int)
   SOS = compute_SOS(Q, parent(elt), length(elt.coeffs))
   SOS_diff = elt - SOS

   ɛ_dist = GroupRings.augmentation(SOS_diff)
   info(logger, "ɛ(∑ξᵢ*ξᵢ) ∈ $(ɛ_dist)")

   eoi_SOS_L1_dist = norm(SOS_diff,1)
   info(logger, "‖Δ² - λΔ - ∑ξᵢ*ξᵢ‖₁ ∈ $(eoi_SOS_L1_dist)")

   dist = 2^(wlen-1)*eoi_SOS_L1_dist
   return dist
end

function distance_to_cone{T}(elt::GroupRingElem, Q::AbstractArray{T,2}, wlen::Int)
   SOS = compute_SOS(Q, parent(elt), length(elt.coeffs))
   SOS_diff = elt - SOS

   ɛ_dist = GroupRings.augmentation(SOS_diff)
   info(logger, "ɛ(Δ² - λΔ - ∑ξᵢ*ξᵢ) ≈ $(@sprintf("%.10f", ɛ_dist))")

   eoi_SOS_L1_dist = norm(SOS_diff,1)
   info(logger, "‖Δ² - λΔ - ∑ξᵢ*ξᵢ‖₁ ≈ $(@sprintf("%.10f", eoi_SOS_L1_dist))")

   dist = 2^(wlen-1)*eoi_SOS_L1_dist
   return dist
end

function augIdproj{T, I<:AbstractInterval}(S::Type{I}, Q::AbstractArray{T,2})
   l = size(Q, 2)
   R = zeros(S, (l,l))
   Threads.@threads for j in 1:l
      col = sum(view(Q, :,j))/l
      for i in 1:l
         R[i,j] = Q[i,j] - col ± eps(0.0)
      end
   end
   return R
end

function augIdproj{T}(Q::AbstractArray{T,2}, logger)
   info(logger, "Projecting columns of Q to the augmentation ideal...")
   @logtime logger Q = augIdproj(Interval{T}, Q)

   info(logger, "Checking that sum of every column contains 0.0... ")
   check = all([0.0 in sum(view(Q, :, i)) for i in 1:size(Q, 2)])
   info(logger, (check? "They do." : "FAILED!"))

   @assert check

   return Q
end

function distance_to_positive_cone(Δ::GroupRingElem, λ, Q, wlen::Int)
    info(logger, "------------------------------------------------------------")
    info(logger, "λ = $λ")
    info(logger, "Checking in floating-point arithmetic...")
    Δ²_λΔ = EOI(Δ, λ)
    @logtime logger fp_distance = λ - distance_to_cone(Δ²_λΔ, Q, wlen)
    info(logger, "Floating point distance (to positive cone) ≈ $(@sprintf("%.10f", fp_distance))")
    info(logger, "------------------------------------------------------------")

    if fp_distance ≤ 0
        return fp_distance
    end

    info(logger, "")
    Q = augIdproj(Q, logger)

    info(logger, "Checking in interval arithmetic")
    λ = @interval(λ)
    Δ = GroupRingElem([@interval(c) for c in Δ.coeffs], parent(Δ))
    Δ²_λΔ = EOI(Δ, λ)

    @logtime logger Interval_dist_to_ΣSq = λ - distance_to_cone(Δ²_λΔ, Q, wlen)
    info(logger, "The Augmentation-projected actual distance (to positive cone) ∈ $(Interval_dist_to_ΣSq)")
    info(logger, "------------------------------------------------------------")

    return Interval_dist_to_ΣSq
end