"""
The doubly-nonnegative cone in `S^{n}` is the set of all such matrices
that both,

  a) are positive semidefinite

  b) have only nonnegative entries

It is represented typically by either `\mathcal{D}^{n}` or
`\mathcal{DNN}`.

"""

from sage.all import *

# Sage doesn't load ~/.sage/init.sage during testing (sage -t), so we
# have to explicitly mangle our sitedir here so that "mjo.cone"
# resolves.
from os.path import abspath
from site import addsitedir
addsitedir(abspath('../../'))
from mjo.cone.symmetric_psd import factor_psd, is_symmetric_psd



def is_doubly_nonnegative(A):
    """
    Determine whether or not the matrix ``A`` is doubly-nonnegative.

    INPUT:

    - ``A`` - The matrix in question

    OUTPUT:

    Either ``True`` if ``A`` is doubly-nonnegative, or ``False``
    otherwise.

    EXAMPLES:

    Every completely positive matrix is doubly-nonnegative::

        sage: v = vector(map(abs, random_vector(ZZ, 10)))
        sage: A = v.column() * v.row()
        sage: is_doubly_nonnegative(A)
        True

    The following matrix is nonnegative but non positive semidefinite::

        sage: A = matrix(ZZ, [[1, 2], [2, 1]])
        sage: is_doubly_nonnegative(A)
        False

    """

    if A.base_ring() == SR:
        msg = 'The matrix ``A`` cannot be the symbolic.'
        raise ValueError.new(msg)

    # Check that all of the entries of ``A`` are nonnegative.
    if not all([ a >= 0 for a in A.list() ]):
        return False

    # It's nonnegative, so all we need to do is check that it's
    # symmetric positive-semidefinite.
    return is_symmetric_psd(A)



def has_admissible_extreme_rank(A):
    """
    The extreme matrices of the doubly-nonnegative cone have some
    restrictions on their ranks. This function checks to see whether or
    not ``A`` could be extreme based on its rank.

    INPUT:

    - ``A`` - The matrix in question

    OUTPUT:

    ``False`` if the rank of ``A`` precludes it from being an extreme
    matrix of the doubly-nonnegative cone, ``True`` otherwise.

    REFERENCE:

    Hamilton-Jester, Crista Lee; Li, Chi-Kwong. Extreme Vectors of
    Doubly Nonnegative Matrices. Rocky Mountain Journal of Mathematics
    26 (1996), no. 4, 1371--1383. doi:10.1216/rmjm/1181071993.
    http://projecteuclid.org/euclid.rmjm/1181071993.

    EXAMPLES:

    The zero matrix has rank zero, which is admissible::

    sage: A = zero_matrix(QQ, 5, 5)
    sage: has_admissible_extreme_rank(A)
    True

    """
    if not A.is_symmetric():
        # This function is more or less internal, so blow up if passed
        # something unexpected.
        raise ValueError('The matrix ``A`` must be symmetric.')

    r = rank(A)
    n = ZZ(A.nrows()) # Columns would work, too, since ``A`` is symmetric.

    if r == 0:
        # Zero is in the doubly-nonnegative cone.
        return True

    # See Theorem 3.1 in the cited reference.
    if r == 2:
        return False

    if n.mod(2) == 0:
        # n is even
        return r <= max(1, n-3)
    else:
        # n is odd
        return r <= max(1, n-2)


def is_extreme_doubly_nonnegative(A):
    """
    Returns ``True`` if the given matrix is an extreme matrix of the
    doubly-nonnegative cone, and ``False`` otherwise.

    EXAMPLES:

    The zero matrix is an extreme matrix::

        sage: A = zero_matrix(QQ, 5, 5)
        sage: is_extreme_doubly_nonnegative(A)
        True

    """

    r = A.rank()

    if r == 0:
        # Short circuit, we know the zero matrix is extreme.
        return True

    if not is_admissible_extreme_rank(r):
        return False

    raise NotImplementedError()