Triangles.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 TrianglesGeomType : public GeomType {
    typedef std::shared_ptr<TrianglesGeomType> SP;
    
    TrianglesGeomType(Context *const context,
                      size_t varStructSize,
                      const std::vector<OWLVarDecl> &varDecls);
    
    std::string toString() const override { return "TriangleGeomType"; }
    
    std::shared_ptr<Geom> createGeom() override;
  };
 
  struct TrianglesGeom : public Geom {
 
    typedef std::shared_ptr<TrianglesGeom> SP;
 
    struct DeviceData : public Geom::DeviceData {
      DeviceData(const DeviceContext::SP &device);
 
      std::vector<CUdeviceptr> vertexPointers;
 
      CUdeviceptr indexPointer  = (CUdeviceptr)0;
    };
 
    TrianglesGeom(Context *const context,
                  GeomType::SP geometryType);
 
    RegisteredObject::DeviceData::SP createOn(const DeviceContext::SP &device) override;
    inline DeviceData &getDD(const DeviceContext::SP &device) const;
 
    void setVertices(const std::vector<Buffer::SP> &vertices,
                     size_t count,
                     size_t stride,
                     size_t offset);
 
    void setIndices(Buffer::SP indices,
                    size_t count,
                    size_t stride,
                    size_t offset);
 
    void computeBounds(box3f bounds[2]);
 
    std::string toString() const override;
 
    struct {
      size_t count  = 0;
      size_t stride = 0;
      size_t offset = 0;
      Buffer::SP buffer;
    } index;
    struct {
      size_t count  = 0;
      size_t stride = 0;
      size_t offset = 0;
      std::vector<Buffer::SP> buffers;
    } vertex;
  };
 
  // ------------------------------------------------------------------
  // implementation section
  // ------------------------------------------------------------------
  
  inline TrianglesGeom::DeviceData &TrianglesGeom::getDD(const DeviceContext::SP &device) const
  {
    assert(device->ID < (int)deviceData.size());
    return deviceData[device->ID]->as<TrianglesGeom::DeviceData>();
  }
  
} // ::owl