colorHarmonizeEngineGlobal.hpp

// This file is part of the AliceVision project.
// Copyright (c) 2017 AliceVision contributors.
// Copyright (c) 2013 openMVG contributors.
// This Source Code Form is subject to the terms of the Mozilla Public License,
// v. 2.0. If a copy of the MPL was not distributed with this file,
// You can obtain one at https://mozilla.org/MPL/2.0/.
 
#pragma once
 
#include <aliceVision/numeric/numeric.hpp>
#include <aliceVision/feature/feature.hpp>
#include <aliceVision/feature/RegionsPerView.hpp>
#include <aliceVision/track/TracksBuilder.hpp>
 
#include <memory>
 
namespace aliceVision {
 
enum class EHistogramSelectionMethod
{
    eHistogramHarmonizeFullFrame = 0,
    eHistogramHarmonizeMatchedPoints,
    eHistogramHarmonizeVLDSegment
};
 
inline std::string EHistogramSelectionMethod_description()
{
    return "Histogram selection method: \n"
           "* full_frame \n"
           "* matched_points \n"
           "* VLD_segments\n";
}
 
EHistogramSelectionMethod EEHistogramSelectionMethod_stringToEnum(const std::string& histogramSelectionMethod);
std::string EHistogramSelectionMethod_enumToString(const EHistogramSelectionMethod histogramSelectionMethod);
std::ostream& operator<<(std::ostream& os, EHistogramSelectionMethod p);
std::istream& operator>>(std::istream& in, EHistogramSelectionMethod& p);
 
class ColorHarmonizationEngineGlobal
{
  public:
    ColorHarmonizationEngineGlobal(const std::string& sfmDataFilename,
                                   const std::vector<std::string>& featuresFolders,
                                   const std::vector<std::string>& matchesFolders,
                                   const std::string& outputDirectory,
                                   const std::vector<feature::EImageDescriberType>& descTypes,
                                   EHistogramSelectionMethod selectionMethod,
                                   int imgRef = 0);
 
    ~ColorHarmonizationEngineGlobal();
 
    virtual bool process();
 
  private:
    EHistogramSelectionMethod _selectionMethod;
    int _imgRef;
 
    // Input data
 
    feature::RegionsPerView _regionsPerView;
    std::vector<std::string> _fileNames;
    std::vector<std::pair<size_t, size_t>> _imageSize;
    aliceVision::matching::PairwiseMatches _pairwiseMatches;
    std::vector<feature::EImageDescriberType> _descTypes;
    std::string _sfmDataFilename;
    std::vector<std::string> _matchesFolders;
    std::vector<std::string> _featuresFolders;
    std::string _outputDirectory;
 
    bool cleanGraph();
 
    bool readInputData();
};
 
}  // namespace aliceVision