SBTObject.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 "Variable.h"
 
namespace owl {
 
  struct SBTObjectType : public RegisteredObject
  {
    typedef std::shared_ptr<SBTObjectType> SP;
 
    SBTObjectType(Context *const context,
                  ObjectRegistry &registry,
                  size_t varStructSize,
                  const std::vector<OWLVarDecl> &varDecls);
 
    virtual ~SBTObjectType();
    
    int getVariableIdx(const std::string &varName);
 
    bool hasVariable(const std::string &varName);
 
    std::string toString() const override;
 
    void declareVariable(const std::string &varName,
                         OWLDataType type,
                         size_t offset);
 
    std::vector<Variable::SP> instantiateVariables();
 
    const size_t         varStructSize;
 
    const std::vector<OWLVarDecl> varDecls;
  };
 
 
 
  struct SBTObjectBase : public RegisteredObject
  {
    SBTObjectBase(Context *const context,
                  ObjectRegistry &registry,
                  std::shared_ptr<SBTObjectType> type);
 
    inline bool hasVariable(const std::string &name);
    
    inline Variable::SP getVariable(const std::string &name);
 
    void writeVariables(uint8_t *sbtEntry,
                        const DeviceContext::SP &device) const;
    
    std::shared_ptr<SBTObjectType> const type;
    
    const std::vector<Variable::SP> variables;
  };
 
 
  template<typename ObjectType>
  struct SBTObject : public SBTObjectBase
  {
    typedef std::shared_ptr<SBTObject> SP;
 
    SBTObject(Context *const context,
              ObjectRegistry &registry,
              std::shared_ptr<ObjectType> type)
      : SBTObjectBase(context,registry,type),
        type(type)
    {}
    
    virtual std::string toString() const { return "SBTObject<"+type->toString()+">"; }
    
    std::shared_ptr<ObjectType> const type;
  };
 
 
 
  // ------------------------------------------------------------------
  // implementation section
  // ------------------------------------------------------------------
  
  inline bool SBTObjectBase::hasVariable(const std::string &name)
  {
    return type->hasVariable(name);
  }
  
  inline Variable::SP SBTObjectBase::getVariable(const std::string &name)
  {
    int varID = type->getVariableIdx(name);
    assert(varID >= 0);
    assert(varID < (int)variables.size());
    Variable::SP var = variables[varID];
    assert(var);
    return var;
  }
 
} // ::owl