/* OpenIDA - Outil d'analyse de fichiers binaires
 * layout.c - mise en place de graphique
 *
 * Copyright (C) 2009 Cyrille Bagard
 *
 *  This file is part of OpenIDA.
 *
 *  OpenIDA 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.
 *
 *  OpenIDA 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 Foobar.  If not, see <http://www.gnu.org/licenses/>.
 */


#include "layout.h"


#include <malloc.h>
#include <string.h>
#include <gtk/gtkfixed.h>


#include "dot.h"
#include "node.h"
#include "../common/extstr.h"



/* Taille maximale des descriptions de liens */
#define LINKS_DESC_LEN 128


/* Etablit tous les liens entre les différents morceaux de code. */
static char *complete_graph_links(const GtkGraphView *, GtkBinView **, size_t, char *);



/******************************************************************************
*                                                                             *
*  Paramètres  : view  = support où placer les différents éléments.           *
*                views = morceaux de code à afficher de façon organisée.      *
*                count = quantité de ces morceaux de code.                    *
*                                                                             *
*  Description : Dispose une série de morceaux d'affichage en graphique.      *
*                                                                             *
*  Retour      : -                                                            *
*                                                                             *
*  Remarques   : -                                                            *
*                                                                             *
******************************************************************************/

bool build_graph_view(GtkGraphView *view, GtkBinView **views, size_t count)
{
    GGraphNode **nodes;                     /* Intermédiaires en place     */
    size_t i;                               /* Boucle de parcours          */
    char *cmds;                             /* Description à envoyer à dot */
    graph_layout *layout;                   /* Graphique construit         */
    GtkLinkRenderer **links;                /* Liens graphiques construits */
    size_t links_count;                     /* Quantité de ces liens       */

    /* Création de la glue */

    nodes = (GGraphNode **)calloc(count, sizeof(GGraphNode *));

    for (i = 0; i < count; i++)
        nodes[i] = g_graph_node_new(views[i]);
 
    /* Définition du graphique */

    cmds = strdup("digraph G {\noverlap=false;\n  splines=true;\n");
    //cmds = strdup("digraph G {\n");

    for (i = 0; i < count; i++)
        cmds = g_graph_node_register_for_dot(nodes[i], cmds);

    cmds = complete_graph_links(view, views, count, cmds);

    cmds = stradd(cmds, "}");

    printf("first step :: '%s'\n", cmds);

    layout = create_graph_layout(cmds);

    /* Affichage du graphique */

    place_nodes_of_graph_layout(layout, GTK_FIXED(view),  nodes, count);

    links = create_links_from_graph_layout(layout, &links_count);
    gtk_graph_view_attach_links(view, links, links_count);

    gtk_widget_queue_draw(GTK_WIDGET(view));

    delete_graph_layout(layout);

    /* TODO : free nodes */

    return true;

}


/******************************************************************************
*                                                                             *
*  Paramètres  : view  = support contenant les différentes lignes.            *
*                views = morceaux de code à afficher de façon organisée.      *
*                count = quantité de ces morceaux de code.                    *
*                desc  = description du graphique à compléter.                *
*                                                                             *
*  Description : Etablit tous les liens entre les différents morceaux de code.*
*                                                                             *
*  Retour      : -                                                            *
*                                                                             *
*  Remarques   : -                                                            *
*                                                                             *
******************************************************************************/

static char *complete_graph_links(const GtkGraphView *view, GtkBinView **views, size_t count, char *desc)
{
    size_t i;                               /* Boucle de parcours #1       */
    GRenderingLine *last;                   /* Dernière ligne d'un bloc    */
    GRenderingLine *dest;                   /* Ligne visée par une autre   */
    InstructionLinkType type;               /* Type de lien entre lignes   */
    bool link_next;                         /* Lien auto. avec la suivante */
    vmpa_t addr;                            /* Addresse de destination     */
    size_t k;                               /* Boucle de parcours #2       */
    char buffer[LINKS_DESC_LEN];            /* Tampon pour l'ajout de liens*/

    for (i = 0; i < count; i++)
    {
        last = gtk_bin_view_get_last_line(views[i]);

        if (g_rendering_line_has_destination(last))
        {
            link_next = false;

            /* Premier saut : exigé par l'instruction */

            dest = g_rendering_line_get_destination(last, &type);

            addr = get_rendering_line_address(dest);

            for (k = 0; k < count; k++)
                if (gtk_bin_view_contain_address(views[k], addr))
                    break;

            if (k < count)
                switch (type)
                {
                    case ILT_JUMP:
                        snprintf(buffer, LINKS_DESC_LEN, "_%p -> _%p;\n", views[i], views[k]);
                        desc = stradd(desc, buffer);
                        break;

                    case ILT_JUMP_IF_TRUE:
                        snprintf(buffer, LINKS_DESC_LEN, "_%p -> _%p;\n", views[i], views[k]);
                        desc = stradd(desc, buffer);
                        link_next = true;
                        break;

                    case ILT_JUMP_IF_FALSE:
                        snprintf(buffer, LINKS_DESC_LEN, "_%p -> _%p;\n", views[i], views[k]);
                        desc = stradd(desc, buffer);
                        link_next = true;
                        break;

                    default:
                        break;

                }

            /* Second saut : automatique selon l'instruction */

            if (link_next)
            {
                dest = g_rendering_line_get_next_iter(gtk_bin_view_get_lines(GTK_BIN_VIEW(view)),
                                                      last,
                                                      gtk_bin_view_get_last_line(GTK_BIN_VIEW(view)));

                if (dest == NULL) continue;

                addr = get_rendering_line_address(dest);

                for (k = 0; k < count; k++)
                    if (gtk_bin_view_contain_address(views[k], addr))
                        break;

                if (k < count)
                {
                        snprintf(buffer, LINKS_DESC_LEN, "_%p -> _%p;\n", views[i], views[k]);
                        desc = stradd(desc, buffer);
                }

            }

        }

        /* Si la ligne n'est pas la dernière, on suit le flux normal */
        else if (last != gtk_bin_view_get_last_line(view))
        {
            dest = g_rendering_line_get_next_iter(gtk_bin_view_get_lines(GTK_BIN_VIEW(view)),
                                                  last,
                                                  gtk_bin_view_get_last_line(GTK_BIN_VIEW(view)));

            if (dest == NULL) continue;

            addr = get_rendering_line_address(dest);

            for (k = 0; k < count; k++)
                if (gtk_bin_view_contain_address(views[k], addr))
                    break;

            if (k < count)
            {
                snprintf(buffer, LINKS_DESC_LEN, "_%p -> _%p;\n", views[i], views[k]);
                desc = stradd(desc, buffer);
            }

        }

    }

    return desc;

}