IMP  2.0.1
The Integrative Modeling Platform
DistributePairsScoreState.h
Go to the documentation of this file.
1 /**
2  * \file IMP/container/DistributePairsScoreState.h
3  * \brief Apply a PairScore to each Pair in a list.
4  *
5  * WARNING This file was generated from DistributePairsScoreState.h
6  * in /tmp/nightly-build-36540/imp-2.0.1/tools/build/container_templates/container
7  * by tools/maintenance/setup_containers.py.
8  *
9  * Copyright 2007-2013 IMP Inventors. All rights reserved.
10  *
11  */
12 
13 #ifndef IMPCONTAINER_DISTRIBUTE_PAIRS_SCORE_STATE_H
14 #define IMPCONTAINER_DISTRIBUTE_PAIRS_SCORE_STATE_H
15 
16 #include <IMP/container/container_config.h>
17 
18 #include "DynamicListPairContainer.h"
19 #include <IMP/PairPredicate.h>
20 #include <IMP/PairContainer.h>
21 #include <IMP/base/Vector.h>
22 #include <IMP/pair_macros.h>
23 #include <IMP/score_state_macros.h>
24 #include <boost/tuple/tuple.hpp>
25 
26 #include <iostream>
27 
28 IMPCONTAINER_BEGIN_NAMESPACE
29 
30 //! Distribute contents of one container into several based on predicates
31 /** This ScoreState takes a list of predicates and values. For
32  each tuple in the input container, it is placed in a given output container
33  if the predicate, when applied, has the passed value.
34 
35  \note The output containers contents are not necessarily disjoint.
36 */
37 class IMPCONTAINEREXPORT DistributePairsScoreState :
38 public ScoreState
39 {
40  base::OwnerPointer<PairContainer> input_;
41  typedef boost::tuple<base::Pointer<DynamicListPairContainer>,
42  base::OwnerPointer<PairPredicate>, int> Data;
43  base::Vector<Data> data_;
44  mutable bool updated_;
45  void update_lists_if_necessary() const;
46 public:
47  DistributePairsScoreState(PairContainerAdaptor input,
48  std::string name="DistributePairsScoreState %1%");
49 
50  /** A given tuple will go into the returned container if \c predicate
51  returns \c value when applied to it.*/
52  DynamicListPairContainer *add_predicate(PairPredicate *predicate,
53  int value) {
54  IMP_NEW(DynamicListPairContainer, c, (input_,
55  predicate->get_name()+ " output"));
56  data_.push_back(Data(c, predicate, value));
57  return c;
58  }
59  virtual void do_before_evaluate() IMP_OVERRIDE;
60  virtual void do_after_evaluate(DerivativeAccumulator *da) IMP_OVERRIDE;
61  virtual ModelObjectsTemp do_get_inputs() const IMP_OVERRIDE;
62  virtual ModelObjectsTemp do_get_outputs() const IMP_OVERRIDE;
63  IMP_OBJECT(DistributePairsScoreState);
64 };
65 
66 IMPCONTAINER_END_NAMESPACE
67 
68 #endif /* IMPCONTAINER_DISTRIBUTE_PAIRS_SCORE_STATE_H */