drwnImageInPainter.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:    drwnImageInPainter.h
** AUTHOR(S):   Robin Liang <robin.gnail@gmail.com>
**              Stephen Gould <stephen.gould@anu.edu.au>
**
*****************************************************************************/

#pragma once

// c++ standard headers
#include <cstdlib>
#include <cstdio>

// opencv library headers
#include "cv.h"

// darwin library headers
#include "drwnBase.h"
#include "drwnIO.h"
#include "drwnVision.h"

using namespace std;
using namespace Eigen;

// drwnImageInPainter --------------------------------------------------------

class drwnImageInPainter {
public:
    static unsigned DEFAULT_RADIUS; 
    static unsigned UPDATE_STEPS;   
    static bool PIXELWISE;          
    static bool PRIORITY_FILLING;   

public:
    unsigned patchRadius;   
    bool bAllowFromFilled;  
    double alphaBlend;      
    bool bVisualize;        

public:
    drwnImageInPainter(unsigned _patchRadius = DEFAULT_RADIUS, bool _bAllowFromFilled = false) :
        patchRadius(_patchRadius), bAllowFromFilled(_bAllowFromFilled),
        alphaBlend(0.125), bVisualize(false) {
        // do nothing
    }
    virtual ~drwnImageInPainter() { /* do nothing */ }

    cv::Mat fill(const cv::Mat& image, const cv::Mat& fillMask) const {
        return fill(image, fillMask, fillMask == 0);
    }
    cv::Mat fill(const cv::Mat& image, const cv::Mat& fillMask, const cv::Mat& copyMask) const {
        return fill(image, fillMask, copyMask, fillMask);
    }
    cv::Mat fill(const cv::Mat& image, const cv::Mat& fillMask, const cv::Mat& copyMask,
        const cv::Mat& ignoreMask) const;

    virtual void monitor(const cv::Mat& image, const cv::Mat& mask) const;

protected:
    virtual cv::Mat featurizeImage(const cv::Mat& image) const;
    double computePixelPriority(const cv::Point& p, const cv::Mat& image, const cv::Mat& mask) const;
    static cv::Mat extractMaskBoundary(const cv::Mat& mask);
    static void blendMaskedImage(cv::Mat img, const cv::Mat& src, const cv::Mat& mask, double alpha);
};