TrackingPointSelector.h

Go to the documentation of this file.

/*
 *  TrackingPointSelector.h
 *  artoolkitX
 *
 *  This file is part of artoolkitX.
 *
 *  artoolkitX is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  artoolkitX is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with artoolkitX.  If not, see <http://www.gnu.org/licenses/>.
 *
 *  As a special exception, the copyright holders of this library give you
 *  permission to link this library with independent modules to produce an
 *  executable, regardless of the license terms of these independent modules, and to
 *  copy and distribute the resulting executable under terms of your choice,
 *  provided that you also meet, for each linked independent module, the terms and
 *  conditions of the license of that module. An independent module is a module
 *  which is neither derived from nor based on this library. If you modify this
 *  library, you may extend this exception to your version of the library, but you
 *  are not obligated to do so. If you do not wish to do so, delete this exception
 *  statement from your version.
 *
 *  Copyright 2018 Realmax, Inc.
 *  Copyright 2015 Daqri, LLC.
 *  Copyright 2010-2015 ARToolworks, Inc.
 *
 *  Author(s): Philip Lamb, Daniel Bell.
 *
 */
 
#ifndef TRACKINGPOINTSELECTOR_H
#define TRACKINGPOINTSELECTOR_H
#include <vector>
#include <map>
#include "OCVConfig.h"
#include "TrackedPoint.h"
 
class TrackingPointSelector
{
public:
    OCV_EXTERN TrackingPointSelector();
    
    OCV_EXTERN TrackingPointSelector(std::vector<cv::Point2f> pts, int width, int height, int markerTemplateWidth);
    
    void DistributeBins(int width, int height, int markerTemplateWidth);
    
    void SetHomography(cv::Mat newHomography);
    
    cv::Mat GetHomography();
    
    void UpdatePointStatus(std::vector<uchar> status);
 
    void SelectPoints();
    
    std::vector<cv::Point2f> GetSelectedFeatures();
    
    std::vector<cv::Point2f> GetTrackedFeatures();
    
    std::vector<cv::Point3f> GetSelectedFeatures3d();
    
    std::vector<cv::Point2f> GetSelectedFeaturesWarped();
 
    OCV_EXTERN std::vector<cv::Point2f> GetAllFeatures();
    
    OCV_EXTERN void CleanUp();
 
private:
    std::vector<TrackedPoint> _selectedPts;
    std::vector<cv::Point2f> _pts;
    std::map<int, std::vector<TrackedPoint> > trackingPointBin;
    cv::Mat homography;
 
};
#endif //TRACKINGPOINTSELECTOR