vcap.hpp

/*
 * Author: Jon Trulson <jtrulson@ics.com>
 * Copyright (c) 2016 Intel Corporation.
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */
#pragma once

#include <string>
#include <iostream>

#include <errno.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <sys/mman.h>
#include <jpeglib.h>
#include <linux/videodev2.h>

#define VCAP_DEFAULT_VIDEODEV "/dev/video0"
#define VCAP_DEFAULT_OUTPUTFILE "vcap.jpg"
#define VCAP_DEFAULT_WIDTH 640
#define VCAP_DEFAULT_HEIGHT 480
#define VCAP_DEFAULT_JPEG_QUALITY 99

namespace upm {
  class VCAP {
  public:

    VCAP(std::string videoDev=VCAP_DEFAULT_VIDEODEV);

    ~VCAP();

    bool setResolution(int width, int height);

    bool captureImage();

    bool saveImage(std::string filename=VCAP_DEFAULT_OUTPUTFILE);

    int getWidth() const
    {
      return m_width;
    };

    int getHeight() const
    {
      return m_height;
    };

    void setJPGQuality(unsigned int quality);

    int getJPGQuality() const
    {
      return m_jpgQuality;
    };

    void setDebug(bool enable)
    {
      m_debugging = enable;
    };
    
  protected:
    // open the device and check that it meats minimum requirements
    bool initVideoDevice();

    // make sure device is streamable, supports mmap and capture
    bool checkCapabilities();

    // read the mmapped buffer in YUYV format and create a jpeg image
    bool YUYV2JPEG(FILE *file);

    // buffer management
    bool allocBuffer();
    void releaseBuffer();

    // does the actual capture
    bool doCaptureImage();
    
  private:
    // internal ioctl
    int xioctl(int fd, int request, void* argp);
               
    std::string m_videoDevice;
    
    // our file descriptor to the video device
    int m_fd;

    // v4l info
    struct v4l2_capability m_caps;
    struct v4l2_format m_format;

    // our mmaped buffer
    unsigned char *m_buffer;
    size_t m_bufferLen;

    // the resolution and quality
    int m_width;
    int m_height;
    int m_jpgQuality;

    // at least one image captured with current settings?
    bool m_imageCaptured;

    // are we debugging?
    bool m_debugging;
  };
}