AstDetector.cc

Go to the documentation of this file.

//
//    AstDetector - the interface class for the AGAST corner detector
//
//    Copyright (C) 2010  Elmar Mair
//
//    This program is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.
//
//    This program 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 General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with this program.  If not, see <http://www.gnu.org/licenses/>.

#include "AstDetector.h"
#include "cvWrapper.h"

using namespace std;
using namespace agast;

void AstDetector::score(const unsigned char* i, const std::vector<CvPoint>& corners_all)
{
        unsigned int n=0;
    unsigned int num_corners=corners_all.size();

        if(num_corners > scores.capacity())
        {
                if(scores.capacity()==0)
                {
                        scores.reserve(512 > num_corners ? 512 : num_corners);
                }
                else
                {
                        unsigned int nScores = scores.capacity()*2;
                        if(num_corners > nScores)
                                nScores = num_corners;
                        scores.reserve(nScores);
                }
        }

    scores.resize(num_corners);

    for(; n < num_corners; n++)
        scores[n] = cornerScore(i + corners_all[n].y*xsize + corners_all[n].x);
}

void AstDetector::nms(const unsigned char* im, const std::vector<CvPoint>& corners_all,
                std::vector<CvPoint>& corners_nms)
{
        score(im,corners_all);
        nonMaximumSuppression(corners_all, corners_nms);
}