InpaintingConverger.cpp

//
// Created by Jonathan on 23/02/2016.
//

#include "InpaintingConverger.h"

InpaintingConverger::InpaintingConverger(int const downsamplingLevels, int const downsamplingRate,
                     int const nbIterations, float const residualThreshold, InpaintingSearcher * const iSearcher,
                     InpaintingReconstructor * const iReconstructor, InpaintingEvaluator * const iEvaluator)
        : downsamplingLevels(downsamplingLevels), downsamplingRate(downsamplingRate),
          nbIterations(nbIterations), residualThreshold(residualThreshold),
          iSearcher(iSearcher), iReconstructor(iReconstructor), iEvaluator(iEvaluator)
{}

InpaintingConverger::~InpaintingConverger() {}

void InpaintingConverger::Inpaint(InpaintingDataSource * const iDataSource, InpaintingProfiler * const iProfiler) {
    IDSDDownsampling downsampledDataSource = IDSDDownsampling(iDataSource, downsamplingLevels, downsamplingRate);

    InpaintingOpenCVIO openCVIO = InpaintingOpenCVIO();

    for(int curLevel = downsamplingLevels - 1; curLevel >= 0; curLevel--){
        iProfiler->BeforeInpaintLevel(curLevel);

        downsampledDataSource.SetCurLevel(curLevel);
        IDSDDetailedOcclusion detailedOcclusionDataSource = IDSDDetailedOcclusion(&downsampledDataSource,\
          iEvaluator->GetPatchDim().GetSizeX(), iEvaluator->GetPatchDim().GetSizeY(),\
          iEvaluator->GetPatchDim().GetSizeZ());

        if(curLevel < downsamplingLevels - 1)
            iReconstructor->Reconstruct(&detailedOcclusionDataSource, false);

        float residual = FLT_MAX;
        iProfiler->BeforeInpaintingIterations(curLevel);
        for(int iteration = 0; iteration < nbIterations && residual > residualThreshold; iteration++){
            iProfiler->BeforeInpaintingIteration(curLevel, iteration);
            if(curLevel == downsamplingLevels-1 && iteration == 0)
            {
              OnionPeelInitialisation(&downsampledDataSource, iProfiler);
              // LoadingInitialisation(&downsampledDataSource,iProfiler);
            }
            else{
                iSearcher->Search(&detailedOcclusionDataSource, false);

                residual = iReconstructor->Reconstruct(&detailedOcclusionDataSource, false);
            }
            iProfiler->AfterInpaintingIteration(curLevel, iteration);
        }

        if(curLevel != 0)
            //downsampledDataSource.UpsampleShiftMatrices();
            downsampledDataSource.MatlabUpsampleShiftMatrix();

        iProfiler->AfterInpaintingIterations(curLevel);
        iProfiler->AfterInpaintLevel(curLevel);
    }
}

void InpaintingConverger::OnionPeelInitialisation(InpaintingDataSource * const iDataSource, InpaintingProfiler * const iProfiler) {
    iProfiler->BeforeInitialisation();

    IDSDDetailedOcclusionPeeling peelingDataSource = \
    IDSDDetailedOcclusionPeeling(iDataSource, iEvaluator->GetPatchDim().GetSizeX(),\
    iEvaluator->GetPatchDim().GetSizeY(), iEvaluator->GetPatchDim().GetSizeZ());

    InpaintingOpenCVIO openCVIO = InpaintingOpenCVIO();

    // Slowly peel back the layers!
    int peelNo = 0;
    do{
        iProfiler->BeforeInitialisationPeel(peelNo);


        iSearcher->Search(&peelingDataSource, true);
        iReconstructor->Reconstruct(&peelingDataSource, true);

        iProfiler->AfterInitialisationPeel(peelNo++);
    }while(peelingDataSource.Peel());

    iProfiler->AfterInitialisation();
}


void InpaintingConverger::LoadingInitialisation(InpaintingDataSource * const iDataSource,InpaintingProfiler * const iProfiler){
  iProfiler->BeforeInitialisation();
  InpaintingOpenCVIO openCVIO = InpaintingOpenCVIO();

  openCVIO.LoadRGBMatrixFromVideo("Data/onion_debug/peel_compare/umbrella_after_peel.avi",\
  iDataSource->GetRed(), iDataSource->GetGreen(), iDataSource->GetBlue());

  PaddedMatrix<int> * const frame_x = new PaddedMatrix<int>(iDataSource->GetDim());
  PaddedMatrix<int> * const frame_y = new PaddedMatrix<int>(iDataSource->GetDim());
  PaddedMatrix<int> * const frame_z = new PaddedMatrix<int>(iDataSource->GetDim());
  PaddedMatrix<float> * const frame_a = new PaddedMatrix<float>(iDataSource->GetDim());
  PaddedMatrix<float> * const frame_gx = new PaddedMatrix<float>(iDataSource->GetDim());
  PaddedMatrix<float> * const frame_gy = new PaddedMatrix<float>(iDataSource->GetDim());

  for(int nbFrame=0;nbFrame<iDataSource->GetDim()->GetSizeZ();nbFrame++)
  {
    openCVIO.LoadMatrixFromCSV("Data/onion_debug/peel_compare/end_peel_disp/M_EndPeel_DispX_"\
    +std::to_string(nbFrame)+".csv",frame_x,1);
    iDataSource->GetDispX()->SetFrame(frame_x,nbFrame);

    openCVIO.LoadMatrixFromCSV("Data/onion_debug/peel_compare/end_peel_disp/M_EndPeel_DispY_"\
    +std::to_string(nbFrame)+".csv",frame_y,1);
    iDataSource->GetDispY()->SetFrame(frame_y,nbFrame);

    openCVIO.LoadMatrixFromCSV("Data/onion_debug/peel_compare/end_peel_disp/M_EndPeel_DispZ_"\
    +std::to_string(nbFrame)+".csv",frame_z,1);
    iDataSource->GetDispZ()->SetFrame(frame_z,nbFrame);

    openCVIO.LoadMatrixFromCSV("Data/onion_debug/peel_compare/end_peel_disp/M_EndPeel_DispA_"\
    +std::to_string(nbFrame)+".csv",frame_a,1);
    iDataSource->GetDispA()->SetFrame(frame_a,nbFrame);

    openCVIO.LoadMatrixFromCSV("Data/onion_debug/peel_compare/end_peel_disp/M_EndPeel_GradX_"\
    +std::to_string(nbFrame)+".csv",frame_gx,1);
    iDataSource->GetGradX()->SetFrame(frame_gx,nbFrame);

    openCVIO.LoadMatrixFromCSV("Data/onion_debug/peel_compare/end_peel_disp/M_EndPeel_GradY_"\
    +std::to_string(nbFrame)+".csv",frame_gy,1);
    iDataSource->GetGradY()->SetFrame(frame_gy,nbFrame);

  }

  //iDataSource->GetDispX()->GetFrame(27)->Mprint();

  iProfiler->AfterInitialisation();

}