/*****************************************************************
 * gmerlin - a general purpose multimedia framework and applications
 *
 * Copyright (c) 2001 - 2024 Members of the Gmerlin project
 * http://github.com/bplaum
 *
 * 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 2 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 <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>

#include <config.h>
#include <gmerlin/translation.h>

#include <gmerlin/log.h>
#define LOG_DOMAIN "pngwriter"

#include <gmerlin/plugin.h>
#include <gmerlin/utils.h>
#include <gavl/metatags.h>

#include "pngwriter.h"

typedef struct
  {
  bg_pngwriter_t png;
  gavl_packet_sink_t * psink;
  gavl_video_sink_t * vsink;
  gavl_video_format_t fmt;
  int have_header;
  gavl_packet_t p;

  int overlay;
  
  gavl_video_frame_t * frame;
  
  } stream_codec_t;

static void * create_codec()
  {
  stream_codec_t * ret = calloc(1, sizeof(*ret));
  return ret;
  }

static void destroy_codec(void * priv)
  {
  stream_codec_t * c = priv;
  if(c->frame)
    {
    gavl_video_frame_null(c->frame);
    gavl_video_frame_destroy(c->frame);
    }
  free(c);
  }

static const bg_parameter_info_t parameters[] =
  {
    {
      .name =        "compression",
      .long_name =   TRS("Compression level"),
      .type =        BG_PARAMETER_SLIDER_INT,
      .val_min =     GAVL_VALUE_INIT_INT(0),
      .val_max =     GAVL_VALUE_INIT_INT(9),
      .val_default = GAVL_VALUE_INIT_INT(9),
    },
    {
      .name =        "bit_mode",
      .long_name =   TRS("Bits per channel"),
      .type =        BG_PARAMETER_STRINGLIST,
      .multi_names = (char const *[]){ "Auto", "8", "16" },
      .val_default = GAVL_VALUE_INIT_STRING("8"),
      .help_string = TRS("If you select auto, the depth will be chosen according to the input format")
    },
    { /* End of parameters */ }
  };

static const bg_parameter_info_t * get_parameters(void * priv)
  {
  return parameters;
  }

static void set_parameter(void * priv, const char * name,
                          const gavl_value_t * val)
  {
  stream_codec_t * c = priv;
  bg_pngwriter_set_parameter(&c->png, name, val);
  }

static gavl_sink_status_t put_frame(void * priv,
                                    gavl_video_frame_t * f)
  {
  gavl_video_frame_t * write_frame;
  gavl_video_format_t fmt;
  stream_codec_t * c = priv;
  
  gavl_packet_reset(&c->p);

  gavl_video_format_copy(&fmt, &c->fmt);
  
  if(c->overlay)
    {
    /* Take a subframe */
    gavl_video_frame_get_subframe(fmt.pixelformat,
                                  f, c->frame,
                                  &f->src_rect);
    fmt.image_width = f->src_rect.w;
    fmt.image_height = f->src_rect.h;
    write_frame = c->frame;
    }
  else
    {
    /* Take the whole frame */
    write_frame = f;
    }
  
  if(!bg_pngwriter_write_header(&c->png, NULL,
                                &c->p,
                                &fmt,
                                NULL))
    return GAVL_SINK_ERROR;
  
  bg_pngwriter_write_image(&c->png, write_frame);
  gavl_video_frame_to_packet_metadata(f, &c->p);

  if(c->overlay)
    {
    memset(&c->p.src_rect, 0, sizeof(c->p.src_rect));
    c->p.dst_x = f->dst_x;
    c->p.dst_y = f->dst_y;
    }
  
  c->have_header = 0;
  c->p.flags |= GAVL_PACKET_KEYFRAME;
  return gavl_packet_sink_put_packet(c->psink, &c->p);
  }

static gavl_video_sink_t * open_video(void * priv, gavl_dictionary_t * s)
  {
  gavl_compression_info_t ci;
  gavl_video_format_t * fmt;
  gavl_dictionary_t * m;
  stream_codec_t * c = priv;
  gavl_packet_reset(&c->p);
  
  fmt = gavl_stream_get_video_format_nc(s);
  m = gavl_dictionary_get_dictionary_create(s, GAVL_META_METADATA);
  
  bg_pngwriter_adjust_format(&c->png, fmt);
  gavl_video_format_copy(&c->fmt, fmt);
  
  gavl_dictionary_set_string(m, GAVL_META_SOFTWARE, "libpng-" PNG_LIBPNG_VER_STRING);
  c->vsink = gavl_video_sink_create(NULL, put_frame, c, &c->fmt);

  gavl_compression_info_init(&ci);
  ci.id = GAVL_CODEC_ID_PNG;
  gavl_stream_set_compression_info(s, &ci);
  
  return c->vsink;
  }

static gavl_video_sink_t * open_overlay(void * priv,
                                        gavl_dictionary_t * s)
  {
  gavl_video_format_t * fmt;
  stream_codec_t * c = priv;
  gavl_video_sink_t * ret;

  fmt = gavl_stream_get_video_format_nc(s);
  
  if(fmt->pixelformat == GAVL_PIXELFORMAT_NONE)
    fmt->pixelformat = GAVL_RGBA_32;
  
  ret = open_video(priv, s);
  c->overlay = 1;
  c->frame = gavl_video_frame_create(NULL);
  return ret;
  }

static void set_packet_sink(void * priv, gavl_packet_sink_t * s)
  {
  stream_codec_t * c = priv;
  c->psink = s;
  }

gavl_codec_id_t compressions[] =
  {
    GAVL_CODEC_ID_PNG,
    GAVL_CODEC_ID_NONE
  };

static const gavl_codec_id_t * get_compressions(void * priv)
  {
  return compressions;
  }


const bg_codec_plugin_t the_plugin =
  {
    .common =
    {
      BG_LOCALE,
      .name =            "c_pngenc",       /* Unique short name */
      .long_name =       "PNG",
      .description =     "libpng based encoder",
      .type =            BG_PLUGIN_COMPRESSOR_VIDEO,
      .flags =           BG_PLUGIN_HANDLES_OVERLAYS,
      .priority =        5,
      .create =          create_codec,
      .destroy =         destroy_codec,
      .get_parameters =    get_parameters,
      .set_parameter =     set_parameter,
    },
    .open_encode_video = open_video,
    .open_encode_overlay = open_overlay,
    .set_packet_sink = set_packet_sink,
    .get_compressions = get_compressions,
  };
/* Include this into all plugin modules exactly once
   to let the plugin loader obtain the API version */
BG_GET_PLUGIN_API_VERSION;
