drwnTemplateMatcher.h

/*****************************************************************************
** DARWIN: A FRAMEWORK FOR MACHINE LEARNING RESEARCH AND DEVELOPMENT
** Distributed under the terms of the BSD license (see the LICENSE file)
** Copyright (c) 2007-2015, Stephen Gould
** All rights reserved.
**
******************************************************************************
** FILENAME:    drwnTemplateMatcher.h
** AUTHOR(S):   Stephen Gould <stephen.gould@anu.edu.au>
**
*****************************************************************************/

#pragma once

#include <list>
#include <vector>

#include "Eigen/Core"

#include "cv.h"

using namespace std;
using namespace Eigen;

// drwnTemplateMatcher ---------------------------------------------------------

class drwnTemplateMatcher {
 protected:
    vector<cv::Mat> _templates;    
    vector<pair<double, double> > _templateStats; 
    cv::Size _largestTemplate;     

    // buffered data (clear with reset())
    cv::Size _imgSize;             
    vector<cv::Mat> _dftTemplates; 
    cv::Mat _dftImage;             
    cv::Mat _dftResponse;          
    cv::Mat _imgSum;               
    cv::Mat _imgSqSum;             

 public:
    drwnTemplateMatcher();
    drwnTemplateMatcher(const drwnTemplateMatcher& tm);
    ~drwnTemplateMatcher();

    void clear();
    inline bool empty() const { return _templates.empty(); }
    inline int size() const { return (int)_templates.size(); }
    inline int width() const { return _largestTemplate.width; }
    inline int height() const { return _largestTemplate.height; }

    void reset();

    void addTemplate(cv::Mat& t);
    void addTemplates(vector<cv::Mat>& t);
    void copyTemplate(const cv::Mat& t);
    void copyTemplates(const vector<cv::Mat>& t);

    vector<cv::Mat> responses(const cv::Mat& img, int method = CV_TM_CCORR);
    vector<cv::Mat> responses(int method = CV_TM_CCORR);
    cv::Mat response(const cv::Mat& img, unsigned tid, int method = CV_TM_CCORR);
    cv::Mat response(unsigned tid, int method = CV_TM_CCORR);

    // operators
    drwnTemplateMatcher& operator=(const drwnTemplateMatcher& tm);
    const cv::Mat& operator[](unsigned i) const;

 protected:
    void cacheDFTAndIntegrals(const cv::Mat& image);
};