bvh_pbrt.h

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2015-2016 Mario Luzeiro <[email protected]>
 * Copyright The KiCad Developers, see AUTHORS.txt for contributors.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

 
 
#ifndef _BVH_PBRT_H_
#define _BVH_PBRT_H_
 
#include "accelerator_3d.h"
#include <cstdint>
#include <list>
 
// Forward Declarations
struct BVHBuildNode;
struct BVHPrimitiveInfo;
struct MortonPrimitive;
 
struct LinearBVHNode
{
    // 24 bytes
    BBOX_3D bounds;
 
    // 4 bytes
    union
    {
        int primitivesOffset;   
        int secondChildOffset;  
    };
 
    // 4 bytes
    uint16_t nPrimitives;  
    uint8_t  axis;         
    uint8_t  pad[1];       
};
 
 
enum class SPLITMETHOD
{
    MIDDLE,
    EQUALCOUNTS,
    SAH,
    HLBVH
};
 
 
class BVH_PBRT : public ACCELERATOR_3D
{
public:
    BVH_PBRT( const CONTAINER_3D_BASE& aObjectContainer, int aMaxPrimsInNode = 4,
              SPLITMETHOD aSplitMethod = SPLITMETHOD::SAH );
 
    ~BVH_PBRT();
 
    // We own at least one list of raw pointers.  Don't let the compiler fill in copy c'tors that
    // will only land us in trouble.
    BVH_PBRT( const BVH_PBRT& ) = delete;
    BVH_PBRT& operator=( const BVH_PBRT& ) = delete;
 
    bool Intersect( const RAY& aRay, HITINFO& aHitInfo ) const override;
    bool Intersect( const RAY& aRay, HITINFO& aHitInfo, unsigned int aAccNodeInfo ) const override;
    bool Intersect( const RAYPACKET& aRayPacket, HITINFO_PACKET* aHitInfoPacket ) const override;
    bool IntersectP( const RAY& aRay, float aMaxDistance ) const override;
 
private:
    BVHBuildNode* recursiveBuild( std::vector<BVHPrimitiveInfo>& primitiveInfo, int start, int end, int* totalNodes,
                                  std::vector<const OBJECT_3D*>& orderedPrims );
 
    BVHBuildNode* HLBVHBuild( const std::vector<BVHPrimitiveInfo>& primitiveInfo, int* totalNodes,
                              std::vector<const OBJECT_3D*>& orderedPrims );

    BVHBuildNode* emitLBVH( BVHBuildNode* &buildNodes, const std::vector<BVHPrimitiveInfo>& primitiveInfo,
                            MortonPrimitive* mortonPrims, int nPrimitives, int* totalNodes,
                            std::vector<const OBJECT_3D*>& orderedPrims, int* orderedPrimsOffset, int bit );
 
    BVHBuildNode* buildUpperSAH( std::vector<BVHBuildNode*>& treeletRoots, int start, int end, int* totalNodes );
 
    int flattenBVHTree( BVHBuildNode* node, uint32_t* offset );
 
    // BVH Private Data
    const int                     m_maxPrimsInNode;
    SPLITMETHOD                   m_splitMethod;
    std::vector<const OBJECT_3D*> m_primitives;
    LinearBVHNode*                m_nodes;
 
    std::list<void*>              m_nodesToFree;
 
    // Partition traversal
    unsigned int m_I[RAYPACKET_RAYS_PER_PACKET];
};
 
#endif  // _BVH_PBRT_H_