vtkOSPRayCache.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkOSPRayCache_h
#define vtkOSPRayCache_h
 
#include "vtkRenderingRayTracingModule.h" // For export macro
#include "vtkSystemIncludes.h"            //dll warning suppression
#include <map>                            // for stl
#include <memory>
 
#include "RTWrapper/RTWrapper.h" // for handle types
 
VTK_ABI_NAMESPACE_BEGIN
template <class T>
class VTKRENDERINGRAYTRACING_EXPORT vtkOSPRayCache
{
public:
  vtkOSPRayCache() { this->Size = 0; }
 
  ~vtkOSPRayCache() { this->Empty(); }
 
  void Set(double tstep, std::shared_ptr<T> payload)
  {
    if (this->Contents.size() >= this->Size)
    {
      return;
    }
    this->Contents[tstep] = payload;
  }
 
  std::shared_ptr<T> Get(double tstep)
  {
    auto ret = this->Contents.find(tstep);
    if (ret != this->Contents.end())
    {
      return ret->second;
    }
    return nullptr;
  }
 
 
  void SetSize(size_t sz)
  {
    if (sz == this->Size)
    {
      return;
    }
    if (sz < this->Size)
    {
      this->Empty();
    }
    this->Size = sz;
  }
  size_t GetSize() { return this->Size; }
 
  bool Contains(double tstep) { return this->Get(tstep) != nullptr; }
 
  bool HasRoom() { return this->Contents.size() < this->Size; }
 
private:
  // deletes all of the content in the cache
  void Empty()
  {
    this->Contents.clear();
    this->Size = 0;
  }
 
  size_t Size;
 
  std::map<double, std::shared_ptr<T>> Contents;
};
 
class vtkOSPRayCacheItemObject
{
public:
  vtkOSPRayCacheItemObject(RTW::Backend* be, OSPObject obj)
    : backend(be)
  {
    object = obj;
  }
  ~vtkOSPRayCacheItemObject() { ospRelease(object); }
  OSPObject object{ nullptr };
  size_t size{ 0 };
  RTW::Backend* backend = nullptr;
};
 
VTK_ABI_NAMESPACE_END
#endif // vtkOSPRayCache_h
// VTK-HeaderTest-Exclude: vtkOSPRayCache.h