/**
 *  \file IMP/kernel/ClassnamePredicate.h
 *  \brief Define ClassnamePredicate.
 *
 *  BLURB
 *
 *  Copyright 2007-2013 IMP Inventors. All rights reserved.
 */

#ifndef IMPKERNEL_CLASSNAME_PREDICATE_H
#define IMPKERNEL_CLASSNAME_PREDICATE_H

#include <IMP/kernel/kernel_config.h>
#include "base_types.h"
#include "ParticleTuple.h"
#include "DerivativeAccumulator.h"
#include "internal/container_helpers.h"
#include "input_output_macros.h"

IMPKERNEL_BEGIN_NAMESPACE

//! Abstract predicate function
/** A predicate is a function which returns one of a discrete set of
    values (eg -1, 0, 1 depending on whether a value is negative, zero
    or positive). ClassnamePredicates will evaluate the predicate for the passed
    particles.

    Implementers should check out IMP_CLASSNAME_PREDICATE().
*/
class IMPKERNELEXPORT ClassnamePredicate : public base::Object
{
 public:
  typedef VARIABLETYPE Argument;
  typedef INDEXTYPE IndexArgument;
  ClassnamePredicate(std::string name="ClassnamePredicate %1%");
  //! Compute the predicate.
  virtual int get_value(ARGUMENTTYPE vt) const =0;

  /** Implementations
      for these are provided by the IMP_CLASSNAME_PREDICATE()
      macro.
  */
  virtual Ints get_value(const PLURALVARIABLETYPE &o) const {
    Ints ret(o.size());
    for (unsigned int i=0; i< o.size(); ++i) {
      ret[i]+= get_value(o[i]);
    }
    return ret;
  }

#if !defined(IMP_DOXYGEN) && !defined(SWIG)
  virtual void remove_if_equal(Model *m,
                               PLURALINDEXTYPE& ps, int v) const;
  virtual void remove_if_not_equal(Model *m,
                                   PLURALINDEXTYPE& ps, int v) const;
#endif

  //! Compute the predicate and the derivative if needed.
  virtual int get_value_index(Model *m, PASSINDEXTYPE vt) const {
    return get_value(internal::get_particle(m, vt));
  }

  //! Enable them to be use as functors
  /** But beware of slicing.
   */
  int operator()(Model *m, PASSINDEXTYPE vt) const {
    return get_value_index(m, vt);
  }

  /** Implementations
      for these are provided by the IMP_CLASSNAME_PREDICATE()
      macro.
  */
  virtual Ints get_value_index(Model *m,
                                const PLURALINDEXTYPE &o) const {
    Ints ret(o.size());
    for (unsigned int i=0; i< o.size(); ++i) {
      ret[i]+= get_value_index(m, o[i]);
    }
    return ret;
  }

  IMP_INPUTS_DECL(ClassnamePredicate);

  IMP_REF_COUNTED_DESTRUCTOR(ClassnamePredicate);
};


IMPKERNEL_END_NAMESPACE

#endif  /* IMPKERNEL_CLASSNAME_PREDICATE_H */