ArnoldMemoryFrameManager.h

Go to the documentation of this file.

#ifndef _ArnoldMemoryFRAMEMANAGER_H_
#define _ArnoldMemoryFRAMEMANAGER_H_

#pragma once

#include "LumiverseCoreConfig.h"
#ifdef USE_ARNOLD

#include "../lib/libjson/libjson.h"
#include "ArnoldFrameManager.h"

#include <mutex>
#include <functional>
#include <set>
#include <chrono>
#include <iostream>

namespace Lumiverse {
    struct FrameData {
  // The time point
  time_t time;
  // RGBA frame buffer
  float *buffer;

  // To make the ordered set.
  bool operator<(FrameData other) const {
      return time < other.time;
  }
    };

    class ArnoldMemoryFrameManager : public ArnoldFrameManager
    {
      public:
  ArnoldMemoryFrameManager(int fps = 24) :
    ArnoldFrameManager(fps), m_bufferSet([](FrameData l, FrameData r) { return l < r; }) { }

  virtual ~ArnoldMemoryFrameManager();

  virtual void dump(time_t time, float *frame, size_t width, size_t height);

  virtual float *getCurrentFrameBuffer();

  virtual time_t getCurrentTime() const;

  virtual time_t getNextTime() const;
      
  virtual bool hasNext() const;
        
    virtual bool isEmpty() const { return m_bufferSet.size() == 0; }

    virtual size_t getFrameNum() const { return m_bufferSet.size(); }
        
  virtual void clear();

  virtual std::string getType() { return "ArnoldMemoryFrameManager"; }

      private:
  // A set for frame data ordered by time ascendingly.
  std::set<FrameData, std::function<bool(FrameData, FrameData)>> m_bufferSet;
  // Lock for the set.
  std::mutex m_buffer;
    };
    
}

#endif

#endif