Group.h

// ======================================================================== //
// Copyright 2019-2020 Ingo Wald                                            //
//                                                                          //
// Licensed under the Apache License, Version 2.0 (the "License");          //
// you may not use this file except in compliance with the License.         //
// You may obtain a copy of the License at                                  //
//                                                                          //
//     http://www.apache.org/licenses/LICENSE-2.0                           //
//                                                                          //
// Unless required by applicable law or agreed to in writing, software      //
// distributed under the License is distributed on an "AS IS" BASIS,        //
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. //
// See the License for the specific language governing permissions and      //
// limitations under the License.                                           //
// ======================================================================== //
 
#pragma once
 
#include "RegisteredObject.h"
#include "Geometry.h"
#include "LaunchParams.h"
// #include "ll/DeviceMemory.h"
// #include "ll/Device.h"
 
namespace owl {
 
  struct Group : public RegisteredObject {
    typedef std::shared_ptr<Group> SP;
 
    struct DeviceData : public RegisteredObject::DeviceData {
      typedef std::shared_ptr<DeviceData> SP;
 
      DeviceData(const DeviceContext::SP &device);
 
      OptixTraversableHandle traversable = 0;
 
      DeviceMemory           bvhMemory;
 
      size_t memFinal = 0;
      
      size_t memPeak = 0;
    };
 
    Group(Context *const context,
          ObjectRegistry &registry);
 
    std::string toString() const override;
 
    virtual void buildAccel(LaunchParams::SP launchParams = nullptr) = 0;
    
    virtual void refitAccel(LaunchParams::SP launchParams = nullptr) = 0;
    
    virtual int getSBTOffset() const = 0;
    
    inline DeviceData &getDD(const DeviceContext::SP &device) const;
 
    RegisteredObject::DeviceData::SP createOn(const DeviceContext::SP &device) override;
 
    inline OptixTraversableHandle getTraversable(const DeviceContext::SP &device) const;
 
    void getAccelSize(size_t &memFinal, size_t &memPeak)
    {
      DeviceData &dd = deviceData[0]->as<DeviceData>();
      memFinal = dd.memFinal;
      memPeak  = dd.memPeak;
    }
 
    box3f bounds[2];
  };
 
  struct GeomGroup : public Group {
    typedef std::shared_ptr<GeomGroup> SP;
 
    GeomGroup(Context *const context, size_t numChildren);
 
    virtual ~GeomGroup();
 
    void setChild(size_t childID, Geom::SP child);
    
    int getSBTOffset() const override { return sbtOffset; }
    
    std::string toString() const;
    
    std::vector<Geom::SP> geometries;
 
    int sbtOffset;
  };
 
  
  // ------------------------------------------------------------------
  // implementation section
  // ------------------------------------------------------------------
  
  inline Group::DeviceData &Group::getDD(const DeviceContext::SP &device) const
  {
    assert(device->ID < (int)deviceData.size());
    return deviceData[device->ID]->as<DeviceData>();
  }
  
  inline OptixTraversableHandle Group::getTraversable(const DeviceContext::SP &device) const
  { return getDD(device).traversable; }
  
} // ::owl