coal/internal/BV_fitter.h

Go to the documentation of this file.

/*
 * Software License Agreement (BSD License)
 *
 *  Copyright (c) 2011-2014, Willow Garage, Inc.
 *  Copyright (c) 2014-2015, Open Source Robotics Foundation
 *  All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions
 *  are met:
 *
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *   * Redistributions in binary form must reproduce the above
 *     copyright notice, this list of conditions and the following
 *     disclaimer in the documentation and/or other materials provided
 *     with the distribution.
 *   * Neither the name of Open Source Robotics Foundation nor the names of its
 *     contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 *  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 *  COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 *  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 *  BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 *  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 *  CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 *  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 *  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *  POSSIBILITY OF SUCH DAMAGE.
 */
 
#ifndef COAL_BV_FITTER_H
#define COAL_BV_FITTER_H
 
#include "coal/BVH/BVH_internal.h"
#include "coal/BV/kIOS.h"
#include "coal/BV/OBBRSS.h"
#include "coal/BV/AABB.h"
#include <iostream>
 
namespace coal {
 
template <typename BV>
void fit(Vec3s* ps, unsigned int n, BV& bv) {
  for (unsigned int i = 0; i < n; ++i)  // TODO(jcarpent): vectorize
  {
    bv += ps[i];
  }
}
 
template <>
void fit<OBB>(Vec3s* ps, unsigned int n, OBB& bv);
 
template <>
void fit<RSS>(Vec3s* ps, unsigned int n, RSS& bv);
 
template <>
void fit<kIOS>(Vec3s* ps, unsigned int n, kIOS& bv);
 
template <>
void fit<OBBRSS>(Vec3s* ps, unsigned int n, OBBRSS& bv);
 
template <>
void fit<AABB>(Vec3s* ps, unsigned int n, AABB& bv);
 
template <typename BV>
class COAL_DLLAPI BVFitterTpl {
 public:
  virtual ~BVFitterTpl() {}
 
  void set(Vec3s* vertices_, Triangle* tri_indices_, BVHModelType type_) {
    vertices = vertices_;
    prev_vertices = NULL;
    tri_indices = tri_indices_;
    type = type_;
  }
 
  void set(Vec3s* vertices_, Vec3s* prev_vertices_, Triangle* tri_indices_,
           BVHModelType type_) {
    vertices = vertices_;
    prev_vertices = prev_vertices_;
    tri_indices = tri_indices_;
    type = type_;
  }
 
  virtual BV fit(unsigned int* primitive_indices,
                 unsigned int num_primitives) = 0;
 
  void clear() {
    vertices = NULL;
    prev_vertices = NULL;
    tri_indices = NULL;
    type = BVH_MODEL_UNKNOWN;
  }
 
 protected:
  Vec3s* vertices;
  Vec3s* prev_vertices;
  Triangle* tri_indices;
  BVHModelType type;
};
 
template <typename BV>
class COAL_DLLAPI BVFitter : public BVFitterTpl<BV> {
  typedef BVFitterTpl<BV> Base;
 
 public:
  BV fit(unsigned int* primitive_indices, unsigned int num_primitives) {
    BV bv;
 
    if (type == BVH_MODEL_TRIANGLES)  
    {
      for (unsigned int i = 0; i < num_primitives; ++i) {
        Triangle t = tri_indices[primitive_indices[i]];
        bv += vertices[t[0]];
        bv += vertices[t[1]];
        bv += vertices[t[2]];
 
        if (prev_vertices)  
        {
          bv += prev_vertices[t[0]];
          bv += prev_vertices[t[1]];
          bv += prev_vertices[t[2]];
        }
      }
    } else if (type == BVH_MODEL_POINTCLOUD)  
    {
      for (unsigned int i = 0; i < num_primitives; ++i) {
        bv += vertices[primitive_indices[i]];
 
        if (prev_vertices)  
        {
          bv += prev_vertices[primitive_indices[i]];
        }
      }
    }
 
    return bv;
  }
 
 protected:
  using Base::prev_vertices;
  using Base::tri_indices;
  using Base::type;
  using Base::vertices;
};
 
template <>
class COAL_DLLAPI BVFitter<OBB> : public BVFitterTpl<OBB> {
 public:
  OBB fit(unsigned int* primitive_indices, unsigned int num_primitives);
};
 
template <>
class COAL_DLLAPI BVFitter<RSS> : public BVFitterTpl<RSS> {
 public:
  RSS fit(unsigned int* primitive_indices, unsigned int num_primitives);
};
 
template <>
class COAL_DLLAPI BVFitter<kIOS> : public BVFitterTpl<kIOS> {
 public:
  kIOS fit(unsigned int* primitive_indices, unsigned int num_primitives);
};
 
template <>
class COAL_DLLAPI BVFitter<OBBRSS> : public BVFitterTpl<OBBRSS> {
 public:
  OBBRSS fit(unsigned int* primitive_indices, unsigned int num_primitives);
};
 
template <>
class COAL_DLLAPI BVFitter<AABB> : public BVFitterTpl<AABB> {
 public:
  AABB fit(unsigned int* primitive_indices, unsigned int num_primitives);
};
 
}  // namespace coal
 
#endif