5-unify-multiple-graphframes-api.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "    __          __       __         __ \n",
      "   / /_  ____ _/ /______/ /_  ___  / /_\n",
      "  / __ \\/ __ `/ __/ ___/ __ \\/ _ \\/ __/\n",
      " / / / / /_/ / /_/ /__/ / / /  __/ /_  \n",
      "/_/ /_/\\__,_/\\__/\\___/_/ /_/\\___/\\__/  v1.3.1a0\n",
      "\n",
      "\u001b[38;5;196m6.000\u001b[0m foo\u001b[0m\n",
      "├─ \u001b[38;5;220m4.000\u001b[0m bar\u001b[0m\n",
      "│  └─ \u001b[38;5;46m3.000\u001b[0m baz\u001b[0m\n",
      "│     └─ \u001b[34mnan\u001b[0m garply\u001b[0m\n",
      "└─ \u001b[38;5;22m1.000\u001b[0m baz\u001b[0m\n",
      "\n",
      "\u001b[4mLegend\u001b[0m (Metric: time Min: 1.00 Max: 6.00)\n",
      "\u001b[38;5;196m█ \u001b[0m5.50 - 6.00\n",
      "\u001b[38;5;208m█ \u001b[0m4.50 - 5.50\n",
      "\u001b[38;5;220m█ \u001b[0m3.50 - 4.50\n",
      "\u001b[38;5;46m█ \u001b[0m2.50 - 3.50\n",
      "\u001b[38;5;34m█ \u001b[0m1.50 - 2.50\n",
      "\u001b[38;5;22m█ \u001b[0m1.00 - 1.50\n",
      "\n",
      "name\u001b[0m User code    \u001b[38;5;160m◀ \u001b[0m Only in left graph    \u001b[38;5;28m▶ \u001b[0m Only in right graph\n",
      "     __          __       __         __ \n",
      "   / /_  ____ _/ /______/ /_  ___  / /_\n",
      "  / __ \\/ __ `/ __/ ___/ __ \\/ _ \\/ __/\n",
      " / / / / /_/ / /_/ /__/ / / /  __/ /_  \n",
      "/_/ /_/\\__,_/\\__/\\___/_/ /_/\\___/\\__/  v1.3.1a0\n",
      "\n",
      "\u001b[38;5;22m2.000\u001b[0m foo\u001b[0m\n",
      "├─ \u001b[34mnan\u001b[0m bar\u001b[0m\n",
      "│  └─ \u001b[34mnan\u001b[0m baz\u001b[0m\n",
      "│     └─ \u001b[34mnan\u001b[0m garply\u001b[0m\n",
      "└─ \u001b[38;5;196m6.000\u001b[0m baz\u001b[0m\n",
      "\n",
      "\u001b[4mLegend\u001b[0m (Metric: time Min: 2.00 Max: 6.00)\n",
      "\u001b[38;5;196m█ \u001b[0m5.60 - 6.00\n",
      "\u001b[38;5;208m█ \u001b[0m4.80 - 5.60\n",
      "\u001b[38;5;220m█ \u001b[0m4.00 - 4.80\n",
      "\u001b[38;5;46m█ \u001b[0m3.20 - 4.00\n",
      "\u001b[38;5;34m█ \u001b[0m2.40 - 3.20\n",
      "\u001b[38;5;22m█ \u001b[0m2.00 - 2.40\n",
      "\n",
      "name\u001b[0m User code    \u001b[38;5;160m◀ \u001b[0m Only in left graph    \u001b[38;5;28m▶ \u001b[0m Only in right graph\n",
      "     __          __       __         __ \n",
      "   / /_  ____ _/ /______/ /_  ___  / /_\n",
      "  / __ \\/ __ `/ __/ ___/ __ \\/ _ \\/ __/\n",
      " / / / / /_/ / /_/ /__/ / / /  __/ /_  \n",
      "/_/ /_/\\__,_/\\__/\\___/_/ /_/\\___/\\__/  v1.3.1a0\n",
      "\n",
      "\u001b[38;5;22m0.000\u001b[0m foo\u001b[0m\n",
      "├─ \u001b[38;5;220m5.000\u001b[0m bar\u001b[0m\n",
      "│  └─ \u001b[38;5;196m8.000\u001b[0m baz\u001b[0m\n",
      "│     └─ \u001b[38;5;46m3.000\u001b[0m garply\u001b[0m\n",
      "└─ \u001b[34mnan\u001b[0m baz\u001b[0m\n",
      "\n",
      "\u001b[4mLegend\u001b[0m (Metric: time Min: 0.00 Max: 8.00)\n",
      "\u001b[38;5;196m█ \u001b[0m7.20 - 8.00\n",
      "\u001b[38;5;208m█ \u001b[0m5.60 - 7.20\n",
      "\u001b[38;5;220m█ \u001b[0m4.00 - 5.60\n",
      "\u001b[38;5;46m█ \u001b[0m2.40 - 4.00\n",
      "\u001b[38;5;34m█ \u001b[0m0.80 - 2.40\n",
      "\u001b[38;5;22m█ \u001b[0m0.00 - 0.80\n",
      "\n",
      "name\u001b[0m User code    \u001b[38;5;160m◀ \u001b[0m Only in left graph    \u001b[38;5;28m▶ \u001b[0m Only in right graph\n",
      "\n"
     ]
    }
   ],
   "source": [
    "import hatchet as ht\n",
    "\n",
    "g_toy1 = [\n",
    "    {\n",
    "        \"frame\": {\"name\": \"foo\", \"type\": \"function\"},\n",
    "        \"metrics\": {\"time (inc)\": 16.0, \"time\": 6.0},\n",
    "        \"children\": [\n",
    "            {\n",
    "                \"frame\": {\"name\": \"bar\"},\n",
    "                \"metrics\": {\"time (inc)\": 9.0, \"time\": 4.0},\n",
    "                \"children\": [\n",
    "                    {\n",
    "                        \"frame\": {\"name\": \"baz\", \"type\": \"function\"},\n",
    "                        \"metrics\": {\"time (inc)\": 3.0, \"time\": 3.0},\n",
    "                    },\n",
    "                ],\n",
    "            },\n",
    "            {\n",
    "                \"frame\": {\"name\": \"baz\"},\n",
    "                \"metrics\": {\"time (inc)\": 1.0, \"time\": 1.0},\n",
    "            },\n",
    "        ],\n",
    "    },\n",
    "]\n",
    "\n",
    "g_toy2 = [\n",
    "    {\n",
    "        \"frame\": {\"name\": \"foo\", \"type\": \"function\"},\n",
    "        \"metrics\": {\"time (inc)\": 8.0, \"time\": 2.0},\n",
    "        \"children\": [\n",
    "            {\n",
    "                \"frame\": {\"name\": \"baz\"},\n",
    "                \"metrics\": {\"time (inc)\": 6.0, \"time\": 6.0},\n",
    "            },\n",
    "        ],\n",
    "    },\n",
    "]\n",
    "\n",
    "g_toy3 = [\n",
    "    {\n",
    "        \"frame\": {\"name\": \"foo\", \"type\": \"function\"},\n",
    "        \"metrics\": {\"time (inc)\": 16.0, \"time\": 0.0},\n",
    "        \"children\": [\n",
    "            {\n",
    "                \"frame\": {\"name\": \"bar\"},\n",
    "                \"metrics\": {\"time (inc)\": 16.0, \"time\": 5.0},\n",
    "                \"children\": [\n",
    "                    {\n",
    "                        \"frame\": {\"name\": \"baz\", \"type\": \"function\"},\n",
    "                        \"metrics\": {\"time (inc)\": 11.0, \"time\": 8.0},\n",
    "                        \"children\" : [\n",
    "                            {\n",
    "                                \"frame\" : {\"name\": \"garply\", \"type\": \"function\"},\n",
    "                                \"metrics\": {\"time (inc)\": 3.0, \"time\": 3.0},\n",
    "                            }\n",
    "                        ]\n",
    "                    },\n",
    "                ],\n",
    "            },\n",
    "        ],\n",
    "    },\n",
    "]\n",
    "\n",
    "gf_toy1 = ht.GraphFrame.from_literal(g_toy1)\n",
    "gf_toy1.update_metadata(1)\n",
    "gf_toy2 = ht.GraphFrame.from_literal(g_toy2)\n",
    "gf_toy2.update_metadata(2)\n",
    "gf_toy3 = ht.GraphFrame.from_literal(g_toy3)\n",
    "gf_toy3.update_metadata(3)\n",
    "gf_toys = [gf_toy1, gf_toy2, gf_toy3]\n",
    "\n",
    "ht.GraphFrame.unify_multiple_graphframes(gf_toys)\n",
    "print(gf_toy1.tree(), gf_toy2.tree(), gf_toy3.tree())"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}