Weight.h

Go to the documentation of this file.

/**
 *  \file isd/Weight.h
 *  \brief Add weights for a set of states to a particle.
 *
 *  Copyright 2007-2013 IMP Inventors. All rights reserved.
 *
 */

#ifndef IMPISD_WEIGHT_H
#define IMPISD_WEIGHT_H

#include "isd_config.h"

#include <IMP/kernel/Particle.h>
#include <IMP/decorator_macros.h>
#include <IMP/kernel/Model.h>
#include <IMP/Decorator.h>
#include <IMP/base/exception.h>
#include <sstream>

IMPISD_BEGIN_NAMESPACE

//! Add weights for a set of states to a particle.
class IMPISDEXPORT Weight: public Decorator
{

  static const int nstates_max = 20;
  static void do_setup_particle(kernel::Model *m,
                                kernel::ParticleIndex pi);
public:

  IMP_DECORATOR_METHODS(Weight, Decorator);
  IMP_DECORATOR_SETUP_0(Weight);

//! Get number of states key
  static IntKey get_nstates_key();

//! Get i-th weight key
  static FloatKey get_weight_key(int i);

//! Set all the weights
  void set_weights(algebra::VectorKD w);

//! Add one weight
  void add_weight();

//! Get the i-th weight
  Float get_weight(int i);

//! Get all the weights
  algebra::VectorKD get_weights();

//! Set weights are optimized
  void set_weights_are_optimized(bool tf);

//! Get number of states
  Int get_number_of_states();

  static bool get_is_setup(kernel::Model *m,
                           kernel::ParticleIndex pi) {
    return m->get_has_attribute(get_nstates_key(), pi);
  }

};

IMP_VALUES(Weight, Weights);

IMPISD_END_NAMESPACE

#endif  /* IMPISD_WEIGHT_H */