/***

  Olive - Non-Linear Video Editor
  Copyright (C) 2019 Olive Team

  This program is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program.  If not, see <http://www.gnu.org/licenses/>.

***/

#include "render.h"

#include "common/timecodefunctions.h"
#include "render/backend/opengl/openglbackend.h"

OLIVE_NAMESPACE_ENTER

RenderTask::RenderTask(ViewerOutput* viewer, const VideoRenderingParams &vparams, const AudioRenderingParams &aparams) :
  viewer_(viewer),
  video_params_(vparams),
  audio_params_(aparams)
{
}

struct TimeHashFuturePair {
  rational time;
  QFuture<QByteArray> hash_future;
};

struct HashTimePair {
  rational time;
  QByteArray hash;
};

struct HashFrameFuturePair {
  QByteArray hash;
  QFuture<FramePtr> frame_future;
};

struct RangeSampleFuturePair {
  TimeRange range;
  QFuture<SampleBufferPtr> sample_future;
};

struct HashDownloadFuturePair {
  QByteArray hash;
  QFuture<void> download_future;
};

void RenderTask::Render(const TimeRangeList& range_to_cache,
                        const QMatrix4x4& mat,
                        bool audio_enabled)
{
  OpenGLBackend backend;

  backend.SetAutomaticAudio(false);
  backend.SetViewerNode(viewer_);
  backend.SetVideoParams(video_params_);
  backend.SetAudioParams(audio_params_);
  backend.SetVideoDownloadMatrix(mat);

  // Get hashes for each frame
  QLinkedList<TimeHashFuturePair> hash_list;

  {
    TimeRangeList range_list = range_to_cache;
    while (!range_list.isEmpty()) {
      const TimeRange& range = range_list.first();

      const rational& timebase = viewer_->video_params().time_base();

      rational time = range.in();
      rational snapped = Timecode::snap_time_to_timebase(time, timebase);
      rational next;

      if (snapped > time) {
        next = snapped;
        snapped -= timebase;
      } else {
        next = snapped + timebase;
      }

      hash_list.append({snapped, backend.Hash(snapped, false)});
      range_list.RemoveTimeRange(TimeRange(snapped, next));
    }
  }

  // Determine any duplicates
  QMap< QByteArray, QLinkedList<rational> > times_to_render;
  foreach (const TimeHashFuturePair& i, hash_list) {
    times_to_render[i.hash_future.result()].append(i.time);
  }

  // Render all frames necessary
  QLinkedList<HashFrameFuturePair> render_lookup_table;
  {
    QLinkedList<HashTimePair> sorted_times;
    QLinkedList<HashTimePair>::iterator sorted_iterator;

    // Rendering is more efficient if we cache in order
    QMap< QByteArray, QLinkedList<rational> >::const_iterator i;
    for (i=times_to_render.constBegin(); i!=times_to_render.constEnd(); i++) {
      const QByteArray& hash = i.key();
      const rational& time = i.value().first();

      bool inserted = false;

      for (sorted_iterator=sorted_times.begin(); sorted_iterator!=sorted_times.end(); sorted_iterator++) {
        if (sorted_iterator->time > time) {
          sorted_times.insert(sorted_iterator, {time, hash});
          inserted = true;
          break;
        }
      }

      if (!inserted) {
        sorted_times.append({time, hash});
      }
    }

    foreach (const HashTimePair& p, sorted_times) {
      render_lookup_table.append({p.hash, backend.RenderFrame(p.time, false, false)});
    }
  }

  QLinkedList<RangeSampleFuturePair> audio_lookup_table;
  if (audio_enabled) {
    foreach (const TimeRange& r, range_to_cache) {
      QList<TimeRange> ranges = RenderBackend::SplitRangeIntoChunks(r);

      foreach (const TimeRange& split, ranges) {
        audio_lookup_table.append({split, backend.RenderAudio(split, false)});
      }
    }
  }

  // Start downloading frames that have finished

  int counter = 0;
  int nb_frames = render_lookup_table.size();

  QLinkedList<HashDownloadFuturePair> download_futures;

  // Iterators
  QLinkedList<HashFrameFuturePair>::iterator i;
  QLinkedList<HashDownloadFuturePair>::iterator j;
  QLinkedList<RangeSampleFuturePair>::iterator k;

  while (!render_lookup_table.isEmpty()
         || !download_futures.isEmpty()
         || !audio_lookup_table.isEmpty()) {

    i = render_lookup_table.begin();

    while (i != render_lookup_table.end()) {
      if (i->frame_future.isFinished()) {
        FramePtr f = i->frame_future.result();

        // Start multithreaded download here
        download_futures.append({i->hash, DownloadFrame(f, i->hash)});

        i = render_lookup_table.erase(i);
      } else {
        i++;
      }
    }

    j = download_futures.begin();

    while (j != download_futures.end()) {
      if (j->download_future.isFinished()) {
        // Place it in the cache
        FrameDownloaded(j->hash, times_to_render.value(j->hash));

        // Signal process
        counter++;
        emit ProgressChanged(static_cast<double>(counter) / static_cast<double>(nb_frames));

        j = download_futures.erase(j);
      } else {
        j++;
      }
    }

    if (audio_enabled) {
      k = audio_lookup_table.begin();

      while (k != audio_lookup_table.end()) {
        if (k->sample_future.isFinished()) {
          AudioDownloaded(k->range, k->sample_future.result());

          k = audio_lookup_table.erase(k);
        } else {
          k++;
        }
      }
    }
  }
}

void RenderTask::AudioDownloaded(const TimeRange &, SampleBufferPtr)
{
}

OLIVE_NAMESPACE_EXIT