src/rcrt/BIH.hpp

Go to the documentation of this file.

#ifndef BIH_HPP_
#define BIH_HPP_

#include "Traceable.h"
#include <vector>
#include "primitives/AABB.h"
#include "BIHNode.hpp"

namespace rcrt
{

template<class T>
class BIH : public Traceable
{
private:
        AABB boundingBox;
        BIHNode<T>* rootNode;
        
public:
        BIH<T>()
        {
                rootNode = 0;
        }
        
        BIH<T>(std::vector<T*>* trList)
        {
                rootNode = 0;
                setTraceables(trList);          
        }
        
        void setTraceables(std::vector<T*>* trList)
        {
                if(rootNode)
                        delete rootNode;
                if(trList->empty())
                        return;
                std::cout << "[BIH]: building BIH size = "<< trList->size() << std::endl;
                
                boundingBox = calcBoundingBox(trList);
                
                rootNode = new BIHNode<T>(boundingBox, trList, 0, trList->size()-1);
                std::cout << "       built BIH" << std::endl;
        }
        
        virtual ~BIH()
        {
                if(rootNode)
                        delete rootNode;
        }
        
        virtual Intersection intersect(Ray& r) const
        {
                if(rootNode)
                        return rootNode->intersect(r, r.minDist(), r.maxDist());
                else
                        return Intersection();
        }
                
        virtual const AABB& getBoundingBox() const
        {
                return boundingBox;
        }
        
        virtual const Point3D& getCentroid() const
        {
                return boundingBox.getCentroid();
        }
        
        std::vector<T*>* getTraceables() const
        {
                if(rootNode)
                        return rootNode->getTraceables();
                else
                        return 0;
        }
        
        static AABB calcBoundingBox(std::vector<T*>* trList)
        {
                AABB box;
                for(unsigned int i = 0; i < trList->size(); i++){
                        box.extend(trList->at(i)->getBoundingBox());
                }
                return box;
        }
        
        AABB clipBox(AABB& cbox) const {
                return getBoundingBox();
        }
        
};


}


#endif /*BIH_HPP_*/

---