Frame.hpp

Go to the documentation of this file.

/*********************************************************************************
 *  OKVIS - Open Keyframe-based Visual-Inertial SLAM
 *  Copyright (c) 2015, Autonomous Systems Lab / ETH Zurich
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 * 
 *   * Redistributions of source code must retain the above copyright notice,
 *     this list of conditions and the following disclaimer.
 *   * Redistributions in binary form must reproduce the above copyright notice,
 *     this list of conditions and the following disclaimer in the documentation
 *     and/or other materials provided with the distribution.
 *   * Neither the name of Autonomous Systems Lab / ETH Zurich nor the names of
 *     its contributors may be used to endorse or promote products derived from
 *     this software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 *  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 *  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 *  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 *  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 *  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 *  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *  POSSIBILITY OF SUCH DAMAGE.
 *
 *  Created on: Mar 30, 2015
 *      Author: Stefan Leutenegger (s.leutenegger@imperial.ac.uk)
 *    Modified: Andreas Forster (an.forster@gmail.com)
 *********************************************************************************/

#ifndef INCLUDE_OKVIS_FRAME_HPP_
#define INCLUDE_OKVIS_FRAME_HPP_

#include <Eigen/StdVector>
#include <Eigen/Core>
#include <memory>
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wnon-virtual-dtor"
#include <opencv2/core/core.hpp> // Code that causes warning goes here
#pragma GCC diagnostic pop
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Woverloaded-virtual"
#include <opencv2/features2d/features2d.hpp> // Code that causes warning goes here
#pragma GCC diagnostic pop
#include <okvis/Time.hpp>
#include <okvis/assert_macros.hpp>
#include "okvis/cameras/CameraBase.hpp"

namespace okvis {

class Frame
{
 public:
  EIGEN_MAKE_ALIGNED_OPERATOR_NEW
  OKVIS_DEFINE_EXCEPTION(Exception, std::runtime_error)

  
  inline Frame()
  {
  }

  inline Frame(const cv::Mat & image,
        std::shared_ptr<cameras::CameraBase> & cameraGeometry,
        std::shared_ptr<cv::FeatureDetector> & detector,
        std::shared_ptr<cv::DescriptorExtractor> & extractor);

  inline virtual ~Frame()
  {
  }

  inline void setImage(const cv::Mat & image);

  inline void setGeometry(std::shared_ptr<const cameras::CameraBase> cameraGeometry);

  inline void setDetector(std::shared_ptr<cv::FeatureDetector> detector);

  inline void setExtractor(std::shared_ptr<cv::DescriptorExtractor> extractor);

  inline const cv::Mat & image() const;

  inline std::shared_ptr<const cameras::CameraBase> geometry() const;

  template<class GEOMETRY_T>
  inline std::shared_ptr<const GEOMETRY_T> geometryAs() const;

  inline int detect();

  inline int describe(
      const Eigen::Vector3d & extractionDirection = Eigen::Vector3d(0, 0, 1));

  template<class GEOMETRY_T>
  inline int describeAs(
      const Eigen::Vector3d & extractionDirection = Eigen::Vector3d(0, 0, 1));

  inline bool getCvKeypoint(size_t keypointIdx, cv::KeyPoint & keypoint) const;

  inline bool getKeypoint(size_t keypointIdx, Eigen::Vector2d & keypoint) const;

  inline bool getKeypointSize(size_t keypointIdx, double & keypointSize) const;

  inline const unsigned char * keypointDescriptor(size_t keypointIdx);

  inline bool setLandmarkId(size_t keypointIdx, uint64_t landmarkId);

  inline uint64_t landmarkId(size_t keypointIdx) const;

  inline bool resetKeypoints(const std::vector<cv::KeyPoint> & keypoints);

  inline bool resetDescriptors(const cv::Mat & descriptors);

  inline size_t numKeypoints() const;

 protected:
  cv::Mat image_;  
  std::shared_ptr<const cameras::CameraBase> cameraGeometry_;  
  std::shared_ptr<cv::FeatureDetector> detector_;  
  std::shared_ptr<cv::DescriptorExtractor> extractor_;  
  std::vector<cv::KeyPoint> keypoints_;  
  cv::Mat descriptors_;  
  std::vector<uint64_t> landmarkIds_;  
};

}  // namespace okvis

#include "implementation/Frame.hpp"

#endif /* INCLUDE_OKVIS_FRAME_HPP_ */