// This file is part of the AliceVision project.
// Copyright (c) 2019 AliceVision 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 "rgbCurve.hpp"
#include <aliceVision/image/Image.hpp>
#include <cmath>

namespace aliceVision {
namespace hdr {

struct MergingParams
{
    double minSignificantValue = 0.05;
    double maxSignificantValue = 0.995;
    float targetCameraExposure;
    int refImageIndex;
    bool computeLightMasks = false;
};

class hdrMerge
{
  public:
    /**
     * @brief
     * @param images
     * @param radiance
     * @param times
     * @param targetCameraExposure
     * @param response
     */
    void process(const std::vector<image::Image<image::RGBfColor>>& images,
                 const std::vector<double>& times,
                 const rgbCurve& weight,
                 const rgbCurve& response,
                 image::Image<image::RGBfColor>& radiance,
                 image::Image<image::RGBfColor>& lowLight,
                 image::Image<image::RGBfColor>& highLight,
                 image::Image<image::RGBfColor>& noMidLight,
                 MergingParams& mergingParams);

    void postProcessHighlight(const std::vector<image::Image<image::RGBfColor>>& images,
                              const std::vector<double>& times,
                              const rgbCurve& weight,
                              const rgbCurve& response,
                              image::Image<image::RGBfColor>& radiance,
                              float clampedValueCorrection,
                              float targetCameraExposure,
                              float highlightMaxLumimance);
};

}  // namespace hdr
}  // namespace aliceVision
