CurvesGeom.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 CurvesGeomType : public GeomType {
    typedef std::shared_ptr<CurvesGeomType> 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;
    };
    
    CurvesGeomType(Context *const context,
                   size_t varStructSize,
                   const std::vector<OWLVarDecl> &varDecls);
    
    std::string toString() const override { return "CurvesGeomType"; }
    
    std::shared_ptr<Geom> createGeom() override;
    
    inline DeviceData &getDD(const DeviceContext::SP &device) const;
    
    RegisteredObject::DeviceData::SP createOn(const DeviceContext::SP &device) override;
 
    void setDegree(int degree, bool force_caps);
    
    int degree = 3;
    
    bool forceCaps = false;
   };
 
  struct CurvesGeom : public Geom {
    
    typedef std::shared_ptr<CurvesGeom> SP;
 
    struct DeviceData : public Geom::DeviceData {
      DeviceData(const DeviceContext::SP &device);
 
      std::vector<CUdeviceptr> verticesPointers;
 
      std::vector<CUdeviceptr> widthsPointers;
 
      CUdeviceptr indicesPointer  = (CUdeviceptr)0;
    };
 
    CurvesGeom(Context *const context,
               GeomType::SP geometryType);
 
    RegisteredObject::DeviceData::SP createOn(const DeviceContext::SP &device) override;
    inline DeviceData &getDD(const DeviceContext::SP &device);
                        
    CurvesGeomType::DeviceData &getTypeDD(const DeviceContext::SP &device) const;
 
    void setVertices(const std::vector<Buffer::SP> &vertices,
                     const std::vector<Buffer::SP> &widths,
                     size_t count);
 
    void setSegmentIndices(Buffer::SP indices,
                           size_t count);
 
    std::string toString() const override;
 
    int segmentIndicesCount = 0;
    Buffer::SP segmentIndicesBuffer;
    
    int vertexCount = 0;
    std::vector<Buffer::SP> verticesBuffers;
    std::vector<Buffer::SP> widthsBuffers;
  };
 
  // ------------------------------------------------------------------
  // implementation section
  // ------------------------------------------------------------------
  
  inline CurvesGeomType::DeviceData &
  CurvesGeomType::getDD(const DeviceContext::SP &device) const
  {
    assert(device && device->ID >= 0 && device->ID < (int)deviceData.size());
    return deviceData[device->ID]->as<CurvesGeomType::DeviceData>();
  }
 
  inline CurvesGeom::DeviceData &
  CurvesGeom::getDD(const DeviceContext::SP &device)
  {
    assert(device && device->ID >= 0 && device->ID < (int)deviceData.size());
    return deviceData[device->ID]->as<CurvesGeom::DeviceData>();
  }
 
  inline CurvesGeomType::DeviceData &
  CurvesGeom::getTypeDD(const DeviceContext::SP &device) const
  {
    return (CurvesGeomType::DeviceData &)type->getDD(device);
  }
  
} // ::owl