/**
 *  \file CartesianProductSampler.h
 *  \brief The class samples all combinations of a discrete set to particles.
 *  Copyright 2007-2010 IMP Inventors. All rights reserved.
 */
#include "IMP/domino/CartesianProductSampler.h"

IMPDOMINO_BEGIN_NAMESPACE

CartesianProductSampler::CartesianProductSampler(MappedDiscreteSet *ds,
                                                 const Particles &ps)
{
  ds_ = ds;
  ps_=ps;
}

void CartesianProductSampler::show(std::ostream& out) const
{
  out <<"CartesianProductSampler with " << ds_->get_number_of_states()
  << " states in the set and " << ps_.size() << "particles: "<< std::endl;
  for(Particles::const_iterator it = ps_.begin(); it != ps_.end(); it++) {
    out<<**it<<",";
  }
  out<<std::endl;
}
void CartesianProductSampler::populate_states_of_particles(
   container::ListSingletonContainer *particles,
   std::map<std::string,CombState *> *states) const
{
  IMP_LOG(IMP::VERBOSE,
    "start CartesianProductSampler::populate_states_of_particles"<<std::endl);
  //CombState *calc_state;
  //the size of the combination:
  unsigned int comb_size = particles->get_number_of_particles();
  IMP_LOG(IMP::VERBOSE,
          "number of particles: "<<comb_size<<std::endl);
  std::vector<int> v_int(comb_size);//
  std::vector<int> c_int(comb_size);//
  unsigned int i;
  for(i=0;i<comb_size;i++){
    IMP_LOG(IMP::VERBOSE,"i:"<<i<<" number of mapped states: " <<
            ds_->get_number_of_mapped_states(particles->get_particle(i))
            <<std::endl);

    //TODO - return!
    // IMP_check(ds_->get_number_of_mapped_states((*particles)[i])>0,
    //"CartesianProductSampler::populate_states_of_particles the "
    //<< i <<"'th particle has no state",
    //ValueException);
    v_int[i] = ds_->get_number_of_mapped_states(particles->get_particle(i));
    c_int[i] = 0;
  }
  while(c_int[0] != v_int[0]) {
    CombState *calc_state = new CombState();
    for (i = 0; i < c_int.size(); i++) {
      calc_state->add_data_item(particles->get_particle(i), c_int[i]);
    }
    (*states)[calc_state->get_partial_key(particles)]=calc_state;
    //update the indexes
    i=c_int.size()-1;
    if(c_int[i]!=v_int[i]) {
      c_int[i]++;
    }
    while (c_int[i] == v_int[i] && i>0) {
      c_int[i]=0;
      c_int[i-1]++;
      i--;
    }
  }
  IMP_LOG(IMP::VERBOSE,
    "end CartesianProductSampler::populate_states_of_particles"<<std::endl);
}

void CartesianProductSampler::move2state(const CombState *cs){
  Particle *p;
  FloatKeys atts = ds_->get_att_keys();
  for (std::map<Particle *,unsigned int>::const_iterator
         it = cs->get_data()->begin();it != cs->get_data()->end(); it++) {
    p = it->first;
    for (FloatKeys::const_iterator k_iter = atts.begin();
         k_iter != atts.end(); k_iter++) {
      p->set_value(*k_iter,
                   ds_->get_mapped_state_val(p,it->second, *k_iter));
    }
  }
}

DiscreteSet* CartesianProductSampler::get_space(Particle *p) const {
  DiscreteSet *ds = new DiscreteSet(ds_->get_att_keys());
  for (int i=0;i<ds_->get_number_of_mapped_states(p);i++) {
    ds->add_state(ds_->get_mapped_state(p,i));
  }
  return ds;
}
IMPDOMINO_END_NAMESPACE