function varargout = nnz(varargin)
%NNZ (overloaded)
%
%    n = nnz(X)
%
% NNZ applied to a constraint counts the number of satisfied constraints.
%
% The NNZ operator is implemented using the concept of nonlinear operators
% in YALMIP. NNZ(X) creates a new so called derived variable that can be
% treated as any other variable in YALMIP. When SOLVESDP is issued,
% logic constraints are added to the problem to model the NNZ operator.


% Author Johan Löfberg
% $Id: nnz.m,v 1.6 2006/05/15 20:00:39 joloef Exp $

switch class(varargin{1})

    case 'constraint'
        varargout{1} = yalmip('addextendedvariable','nnz',varargin{1});

    case 'char'
        t = varargin{2};
        x = varargin{3};
        switch varargin{1}
            case 'graph'
                x = set(x);
                if is(x,'elementwise')
                    x = sdpvar(x);
                    x = x(:);
                    [M,m] = derivebounds(x);
                    delta = binvar(length(x),1);
                    F = set(x >= m.*(1-delta)) + set(sum(delta) == t);
                elseif is(x,'equality')
                    x = sdpvar(x);
                    x = x(:);
                    [M,m,infbound] = derivebounds(x);
                    if infbound
                        warning('You have unbounded variables in IMPLIES leading to a lousy big-M relaxation.');
                    end
                    delta = binvar(length(x),1);
                    F = set(M.*(1-delta) >= x >= m.*(1-delta)) + set(sum(delta) == t);
                else
                    error('Constraint type not supported in cardinality. Make a feature request');
                end
                varargout{1} = F;
                varargout{2} = struct('convexity','concave','monotonicity','none','definiteness','none');
                varargout{3} = x;

            otherwise
                x = set(x);
                if is(x,'elementwise')
                    x = sdpvar(x);
                    x = x(:);
                    [M,m,infbound] = derivebounds(x);
                    if infbound
                            warning('You have unbounded variables in NNZ leading to a lousy big-M relaxation.');
                    end
                    n = length(x);
                    deltaD = binvar(n,1);
                    deltaU = binvar(n,1);
                    if  all(ismember(getvariables(x),[yalmip('intvariables') yalmip('binvariables')]))
                        % If an integer constraint not holds, it has to  be
                        % epsilon violated
                        F = set((M+1e-4).*(1-deltaU)-1e-4 >= x >= m.*(1-deltaD));
                    else
                        % User has to think harder him self
                        F = set((M).*(1-deltaU) >= x >= m.*(1-deltaD));
                    end
                    F = F + set(sum(deltaD) == t);
                    F = F + set(deltaU + deltaD == 1);
                    F = F + set(0 <= t <= n);
                elseif is(x,'equality')
                    x = sdpvar(x);
                    x = x(:);
                    n = length(x);
                    [M,m] = derivebounds(x);
                    deltaZ   = binvar(n,1);
                    deltaNZU = binvar(n,1);
                    deltaNZD = binvar(n,1);
                    eps = 1e-4;
                    % deltaZ forces x to be zero
                    % deltaZU 1 is x > eps
                    % deltaZU 1 is x < -eps
                    F = set(M.*(1-deltaZ) >= x >= m.*(1-deltaZ));
                    F = set((M+eps).*deltaNZU-eps>= x >= eps + (m-eps).*deltaNZD);
                    F = F + set(sum(deltaZ) == t);
                    F = F + set(deltaZ + deltaNZU + deltaNZD == 1);
                    F = F + set(0 <= t <= n);
                else
                    error('Constraint type not supported in cardinality. Make a feature request');
                end
                varargout{1} = F;
                varargout{2} = struct('convexity','milp','monotonicity','milp','definiteness','milp');
                varargout{3} = x;
        end
    otherwise
end