require 'hpsqrt/inspect_mode'
require 'hpsqrt/term'
require 'hpsqrt/version'


class HpSqrt < Numeric

  @@inspect_mode = INSPECT_MODE::VALUE

  def self.inspect_mode
    @@inspect_mode
  end

  def self.inspect_mode=(v)
    @@inspect_mode = v
  end


  attr_reader :terms

  def initialize(terms)
    @terms = terms.freeze
    freeze
  end

  def -@
    terms = @terms.map{|t,c| [t, -c]}.to_h
    self.class.new(terms)
  end

  def +(other)
    other = self.class.create(other)

    terms = @terms.merge(other.terms) {|t, c1, c2|
      c1 + c2
    }
    terms.delete_if {|t,c| c==0}

    self.class.new(terms)
  end

  def -(other)
    other = self.class.create(other)

    other_terms = other.terms.map{|t,c| [t, -c]}.to_h
    terms = @terms.merge(other_terms) {|t, c1, c2|
      c1 + c2
    }
    terms.delete_if {|t,c| c==0}

    self.class.new(terms)
  end

  def *(other)
    other = self.class.create(other)

    terms = {}
    @terms.each {|t1, c1|
      other.terms.each {|t2, c2|
        t = t1 * t2
        c = c1 * c2

        terms[t] ||= 0
        terms[t] += c
      }
    }
    terms.delete_if {|t,c| c==0}

    self.class.new(terms)
  end

  def /(other)
    other = self.class.create(other)
    other_inv = Term.new(number: Rational(1, other.to_c))

    terms = {}
    @terms.each {|t, c|
      t *= other_inv

      terms[t] ||= 0
      terms[t] += c
    }
    terms.delete_if {|t,c| c==0}

    self.class.new(terms)
  end

  def **(other)
    other = self.class.create(other)

    if other.int?
      result = self.class.create(1)
      other_i = other.real.to_i
      other_i.abs.times {|i|
        result *= self
      }
      if other_i<0
        result = Rational(1, result)
      end
      result
    else
      self.class.number(self.to_c ** other.to_c)
    end
  end

  def coerce(other)
    [self.class.create(other), self]
  end

  def ==(other)
    if self.class==other
      self.to_c==other.to_c
    elsif Numeric===other
      self.to_c==other
    else
      super.==(other)
    end
  end

  def rect
    to_c.rect
  end

  def polar
    to_c.polar
  end

  def arg
    to_c.arg
  end
  alias_method :angle, :arg
  alias_method :phase, :arg

  def to_c
    @terms.map {|t, c|
      t.number * Math.sqrt(Complex(t.sqrt)) * c
    }.sum.to_c
  end

  def real
    to_c.real
  end

  def imag
    to_c.imag
  end
  alias_method :imaginary, :imag

  def to_i
    c = to_c
    if c.imag.zero?
      c.real.to_i
    else
      raise RangeError, "can't convert %s into Integer" % c
    end
  end

  def to_f
    c = to_c
    if c.imag.zero?
      c.real.to_f
    else
      raise RangeError, "can't convert %s into Float" % c
    end
  end

  def to_r
    c = to_c
    if c.imag.zero?
      Rational(c.real, 1)
    else
      raise RangeError, "can't convert %s into Rational" % c
    end
  end

  def expr
    value_to_s = -> (v) {
      if Complex===v && v.imag.zero?
        v = v.real
      end
      if Rational===v
        v = v.to_s.sub(/\/1$/, "")
      end
      v = v.to_s
      if v !~ /^[\d\.]+$/
        v = "(%s)" % v
      end
      v
    }

    result = @terms.map {|t, c|
      n = t.number * c
      s = t.sqrt

      if s!=1
        if n==1
          "\u221A%s" % value_to_s[s]
        elsif 0<n.real
          "%s\u221A%s" % [value_to_s[n], value_to_s[s]]
        elsif n==-1
          "(-\u221A%s)" % value_to_s[s]
        else
          "(%s\u221A%s)" % [value_to_s[n], value_to_s[s]]
        end
      else
        value_to_s[n]
      end
    }

    if 0<result.length
      result.join(" + ")
    else
      "0"
    end
  end

  def inspect
    to_s
  end

  def to_s
    case @@inspect_mode
    when INSPECT_MODE::VALUE
      to_c.to_s
    when INSPECT_MODE::EXPR
      expr
    else
      "#<%s:0x%016x value=(%s) expr=(%s)>" % [self.class.name, self.object_id, to_c, expr]
    end
  end

  def real?
    imag.zero?
  end

  def imag?
    !real?
  end

  def int?
    c = to_c
    c.imag.zero? && c.real==c.real.to_i
  end

  def float?
    c = to_c
    c.imag.zero? && Float===c.real && c.real!=c.real.to_i
  end

  def self.create(v)
    if self===v
      v
    else
      number(v)
    end
  end

  def self.number(v)
    if v!=0
      new({Term.new(number: v) => 1})
    else
      zero
    end
  end

  def self.sqrt(v)
    if self===v
      v = v.to_c
    end
    if v!=0
      new({Term.new(sqrt: v) => 1})
    else
      zero
    end
  end

  def self.zero
    new({})
  end
end