pub.hpp

Go to the documentation of this file.

#ifndef NGC_GATEWAY_DDT_PUB_H_
#define NGC_GATEWAY_DDT_PUB_H_
 
#include <string>
#include <mutex>
#include <concepts>
 
#define BOOST_BIND_GLOBAL_PLACEHOLDERS
#include <ddt/ddtDataPublisherApplication.hpp>
#include <ddt/ddtDataPublisher.hpp>
#include <ddt/ddtDataTransferFactory.hpp>
#include <ddt/ddtEncDecImage2D.hpp>
#include <ddt/ddtEncDecImage3D.hpp>
#include <ddt/ddtLogger.hpp>
 
#include <ngc/core/ngcdtt.hpp>
#include <ngc/gateway/base/wcs.hpp>
 
namespace ngc::gateway::ddt {

  class Publisher: public ngcdttPUBLISH {
  public:
    explicit Publisher(std::mutex &m, ::ddt::DdtLogger *logger = nullptr);
    
    explicit Publisher(std::mutex &m, int n, ::ddt::DdtLogger *logger = nullptr);
    
    virtual ~Publisher();
    

    Publisher(Publisher&) = delete;
    Publisher(Publisher&&) = delete;
    Publisher& operator=(Publisher&) = delete;
    
    virtual int Configure(const char *name, int nx, int ny) override;
    
    virtual void Shutdown() override;
    
    virtual int IdxOut() override;
    
    virtual int IdxIn() override;
    
    virtual int Locked(int idx) override;
    
    virtual int Lock(int idx) override;
    
    virtual int TryLock(int idx) override;
    
    virtual int Unlock(int idx) override;
    
    virtual int Wait(int idx) override;
    
    virtual int Publish(int idx) override;
    
    virtual void Invalidate(int idx) override;
    
  private:
    int m_registered;
    
    int m_num_buf;
    
    int m_idx_in; 
    
    int m_idx_out; 
    
    uint32_t m_image_id;
    
    int m_endian; 
    
    std::mutex &m_gw_mutex;
    
    ::ddt::DdtLogger *m_logger;
    
    std::unique_ptr<::ddt::DdtDataPublisher> m_publisher;
    
    DdtEncDecImage2D m_encoder;
    //DdtEncDecImage3D encoder_;
    
    void Init();
    
    void Free();

    template <typename T>
    void SetWcs(T& obj, ngc::gateway::base::WcsInformation wcs) {
      if constexpr (requires { obj.arcsec_pixel_x; }) {
        obj.arcsec_pixel_x = wcs.arcsec_pixel_x; 
      } 
      if constexpr (requires { obj.arcsec_pixel_y; }) {
        obj.arcsec_pixel_y = wcs.arcsec_pixel_y; 
      }
      if constexpr (requires { obj.rotation_x_axis; }) {
        obj.rotation_x_axis = wcs.rotation_x_axis; 
      }
      if constexpr (requires { obj.epoch_equinox; }) {
        obj.epoch_equinox = wcs.epoch_equinox; 
      }
      obj.reference_point_x = wcs.reference_point_x;
      obj.reference_point_y = wcs.reference_point_y;
      obj.ra_reference_point = wcs.ra_reference_point;
      obj.dec_reference_point = wcs.dec_reference_point;
      obj.cd1_1 = wcs.cd1_1;
      obj.cd1_2 = wcs.cd1_2;
      obj.cd2_1 = wcs.cd2_1;
      obj.cd2_2 = wcs.cd2_2;
      obj.type_1 = wcs.type_1;
      obj.type_2 = wcs.type_2;
    }
  };    
}
 
#endif /* NGC_GATEWAY_DTT_PUB_H_ */