kernel/TripletPredicate.h

Go to the documentation of this file.

/**
 *  \file IMP/kernel/TripletPredicate.h
 *  \brief Define TripletPredicate.
 *
 *  WARNING This file was generated from TripletPredicate.h
 *  in /tmp/nightly-build-5687/imp-2.0.0/tools/build/container_templates/kernel
 *  by tools/maintenance/setup_containers.py.
 *
 *  Copyright 2007-2013 IMP Inventors. All rights reserved.
 */

#ifndef IMPKERNEL_TRIPLET_PREDICATE_H
#define IMPKERNEL_TRIPLET_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). TripletPredicates will evaluate the predicate for the passed
    particles.

    Implementers should check out IMP_TRIPLET_PREDICATE().
*/
class IMPKERNELEXPORT TripletPredicate : public base::Object
{
 public:
  typedef ParticleTriplet Argument;
  typedef ParticleIndexTriplet IndexArgument;
  TripletPredicate(std::string name="TripletPredicate %1%");
  //! Compute the predicate.
  virtual int get_value(const ParticleTriplet& vt) const =0;

  /** Implementations
      for these are provided by the IMP_TRIPLET_PREDICATE()
      macro.
  */
  virtual Ints get_value(const ParticleTripletsTemp &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,
                               ParticleIndexTriplets& ps, int v) const;
  virtual void remove_if_not_equal(Model *m,
                                   ParticleIndexTriplets& ps, int v) const;
#endif

  //! Compute the predicate and the derivative if needed.
  virtual int get_value_index(Model *m, const ParticleIndexTriplet& 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, const ParticleIndexTriplet& vt) const {
    return get_value_index(m, vt);
  }

  /** Implementations
      for these are provided by the IMP_TRIPLET_PREDICATE()
      macro.
  */
  virtual Ints get_value_index(Model *m,
                                const ParticleIndexTriplets &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(TripletPredicate);

  IMP_REF_COUNTED_DESTRUCTOR(TripletPredicate);
};


IMPKERNEL_END_NAMESPACE

#endif  /* IMPKERNEL_TRIPLET_PREDICATE_H */