DetectionFrameImages.h

Go to the documentation of this file.

 
#ifndef _SEIMPLEMENTATION_PLUGIN_DETECTIONFRAMEIMAGES_DETECTIONFRAMEIMAGES_H_
#define _SEIMPLEMENTATION_PLUGIN_DETECTIONFRAMEIMAGES_DETECTIONFRAMEIMAGES_H_
 
#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/ImageAccessor.h"
#include "SEFramework/Image/ImageChunk.h"
#include "SEFramework/Property/Property.h"
#include "SEFramework/Frame/Frame.h"
 
 
namespace SourceXtractor {
 
class DetectionFrameImages : public Property {
 
public:
  virtual ~DetectionFrameImages() = default;
 
  DetectionFrameImages(std::shared_ptr<DetectionImageFrame> frame, int width, int height)
    : m_width(width), m_height(height), m_frame(frame) {}
 
  std::shared_ptr<ImageAccessor<SeFloat>> getLockedImage(FrameImageLayer layer) const {
    return std::make_shared<ImageAccessor<SeFloat>>(m_frame->getImage(layer));
  }
 
  std::shared_ptr<ImageChunk<DetectionImage::PixelType>> getImageChunk(FrameImageLayer layer, int x, int y, int width, int height) const {
    return m_frame->getImage(layer)->getChunk(x, y, width, height);
  }
 
  int getWidth() const {
    return m_width;
  }
 
  int getHeight() const {
    return m_height;
  }
 
private:
  int m_width;
  int m_height;
  std::shared_ptr<DetectionImageFrame> m_frame;
};
 
}
 
#endif /* _SEIMPLEMENTATION_PLUGIN_DETECTIONFRAMEIMAGES_DETECTIONFRAMEIMAGES_H_ */