drwnPixelNeighbourContrasts.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:    drwnPixelNeighbourContrasts.h
** AUTHOR(S):   Stephen Gould <stephen.gould@anu.edu.au>
**
*****************************************************************************/

#pragma once
#include "Eigen/Core"

#include "cv.h"

using namespace std;
using namespace Eigen;

// drwnPixelNeighbourContrasts class ----------------------------------------

class drwnPixelNeighbourContrasts : public drwnStdObjIface {
protected:
    MatrixXd _horzContrast;  
    MatrixXd _vertContrast;  
    MatrixXd _nwContrast;    
    MatrixXd _swContrast;    

public:
    drwnPixelNeighbourContrasts();
    drwnPixelNeighbourContrasts(const cv::Mat& img);
    virtual ~drwnPixelNeighbourContrasts();

    // i/o
    const char *type() const { return "drwnPixelNeighbourContrasts"; }
    drwnPixelNeighbourContrasts *clone() const { return new drwnPixelNeighbourContrasts(*this); }

    virtual void clear();
    virtual void initialize(const cv::Mat& img);

    bool save(drwnXMLNode& xml) const;
    bool load(drwnXMLNode& xml);

    int width() const { return _horzContrast.rows(); }
    int height() const { return _horzContrast.cols(); }

    // access
    double contrastN(int x, int y) const { return _vertContrast(x, y); }
    double contrastS(int x, int y) const { return _vertContrast(x, y + 1); }
    double contrastE(int x, int y) const { return _horzContrast(x + 1, y); }
    double contrastW(int x, int y) const { return _horzContrast(x, y); }
    double contrastNW(int x, int y) const { return _nwContrast(x, y); }
    double contrastSW(int x, int y) const { return _swContrast(x, y); }
    double contrastNE(int x, int y) const { return _swContrast(x + 1, y - 1); }
    double contrastSE(int x, int y) const { return _nwContrast(x + 1, y + 1); }

    // visualization
    cv::Mat visualize(bool bComposite = true) const;

protected:
    static double pixelContrast(const cv::Mat& img, const cv::Point &p, const cv::Point& q);
};