/**
 * \file  MappedDiscreteSet.cpp
 * \brief Holds a mapped discrete sampling space.
 *
 *  Copyright 2007-2010 IMP Inventors. All rights reserved.
 */

#include "IMP/domino/MappedDiscreteSet.h"

IMPDOMINO_BEGIN_NAMESPACE
MappedDiscreteSet::MappedDiscreteSet(const Particles &ps_target)
{
  for (Particles::const_iterator it = ps_target.begin();
       it != ps_target.end();it++) {
    states_map_[*it] = Particles();
  }
}
MappedDiscreteSet::MappedDiscreteSet(Particles *ps_target,
        const std::vector<FloatKey> &atts) : DiscreteSet(atts){
  for (Particles::const_iterator it = ps_target->begin();
       it != ps_target->end();it++) {
    states_map_[*it] = Particles();
  }
}
void MappedDiscreteSet::add_mapped_state(Particle* sampled_p,Particle *state)
{
  IMP_INTERNAL_CHECK(states_map_.find(sampled_p) != states_map_.end(),
     "The model particle is not part of the mapping particles.");
  states_map_[sampled_p].push_back(state);
}

long MappedDiscreteSet::get_number_of_mapped_states(Particle *p_target) const
{
  return states_map_.find(p_target)->second.size();
}

Particle * MappedDiscreteSet::get_mapped_state(Particle *p_target,
                                               long state_ind) const {
  IMP_INTERNAL_CHECK(static_cast<unsigned int>(state_ind)
        <states_map_.find(p_target)->second.size(),
        "MappedDiscreteSet::get_mapped_state the input state index: "
        << state_ind << " is out of range ( " << states_.size() << " ) ");
  return (states_map_.find(p_target)->second)[state_ind];
}

Float MappedDiscreteSet::get_mapped_state_val(Particle* p_target,
                           long state_ind, IMP::FloatKey key) const {
  std::stringstream err_msg;
  err_msg <<"MappedDiscreteSet::get_mapped_state_val the input state index: ";
  err_msg <<state_ind <<" for particle : "<<
          p_target->get_value(node_name_key());
  err_msg <<" is out of range ( " << get_number_of_mapped_states(p_target);
  err_msg << " ) " <<std::endl;
  IMP_INTERNAL_CHECK(state_ind<get_number_of_mapped_states(p_target),
                     err_msg.str());
  return (states_map_.find(p_target)->second)[state_ind]->get_value(key);
  // if the key does not exist Particle will raise an exception
}

void MappedDiscreteSet::show(std::ostream& out) const {
  for(std::map<Particle*, Particles >::const_iterator
      it1 = states_map_.begin(); it1 != states_map_.end(); it1++) {
    out << "There are " << get_number_of_mapped_states(it1->first)
        << " states for : "<< it1->first->get_name()<<"  :: " << std::endl;
    /*    for(std::vector<Particle *> ::const_iterator
      it2 = it1->second.begin(); it2 != it1->second.end(); it2++) {
      (*it2)->show(out);
      }*/
  }
}

//! Check if each particle has mapped states
bool MappedDiscreteSet::is_valid() const {
  for(std::map<Particle*, Particles >::const_iterator
      it1 = states_map_.begin(); it1 != states_map_.end(); it1++) {
    if(it1->second.size()==0){
      return false;
    }
  }
  return true;
}


IMPDOMINO_END_NAMESPACE