Depletion chain ancestors.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "4c247d71",
   "metadata": {},
   "outputs": [],
   "source": [
    "import openmc.deplete\n",
    "import networkx as nx\n",
    "import matplotlib.pyplot as plt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "6fceaca8",
   "metadata": {},
   "outputs": [],
   "source": [
    "chain = openmc.deplete.Chain.from_xml('/home/romano/openmc-data/depletion/chain_endfb71_pwr.xml')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 40,
   "id": "36ed7b9a",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Create directed graph based on reactions/decays\n",
    "dg = nx.DiGraph()\n",
    "for nuc in chain.nuclides:\n",
    "    for mode in nuc.decay_modes:\n",
    "        if mode.target is not None:\n",
    "            dg.add_edge(nuc.name, mode.target)\n",
    "            \n",
    "    \"\"\"\n",
    "    for rx in nuc.reactions:\n",
    "        if rx.target is not None:\n",
    "            dg.add_edge(nuc.name, mode.target)\n",
    "    \n",
    "    if nuc.yield_data is not None:\n",
    "        for product in nuc.yield_data.products:\n",
    "            dg.add_edge(nuc.name, product)\n",
    "    \"\"\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "id": "edc56fb5",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Sb135', 'Sb136', 'Sb137', 'Sb138', 'Sn136', 'Sn137', 'Te135', 'Te136'}"
      ]
     },
     "execution_count": 42,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Get ancestors of a given nuclide\n",
    "nx.ancestors(dg, \"I135\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "572f03d3",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.9.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}