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

#pragma once

#include <cstdlib>
#include <cassert>
#include <set>

#include "drwnBase.h"
#include "drwnIO.h"
#include "drwnVision.h"

#include "drwnPatchMatch.h"

using namespace std;

// drwnPatchMatchUtils -------------------------------------------------------

namespace drwnPatchMatchUtils {

    inline cv::Point transformPixel(const cv::Point& srcPixel,
        const cv::Size& srcSize, const cv::Size& dstSize)
    {
        return cv::Point(srcPixel.x * dstSize.width / srcSize.width,
            srcPixel.y * dstSize.height / dstSize.height);
    }

    inline cv::Rect transformRect(const cv::Rect& srcRect,
        const cv::Size& srcSize, const cv::Size& dstSize)
    {
        cv::Rect dstRect;

        dstRect.x = srcRect.x * dstSize.width / srcSize.width;
        dstRect.y = srcRect.y * dstSize.height / srcSize.height;
        dstRect.width = srcRect.width * dstSize.width / srcSize.width;
        dstRect.height = srcRect.height * dstSize.height / srcSize.height;

        return dstRect;
    }

    void loadLabels(const set<string>& baseNames, map<string, MatrixXi>& labels);

    vector<cv::Point> sortPixelsByVariance(const cv::Mat& img, const cv::Size& patchSize);

    size_t countMatchablePixels(const drwnPatchMatchGraph& graph);

    double overlap(const drwnPatchMatchEdgeList& edgesA,
        const drwnPatchMatchEdgeList& edgesB);

    bool isValidNode(const drwnPatchMatchGraph& graph, const drwnPatchMatchNode& node);
};