Provided access to the graph layout from Python.

1 files changed, 140 insertions, 0 deletions

diff --git a/plugins/pychrysalide/analysis/block.c b/plugins/pychrysalide/analysis/block.c
index ca4ac09..e282d14 100644
--- a/plugins/pychrysalide/analysis/block.c
+++ b/plugins/pychrysalide/analysis/block.c
@@ -332,6 +332,51 @@ bool ensure_python_code_block_is_registered(void)
 }
 
 
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : arg = argument quelconque à tenter de convertir.             *
+*                dst = destination des valeurs récupérées en cas de succès.   *
+*                                                                             *
+*  Description : Tente de convertir en bloc de code.                          *
+*                                                                             *
+*  Retour      : Bilan de l'opération, voire indications supplémentaires.     *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+int convert_to_code_block(PyObject *arg, void *dst)
+{
+    int result;                             /* Bilan à retourner           */
+
+    result = PyObject_IsInstance(arg, (PyObject *)get_python_code_block_type());
+
+    switch (result)
+    {
+        case -1:
+            /* L'exception est déjà fixée par Python */
+            result = 0;
+            break;
+
+        case 0:
+            PyErr_SetString(PyExc_TypeError, "unable to convert the provided argument to code block");
+            break;
+
+        case 1:
+            *((GCodeBlock **)dst) = G_CODE_BLOCK(pygobject_get(arg));
+            break;
+
+        default:
+            assert(false);
+            break;
+
+    }
+
+    return result;
+
+}
+
+
 
 /* ---------------------------------------------------------------------------------- */
 /*                           REGROUPEMENT EN LISTE DE BLOCS                           */
@@ -542,3 +587,98 @@ bool ensure_python_block_list_is_registered(void)
     return true;
 
 }
+
+
+/******************************************************************************
+*                                                                             *
+*  Paramètres  : arg = argument quelconque à tenter de convertir.             *
+*                dst = destination des valeurs récupérées en cas de succès.   *
+*                                                                             *
+*  Description : Tente de convertir en liste de blocs de code.                *
+*                                                                             *
+*  Retour      : Bilan de l'opération, voire indications supplémentaires.     *
+*                                                                             *
+*  Remarques   : -                                                            *
+*                                                                             *
+******************************************************************************/
+
+int convert_to_block_list_with_ref(PyObject *arg, void *dst)
+{
+    int result;                             /* Bilan à retourner           */
+    GBlockList *new;                        /* Nouvelle liste à constituer */
+    PyObject *item;                         /* Elément issu de l'itération */
+    int ret;                                /* Bilan d'une conversion      */
+    GCodeBlock *block;                      /* Bloc de code à intégrer     */
+
+    if (arg == NULL)
+    {
+        g_clear_object((void **)dst);
+        result = 1;
+    }
+
+    else
+    {
+        result = PyObject_IsInstance(arg, (PyObject *)get_python_block_list_type());
+
+        switch (result)
+        {
+            case -1:
+                /* L'exception est déjà fixée par Python */
+                result = 0;
+                break;
+
+            case 0:
+
+                if (PyIter_Check(arg))
+                {
+                    new = g_block_list_new(0);
+
+                    result = Py_CLEANUP_SUPPORTED;
+
+                    for (item = PyIter_Next(arg); item != NULL; item = PyIter_Next(arg))
+                    {
+                        ret = convert_to_code_block(item, &block);
+
+                        if (ret == 1)
+                            g_object_ref(G_OBJECT(block));
+
+                        Py_DECREF(item);
+
+                        if (ret != 1)
+                        {
+                            result = 0;
+                            break;
+                        }
+
+                        g_block_list_append_block(new, block);
+
+                    }
+
+                    if (result != Py_CLEANUP_SUPPORTED)
+                        g_object_unref(G_OBJECT(new));
+
+                    else
+                        *((GBlockList **)dst) = new;
+
+                }
+
+                else
+                    PyErr_SetString(PyExc_TypeError, "unable to convert the provided argument to graph cluster");
+
+                break;
+
+            case 1:
+                *((GBlockList **)dst) = G_BLOCK_LIST(pygobject_get(arg));
+                break;
+
+            default:
+                assert(false);
+                break;
+
+        }
+
+    }
+
+    return result;
+
+}