/* GstShark - A Front End for GstTracer
 * Copyright (C) 2016 RidgeRun Engineering <michael.gruner@ridgerun.com>
 *
 * This file is part of GstShark.
 *
 * This library 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.
 *
 * This library 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 this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

#ifdef HAVE_CONFIG_H
#  include "config.h"
#endif /*  */

#include <gst/gst.h>
#include <glib.h>
#include <glib/gstdio.h>

#ifdef ENABLE_GRAPHVIZ
#  include <cgraph.h>
#  include <gvc.h>
#endif

#include "gstctf.h"
#include "gstdot.h"

typedef struct _GstDotRenderThread GstDotRenderThread;

struct _GstDotRenderThread
{
  GstDotRender render;
  gchar *dot_string;
  gpointer args;
};

static void gst_dot_pipeline_to_file (GstBin * bin, GstDebugGraphDetails flags);

#define MAX_SUFFIX_LEN (32)

void
gst_dot_pipeline_to_file (GstBin * bin, GstDebugGraphDetails flags)
{
  gchar *trace_dir;
  gchar *full_trace_dir = NULL;
  gchar *full_file_name = NULL;
  FILE *out;
  gchar time_suffix[MAX_SUFFIX_LEN];
  time_t now = time (NULL);

  trace_dir = get_ctf_path_name ();

  full_trace_dir = g_strdup_printf ("%s" G_DIR_SEPARATOR_S "graphic",
      trace_dir);

  if (!g_file_test (full_trace_dir, G_FILE_TEST_EXISTS)) {
    if (g_mkdir (full_trace_dir, 0775) == 0) {
      GST_INFO ("Directory %s did not exist and was created sucessfully.",
          full_trace_dir);
    } else {
      GST_ERROR ("Directory %s could not be created.", full_trace_dir);
    }
  } else {
    GST_INFO ("Directory %s already exists in the current path.",
        full_trace_dir);
  }

  strftime (time_suffix, MAX_SUFFIX_LEN, "%F_%T", localtime (&now));

  full_file_name = g_strdup_printf ("%s" G_DIR_SEPARATOR_S "%s_%s.dot",
      full_trace_dir, GST_OBJECT_NAME (bin), time_suffix);

  out = g_fopen (full_file_name, "w");
  if (out != NULL) {
    gchar *buf;

    buf = gst_debug_bin_to_dot_data (bin, flags);
    fputs (buf, out);

    g_free (buf);
    fclose (out);

    GST_INFO ("Wrote bin graph to : '%s'", full_file_name);
  } else {
    GST_WARNING ("Failed to open file '%s' for writing: %s", full_file_name,
        g_strerror (errno));
  }

  g_free (full_file_name);
  g_free (full_trace_dir);
}

gchar *
gst_dot_pipeline_to_string (const GstPipeline * pipe)
{
  GstBin *bin;
  GstDebugGraphDetails flags;

  g_return_val_if_fail (pipe, NULL);
  bin = GST_BIN (pipe);
  flags = GST_DEBUG_GRAPH_SHOW_ALL;

  gst_dot_pipeline_to_file (bin, flags);

  return gst_debug_bin_to_dot_data (bin, flags);
}

static gpointer
gst_dot_render_thread (gpointer data)
{
  GstDotRenderThread *thread_args;
  GstDotRender render;
  gchar *dot_string;
  gpointer args;

  g_return_val_if_fail (data, NULL);
  thread_args = (GstDotRenderThread *) data;

  /* Breaking the structure passed as a parameter */
  render = thread_args->render;
  dot_string = thread_args->dot_string;
  args = thread_args->args;

  /* Calling the function to get the pipeline graphic */
  render (dot_string, args);

  /* Freeing the memory allocated and killing the graphic thread */
  g_free (dot_string);
  g_free (thread_args);
  g_thread_exit (0);

  return NULL;
}

gboolean
gst_dot_do_render (const gchar * dot_string, GstDotRender render, gpointer args)
{
  GstDotRenderThread *thread_args;

  g_return_val_if_fail (dot_string, FALSE);
  g_return_val_if_fail (render, FALSE);

  /* Allocating memory with the parameters for the graphic pipeline */
  thread_args = g_malloc (sizeof (GstDotRenderThread));

  /* Filling the structure that is going to be used to get a new thread
     for displaying the pipeline graphic */
  thread_args->render = render;
  thread_args->dot_string = g_strdup (dot_string);
  thread_args->args = args;

  g_thread_new ("GstDotRender", gst_dot_render_thread, thread_args);

  return TRUE;
}

#ifdef ENABLE_GRAPHVIZ
gboolean
gst_dot_x11_render (const gchar * dot_string, gpointer args)
{
  Agraph_t *G;
  GVC_t *gvc;

  g_return_val_if_fail (dot_string, FALSE);
  gvc = gvContext ();
  G = agmemread (dot_string);
  gvLayout (gvc, G, "dot");
  gvRender (gvc, G, "x11", NULL);
  gvFreeLayout (gvc, G);
  agclose (G);
  gvFreeContext (gvc);

  return TRUE;
}
#else
gboolean
gst_dot_x11_render (const gchar * dot_string, gpointer args)
{
  return TRUE;
}
#endif