IMP logo
IMP Reference Guide  2.5.0
The Integrative Modeling Platform
PredicateSingletonsRestraint.h
Go to the documentation of this file.
1 // Autogenerated by ../../../../tmp/nightly-build-6636/imp-2.5.0/tools/build/make_containers.py
2 // from ../../../../tmp/nightly-build-6636/imp-2.5.0/tools/build/container_templates/container/PredicateClassnamesRestraint.h
3 // Do not edit - any changes will be lost!
4 
5 /**
6  * \file IMP/container/PredicateSingletonsRestraint.h
7  * \brief Apply a SingletonScore to each Singleton in a list.
8  *
9  * Copyright 2007-2015 IMP Inventors. All rights reserved.
10  *
11  */
12 
13 #ifndef IMPCONTAINER_PREDICATE_SINGLETONS_RESTRAINT_H
14 #define IMPCONTAINER_PREDICATE_SINGLETONS_RESTRAINT_H
15 
16 #include <IMP/container/container_config.h>
17 #include <boost/unordered_map.hpp>
18 #include <IMP/SingletonPredicate.h>
19 #include "generic.h"
20 
21 #include <iostream>
22 
23 IMPCONTAINER_BEGIN_NAMESPACE
24 
25 //! Applies a SingletonScore to each Singleton in a list based on a predicate
26 /** This restraint uses a passed predicate to choose which score to apply
27  to each tuple in the input container. The selections are cached, making it
28  substantially faster than using a core::TypedPairScore.
29 
30  \note The ordering of particles within a tuple may vary depending on the
31  input container used. You may need to call set_score() with several
32  different predicate values for different orderings.
33 */
34 class IMPCONTAINEREXPORT PredicateSingletonsRestraint : public Restraint {
37  mutable boost::unordered_map<int, ParticleIndexes> lists_;
38  boost::unordered_map<int, PointerMember<SingletonScore> > scores_;
39  mutable std::size_t input_version_;
40  bool error_on_unknown_;
41  void update_lists_if_necessary() const;
42 
43  public:
46  std::string name =
47  "PredicateSingletonsRestraint %1%");
48 
49  /** Apply the passed score to all pairs whose predicate values match
50  the passed value.*/
51  void set_score(int predicate_value, SingletonScore *score);
52 
53  /** Apply this score to any pair whose predicate value does not match
54  one passed to set_score().*/
55  void set_unknown_score(SingletonScore *score);
56 
57  /** By default, it is an error if the predicate returns a value that is
58  not known. If this is false, then they are silently skipped.
59  */
60  void set_is_complete(bool tf) { error_on_unknown_ = tf; }
61 
62  /** return the indexes of all particles for a given predicate value.*/
63  ParticleIndexes get_indexes(int predicate_value) const {
64  return lists_.find(predicate_value)->second;
65  }
66 
67  public:
72 
73  private:
74  Restraints do_create_current_decomposition() const;
75 };
76 
77 IMPCONTAINER_END_NAMESPACE
78 
79 #endif /* IMPCONTAINER_PREDICATE_SINGLETONS_RESTRAINT_H */
#define IMP_OBJECT_METHODS(Name)
Define the basic things needed by any Object.
Definition: object_macros.h:25
virtual void do_add_score_and_derivatives(ScoreAccumulator sa) const
A restraint should override this to compute the score and derivatives.
Define SingletonPredicate.
Abstract class for scoring object(s) of type ParticleIndex.
A smart pointer to a ref-counted Object that is a class member.
Definition: Pointer.h:146
Class for adding up scores during ScoringFunction evaluation.
Applies a SingletonScore to each Singleton in a list based on a predicate.
Abstract predicate function.
ParticleIndexes get_indexes(int predicate_value) const
virtual ModelObjectsTemp do_get_inputs() const =0
#define IMP_OVERRIDE
Cause a compile error if this method does not override a parent method.
A restraint is a term in an IMP ScoringFunction.
Definition: Restraint.h:52