DistributedCachingArnoldInterface.h

Go to the documentation of this file.

#ifndef _Arnold_DISTRIBUTED_CACHE_INTERFACE_H_
#define _Arnold_DISTRIBUTED_CACHE_INTERFACE_H_

#pragma once

#include "LumiverseCoreConfig.h"

#ifdef USE_ARNOLD

#ifdef USE_DUMIVERSE

#ifdef USE_ARNOLD_CACHING

#include "ArnoldInterface.h"
#include <ai.h>
#include "ArnoldParameterVector.h"
#include "Compositor.h"
#include "ToneMapper.h"
#include "EXRLayer.h"
#include <thread>
#include <algorithm>
#include <unordered_map>
#include <set>
#include "CachingArnoldInterface.h"
#include "DistributedArnoldInterface.h"

namespace Lumiverse {

  class DistributedCachingArnoldInterface : public DistributedArnoldInterface, public CachingArnoldInterface
  {
  public:
    DistributedCachingArnoldInterface(string host, int port, string outputPath) : DistributedArnoldInterface(host, port, outputPath) {}

    void init(const JSONNode jsonPatch) override;

    void close() override;

    int render(const std::set<Device *> &devices);

    bool setDims(int w, int h) override;

    virtual float getPercentage() override {
      return CachingArnoldInterface::getPercentage();
    }

    void setSamples(int samples) override;

    void setOptionParameter(const std::string &paramName, int val) override;
    void setOptionParameter(const std::string &paramName, float val) override;

    float *getBufferPointer() override;

  protected:
    Compositor compositor;
    
    void dumpHDRToBuffer(const std::set<Device *> &devices);
    void updateDevicesLayers(const std::set<Device *> &devices) override;
    void setHDROutputBuffer() override;
  };
}

#endif // USE_ARNOLD_CACHING

#endif // USE_DUMIVERSE

#endif // USE_ARNOLD

#endif // _Arnold_CACHE_INTERFACE_H_