Inpaint Class Reference

Public Member Functions
	Inpaint (KisPaintDeviceSP dev, KisPaintDeviceSP devMask, int _radius, QRect maskRect)
MaskedImageSP	patch (void)
MaskedImageSP	patch_simple (void)

Private Attributes
KisPaintDeviceSP	devCache
MaskedImageSP	initial
NearestNeighborFieldSP	nnf_SourceToTarget
NearestNeighborFieldSP	nnf_TargetToSource
QList< MaskedImageSP >	pyramid
int	radius

Detailed Description

Definition at line 755 of file kis_inpaint.cpp.

Constructor & Destructor Documentation

Inpaint()

Inpaint::Inpaint ( KisPaintDeviceSP dev, KisPaintDeviceSP devMask, int _radius, QRect maskRect )

inline

Definition at line 767 of file kis_inpaint.cpp.

    : devCache(dev)
    , initial(new MaskedImage(dev, devMask, maskRect))
    , radius(_radius)
    {
    }

Member Function Documentation

patch()

MaskedImageSP Inpaint::patch

(

void

)

Definition at line 779 of file kis_inpaint.cpp.

{
    MaskedImageSP source = initial->copy();
 
    pyramid.append(initial);
 
    QRect size = source->size();
 
    //qDebug() << "countMasked: " <<  source->countMasked() << "\n";
    while ((size.width() > radius) && (size.height() > radius) && source->countMasked() > 0) {
        source->downsample2x();
        //source->DebugDump("Pyramid");
        //qDebug() << "countMasked1: " <<  source->countMasked() << "\n";
        pyramid.append(source->copy());
        size = source->size();
    }
    int maxlevel = pyramid.size();
    //qDebug() << "MaxLevel: " <<  maxlevel << "\n";
 
    // The initial target is the same as the smallest source.
    // We consider that this target contains no masked pixels
    MaskedImageSP target = source->copy();
    target->clearMask();
 
    //recursively building nearest neighbor field
    for (int level = maxlevel - 1; level > 0; level--) {
        source = pyramid.at(level);
 
        if (level == maxlevel - 1) {
            //random initial guess
            nnf_TargetToSource = new NearestNeighborField(target, source, radius);
            nnf_TargetToSource->randomize();
        } else {
            // then, we use the rebuilt (upscaled) target
            // and reuse the previous NNF as initial guess
 
            NearestNeighborFieldSP new_nnf_rev = new NearestNeighborField(target, source, radius);
            new_nnf_rev->initialize(*nnf_TargetToSource);
            nnf_TargetToSource = new_nnf_rev;
        }
 
        //Build an upscaled target by EM-like algorithm (see "PatchMatch" paper referenced above - page 6)
        target = NearestNeighborField::ExpectationMaximization(nnf_TargetToSource, level, radius, pyramid);
        //target->DebugDump( "target" );
    }
    return target;
}

References MaskedImage::copy(), NearestNeighborField::ExpectationMaximization(), initial, NearestNeighborField::initialize(), nnf_TargetToSource, pyramid, radius, NearestNeighborField::randomize(), source(), and target().

patch_simple()

MaskedImageSP Inpaint::patch_simple

(

void

)

Member Data Documentation

devCache

KisPaintDeviceSP Inpaint::devCache

private

Definition at line 758 of file kis_inpaint.cpp.

initial

MaskedImageSP Inpaint::initial

private

Definition at line 759 of file kis_inpaint.cpp.

nnf_SourceToTarget

NearestNeighborFieldSP Inpaint::nnf_SourceToTarget

private

Definition at line 761 of file kis_inpaint.cpp.

nnf_TargetToSource

NearestNeighborFieldSP Inpaint::nnf_TargetToSource

private

Definition at line 760 of file kis_inpaint.cpp.

pyramid

QList<MaskedImageSP> Inpaint::pyramid

private

Definition at line 763 of file kis_inpaint.cpp.

radius

int Inpaint::radius

private

Definition at line 762 of file kis_inpaint.cpp.

---

The documentation for this class was generated from the following file:

• plugins/tools/tool_smart_patch/kis_inpaint.cpp