//
//  Created by Samuel Gateau on 2018/12/06
//  Copyright 2013-2018 High Fidelity, Inc.
//
//  Distributed under the Apache License, Version 2.0.
//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
//

#ifndef hifi_AssembleLightingStageTask_h
#define hifi_AssembleLightingStageTask_h

#include <render/RenderFetchCullSortTask.h>
#include "LightingModel.h"

#include "LightStage.h"
#include "BackgroundStage.h"
#include "HazeStage.h"
#include "BloomStage.h"
#include "TonemappingStage.h"
#include "AmbientOcclusionStage.h"
#include "NormalMapAttenuationStage.h"

#include "ZoneRenderer.h"


class FetchCurrentFrames {
public:
    using Output = render::VaryingSet7<LightStage::FramePointer, BackgroundStage::FramePointer, HazeStage::FramePointer,
        BloomStage::FramePointer, TonemappingStage::FramePointer, AmbientOcclusionStage::FramePointer, NormalMapAttenuationStage::FramePointer>;
    using JobModel = render::Job::ModelO<FetchCurrentFrames, Output>;

    FetchCurrentFrames() {}

    void run(const render::RenderContextPointer& renderContext, Output& output);
};

class AssembleLightingStageTask {
public:
    using Input = RenderFetchCullSortTask::Output;
    using Output = render::VaryingSet2<FetchCurrentFrames::Output, ZoneRendererTask::Output>;
    using JobModel = render::Task::ModelIO<AssembleLightingStageTask, Input, Output, render::Task::Config>;

    AssembleLightingStageTask() {}

    void build(JobModel& task, const render::Varying& input, render::Varying& output);
};

#endif