SphereGeom.h

// ======================================================================== //
// Copyright 2019-2021 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 "Geometry.h"
 
namespace owl {
 
  struct SphereGeomType : public GeomType {
    typedef std::shared_ptr<SphereGeomType> SP;
    
    struct DeviceData : public GeomType::DeviceData {
      typedef std::shared_ptr<DeviceData> SP;
 
      DeviceData(const DeviceContext::SP &device);
 
      void fillPGDesc(OptixProgramGroupDesc &pgDesc,
                      GeomType *gt,
                      int rayType) override;
    };
    
    SphereGeomType(Context *const context,
                   size_t varStructSize,
                   const std::vector<OWLVarDecl> &varDecls);
    
    std::string toString() const override { return "SphereGeomType"; }
    
    std::shared_ptr<Geom> createGeom() override;
    
    inline DeviceData &getDD(const DeviceContext::SP &device) const;
    
    RegisteredObject::DeviceData::SP createOn(const DeviceContext::SP &device) override;
   };
 
  struct SphereGeom : public Geom {
    
    typedef std::shared_ptr<SphereGeom> SP;
 
    struct DeviceData : public Geom::DeviceData {
      DeviceData(const DeviceContext::SP &device);
 
      std::vector<CUdeviceptr> verticesPointers;
 
      std::vector<CUdeviceptr> radiusPointers;
    };
 
    SphereGeom(Context *const context,
               GeomType::SP geometryType);
 
    RegisteredObject::DeviceData::SP createOn(const DeviceContext::SP &device) override;
    inline DeviceData &getDD(const DeviceContext::SP &device);
                        
    SphereGeomType::DeviceData &getTypeDD(const DeviceContext::SP &device) const;
 
    void setVertices(const std::vector<Buffer::SP> &vertices,
                     const std::vector<Buffer::SP> &radii,
                     size_t count);
 
    std::string toString() const override;
 
    int vertexCount = 0;
    std::vector<Buffer::SP> verticesBuffers;
    std::vector<Buffer::SP> radiusBuffers;
  };
 
  // ------------------------------------------------------------------
  // implementation section
  // ------------------------------------------------------------------
  
  inline SphereGeomType::DeviceData &
  SphereGeomType::getDD(const DeviceContext::SP &device) const
  {
    assert(device && device->ID >= 0 && device->ID < (int)deviceData.size());
    return deviceData[device->ID]->as<SphereGeomType::DeviceData>();
  }
 
  inline SphereGeom::DeviceData &
  SphereGeom::getDD(const DeviceContext::SP &device)
  {
    assert(device && device->ID >= 0 && device->ID < (int)deviceData.size());
    return deviceData[device->ID]->as<SphereGeom::DeviceData>();
  }
 
  inline SphereGeomType::DeviceData &
  SphereGeom::getTypeDD(const DeviceContext::SP &device) const
  {
    return (SphereGeomType::DeviceData &)type->getDD(device);
  }
  
} // ::owl