/**
 *  \file XYZR.cpp   \brief Simple xyzr decorator.
 *
 *  Copyright 2007-2010 IMP Inventors. All rights reserved.
 *
 */

#include "IMP/core/XYZR.h"
#include <IMP/algebra/Vector3D.h>
#include <IMP/algebra/vector_generators.h>

IMPCORE_BEGIN_NAMESPACE

void XYZR::show(std::ostream &out) const
{
  out << "(" << get_x()<< ", "
      << get_y() << ", " << get_z() << ": " << get_radius() << ")";

}


algebra::SphereD<3> get_enclosing_sphere(const XYZsTemp &v, FloatKey rk) {
  std::vector<algebra::SphereD<3> > ss(v.size());
  for (unsigned int i=0; i< v.size(); ++i) {
    XYZ d(v[i]);
    Float r=0;
    if (v[i].get_particle()->has_attribute(rk)) {
      r= v[i].get_particle()->get_value(rk);
    }
    ss[i]= algebra::SphereD<3>(d.get_coordinates(), r);
  }
  algebra::SphereD<3> s= algebra::get_enclosing_sphere(ss);
  return s;
}

void set_enclosing_sphere(XYZR out,
                          const XYZsTemp &v,
                          double slack)
{
  algebra::SphereD<3> s= get_enclosing_sphere(v, out.get_radius_key());
  algebra::SphereD<3> s2(s.get_center(), s.get_radius()+slack);
  out.set_sphere(s2);
}


void set_enclosing_radius(XYZR out,
                          const XYZsTemp &v)
{
  double r=0;
  for (unsigned int i=0; i< v.size(); ++i) {
    if (XYZR::particle_is_instance(v[i], out.get_radius_key())) {
      XYZR d(v[i]);
      double dist= get_distance(static_cast<XYZ>(out), static_cast<XYZ>(d));
      dist+= d.get_radius();
      r= std::max(dist, r);
    } else {
      XYZ d(v[i]);
      double dist= get_distance(static_cast<XYZ>(out), static_cast<XYZ>(d));
      r= std::max(dist, r);
    }
  }
  out.set_radius(r);
}


XYZRs create_xyzr_particles(Model *m,
                                unsigned int num,
                                Float radius,
                                Float box_side) {
  XYZRs ret;
  for (unsigned int i=0; i< num; ++i) {
    Particle *p= new Particle(m);
    XYZR d= XYZR::setup_particle(p);
    d.set_coordinates(algebra::get_random_vector_in(
                                                    algebra::BoundingBox3D(
                                    algebra::get_ones_vector_d<3>(-box_side),
                                    algebra::get_ones_vector_d<3>(box_side))));
    d.set_radius(radius);
    d.set_coordinates_are_optimized(true);
    ret.push_back(d);
  }
  return ret;
}



IMPCORE_END_NAMESPACE