cvimage.h

Go to the documentation of this file.

// %flair:license{
// This file is part of the Flair framework distributed under the
// CECILL-C License, Version 1.0.
// %flair:license}
#ifndef CVIMAGE_H
#define CVIMAGE_H

#include <cxcore.h>
#include <io_data.h>
#include <stdint.h>

namespace flair {
namespace core {
class cvimage : public io_data {
public:
  class Type : public DataType {
  public:
    enum class Format {
      YUYV, 
      UYVY, 
      BGR,  
      Gray, 
    };
    Type(uint16_t _width, uint16_t _height, Format _format)
        : width(_width), height(_height), format(_format) {}

    size_t GetSize() const {
      size_t pixelSize;
      switch (format) {
      case Format::Gray:
        pixelSize = 1;
        break;
      case Format::YUYV:
      case Format::UYVY:
        pixelSize = 2;
        break;
      case Format::BGR:
        pixelSize = 3;
        break;
      default:
        pixelSize = 0; // TODO should throw an exception instead
      }
      return pixelSize * width * height;
    }
    std::string GetDescription() const { return "cv image"; };

    Format GetFormat() const { return format; };
    uint16_t GetWidth() const { return width; };
    uint16_t GetHeight() const { return height; };

  private:
    uint16_t width;
    uint16_t height;
    Format format;
  };

  cvimage(const Object *parent, uint16_t width, uint16_t height,
          Type::Format format, std::string name = "", bool allocate_data = true,
          int n = 1);

  ~cvimage();

  IplImage *img;

  Type const &GetDataType() const { return dataType; };

private:
  void RawRead(char *dst) const;

  bool allocate_data;
  Type dataType;
};

} // end namespace core
} // end namespace flair

#endif // CVIMAGE_H