MrepSkeleton2d.hpp

// This file is part of the imaging2 class library.
//
// University of Innsbruck, Infmath Imaging, 2009.
// http://infmath.uibk.ac.at
//
// All rights reserved.


#ifndef SHAPE_MREP_MREPSKELETON2D_H
#define SHAPE_MREP_MREPSKELETON2D_H

#include <shape/mrep/MrepModel.hpp>
#include <shape/mrep/Position2d.hpp>
#include <shape/mrep/MrepAtom.hpp>
#include <shape/mrep/MrepConnection.hpp>
#include <shape/mrep/MrepModel.hpp>


namespace imaging
{
  class MrepSkeleton2d : public MrepModel< Position2d, MrepAtom, MrepConnection<2> >
  {
    float_t atom_rotation(size_t atom_index) const;
      
    void set_subtree_geometry(size_t atom_index, float_t rotation, const std::vector< ublas::fixed_vector<float_t, 2> > & atom_centers);

  public:
    MrepSkeleton2d() : MrepModel< Position2d, MrepAtom, MrepConnection<2> >() {}
    MrepSkeleton2d(const Position2d & position,
                size_t n_atoms = 0, size_t n_connections = 0) :
      MrepModel< Position2d, MrepAtom, MrepConnection<2> >(position, n_atoms, n_connections) {}
    
    ublas::fixed_vector<float_t, 2> atom_center(size_t atom_index) const;
    
    void set_atom_center(size_t atom_index, const ublas::fixed_vector<float_t, 2> & center);
    
    void set_atom_radius(size_t atom_index, float_t radius);

    void get_geometry(std::vector< ublas::fixed_vector<float_t, 2> > & atom_centers,
      std::vector<float_t> & atom_radii) const;
    
    void set_geometry(const std::vector< ublas::fixed_vector<float_t, 2> > & atom_centers,
      const std::vector<float_t> & atom_radii);
    
    float_t atom_radius(size_t atom_index) const
    {
      return _atoms[atom_index].radius();
    }
    
    float_t compute_tangent_angle(size_t atom_1, size_t atom_2) const;
  };

}

#endif

---