ObjectRegistry.h

// ======================================================================== //
// Copyright 2019 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 "Object.h"
 
namespace owl {
 
  struct RegisteredObject;
  struct Context;
  struct ObjectRegistry;
 
  struct ObjectRegistry {
    // ReallocContextIDsCB reallocContextIDs,
    // const char *typeDescription);
    inline size_t size()  const { return objects.size(); }
    inline bool   empty() const { return objects.empty(); }
 
    // virtual void reallocContextIDs(int newMaxIDs) = 0;
    
    void forget(RegisteredObject *object);
    void track(RegisteredObject *object);
    int allocID();
    RegisteredObject *getPtr(size_t ID);
  private:
    std::vector<RegisteredObject *> objects;
 
    int numIDsAllocedInContext = 0;
    
    std::stack<int> previouslyReleasedIDs;
    std::mutex mutex;
  };
 
 
  template<typename T>
  struct ObjectRegistryT : public ObjectRegistry {
    ObjectRegistryT(Context *context)
      : context(context)
    {};
      
    // void reallocContextIDs(int newMaxIDs) override;
    
    inline T* getPtr(size_t ID)
    {
        return (T*)ObjectRegistry::getPtr(ID);
    }
 
    inline typename T::SP getSP(size_t ID)
    {
      T *ptr = getPtr(ID);
      if (!ptr) return {};
      Object::SP object = ptr->shared_from_this();
      assert(object);
      return object->as<T>();
    }
    
    Context *const context;
  };
    
} // ::owl