/*
  * FIFO test pseudo-muxer
  * Copyright (c) 2016 Jan Sebechlebsky
  *
  * This file is part of FFmpeg.
  *
  * FFmpeg is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public License
  * as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
  * FFmpeg 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 Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public License
  * along with FFmpeg; if not, write to the Free Software * Foundation, Inc.,
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
 #include <stdlib.h>

 #include "libavutil/opt.h"
 #include "libavutil/time.h"
 #include "libavutil/avassert.h"

 
 #include "avformat.h"
 #include "url.h"

 
 /* Implementation of mock muxer to simulate real muxer failures */
 
 #define MAX_TST_PACKETS 128
 #define SLEEPTIME_50_MS 50000
 #define SLEEPTIME_10_MS 10000
 
 /* Implementation of mock muxer to simulate real muxer failures */
 
 /* This is structure of data sent in packets to
  * failing muxer */
 typedef struct FailingMuxerPacketData {
     int ret;             /* return value of write_packet call*/
     int recover_after;   /* set ret to zero after this number of recovery attempts */
     unsigned sleep_time; /* sleep for this long in write_packet to simulate long I/O operation */
 } FailingMuxerPacketData;
 
 
 typedef struct FailingMuxerContext {
     AVClass *class;
     int write_header_ret;
     int write_trailer_ret;
     /* If non-zero, summary of processed packets will be printed in deinit */
     int print_deinit_summary;
 
     int flush_count;
     int pts_written[MAX_TST_PACKETS];
     int pts_written_nr;
 } FailingMuxerContext;
 
 static int failing_write_header(AVFormatContext *avf)
 {
     FailingMuxerContext *ctx = avf->priv_data;
     return ctx->write_header_ret;
 }
 
 static int failing_write_packet(AVFormatContext *avf, AVPacket *pkt)
 {
     FailingMuxerContext *ctx = avf->priv_data;
     int ret = 0;
     if (!pkt) {
         ctx->flush_count++;
     } else {
         FailingMuxerPacketData *data = (FailingMuxerPacketData*) pkt->data;
 
         if (!data->recover_after) {
             data->ret = 0;
         } else {
             data->recover_after--;
         }
 
         ret = data->ret;
 
         if (data->sleep_time) {
             int64_t slept = 0;
             while (slept < data->sleep_time) {
                 if (ff_check_interrupt(&avf->interrupt_callback))
                     return AVERROR_EXIT;
                 av_usleep(SLEEPTIME_10_MS);
                 slept += SLEEPTIME_10_MS;
             }
         }
 
         if (!ret) {
             ctx->pts_written[ctx->pts_written_nr++] = pkt->pts;
             av_packet_unref(pkt);
         }
     }
     return ret;
 }
 
 static int failing_write_trailer(AVFormatContext *avf)
 {
     FailingMuxerContext *ctx = avf->priv_data;
     return ctx->write_trailer_ret;
 }
 
 static void failing_deinit(AVFormatContext *avf)
 {
     int i;
     FailingMuxerContext *ctx = avf->priv_data;
 
     if (!ctx->print_deinit_summary)
         return;
 
     printf("flush count: %d\n", ctx->flush_count);
     printf("pts seen nr: %d\n", ctx->pts_written_nr);
     printf("pts seen: ");
     for (i = 0; i < ctx->pts_written_nr; ++i ) {
         printf(i ? ",%d" : "%d", ctx->pts_written[i]);
     }
     printf("\n");
 }
 #define OFFSET(x) offsetof(FailingMuxerContext, x)
 static const AVOption options[] = {
         {"write_header_ret", "write_header() return value", OFFSET(write_header_ret),
          AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM},
         {"write_trailer_ret", "write_trailer() return value", OFFSET(write_trailer_ret),
          AV_OPT_TYPE_INT, {.i64 = 0}, INT_MIN, INT_MAX, AV_OPT_FLAG_ENCODING_PARAM},
         {"print_deinit_summary", "print summary when deinitializing muxer", OFFSET(print_deinit_summary),
          AV_OPT_TYPE_BOOL, {.i64 = 1}, 0, 1, AV_OPT_FLAG_ENCODING_PARAM},
         {NULL}
     };
 
 static const AVClass failing_muxer_class = {
     .class_name = "Fifo test muxer",
     .item_name  = av_default_item_name,
     .option     = options,
     .version    = LIBAVUTIL_VERSION_INT,
 };
 
 AVOutputFormat ff_fifo_test_muxer = {
     .name           = "fifo_test",
     .long_name      = NULL_IF_CONFIG_SMALL("Fifo test muxer"),
     .priv_data_size = sizeof(FailingMuxerContext),
     .write_header   = failing_write_header,
     .write_packet   = failing_write_packet,
     .write_trailer  = failing_write_trailer,
     .deinit         = failing_deinit,
     .priv_class     = &failing_muxer_class,
     .flags          = AVFMT_NOFILE | AVFMT_ALLOW_FLUSH,
 };