TrackerFeatures.h

Go to the documentation of this file.

/*
 * This file is part of ALVAR, A Library for Virtual and Augmented Reality.
 *
 * Copyright 2007-2012 VTT Technical Research Centre of Finland
 *
 * Contact: VTT Augmented Reality Team <alvar.info@vtt.fi>
 *          <http://www.vtt.fi/multimedia/alvar.html>
 *
 * ALVAR 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 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library 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 ALVAR; if not, see
 * <http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html>.
 */

#ifndef TRACKERFEATURES_H
#define TRACKERFEATURES_H

#include "Tracker.h"

namespace alvar {

class ALVAR_EXPORT TrackerFeatures : public Tracker {
protected:
        int x_res, y_res;
        int frame_count;
        double quality_level;
        double min_distance;
        int min_features;
        int max_features;
        char *status;
        IplImage *img_eig;
        IplImage *img_tmp;
        IplImage *gray;
        IplImage *prev_gray;
        IplImage *pyramid;
        IplImage *prev_pyramid;
        IplImage *mask;
        int next_id;
        int win_size;
        int pyr_levels;

        double TrackHid(IplImage *img, IplImage *mask=NULL, bool add_features=true);
public:
        CvPoint2D32f *prev_features;
        CvPoint2D32f *features;
        int prev_feature_count;
        int feature_count;
        int *prev_ids;
        int *ids;
        TrackerFeatures(int _max_features=100, int _min_features=90, double _quality_level=0.01, double _min_distance=10, int _pyr_levels=1, int _win_size=3);
        ~TrackerFeatures(); 
        void ChangeSettings(int _max_features=100, int _min_features=90, double _quality_level=0.01, double _min_distance=10);
        void Reset(); 
        double Reset(IplImage *img, IplImage *mask); 
        bool DelFeature(int index);
        bool DelFeatureId(int id);
        double Track(IplImage *img) { return Track(img, true); }
        double Track(IplImage *img, bool add_features); 
    double Track(IplImage* img, IplImage* mask);
        int AddFeatures(IplImage *mask=NULL);
        IplImage *NewFeatureMask();
        int Purge();
};

} // namespace alvar

#endif