/**
 *  \file BondPairContainer.cpp
 *  \brief A fake container that returns true if a pair of particles are bonded
 *
 *  Copyright 2007-2010 IMP Inventors. All rights reserved.
 *
 */

#include "IMP/atom/BondPairContainer.h"
#include "IMP/PairModifier.h"
#include "IMP/PairScore.h"

#define FOREACH(expr)\
  unsigned int sz= BondPairContainer::get_number_of_particle_pairs();   \
  for (unsigned int i=0; i< sz; ++i) {                                  \
    ParticlePair p= BondPairContainer::get_particle_pair(i);           \
    expr;                                                               \
  }


IMPATOM_BEGIN_NAMESPACE

BondPairContainer
::BondPairContainer(SingletonContainer *sc, bool): sc_(sc){
}

BondPairContainer
::BondPairContainer(SingletonContainer *sc): sc_(sc){
  set_added_and_removed_containers(
       create_untracked_container(sc_->get_removed_singletons_container()),
       create_untracked_container(sc_->get_added_singletons_container()));
}

bool BondPairContainer
::get_contains_particle_pair(const ParticlePair& pp) const {
  if (!Bonded::particle_is_instance(pp[0])
      || ! Bonded::particle_is_instance(pp[1])) {
    return false;
  }

  Bonded ba(pp[0]);
  Bonded bb(pp[1]);
  Bond bd=get_bond(ba, bb);
  return sc_->get_contains_particle(bd);
}

unsigned int BondPairContainer
::get_number_of_particle_pairs() const {
  return sc_->get_number_of_particles();
}

ParticlePair BondPairContainer
::get_particle_pair(unsigned int i) const {
  Bond bd(sc_->get_particle(i));
  return ParticlePair(bd.get_bonded(0), bd.get_bonded(1));
}


void BondPairContainer::do_show(std::ostream &out) const {
  out << "container " << *sc_ << std::endl;
}

IMP_PAIR_CONTAINER_METHODS_FROM_FOREACH(BondPairContainer);

ParticlesTemp BondPairContainer::get_contained_particles() const {
  ParticlesTemp ret(3*sc_->get_number_of_particles());
  for (unsigned int i=0; i< sc_->get_number_of_particles(); ++i) {
    ret[i*3]= sc_->get_particle(i);
    ret[i*3+1]= Bond(sc_->get_particle(i)).get_bonded(0);
    ret[i*3+2]= Bond(sc_->get_particle(i)).get_bonded(1);
  }
  return ret;
}

bool BondPairContainer::get_contained_particles_changed() const {
  return sc_->get_contained_particles_changed();
}

ContainersTemp BondPairContainer::get_input_containers() const {
  ContainersTemp ret(1, sc_);
  return ret;
}

IMPATOM_END_NAMESPACE