particles.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "view-in-github"
   },
   "source": [
    "<a href=\"https://colab.research.google.com/github/restrepo/BSM-Submodules/blob/master/particles.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "MAjFAV9yjPxG"
   },
   "source": [
    "# Particles generator"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "PFBjQLsGjwfn"
   },
   "source": [
    "## Dictionary design for JSON output"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "o2ePEeiKjPxI"
   },
   "source": [
    "The SM will be defined as pandas file in JSON format.\n",
    "In this file it will be shown the definition of such a file from a dictionary"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "y7FQ8KB3jPxJ"
   },
   "outputs": [],
   "source": [
    "#TODO: Initialize objects\n",
    "particles=[]\n",
    "particlessons=[]\n",
    "\n",
    "Fields=['Gauge','FermionFields','ScalarFields']\n",
    "NAME='DEFINITION'\n",
    "KEY='EWSB'\n",
    "# Get also particles from\n",
    "#NAME[KEY] dictionary!"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "SUfSThQijPxM"
   },
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "import json\n",
    "# import re, cmdlike as cmd, OrdereDict, json\n",
    "from SARAH import *\n",
    "pd.set_option('display.max_colwidth',200)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "ZsHSZkarjPxR",
    "outputId": "f3f7fca7-295e-4f7c-b978-e6191db78a40"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "cat: kk.json: No such file or directory\r\n"
     ]
    }
   ],
   "source": [
    "cat kk.json | cut -c1-80"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "VXpb3AJyjPxW"
   },
   "source": [
    "Build SM as dictionary to feed pandas"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "uB4IhO9cjPxX"
   },
   "outputs": [],
   "source": [
    "#Matter fields\n",
    "# LaTeX backslash as \\\\\\\\\n",
    "str_keys=['LaTeX','OutputName','Description']\n",
    "lst_keys=['PDG','PDG.IX']\n",
    "ParticleDefinitions={'GaugeES':\n",
    "                             {'H0':{'PDG'  :[0],\n",
    "                                    'Width':0,\n",
    "                                    'Mass' : 'Automatic',\n",
    "                                    'FeynArtsNr' : 1,\n",
    "                                    'LaTeX' : 'H^0',\n",
    "                                    'OutputName' : 'H0'                                   \n",
    "                                   },\n",
    "                              'Hp':{'PDG'  :[0],\n",
    "                                    'Width':0,\n",
    "                                    'Mass' : 'Automatic',\n",
    "                                    'FeynArtsNr' : 2,\n",
    "                                    'LaTeX' : 'H^+',\n",
    "                                    'OutputName' : 'Hp'                                   \n",
    "                                   },\n",
    "                              'VB' : {'Description' :'B-Boson'      },                                                   \n",
    "                              'VG' : {'Description' :'Gluon'        },          \n",
    "                              'VWB': {'Description' :'W-Bosons'     },          \n",
    "                              'gB' : {'Description' :'B-Boson Ghost'},                                                   \n",
    "                              'gG' : {'Description' :'Gluon Ghost'  },          \n",
    "                              'gWB': {'Description' :'W-Boson Ghost'}\n",
    "                             },\n",
    "                            'EWSB':\n",
    "                            {                                \n",
    "                            },\n",
    "                            'WeylFermionAndIndermediate':\n",
    "                            {   \n",
    "                            }\n",
    "                           }\n",
    "\n",
    "ParticleDefinitions['EWSB']['hh']={'Description' : 'Higgs',\n",
    "                                   'PDG'         : [25],\n",
    "                                   'PDG.IX'      : [101000001]}\n",
    "ParticleDefinitions['EWSB']['Ah']={'Description' : \"Pseudo-Scalar Higgs\",\n",
    "                                   'PDG'         : [0],\n",
    "                                   'PDG.IX'      : [0],\n",
    "                                   'Mass'        : [0],\n",
    "                                   'Width'       : [0]\n",
    "                                   }\n",
    "ParticleDefinitions['EWSB']['Hp']={'Description'    : 'Charged Higgs', \n",
    "                                   'PDG'            : [0],\n",
    "                                   'PDG.IX'         : [0],\n",
    "                                   'Width'          : [0], \n",
    "                                   'Mass'           : [0],\n",
    "                                   'LaTeX'          : ['H^+','H^-' ],\n",
    "                                   'OutputName'     : ['Hp' ,'Hm'  ],\n",
    "                                   'ElectricCharge' : 1\n",
    "                                   }\n",
    "\n",
    "ParticleDefinitions['EWSB']['VP']  =  { 'Description' : 'Photon'}\n",
    "ParticleDefinitions['EWSB']['VZ']  =  { 'Description' : 'Z-Boson',\n",
    "                                        'Goldstone' : 'Ah' }\n",
    "ParticleDefinitions['EWSB']['VG']  =  { 'Description' : 'Gluon' }\n",
    "ParticleDefinitions['EWSB']['VWp'] =  { 'Description' : 'W+ - Boson',\n",
    "                                        'Goldstone': 'Hp' }\n",
    "ParticleDefinitions['EWSB']['gP']  =  { 'Description' : 'Photon Ghost'}\n",
    "ParticleDefinitions['EWSB']['gWp'] =  { 'Description' : 'Positive W+ - Boson Ghost'}\n",
    "ParticleDefinitions['EWSB']['gWpC']=  { 'Description' : 'Negative W+ - Boson Ghost' }\n",
    "ParticleDefinitions['EWSB']['gZ']  =  { 'Description' : 'Z-Boson Ghost' }\n",
    "ParticleDefinitions['EWSB']['gG']  =  { 'Description' : 'Gluon Ghost' }\n",
    "ParticleDefinitions['EWSB']['Fd']  =  { 'Description' : 'Down-Quarks'}\n",
    "ParticleDefinitions['EWSB']['Fu']  =  { 'Description' : 'Up-Quarks'}\n",
    "ParticleDefinitions['EWSB']['Fe']  =  { 'Description' : 'Leptons' }\n",
    "ParticleDefinitions['EWSB']['Fv']  =  { 'Description' : 'Neutrinos' }\n",
    "\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['H' ] = {'PDG'       : [0],\n",
    "                                                           'Width'     : 0,\n",
    "                                                           'Mass'      : 'Automatic',\n",
    "                                                           'LaTeX'     : 'H',\n",
    "                                                           'OutputName': 'H' }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['dR'] = {'LaTeX' : 'd_R'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['eR'] = {'LaTeX' : 'e_R'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['l' ] = {'LaTeX' : 'l'     }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['uR'] = {'LaTeX' : 'u_R'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['q' ] = {'LaTeX' : 'q'     }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['eL'] = {'LaTeX' : 'e_L'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['dL'] = {'LaTeX' : 'd_L'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['uL'] = {'LaTeX' : 'u_L'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['vL'] = {'LaTeX' : '\\\\\\\\nu_L'}\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['DR'] = {'LaTeX' : 'D_R'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['ER'] = {'LaTeX' : 'E_R'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['UR'] = {'LaTeX' : 'U_R'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['EL'] = {'LaTeX' : 'E_L'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['DL'] = {'LaTeX' : 'D_L'   }\n",
    "ParticleDefinitions['WeylFermionAndIndermediate']['UL'] = {'LaTeX' : 'U_L'   }"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "3voQvPF0jPxZ"
   },
   "source": [
    "Example to modify dictionary entry"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "AjhNwiXyjPxa"
   },
   "outputs": [],
   "source": [
    "ParticleDefinitions['GaugeES']['H0']['FeynArtsNr']=1"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "qItPu978jPxc"
   },
   "source": [
    "## Save as JSON"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "7vI2Wq9BjPxd"
   },
   "outputs": [],
   "source": [
    "f=open('JSON/particles.json','w')\n",
    "json.dump(ParticleDefinitions,f)\n",
    "f.close()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "sWh7tIjgjPxf"
   },
   "source": [
    "Check if `pandas` can read it"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "DYNGl7dwjPxg",
    "outputId": "a3944550-2322-4fb3-8181-1c2cee6e1c1e"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Description': 'Charged Higgs',\n",
       " 'ElectricCharge': 1,\n",
       " 'LaTeX': ['H^+', 'H^-'],\n",
       " 'Mass': [0],\n",
       " 'OutputName': ['Hp', 'Hm'],\n",
       " 'PDG': [0],\n",
       " 'PDG.IX': [0],\n",
       " 'Width': [0]}"
      ]
     },
     "execution_count": 7,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pd.read_json('JSON/particles.json').loc['Hp']['EWSB']"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "DLU49UIfjPxk"
   },
   "source": [
    "## Feed a `pandas` `DataFrame`"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "EPSbXGZ4jPxl"
   },
   "outputs": [],
   "source": [
    "SM=pd.DataFrame( ParticleDefinitions  )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "DqKqweXXjPxo",
    "outputId": "83ba7e5d-56f8-4a30-f697-97e22ad8f62b"
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>EWSB</th>\n",
       "      <th>GaugeES</th>\n",
       "      <th>WeylFermionAndIndermediate</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Ah</th>\n",
       "      <td>{'Width': [0], 'PDG.IX': [0], 'PDG': [0], 'Description': 'Pseudo-Scalar Higgs', 'Mass': [0]}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>DL</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'D_L'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>DR</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'D_R'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>EL</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'E_L'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>ER</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'E_R'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Fd</th>\n",
       "      <td>{'Description': 'Down-Quarks'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Fe</th>\n",
       "      <td>{'Description': 'Leptons'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Fu</th>\n",
       "      <td>{'Description': 'Up-Quarks'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Fv</th>\n",
       "      <td>{'Description': 'Neutrinos'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>H</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'H', 'OutputName': 'H', 'Width': 0, 'PDG': [0], 'Mass': 'Automatic'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>H0</th>\n",
       "      <td>NaN</td>\n",
       "      <td>{'FeynArtsNr': 1, 'Width': 0, 'PDG': [0], 'OutputName': 'H0', 'Mass': 'Automatic', 'LaTeX': 'H^0'}</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Hp</th>\n",
       "      <td>{'PDG.IX': [0], 'Width': [0], 'PDG': [0], 'LaTeX': ['H^+', 'H^-'], 'ElectricCharge': 1, 'OutputName': ['Hp', 'Hm'], 'Mass': [0], 'Description': 'Charged Higgs'}</td>\n",
       "      <td>{'FeynArtsNr': 2, 'Width': 0, 'PDG': [0], 'OutputName': 'Hp', 'Mass': 'Automatic', 'LaTeX': 'H^+'}</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>UL</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'U_L'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>UR</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'U_R'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>VB</th>\n",
       "      <td>NaN</td>\n",
       "      <td>{'Description': 'B-Boson'}</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>VG</th>\n",
       "      <td>{'Description': 'Gluon'}</td>\n",
       "      <td>{'Description': 'Gluon'}</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>VP</th>\n",
       "      <td>{'Description': 'Photon'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>VWB</th>\n",
       "      <td>NaN</td>\n",
       "      <td>{'Description': 'W-Bosons'}</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>VWp</th>\n",
       "      <td>{'Goldstone': 'Hp', 'Description': 'W+ - Boson'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>VZ</th>\n",
       "      <td>{'Goldstone': 'Ah', 'Description': 'Z-Boson'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>dL</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'd_L'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>dR</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'd_R'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>eL</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'e_L'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>eR</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'e_R'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>gB</th>\n",
       "      <td>NaN</td>\n",
       "      <td>{'Description': 'B-Boson Ghost'}</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>gG</th>\n",
       "      <td>{'Description': 'Gluon Ghost'}</td>\n",
       "      <td>{'Description': 'Gluon Ghost'}</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>gP</th>\n",
       "      <td>{'Description': 'Photon Ghost'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>gWB</th>\n",
       "      <td>NaN</td>\n",
       "      <td>{'Description': 'W-Boson Ghost'}</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>gWp</th>\n",
       "      <td>{'Description': 'Positive W+ - Boson Ghost'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>gWpC</th>\n",
       "      <td>{'Description': 'Negative W+ - Boson Ghost'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>gZ</th>\n",
       "      <td>{'Description': 'Z-Boson Ghost'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>hh</th>\n",
       "      <td>{'PDG.IX': [101000001], 'PDG': [25], 'Description': 'Higgs'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>l</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'l'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>q</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'q'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>uL</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'u_L'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>uR</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': 'u_R'}</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>vL</th>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>{'LaTeX': '\\\\nu_L'}</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                                                                                                                                  EWSB  \\\n",
       "Ah                                                                        {'Width': [0], 'PDG.IX': [0], 'PDG': [0], 'Description': 'Pseudo-Scalar Higgs', 'Mass': [0]}   \n",
       "DL                                                                                                                                                                 NaN   \n",
       "DR                                                                                                                                                                 NaN   \n",
       "EL                                                                                                                                                                 NaN   \n",
       "ER                                                                                                                                                                 NaN   \n",
       "Fd                                                                                                                                      {'Description': 'Down-Quarks'}   \n",
       "Fe                                                                                                                                          {'Description': 'Leptons'}   \n",
       "Fu                                                                                                                                        {'Description': 'Up-Quarks'}   \n",
       "Fv                                                                                                                                        {'Description': 'Neutrinos'}   \n",
       "H                                                                                                                                                                  NaN   \n",
       "H0                                                                                                                                                                 NaN   \n",
       "Hp    {'PDG.IX': [0], 'Width': [0], 'PDG': [0], 'LaTeX': ['H^+', 'H^-'], 'ElectricCharge': 1, 'OutputName': ['Hp', 'Hm'], 'Mass': [0], 'Description': 'Charged Higgs'}   \n",
       "UL                                                                                                                                                                 NaN   \n",
       "UR                                                                                                                                                                 NaN   \n",
       "VB                                                                                                                                                                 NaN   \n",
       "VG                                                                                                                                            {'Description': 'Gluon'}   \n",
       "VP                                                                                                                                           {'Description': 'Photon'}   \n",
       "VWB                                                                                                                                                                NaN   \n",
       "VWp                                                                                                                   {'Goldstone': 'Hp', 'Description': 'W+ - Boson'}   \n",
       "VZ                                                                                                                       {'Goldstone': 'Ah', 'Description': 'Z-Boson'}   \n",
       "dL                                                                                                                                                                 NaN   \n",
       "dR                                                                                                                                                                 NaN   \n",
       "eL                                                                                                                                                                 NaN   \n",
       "eR                                                                                                                                                                 NaN   \n",
       "gB                                                                                                                                                                 NaN   \n",
       "gG                                                                                                                                      {'Description': 'Gluon Ghost'}   \n",
       "gP                                                                                                                                     {'Description': 'Photon Ghost'}   \n",
       "gWB                                                                                                                                                                NaN   \n",
       "gWp                                                                                                                       {'Description': 'Positive W+ - Boson Ghost'}   \n",
       "gWpC                                                                                                                      {'Description': 'Negative W+ - Boson Ghost'}   \n",
       "gZ                                                                                                                                    {'Description': 'Z-Boson Ghost'}   \n",
       "hh                                                                                                        {'PDG.IX': [101000001], 'PDG': [25], 'Description': 'Higgs'}   \n",
       "l                                                                                                                                                                  NaN   \n",
       "q                                                                                                                                                                  NaN   \n",
       "uL                                                                                                                                                                 NaN   \n",
       "uR                                                                                                                                                                 NaN   \n",
       "vL                                                                                                                                                                 NaN   \n",
       "\n",
       "                                                                                                 GaugeES  \\\n",
       "Ah                                                                                                   NaN   \n",
       "DL                                                                                                   NaN   \n",
       "DR                                                                                                   NaN   \n",
       "EL                                                                                                   NaN   \n",
       "ER                                                                                                   NaN   \n",
       "Fd                                                                                                   NaN   \n",
       "Fe                                                                                                   NaN   \n",
       "Fu                                                                                                   NaN   \n",
       "Fv                                                                                                   NaN   \n",
       "H                                                                                                    NaN   \n",
       "H0    {'FeynArtsNr': 1, 'Width': 0, 'PDG': [0], 'OutputName': 'H0', 'Mass': 'Automatic', 'LaTeX': 'H^0'}   \n",
       "Hp    {'FeynArtsNr': 2, 'Width': 0, 'PDG': [0], 'OutputName': 'Hp', 'Mass': 'Automatic', 'LaTeX': 'H^+'}   \n",
       "UL                                                                                                   NaN   \n",
       "UR                                                                                                   NaN   \n",
       "VB                                                                            {'Description': 'B-Boson'}   \n",
       "VG                                                                              {'Description': 'Gluon'}   \n",
       "VP                                                                                                   NaN   \n",
       "VWB                                                                          {'Description': 'W-Bosons'}   \n",
       "VWp                                                                                                  NaN   \n",
       "VZ                                                                                                   NaN   \n",
       "dL                                                                                                   NaN   \n",
       "dR                                                                                                   NaN   \n",
       "eL                                                                                                   NaN   \n",
       "eR                                                                                                   NaN   \n",
       "gB                                                                      {'Description': 'B-Boson Ghost'}   \n",
       "gG                                                                        {'Description': 'Gluon Ghost'}   \n",
       "gP                                                                                                   NaN   \n",
       "gWB                                                                     {'Description': 'W-Boson Ghost'}   \n",
       "gWp                                                                                                  NaN   \n",
       "gWpC                                                                                                 NaN   \n",
       "gZ                                                                                                   NaN   \n",
       "hh                                                                                                   NaN   \n",
       "l                                                                                                    NaN   \n",
       "q                                                                                                    NaN   \n",
       "uL                                                                                                   NaN   \n",
       "uR                                                                                                   NaN   \n",
       "vL                                                                                                   NaN   \n",
       "\n",
       "                                                          WeylFermionAndIndermediate  \n",
       "Ah                                                                               NaN  \n",
       "DL                                                                  {'LaTeX': 'D_L'}  \n",
       "DR                                                                  {'LaTeX': 'D_R'}  \n",
       "EL                                                                  {'LaTeX': 'E_L'}  \n",
       "ER                                                                  {'LaTeX': 'E_R'}  \n",
       "Fd                                                                               NaN  \n",
       "Fe                                                                               NaN  \n",
       "Fu                                                                               NaN  \n",
       "Fv                                                                               NaN  \n",
       "H     {'LaTeX': 'H', 'OutputName': 'H', 'Width': 0, 'PDG': [0], 'Mass': 'Automatic'}  \n",
       "H0                                                                               NaN  \n",
       "Hp                                                                               NaN  \n",
       "UL                                                                  {'LaTeX': 'U_L'}  \n",
       "UR                                                                  {'LaTeX': 'U_R'}  \n",
       "VB                                                                               NaN  \n",
       "VG                                                                               NaN  \n",
       "VP                                                                               NaN  \n",
       "VWB                                                                              NaN  \n",
       "VWp                                                                              NaN  \n",
       "VZ                                                                               NaN  \n",
       "dL                                                                  {'LaTeX': 'd_L'}  \n",
       "dR                                                                  {'LaTeX': 'd_R'}  \n",
       "eL                                                                  {'LaTeX': 'e_L'}  \n",
       "eR                                                                  {'LaTeX': 'e_R'}  \n",
       "gB                                                                               NaN  \n",
       "gG                                                                               NaN  \n",
       "gP                                                                               NaN  \n",
       "gWB                                                                              NaN  \n",
       "gWp                                                                              NaN  \n",
       "gWpC                                                                             NaN  \n",
       "gZ                                                                               NaN  \n",
       "hh                                                                               NaN  \n",
       "l                                                                     {'LaTeX': 'l'}  \n",
       "q                                                                     {'LaTeX': 'q'}  \n",
       "uL                                                                  {'LaTeX': 'u_L'}  \n",
       "uR                                                                  {'LaTeX': 'u_R'}  \n",
       "vL                                                               {'LaTeX': '\\\\nu_L'}  "
      ]
     },
     "execution_count": 9,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "SM"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "K9lvhZ_fjPxq"
   },
   "source": [
    "Checks"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "08I0f95HjPxr",
    "outputId": "2e6c5684-0347-4255-c783-5cf1b4ed2af6"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'FeynArtsNr': 1,\n",
       " 'LaTeX': 'H^0',\n",
       " 'Mass': 'Automatic',\n",
       " 'OutputName': 'H0',\n",
       " 'PDG': [0],\n",
       " 'Width': 0}"
      ]
     },
     "execution_count": 10,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "SM['GaugeES']['H0']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "expY1-fgjPxu",
    "outputId": "f12f13c5-9d88-4d57-f5e7-cf537c3a0587"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Description': 'Z-Boson', 'Goldstone': 'Ah'}"
      ]
     },
     "execution_count": 11,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "SM['EWSB']['VZ']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "CqWtRpTpjPxx",
    "outputId": "20d20141-3bdb-4078-a622-d6159c35a578"
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>EWSB</th>\n",
       "      <th>GaugeES</th>\n",
       "      <th>WeylFermionAndIndermediate</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Ah</th>\n",
       "      <td>{'Width': [0], 'PDG.IX': [0], 'PDG': [0], 'Description': 'Pseudo-Scalar Higgs', 'Mass': [0]}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                                                            EWSB  \\\n",
       "Ah  {'Width': [0], 'PDG.IX': [0], 'PDG': [0], 'Description': 'Pseudo-Scalar Higgs', 'Mass': [0]}   \n",
       "\n",
       "   GaugeES WeylFermionAndIndermediate  \n",
       "Ah     NaN                        NaN  "
      ]
     },
     "execution_count": 12,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "SM[:1]"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "C8IdPTKMjPxz"
   },
   "source": [
    "Show a particle"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "-fQxDfHbjPx0",
    "outputId": "6344156e-46d8-4357-d11d-cb9d45dd83fc"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'FeynArtsNr': 1,\n",
       " 'LaTeX': 'H^0',\n",
       " 'Mass': 'Automatic',\n",
       " 'OutputName': 'H0',\n",
       " 'PDG': [0],\n",
       " 'Width': 0}"
      ]
     },
     "execution_count": 13,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "SM['GaugeES']['H0']"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "L9EgUsifjPx1"
   },
   "source": [
    "Example: Modify particle property"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "LTirvgEUjPx2"
   },
   "outputs": [],
   "source": [
    "SM['WeylFermionAndIndermediate']['dR'] = {'LaTeX' : 'd_R'   }"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "zfnDhGoBjPx6"
   },
   "source": [
    "## Load JSON\n",
    "Directly fromo the `json.dumps...`"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "peoX7__OjPx7"
   },
   "outputs": [],
   "source": [
    "BSM=pd.read_json('JSON/particles.json')#,lines=True,orient='records')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "BKnOyqT_jPx9",
    "outputId": "3e3e6fa3-a5ee-423e-adcf-b3fc2b393798"
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>EWSB</th>\n",
       "      <th>GaugeES</th>\n",
       "      <th>WeylFermionAndIndermediate</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Ah</th>\n",
       "      <td>{'PDG.IX': [0], 'Mass': [0], 'Width': [0], 'PDG': [0], 'Description': 'Pseudo-Scalar Higgs'}</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                                                            EWSB  \\\n",
       "Ah  {'PDG.IX': [0], 'Mass': [0], 'Width': [0], 'PDG': [0], 'Description': 'Pseudo-Scalar Higgs'}   \n",
       "\n",
       "   GaugeES WeylFermionAndIndermediate  \n",
       "Ah     NaN                        NaN  "
      ]
     },
     "execution_count": 16,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "\n",
    "BSM[:1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "ZmdwzMDojPyA",
    "outputId": "613262dd-be6e-49a2-8378-020c2f5b4468"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'FeynArtsNr': 1,\n",
       " 'LaTeX': 'H^0',\n",
       " 'Mass': 'Automatic',\n",
       " 'OutputName': 'H0',\n",
       " 'PDG': [0],\n",
       " 'Width': 0}"
      ]
     },
     "execution_count": 17,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "BSM['GaugeES'].dropna()['H0']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "wNcDrdrhjPyC",
    "outputId": "b98d0a7f-14f9-4adb-850a-e4f4b86088d7"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'FeynArtsNr': 1,\n",
       " 'LaTeX': 'H^0',\n",
       " 'Mass': 'Automatic',\n",
       " 'OutputName': 'H0',\n",
       " 'PDG': [0],\n",
       " 'Width': 0}"
      ]
     },
     "execution_count": 18,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "BSM['GaugeES']['H0']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "leiWdLCajPyE",
    "outputId": "de67383f-7b05-4528-b794-3ff3a733ab2a"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Description': 'Z-Boson', 'Goldstone': 'Ah'}"
      ]
     },
     "execution_count": 19,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "BSM['EWSB']['VZ']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "Yyo64EnsjPyJ"
   },
   "outputs": [],
   "source": [
    "BSM['WeylFermionAndIndermediate']['dR'] = {'LaTeX' : 'd_R'   }"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "EHzpxzjhjPyL",
    "outputId": "305df22d-fb7a-4263-8cb2-06dadd1aa960"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'Description': 'Pseudo-Scalar Higgs',\n",
       " 'Mass': [0],\n",
       " 'PDG': [0],\n",
       " 'PDG.IX': [0],\n",
       " 'Width': [0]}"
      ]
     },
     "execution_count": 21,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "BSM.loc['Ah','EWSB']"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "rWKV63g7jPyN"
   },
   "source": [
    "## Export to files"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "xfhAGdBJjPyN"
   },
   "outputs": [],
   "source": [
    "SM=BSM.loc[['Ah', 'DL', 'DR', 'EL', 'ER', 'Fd', 'Fe', 'Fu', 'Fv', 'H', 'H0', 'Hp',\n",
    "       'UL', 'UR', 'VB', 'VG', 'VP', 'VWB', 'VWp', 'VZ', 'dL', 'dR', 'eL',\n",
    "       'eR', 'gB', 'gG', 'gP', 'gWB', 'gWp', 'gWpC', 'gZ', 'hh', 'l', 'q',\n",
    "       'uL', 'uR', 'vL']]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "OsohFsIBjPyP"
   },
   "outputs": [],
   "source": [
    "CHECK_PANDAS_SAVED_JSON=False\n",
    "if CHECK_PANDAS_SAVED_JSON:\n",
    "    SM=pd.read_json('SM2.json')#,lines=True,orient='records')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "z5tJHnWsjPyR",
    "outputId": "c0872884-ee33-4a6d-f0e1-8ac1503670c0"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "ParticleDefinitions[EWSB] = {\n",
      "    {Ah, { Description -> \"Pseudo-Scalar Higgs\",\n",
      "           PDG.IX -> {0},\n",
      "           Mass -> {0},\n",
      "           Width -> {0},\n",
      "           PDG -> {0}}},\n",
      "    {Fd, { Description -> \"Down-Quarks\"}},\n",
      "    {Fe, { Description -> \"Leptons\"}},\n",
      "    {Fu, { Description -> \"Up-Quarks\"}},\n",
      "    {Fv, { Description -> \"Neutrinos\"}},\n",
      "    {Hp, { Description -> \"Charged Higgs\",\n",
      "           PDG.IX -> {0},\n",
      "           Width -> {0},\n",
      "           PDG -> {0},\n",
      "           ElectricCharge -> 1,\n",
      "           OutputName -> {\"Hp\", \"Hm\"},\n",
      "           Mass -> {0},\n",
      "           LaTeX -> {\"H^+\", \"H^-\"}}},\n",
      "    {VG, { Description -> \"Gluon\"}},\n",
      "    {VP, { Description -> \"Photon\"}},\n",
      "    {VWp, { Description -> \"W+ - Boson\",\n",
      "           Goldstone -> Hp}},\n",
      "    {VZ, { Description -> \"Z-Boson\",\n",
      "           Goldstone -> Ah}},\n",
      "    {gG, { Description -> \"Gluon Ghost\"}},\n",
      "    {gP, { Description -> \"Photon Ghost\"}},\n",
      "    {gWp, { Description -> \"Positive W+ - Boson Ghost\"}},\n",
      "    {gWpC, { Description -> \"Negative W+ - Boson Ghost\"}},\n",
      "    {gZ, { Description -> \"Z-Boson Ghost\"}},\n",
      "    {hh, { Description -> \"Higgs\",\n",
      "           PDG.IX -> {101000001},\n",
      "           PDG -> {25}}}\n",
      "};\n",
      "\n",
      "ParticleDefinitions[GaugeES] = {\n",
      "    {H0, { FeynArtsNr -> 1,\n",
      "           Width -> 0,\n",
      "           PDG -> {0},\n",
      "           OutputName -> H0,\n",
      "           Mass -> Automatic,\n",
      "           LaTeX -> \"H^0\"}},\n",
      "    {Hp, { FeynArtsNr -> 2,\n",
      "           Width -> 0,\n",
      "           PDG -> {0},\n",
      "           OutputName -> Hp,\n",
      "           Mass -> Automatic,\n",
      "           LaTeX -> \"H^+\"}},\n",
      "    {VB, { Description -> \"B-Boson\"}},\n",
      "    {VG, { Description -> \"Gluon\"}},\n",
      "    {VWB, { Description -> \"W-Bosons\"}},\n",
      "    {gB, { Description -> \"B-Boson Ghost\"}},\n",
      "    {gG, { Description -> \"Gluon Ghost\"}},\n",
      "    {gWB, { Description -> \"W-Boson Ghost\"}}\n",
      "};\n",
      "\n",
      "ParticleDefinitions[WeylFermionAndIndermediate] = {\n",
      "    {DL, { LaTeX -> \"D_L\"}},\n",
      "    {DR, { LaTeX -> \"D_R\"}},\n",
      "    {EL, { LaTeX -> \"E_L\"}},\n",
      "    {ER, { LaTeX -> \"E_R\"}},\n",
      "    {H, { PDG -> {0},\n",
      "           OutputName -> H,\n",
      "           Width -> 0,\n",
      "           LaTeX -> \"H\",\n",
      "           Mass -> Automatic}},\n",
      "    {UL, { LaTeX -> \"U_L\"}},\n",
      "    {UR, { LaTeX -> \"U_R\"}},\n",
      "    {dL, { LaTeX -> \"d_L\"}},\n",
      "    {dR, { LaTeX -> \"d_R\"}},\n",
      "    {eL, { LaTeX -> \"e_L\"}},\n",
      "    {eR, { LaTeX -> \"e_R\"}},\n",
      "    {l, { LaTeX -> \"l\"}},\n",
      "    {q, { LaTeX -> \"q\"}},\n",
      "    {uL, { LaTeX -> \"u_L\"}},\n",
      "    {uR, { LaTeX -> \"u_R\"}},\n",
      "    {vL, { LaTeX -> \"\\\\nu_L\"}}\n",
      "};\n",
      "\n",
      "\n"
     ]
    }
   ],
   "source": [
    "print( to_math(SM,'p.m') )"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "RLFquyVYjPyT"
   },
   "source": [
    "# Get particle from model file"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "ZLd2nde4jPyU"
   },
   "source": [
    "Design:\n",
    "* Intialize `particle` as an empty dictionary.\n",
    "* Intialize `particles` as an empty list.\n",
    "* Fill particle properties and build the list: `particles`\n",
    "* Internal defintions\n",
    "  * weyl: Charged Weyl fermion bidiagonal rotation object\n",
    "  * symm: Symmetric diagonalization system object\n",
    "  * vev:  Complex scalar decomposition"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "-dIdbo7cjPyW"
   },
   "source": [
    "Implementation:\n",
    "1. Get particles from particles definitions\n",
    "2. Get particles from `DEFINITION[EWSB]`\n",
    "\n",
    "Assumptions:\n",
    "\n",
    "* SM doublet Higgs field is assigned `H`\n",
    "* Fermion doublets have decreasing Hypercharge, $Y$\n",
    "* $ Q=T_3+Y/2 $"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "0LMMBdupjPyW"
   },
   "source": [
    "## Load Model file"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 51
    },
    "colab_type": "code",
    "id": "9n96DvL55OQ_",
    "outputId": "b1380c74-ed49-4d10-951b-ff917fe7b5bf"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "mkdir: cannot create directory ‘JSON’: File exists\n",
      "/content/JSON\n"
     ]
    }
   ],
   "source": [
    "import os\n",
    "if os.getcwd()=='/content':\n",
    "    !wget -O SARAH.py https://raw.githubusercontent.com/restrepo/BSM-Submodules/master/SARAH.py 2>/dev/null\n",
    "    !wget -O cmdlike.py https://raw.githubusercontent.com/restrepo/BSM-Submodules/master/cmdlike.py 2>/dev/null\n",
    "    !git clone https://github.com/restrepo/SARAH.git 2>/dev/null\n",
    "    !mkdir -p JSON\n",
    "    os.chdir('JSON')\n",
    "    !wget -O fullparticles.json https://raw.githubusercontent.com/restrepo/BSM-Submodules/master/JSON/fullparticles.json 2> /dev/null\n",
    "    os.chdir('../')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "gIOocPRP450c"
   },
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "from SARAH import *\n",
    "pd.set_option('display.max_colwidth',200)\n",
    "#TODO: Initialize objects\n",
    "particles=[]\n",
    "particlessons=[]\n",
    "\n",
    "Fields=['Gauge','FermionFields','ScalarFields']\n",
    "NAME='DEFINITION'\n",
    "KEY='EWSB'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "Q7c9ibrujPyZ"
   },
   "outputs": [],
   "source": [
    "MODEL_PATH=\"./SARAH/Models/SM/SM.m\"\n",
    "f=open(MODEL_PATH,'r')\n",
    "smf=f.read()\n",
    "f.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 54
    },
    "colab_type": "code",
    "id": "X0XuyqCejPye",
    "outputId": "7089718b-d7ee-466f-b3e5-54c5656014d2"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Off[General::spell]\\n\\nModel`Name = \"SM\";\\nModel`NameLaTeX =\"Standard Model\";\\nModel`Authors = \"F.Staub\";\\nModel`Date = \"2014-11-06\";\\n\\n(* 2013-01-24: changed normalization of lambda term to convention of hep-ph/0207271 *)\\n(* 2013-06-24: using new name conventions (without inital \"S\" and \"F\" for scalar and matter fields) *)\\n(* 2013-09-01: changing to new conventions for FermionFields/MatterFields *)\\n(* 2014-11-06: Changed sign in Lagrangian to fit standard conventions *)\\n(* 2016-05-03: Changed sign of Yu *)\\n\\n\\n\\n(*-------------------------------------------*)\\n(*   Particle Content*)\\n(*-------------------------------------------*)\\n\\n(* Gauge Groups *)\\n\\nGauge[[1]]={B,   U[1], hypercharge, g1,False};\\nGauge[[2]]={WB, SU[2], left,        g2,True};\\nGauge[[3]]={G,  SU[3], color,       g3,False};\\n\\n\\n(* Matter Fields *)\\n\\nFermionFields[[1]] = {q, 3, {uL, dL},     1/6, 2,  3};  \\nFermionFields[[2]] = {l, 3, {vL, eL},    -1/2, 2,  1};\\nFermionFields[[3]] = {d, 3, conj[dR],     1/3, 1, -3};\\nFermionFields[[4]] = {u, 3, conj[uR],    -2/3, 1, -3};\\nFermionFields[[5]] = {e, 3, conj[eR],       1, 1,  1};\\n\\nScalarFields[[1]] =  {H, 1, {Hp, H0},     1/2, 2,  1};\\n\\n\\n        \\n(*----------------------------------------------*)\\n(*   DEFINITION                                 *)\\n(*----------------------------------------------*)\\n\\nNameOfStates={GaugeES, EWSB};\\n\\n(* ----- Before EWSB ----- *)\\n\\nDEFINITION[GaugeES][LagrangianInput]= {\\n\\t{LagHC, {AddHC->True}},\\n\\t{LagNoHC,{AddHC->False}}\\n};\\n\\n\\nLagNoHC = -mu2 conj[H].H - 1/2 \\\\[Lambda] conj[H].H.conj[H].H;\\nLagHC =  -(Yd conj[H].d.q + Ye conj[H].e.l + Yu u.q.H);\\n\\n\\n\\t\\t\\t  \\t\\t  \\n\\n(* Gauge Sector *)\\n\\nDEFINITION[EWSB][GaugeSector] =\\n{ \\n  {{VB,VWB[3]},{VP,VZ},ZZ},\\n  {{VWB[1],VWB[2]},{VWp,conj[VWp]},ZW}\\n};     \\n        \\n        \\n          \\t\\n\\n(* ----- VEVs ---- *)\\n\\nDEFINITION[EWSB][VEVs]= \\n{    {H0, {v, 1/Sqrt[2]}, {Ah, \\\\[ImaginaryI]/Sqrt[2]},{hh, 1/Sqrt[2]}}     };\\n \\n\\nDEFINITION[EWSB][MatterSector]=   \\n    {{{{dL}, {conj[dR]}}, {{DL,Vd}, {DR,Ud}}},\\n     {{{uL}, {conj[uR]}}, {{UL,Vu}, {UR,Uu}}},\\n     {{{eL}, {conj[eR]}}, {{EL,Ve}, {ER,Ue}}}};  \\n\\n\\n(*------------------------------------------------------*)\\n(* Dirac-Spinors *)\\n(*------------------------------------------------------*)\\n\\nDEFINITION[EWSB][DiracSpinors]={\\n Fd ->{  DL, conj[DR]},\\n Fe ->{  EL, conj[ER]},\\n Fu ->{  UL, conj[UR]},\\n Fv ->{  vL, 0}};\\n\\nDEFINITION[EWSB][GaugeES]={\\n Fd1 ->{  FdL, 0},\\n Fd2 ->{  0, FdR},\\n Fu1 ->{  Fu1, 0},\\n Fu2 ->{  0, Fu2},\\n Fe1 ->{  Fe1, 0},\\n Fe2 ->{  0, Fe2}};\\n\\n\\n\\n'"
      ]
     },
     "execution_count": 4,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "smf"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "jvF6WrT6jPyf"
   },
   "source": [
    "## General definition"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "ennzIYf1jPyg"
   },
   "source": [
    "## 1) Get particles from particles definitions"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "HfxfrEv9jPyg"
   },
   "source": [
    "### Obtain `particle` from \n",
    "* `Gauge` list\n",
    "* `FermionFields` list, with the components fields (intialized as `particlesson` `dict`) with parents information\n",
    "* `ScalarFields`  list, with the components fields (intialized as `particlesson` `dict`) with parents information\n",
    "\n",
    "and fill `particles`"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 700
    },
    "colab_type": "code",
    "id": "LmEqj2KWjPyh",
    "outputId": "8eced90f-54c1-4f3d-8f1b-0b8ba0bd9dc0"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "('B,   U[1], hypercharge, g1,False',) Gauge 5\n",
      "{'Field': 'VB', 'Parents': None, 'Properties': {'Group': '   U[1]', 'Index': ' hypercharge', 'Coupling': ' g1', 'SSB': ' g1', 'Lorentz': 'Vector'}, 'Definition': 'GaugeES'}\n",
      "('WB, SU[2], left,        g2,True',) Gauge 5\n",
      "{'Field': 'VWB', 'Parents': None, 'Properties': {'Group': ' SU[2]', 'Index': ' left', 'Coupling': '        g2', 'SSB': '        g2', 'Lorentz': 'Vector'}, 'Definition': 'GaugeES'}\n",
      "('G,  SU[3], color,       g3,False',) Gauge 5\n",
      "{'Field': 'VG', 'Parents': None, 'Properties': {'Group': '  SU[3]', 'Index': ' color', 'Coupling': '       g3', 'SSB': '       g3', 'Lorentz': 'Vector'}, 'Definition': 'GaugeES'}\n",
      "('q, 3, uL:: dL,     1/6, 2,  3;  ',) FermionFields 6\n",
      "**********\n",
      " uL:: dL [' uL', ' dL']\n",
      " uL\n",
      " dL\n",
      "**********\n",
      "('l, 3, vL:: eL,    -1/2, 2,  1;',) FermionFields 6\n",
      "**********\n",
      " vL:: eL [' vL', ' eL']\n",
      " vL\n",
      " eL\n",
      "**********\n",
      "('d, 3, conj[dR],     1/3, 1, -3',) FermionFields 6\n",
      "**********\n",
      " conj[dR] [' dR']\n",
      " dR\n",
      "**********\n",
      "('u, 3, conj[uR],    -2/3, 1, -3',) FermionFields 6\n",
      "**********\n",
      " conj[uR] [' uR']\n",
      " uR\n",
      "**********\n",
      "('e, 3, conj[eR],       1, 1,  1',) FermionFields 6\n",
      "**********\n",
      " conj[eR] [' eR']\n",
      " eR\n",
      "**********\n",
      "('H, 1, Hp:: H0,     1/2, 2,  1;',) ScalarFields 6\n",
      "**********\n",
      " Hp:: H0 [' Hp', ' H0']\n",
      " Hp\n",
      " H0\n",
      "**********\n"
     ]
    }
   ],
   "source": [
    "particles=get_particles(smf,Fields,NAME,KEY,particles,particlessons)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 607
    },
    "colab_type": "code",
    "id": "-xaA_G0jjPyi",
    "outputId": "a3a29787-8518-453a-c6cd-51d296ee821e"
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Field</th>\n",
       "      <th>Parents</th>\n",
       "      <th>Properties</th>\n",
       "      <th>Definition</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>VB</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': '   U[1]', 'Index': ' hypercharge', 'Coupling': ' g1', 'SSB': ' g1', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>VWB</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': ' SU[2]', 'Index': ' left', 'Coupling': '        g2', 'SSB': '        g2', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>VG</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': '  SU[3]', 'Index': ' color', 'Coupling': '       g3', 'SSB': '       g3', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>q</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1/6', '2', '3'], 'Lorentz': 'WeylFermion', 'multiplet': ['uL', 'dL']}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>l</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['-1/2', '2', '1'], 'Lorentz': 'WeylFermion', 'multiplet': ['vL', 'eL']}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>d</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1/3', '1', '-3'], 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>u</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['-2/3', '1', '-3'], 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>e</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1', '1', '1'], 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>H</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '1', 'Groups': ['1/2', '2', '1'], 'Lorentz': 'Scalar', 'multiplet': ['Hp', 'H0']}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>uL</td>\n",
       "      <td>q</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>dL</td>\n",
       "      <td>q</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>vL</td>\n",
       "      <td>l</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>eL</td>\n",
       "      <td>l</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>dR</td>\n",
       "      <td>d</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>uR</td>\n",
       "      <td>u</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>eR</td>\n",
       "      <td>e</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Hp</td>\n",
       "      <td>H</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'NF': '1'}</td>\n",
       "      <td>GaugeES</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>H0</td>\n",
       "      <td>H</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'NF': '1'}</td>\n",
       "      <td>GaugeES</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Field  ...                  Definition\n",
       "0     VB  ...                     GaugeES\n",
       "1    VWB  ...                     GaugeES\n",
       "2     VG  ...                     GaugeES\n",
       "3      q  ...  WeylFermionAndIndermediate\n",
       "4      l  ...  WeylFermionAndIndermediate\n",
       "5      d  ...  WeylFermionAndIndermediate\n",
       "6      u  ...  WeylFermionAndIndermediate\n",
       "7      e  ...  WeylFermionAndIndermediate\n",
       "8      H  ...  WeylFermionAndIndermediate\n",
       "9     uL  ...  WeylFermionAndIndermediate\n",
       "10    dL  ...  WeylFermionAndIndermediate\n",
       "11    vL  ...  WeylFermionAndIndermediate\n",
       "12    eL  ...  WeylFermionAndIndermediate\n",
       "13    dR  ...  WeylFermionAndIndermediate\n",
       "14    uR  ...  WeylFermionAndIndermediate\n",
       "15    eR  ...  WeylFermionAndIndermediate\n",
       "16    Hp  ...                     GaugeES\n",
       "17    H0  ...                     GaugeES\n",
       "\n",
       "[18 rows x 4 columns]"
      ]
     },
     "execution_count": 7,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pd.DataFrame( particles )#.loc[8]['Properties']"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "UYjdaJE4jPyk"
   },
   "source": [
    "## 2) Get particles from `DEFINITION[EWSB]`"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 340
    },
    "colab_type": "code",
    "id": "2wuReJa6jPyk",
    "outputId": "f5950125-9e00-4ea1-c101-a8c25082a446"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'DiracSpinors': {'Fd': ['DL', 'conj[DR]'],\n",
       "  'Fe': ['EL', 'conj[ER]'],\n",
       "  'Fu': ['UL', 'conj[UR]'],\n",
       "  'Fv': ['vL', 0]},\n",
       " 'GaugeES': {'Fd1': ['FdL', 0],\n",
       "  'Fd2': [0, 'FdR'],\n",
       "  'Fe1': ['Fe1', 0],\n",
       "  'Fe2': [0, 'Fe2'],\n",
       "  'Fu1': ['Fu1', 0],\n",
       "  'Fu2': [0, 'Fu2']},\n",
       " 'GaugeSector': [[['VB', 'VWB[3]'], ['VP', 'VZ'], 'ZZ'],\n",
       "  [['VWB[1]', 'VWB[2]'], ['VWp', 'conj[VWp]'], 'ZW']],\n",
       " 'MatterSector': [[[['dL'], ['conj[dR]']], [['DL', 'Vd'], ['DR', 'Ud']]],\n",
       "  [[['uL'], ['conj[uR]']], [['UL', 'Vu'], ['UR', 'Uu']]],\n",
       "  [[['eL'], ['conj[eR]']], [['EL', 'Ve'], ['ER', 'Ue']]]],\n",
       " 'VEVs': [['H0',\n",
       "   ['v', '1/Sqrt[2]'],\n",
       "   ['Ah', '\\\\[ImaginaryI]/Sqrt[2]'],\n",
       "   ['hh', '1/Sqrt[2]']]]}"
      ]
     },
     "execution_count": 8,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "dsbd=parse_mathematica_list_of_list(smf,NAME='DEFINITION',KEY='EWSB')\n",
    "dsbd"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "Cf9_8QgPjPyl"
   },
   "outputs": [],
   "source": [
    "DEFINITION={}\n",
    "DEFINITION['EWSB']={}\n",
    "Bidiagonal=[]\n",
    "Diagonal=[]\n",
    "VEV=[]\n",
    "for k in dsbd.keys():\n",
    "    \n",
    "    for w in dsbd[k]:\n",
    "        if k=='GaugeSector' or k=='MatterSector':\n",
    "            if np.array(w).shape==(2,2):\n",
    "                weyl=bidiagonal(w,k)\n",
    "                Bidiagonal.append(weyl)\n",
    "            elif np.array(w).shape==(3,):\n",
    "                symm=diagonal(w,k)\n",
    "                Diagonal.append(symm)\n",
    "        elif k=='VEVs':\n",
    "            VEV.append(get_vev(w,k))\n",
    "\n",
    "DEFINITION['EWSB']['Bidiagonal']=Bidiagonal\n",
    "DEFINITION['EWSB']['Diagonal']=Diagonal\n",
    "DEFINITION['EWSB']['VEVs']=VEV\n",
    "DEFINITION['EWSB']['DiracSpinors']=dsbd['DiracSpinors']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "KSKoARd2jPyq",
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "particles=particles+rotations_to_particles(DEFINITION,key='EWSB',lr='',sep='')\n",
    "particles=particles+rotations_to_particles(DEFINITION,key='EWSB',lr='left',sep='_')\n",
    "particles=particles+rotations_to_particles(DEFINITION,key='EWSB',lr='right',sep='_')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "k_-isjiZjPyt"
   },
   "outputs": [],
   "source": [
    "newparticles=[]\n",
    "for d in DEFINITION['EWSB']['VEVs']:\n",
    "    particle=vev_to_particles(d,cp='Real')\n",
    "    newparticles.append(particle)\n",
    "    particle=vev_to_particles(d,cp='Imaginary')\n",
    "    newparticles.append(particle)\n",
    "particles=particles+newparticles    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "cTBVVqBRjPyw"
   },
   "outputs": [],
   "source": [
    "newparticles=[]\n",
    "dict_of_spinors=DEFINITION['EWSB']['DiracSpinors']\n",
    "for f in dict_of_spinors.keys():\n",
    "    particle=spinor_to_particles(dict_of_spinors,DEFINITION,f)\n",
    "    newparticles.append(particle)\n",
    "particles=particles+newparticles    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "flFKwQnijPyy"
   },
   "outputs": [],
   "source": [
    "f=open('parseparticles.json','w')\n",
    "json.dump(particles,f)\n",
    "f.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 1000
    },
    "colab_type": "code",
    "id": "AHvCHSk_jPyz",
    "outputId": "3b6f728e-90c1-4fe9-d691-aad6332e7b77"
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Field</th>\n",
       "      <th>Parents</th>\n",
       "      <th>Properties</th>\n",
       "      <th>Definition</th>\n",
       "      <th>rotation</th>\n",
       "      <th>Block</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>VB</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': '   U[1]', 'Index': ' hypercharge', 'Coupling': ' g1', 'SSB': ' g1', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>VWB</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': ' SU[2]', 'Index': ' left', 'Coupling': '        g2', 'SSB': '        g2', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>VG</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': '  SU[3]', 'Index': ' color', 'Coupling': '       g3', 'SSB': '       g3', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>q</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1/6', '2', '3'], 'Lorentz': 'WeylFermion', 'multiplet': ['uL', 'dL']}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>l</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['-1/2', '2', '1'], 'Lorentz': 'WeylFermion', 'multiplet': ['vL', 'eL']}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>d</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1/3', '1', '-3'], 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>u</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['-2/3', '1', '-3'], 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>e</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1', '1', '1'], 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>H</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '1', 'Groups': ['1/2', '2', '1'], 'Lorentz': 'Scalar', 'multiplet': ['Hp', 'H0']}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>uL</td>\n",
       "      <td>q</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>dL</td>\n",
       "      <td>q</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>vL</td>\n",
       "      <td>l</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>eL</td>\n",
       "      <td>l</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>dR</td>\n",
       "      <td>d</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>uR</td>\n",
       "      <td>u</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>eR</td>\n",
       "      <td>e</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Hp</td>\n",
       "      <td>H</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'NF': '1'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>H0</td>\n",
       "      <td>H</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'NF': '1'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>VP</td>\n",
       "      <td>VB</td>\n",
       "      <td>{'Mass_basis': ['VP', 'VZ'], 'Interaction_basis': ['VB', 'VWB[3]'], 'Lorentz': 'Vector'}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>ZZ</td>\n",
       "      <td>GaugeSector</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>VZ</td>\n",
       "      <td>VWB[3]</td>\n",
       "      <td>{'Mass_basis': ['VP', 'VZ'], 'Interaction_basis': ['VB', 'VWB[3]'], 'Lorentz': 'Vector'}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>ZZ</td>\n",
       "      <td>GaugeSector</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>VWp</td>\n",
       "      <td>VWB[1]</td>\n",
       "      <td>{'Mass_basis': ['VWp', 'conj[VWp]'], 'Interaction_basis': ['VWB[1]', 'VWB[2]'], 'Lorentz': 'Vector'}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>ZW</td>\n",
       "      <td>GaugeSector</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>DL</td>\n",
       "      <td>dL</td>\n",
       "      <td>{'Mass_basis': 'DL', 'Interaction_basis': ['dL'], 'Chirality': 'left', 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Vd</td>\n",
       "      <td>MatterField</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>UL</td>\n",
       "      <td>uL</td>\n",
       "      <td>{'Mass_basis': 'UL', 'Interaction_basis': ['uL'], 'Chirality': 'left', 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Vu</td>\n",
       "      <td>MatterField</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>EL</td>\n",
       "      <td>eL</td>\n",
       "      <td>{'Mass_basis': 'EL', 'Interaction_basis': ['eL'], 'Chirality': 'left', 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Ve</td>\n",
       "      <td>MatterField</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>DR</td>\n",
       "      <td>conj[dR]</td>\n",
       "      <td>{'Mass_basis': 'DR', 'Interaction_basis': ['conj[dR]'], 'Chirality': 'right', 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Ud</td>\n",
       "      <td>MatterField</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>UR</td>\n",
       "      <td>conj[uR]</td>\n",
       "      <td>{'Mass_basis': 'UR', 'Interaction_basis': ['conj[uR]'], 'Chirality': 'right', 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Uu</td>\n",
       "      <td>MatterField</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>ER</td>\n",
       "      <td>conj[eR]</td>\n",
       "      <td>{'Mass_basis': 'ER', 'Interaction_basis': ['conj[eR]'], 'Chirality': 'right', 'Lorentz': 'WeylFermion'}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Ue</td>\n",
       "      <td>MatterField</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>hh</td>\n",
       "      <td>H0</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'vev': 'v', 'CP': 'Real', 'Coefficient': '1/Sqrt[2]'}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>NaN</td>\n",
       "      <td>VEVs</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>Ah</td>\n",
       "      <td>H0</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'vev': 'v', 'CP': 'Imaginary', 'Coefficient': '\\[ImaginaryI]/Sqrt[2]'}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>NaN</td>\n",
       "      <td>VEVs</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>Fd</td>\n",
       "      <td>DL</td>\n",
       "      <td>{'Lorentz': 'DiracSpinor', 'DiracSpinor': ['DL', 'conj[DR]']}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>NaN</td>\n",
       "      <td>DiracSpinors</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>Fe</td>\n",
       "      <td>EL</td>\n",
       "      <td>{'Lorentz': 'DiracSpinor', 'DiracSpinor': ['EL', 'conj[ER]']}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>NaN</td>\n",
       "      <td>DiracSpinors</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>Fu</td>\n",
       "      <td>UL</td>\n",
       "      <td>{'Lorentz': 'DiracSpinor', 'DiracSpinor': ['UL', 'conj[UR]']}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>NaN</td>\n",
       "      <td>DiracSpinors</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>Fv</td>\n",
       "      <td>vL</td>\n",
       "      <td>{'Lorentz': 'MajoranaSpinor', 'MajoranaSpinor': ['vL', 0]}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>NaN</td>\n",
       "      <td>DiracSpinors</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Field   Parents  ... rotation         Block\n",
       "0     VB      None  ...      NaN           NaN\n",
       "1    VWB      None  ...      NaN           NaN\n",
       "2     VG      None  ...      NaN           NaN\n",
       "3      q      None  ...      NaN           NaN\n",
       "4      l      None  ...      NaN           NaN\n",
       "5      d      None  ...      NaN           NaN\n",
       "6      u      None  ...      NaN           NaN\n",
       "7      e      None  ...      NaN           NaN\n",
       "8      H      None  ...      NaN           NaN\n",
       "9     uL         q  ...      NaN           NaN\n",
       "10    dL         q  ...      NaN           NaN\n",
       "11    vL         l  ...      NaN           NaN\n",
       "12    eL         l  ...      NaN           NaN\n",
       "13    dR         d  ...      NaN           NaN\n",
       "14    uR         u  ...      NaN           NaN\n",
       "15    eR         e  ...      NaN           NaN\n",
       "16    Hp         H  ...      NaN           NaN\n",
       "17    H0         H  ...      NaN           NaN\n",
       "18    VP        VB  ...       ZZ   GaugeSector\n",
       "19    VZ    VWB[3]  ...       ZZ   GaugeSector\n",
       "20   VWp    VWB[1]  ...       ZW   GaugeSector\n",
       "21    DL        dL  ...       Vd   MatterField\n",
       "22    UL        uL  ...       Vu   MatterField\n",
       "23    EL        eL  ...       Ve   MatterField\n",
       "24    DR  conj[dR]  ...       Ud   MatterField\n",
       "25    UR  conj[uR]  ...       Uu   MatterField\n",
       "26    ER  conj[eR]  ...       Ue   MatterField\n",
       "27    hh        H0  ...      NaN          VEVs\n",
       "28    Ah        H0  ...      NaN          VEVs\n",
       "29    Fd        DL  ...      NaN  DiracSpinors\n",
       "30    Fe        EL  ...      NaN  DiracSpinors\n",
       "31    Fu        UL  ...      NaN  DiracSpinors\n",
       "32    Fv        vL  ...      NaN  DiracSpinors\n",
       "\n",
       "[33 rows x 6 columns]"
      ]
     },
     "execution_count": 14,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pd.read_json('parseparticles.json')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "colab_type": "text",
    "id": "sIVy3jl5451r"
   },
   "source": [
    "## Generate `particles.json` from particles dict"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "HIyNIjw5jPy2"
   },
   "outputs": [],
   "source": [
    "prt=pd.DataFrame(particles)\n",
    "prt=prt.where((pd.notnull(prt)), None)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "0PLbjIhH451z"
   },
   "outputs": [],
   "source": [
    "def get_weylfermion_LaTeX(s):\n",
    "    if len(s)==2 and re.search('[LR]$',s):\n",
    "        sini=s[0]\n",
    "        if sini=='v':\n",
    "            sini=r'\\nu'\n",
    "        s=sini+'_'+s[1]\n",
    "    return s\n",
    "\n",
    "def get_abs_groups(grps=['1/2','2','1'],u1_abs='1/2',su2='2',su3_abs='1'):\n",
    "    sgrps=str(grps).replace(' ','')\n",
    "    return re.search(r\"\\['-*{}','{}','{}'\".format(u1_abs,su2,su3_abs), sgrps )\n",
    "\n",
    "def update_EWSB_Fermion_Description(p,prt):\n",
    "    gprt=prt[prt.get('Field')==p.get('Parents')].reset_index(drop=True)\n",
    "    if gprt.loc[0,'rotation']:\n",
    "        gprt=prt[prt.get('Field')==gprt.loc[0,'Parents']].reset_index(drop=True)\n",
    "    cmpt=gprt.loc[0,'Field']\n",
    "    gprt=prt[prt.get('Field')==gprt.loc[0,'Parents']].reset_index(drop=True)\n",
    "    if get_abs_groups( gprt.loc[0,'Properties'].get('Groups'), u1_abs='1/6',su2='2',su3_abs='3' ):\n",
    "        if cmpt==gprt.loc[0,'Properties'].get('multiplet')[0]:\n",
    "            p['Description']='Up-Quarks'\n",
    "        else:\n",
    "            p['Description']='Down-Quarks'\n",
    "    if get_abs_groups( gprt.loc[0,'Properties'].get('Groups'), u1_abs='1/2',su2='2',su3_abs='1' ):\n",
    "        if cmpt==gprt.loc[0,'Properties'].get('multiplet')[0]:\n",
    "            p['Description']='Neutrinos'\n",
    "        else:\n",
    "            p['Description']='Leptons'\n",
    "    return p\n",
    "\n",
    "def get_EWSB_gauge_bosons(p,prt):\n",
    "    mb=p.get('Properties').get('Mass_basis')\n",
    "    #Get basis position\n",
    "    pi=[i for i in range(len(mb)) if p.get('Field')==mb[i] ][-1]\n",
    "    #Get parent particle\n",
    "    ppij=p.get('Properties').get('Interaction_basis')[pi]\n",
    "    #Simplify parent particle name\n",
    "    ppi=re.sub('\\[[0-9]+\\]','',ppij)\n",
    "    #Get Gauge Group from simplified parent particle name\n",
    "    gppi=(prt[prt.get('Field')==ppi].reset_index(drop=True)).loc[0,'Properties'].get('Group').strip()\n",
    "    if gppi=='U[1]':\n",
    "        p['Description']='Photon'\n",
    "    elif gppi=='SU[2]':\n",
    "        #Get SU[2] multiplet position\n",
    "        j=eval(re.search('\\[([0-9]+)\\]',ppij).groups()[0])\n",
    "        if j==3:\n",
    "            p['Description']='Z-Boson'\n",
    "        elif j<3:\n",
    "            p['Description']='W+ - Boson'\n",
    "    return p"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 34
    },
    "colab_type": "code",
    "id": "q53vGzEp452J",
    "outputId": "d708b682-c21f-4548-d3bb-fe2852da26d7",
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "va 0\n"
     ]
    }
   ],
   "source": [
    "#TODO: Make false when there are more scalar doublets\n",
    "Add_Charged_SM_Higgs_to_EWSB=True\n",
    "newparticles=[]\n",
    "for i in prt.index:\n",
    "    p=prt.loc[i].to_dict()\n",
    "    if p.get('Definition')=='GaugeES' and p.get('Properties').get('Lorentz').strip()=='Vector':\n",
    "        if  p.get('Properties').get('Group').strip()=='U[1]':\n",
    "            p['Description']='B-Boson'\n",
    "            newparticles.append(p)\n",
    "            newparticles.append({'Definition':'GaugeES','Field':'gB',\n",
    "                               'Description':'B-Boson Ghost',\n",
    "                               'Parents':p.get('Field'),'Properties':{\n",
    "                               'Group':p.get('Properties').get('Group'),'Lorentz':'Scalar'}})\n",
    "            #break\n",
    "        elif p.get('Properties').get('Group').strip()=='SU[2]':\n",
    "            p['Description']='W-Bosons'\n",
    "            newparticles.append(p)\n",
    "            newparticles.append({'Definition':'GaugeES','Field':'gWB',\n",
    "                               'Description':'W-Boson Ghost',\n",
    "                               'Parents':p.get('Field'),'Properties':{\n",
    "                                'Group':p.get('Properties').get('Group'), 'Lorentz':'Scalar'}})        \n",
    "            \n",
    "        elif p.get('Properties').get('Group').strip()=='SU[3]':\n",
    "            p['Description']='Gluon'\n",
    "            newparticles.append(p)\n",
    "            newparticles.append({'Definition':'GaugeES','Field':'gG',\n",
    "                               'Description':'Gluon Ghost',\n",
    "                               'Parents':p.get('Field'),'Properties':{\n",
    "                               'Group':p.get('Properties').get('Group'),'Lorentz':'Scalar'}})\n",
    "        #Search for daugthers\n",
    "    if p.get('Definition')=='WeylFermionAndIndermediate' and p.get('Properties').get('Lorentz')=='Scalar':\n",
    "        #Generic Doublet scalar: Include SM Higgs\n",
    "        #re.search(r\"\\['-*1/2','2','1'\",str(p.get('Properties').get('Groups')).replace(' ',''))\n",
    "        if get_abs_groups(p.get('Properties').get('Groups'),u1_abs='1/2',su2='2',su3_abs='1'):\n",
    "            p.get('Properties')['update_Description']={'PDG'       : [0],\n",
    "                                                           'Width'     : 0,\n",
    "                                                           'Mass'      : 'Automatic',\n",
    "                                                           'LaTeX'     : p.get('Field'),\n",
    "                                                           'OutputName': p.get('Field') }\n",
    "            newparticles.append(p)\n",
    "            #print(p)\n",
    "            dprt=prt[prt['Parents']==p.get('Field')].reset_index(drop=True)\n",
    "            if dprt.shape[0]==2:\n",
    "                print('va',0)\n",
    "                for j in dprt.index:\n",
    "                    dp=dprt.loc[j].to_dict()\n",
    "                    #Higss components\n",
    "                    charge=[]\n",
    "                    if dp.get('Parents')=='H':\n",
    "                        if p.get('Properties').get('Groups')[0].find('-')==-1:\n",
    "                            charge=['Charged','Neutral']\n",
    "                        else:\n",
    "                            charge=['Neutral','Charged']\n",
    "                        #print('va j',j,charge)\n",
    "                            \n",
    "                        if charge[j]=='Charged':\n",
    "                            #Obtain from general dictionary: Charged Higgs NOT Physical\n",
    "                            dp.get('Properties')['update_Description']={'PDG'  :[0],\n",
    "                                    'Width':0,\n",
    "                                    'Mass' : 'Automatic',\n",
    "                                    'FeynArtsNr' : 2,\n",
    "                                    'LaTeX' : 'H^+',\n",
    "                                    'OutputName' : dp.get('Field')                            \n",
    "                                   }\n",
    "                            dp.get('Properties')['ElectricCharge']=1\n",
    "                            newparticles.append(dp)\n",
    "                            #TODO: Add EWSB charged scalar only if no more scalars\n",
    "                            # with parent Hp\n",
    "                            if Add_Charged_SM_Higgs_to_EWSB:\n",
    "                                gdp={'Definition': 'EWSB',\n",
    "                                     'Field': 'Hp',\n",
    "                                     'Parents': 'Hp',\n",
    "                                     'Description': 'Charged Higgs',\n",
    "                                     'Properties': {'Lorentz': 'Scalar',\n",
    "                                     'NF': '1',\n",
    "                                     'ElectricCharge' : 1,\n",
    "                                     'update_Description': { #Obtain from general dictionary: Charged Higgs\n",
    "                                         'PDG'            : [0],\n",
    "                                         'PDG.IX'         : [0],\n",
    "                                         'Width'          : [0], \n",
    "                                         'Mass'           : [0],\n",
    "                                         'LaTeX'          : ['H^+','H^-' ],\n",
    "                                         'OutputName'     : ['Hp' ,'Hm'  ],\n",
    "                                        'ElectricCharge' : 1\n",
    "                                     }}}\n",
    "                                \n",
    "                                newparticles.append(gdp)                                \n",
    "                        else: #Neutral\n",
    "                            #Obtain from general dictionary: Complex Neutral Higgs\n",
    "                            dp.get('Properties')['update_Description']={'PDG'  :[0], \n",
    "                                    'Width':0,\n",
    "                                    'Mass' : 'Automatic',\n",
    "                                    'FeynArtsNr' : 1,\n",
    "                                    'LaTeX' : 'H^0',\n",
    "                                    'OutputName' : dp.get('Field')                        \n",
    "                                   }\n",
    "                            dp.get('Properties')['ElectricCharge']=0                            \n",
    "                            newparticles.append(dp)\n",
    "                            #print('va H0',j)\n",
    "                            \n",
    "                            #TODO: Define object to capture two daugthers\n",
    "                            gdprt=prt[prt['Parents']==dp.get('Field')].reset_index(drop=True)\n",
    "                            if gdprt.shape[0]==2:\n",
    "                                for k in gdprt.index:\n",
    "                                    gdp=gdprt.loc[k].to_dict()\n",
    "                                    if gdp.get('Properties').get('CP')=='Real':\n",
    "                                        gdp['Description']='Higgs'\n",
    "                                        #Obtain from general dictionary: Higgs\n",
    "                                        gdp.get('Properties')['update_Description']={\n",
    "                                            'PDG'         : [25],\n",
    "                                            'PDG.IX'      : [101000001]\n",
    "                                        }\n",
    "                                        newparticles.append(gdp)\n",
    "                                    else: #'CP' → Imaginary \n",
    "                                        gdp['Description']='Pseudo-Scalar Higgs'\n",
    "                                        #Obtain from general dictionary: Pseudo-Scalar Higgs\n",
    "                                        gdp.get('Properties')['update_Description']={\n",
    "                                            'PDG'         : [0],\n",
    "                                            'PDG.IX'      : [0],\n",
    "                                            'Mass'        : [0],\n",
    "                                            'Width'       : [0]\n",
    "                                        }\n",
    "                                        newparticles.append(gdp)                                        \n",
    "                                    #break\n",
    "                        #break\n",
    "            #break\n",
    "            #Search for daughters: two complex fields\n",
    "    if p.get('Definition')=='WeylFermionAndIndermediate' and p.get('Properties').get('Lorentz')=='WeylFermion':\n",
    "        p.get('Properties')['update_Description']={'LaTeX' : get_weylfermion_LaTeX( p.get('Field') )}            \n",
    "        newparticles.append(p)\n",
    "    #For the following two an or could be used\n",
    "    if p.get('Definition')=='EWSB' and p.get('Properties').get('Lorentz')=='DiracSpinor':\n",
    "        p=update_EWSB_Fermion_Description(p,prt)\n",
    "        newparticles.append(p)\n",
    "    if p.get('Definition')=='EWSB' and p.get('Properties').get('Lorentz')=='MajoranaSpinor':\n",
    "        p=update_EWSB_Fermion_Description(p,prt)\n",
    "        newparticles.append(p)\n",
    "        #break        \n",
    "        #break\n",
    "    if p.get('Definition')=='EWSB' and p.get('Properties').get('Lorentz')=='Vector':\n",
    "        #if p.get('Field')=='VP':#'VP':#'VWp':#'VZ':\n",
    "        #    break\n",
    "        p=get_EWSB_gauge_bosons(p,prt)\n",
    "        newparticles.append(p)        "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 1000
    },
    "colab_type": "code",
    "id": "UES-oBQ4452O",
    "outputId": "866f7e11-41c3-42b8-e73f-58ae3863000d"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'Block': None,\n",
       "  'Definition': 'GaugeES',\n",
       "  'Description': 'B-Boson',\n",
       "  'Field': 'VB',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Coupling': ' g1',\n",
       "   'Group': '   U[1]',\n",
       "   'Index': ' hypercharge',\n",
       "   'Lorentz': 'Vector',\n",
       "   'SSB': ' g1'},\n",
       "  'rotation': None},\n",
       " {'Definition': 'GaugeES',\n",
       "  'Description': 'B-Boson Ghost',\n",
       "  'Field': 'gB',\n",
       "  'Parents': 'VB',\n",
       "  'Properties': {'Group': '   U[1]', 'Lorentz': 'Scalar'}},\n",
       " {'Block': None,\n",
       "  'Definition': 'GaugeES',\n",
       "  'Description': 'W-Bosons',\n",
       "  'Field': 'VWB',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Coupling': '        g2',\n",
       "   'Group': ' SU[2]',\n",
       "   'Index': ' left',\n",
       "   'Lorentz': 'Vector',\n",
       "   'SSB': '        g2'},\n",
       "  'rotation': None},\n",
       " {'Definition': 'GaugeES',\n",
       "  'Description': 'W-Boson Ghost',\n",
       "  'Field': 'gWB',\n",
       "  'Parents': 'VWB',\n",
       "  'Properties': {'Group': ' SU[2]', 'Lorentz': 'Scalar'}},\n",
       " {'Block': None,\n",
       "  'Definition': 'GaugeES',\n",
       "  'Description': 'Gluon',\n",
       "  'Field': 'VG',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Coupling': '       g3',\n",
       "   'Group': '  SU[3]',\n",
       "   'Index': ' color',\n",
       "   'Lorentz': 'Vector',\n",
       "   'SSB': '       g3'},\n",
       "  'rotation': None},\n",
       " {'Definition': 'GaugeES',\n",
       "  'Description': 'Gluon Ghost',\n",
       "  'Field': 'gG',\n",
       "  'Parents': 'VG',\n",
       "  'Properties': {'Group': '  SU[3]', 'Lorentz': 'Scalar'}},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'q',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Groups': ['1/6', '2', '3'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'multiplet': ['uL', 'dL'],\n",
       "   'update_Description': {'LaTeX': 'q'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'l',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Groups': ['-1/2', '2', '1'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'multiplet': ['vL', 'eL'],\n",
       "   'update_Description': {'LaTeX': 'l'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'd',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Groups': ['1/3', '1', '-3'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'd'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'u',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Groups': ['-2/3', '1', '-3'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'u'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'e',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Groups': ['1', '1', '1'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'e'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'H',\n",
       "  'Parents': None,\n",
       "  'Properties': {'Groups': ['1/2', '2', '1'],\n",
       "   'Lorentz': 'Scalar',\n",
       "   'NF': '1',\n",
       "   'multiplet': ['Hp', 'H0'],\n",
       "   'update_Description': {'LaTeX': 'H',\n",
       "    'Mass': 'Automatic',\n",
       "    'OutputName': 'H',\n",
       "    'PDG': [0],\n",
       "    'Width': 0}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'GaugeES',\n",
       "  'Field': 'Hp',\n",
       "  'Parents': 'H',\n",
       "  'Properties': {'ElectricCharge': 1,\n",
       "   'Lorentz': 'Scalar',\n",
       "   'NF': '1',\n",
       "   'update_Description': {'FeynArtsNr': 2,\n",
       "    'LaTeX': 'H^+',\n",
       "    'Mass': 'Automatic',\n",
       "    'OutputName': 'Hp',\n",
       "    'PDG': [0],\n",
       "    'Width': 0}},\n",
       "  'rotation': None},\n",
       " {'Definition': 'EWSB',\n",
       "  'Description': 'Charged Higgs',\n",
       "  'Field': 'Hp',\n",
       "  'Parents': 'Hp',\n",
       "  'Properties': {'ElectricCharge': 1,\n",
       "   'Lorentz': 'Scalar',\n",
       "   'NF': '1',\n",
       "   'update_Description': {'ElectricCharge': 1,\n",
       "    'LaTeX': ['H^+', 'H^-'],\n",
       "    'Mass': [0],\n",
       "    'OutputName': ['Hp', 'Hm'],\n",
       "    'PDG': [0],\n",
       "    'PDG.IX': [0],\n",
       "    'Width': [0]}}},\n",
       " {'Block': None,\n",
       "  'Definition': 'GaugeES',\n",
       "  'Field': 'H0',\n",
       "  'Parents': 'H',\n",
       "  'Properties': {'ElectricCharge': 0,\n",
       "   'Lorentz': 'Scalar',\n",
       "   'NF': '1',\n",
       "   'update_Description': {'FeynArtsNr': 1,\n",
       "    'LaTeX': 'H^0',\n",
       "    'Mass': 'Automatic',\n",
       "    'OutputName': 'H0',\n",
       "    'PDG': [0],\n",
       "    'Width': 0}},\n",
       "  'rotation': None},\n",
       " {'Block': 'VEVs',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'Higgs',\n",
       "  'Field': 'hh',\n",
       "  'Parents': 'H0',\n",
       "  'Properties': {'CP': 'Real',\n",
       "   'Coefficient': '1/Sqrt[2]',\n",
       "   'Lorentz': 'Scalar',\n",
       "   'update_Description': {'PDG': [25], 'PDG.IX': [101000001]},\n",
       "   'vev': 'v'},\n",
       "  'rotation': None},\n",
       " {'Block': 'VEVs',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'Pseudo-Scalar Higgs',\n",
       "  'Field': 'Ah',\n",
       "  'Parents': 'H0',\n",
       "  'Properties': {'CP': 'Imaginary',\n",
       "   'Coefficient': '\\\\[ImaginaryI]/Sqrt[2]',\n",
       "   'Lorentz': 'Scalar',\n",
       "   'update_Description': {'Mass': [0],\n",
       "    'PDG': [0],\n",
       "    'PDG.IX': [0],\n",
       "    'Width': [0]},\n",
       "   'vev': 'v'},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'uL',\n",
       "  'Parents': 'q',\n",
       "  'Properties': {'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'u_L'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'dL',\n",
       "  'Parents': 'q',\n",
       "  'Properties': {'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'd_L'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'vL',\n",
       "  'Parents': 'l',\n",
       "  'Properties': {'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': '\\\\nu_L'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'eL',\n",
       "  'Parents': 'l',\n",
       "  'Properties': {'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'e_L'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'dR',\n",
       "  'Parents': 'd',\n",
       "  'Properties': {'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'd_R'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'uR',\n",
       "  'Parents': 'u',\n",
       "  'Properties': {'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'u_R'}},\n",
       "  'rotation': None},\n",
       " {'Block': None,\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'eR',\n",
       "  'Parents': 'e',\n",
       "  'Properties': {'Lorentz': 'WeylFermion',\n",
       "   'NF': '3',\n",
       "   'update_Description': {'LaTeX': 'e_R'}},\n",
       "  'rotation': None},\n",
       " {'Block': 'GaugeSector',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'Photon',\n",
       "  'Field': 'VP',\n",
       "  'Parents': 'VB',\n",
       "  'Properties': {'Interaction_basis': ['VB', 'VWB[3]'],\n",
       "   'Lorentz': 'Vector',\n",
       "   'Mass_basis': ['VP', 'VZ']},\n",
       "  'rotation': 'ZZ'},\n",
       " {'Block': 'GaugeSector',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'Z-Boson',\n",
       "  'Field': 'VZ',\n",
       "  'Parents': 'VWB[3]',\n",
       "  'Properties': {'Interaction_basis': ['VB', 'VWB[3]'],\n",
       "   'Lorentz': 'Vector',\n",
       "   'Mass_basis': ['VP', 'VZ']},\n",
       "  'rotation': 'ZZ'},\n",
       " {'Block': 'GaugeSector',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'W+ - Boson',\n",
       "  'Field': 'VWp',\n",
       "  'Parents': 'VWB[1]',\n",
       "  'Properties': {'Interaction_basis': ['VWB[1]', 'VWB[2]'],\n",
       "   'Lorentz': 'Vector',\n",
       "   'Mass_basis': ['VWp', 'conj[VWp]']},\n",
       "  'rotation': 'ZW'},\n",
       " {'Block': 'MatterField',\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'DL',\n",
       "  'Parents': 'dL',\n",
       "  'Properties': {'Chirality': 'left',\n",
       "   'Interaction_basis': ['dL'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'Mass_basis': 'DL',\n",
       "   'update_Description': {'LaTeX': 'D_L'}},\n",
       "  'rotation': 'Vd'},\n",
       " {'Block': 'MatterField',\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'UL',\n",
       "  'Parents': 'uL',\n",
       "  'Properties': {'Chirality': 'left',\n",
       "   'Interaction_basis': ['uL'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'Mass_basis': 'UL',\n",
       "   'update_Description': {'LaTeX': 'U_L'}},\n",
       "  'rotation': 'Vu'},\n",
       " {'Block': 'MatterField',\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'EL',\n",
       "  'Parents': 'eL',\n",
       "  'Properties': {'Chirality': 'left',\n",
       "   'Interaction_basis': ['eL'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'Mass_basis': 'EL',\n",
       "   'update_Description': {'LaTeX': 'E_L'}},\n",
       "  'rotation': 'Ve'},\n",
       " {'Block': 'MatterField',\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'DR',\n",
       "  'Parents': 'conj[dR]',\n",
       "  'Properties': {'Chirality': 'right',\n",
       "   'Interaction_basis': ['conj[dR]'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'Mass_basis': 'DR',\n",
       "   'update_Description': {'LaTeX': 'D_R'}},\n",
       "  'rotation': 'Ud'},\n",
       " {'Block': 'MatterField',\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'UR',\n",
       "  'Parents': 'conj[uR]',\n",
       "  'Properties': {'Chirality': 'right',\n",
       "   'Interaction_basis': ['conj[uR]'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'Mass_basis': 'UR',\n",
       "   'update_Description': {'LaTeX': 'U_R'}},\n",
       "  'rotation': 'Uu'},\n",
       " {'Block': 'MatterField',\n",
       "  'Definition': 'WeylFermionAndIndermediate',\n",
       "  'Field': 'ER',\n",
       "  'Parents': 'conj[eR]',\n",
       "  'Properties': {'Chirality': 'right',\n",
       "   'Interaction_basis': ['conj[eR]'],\n",
       "   'Lorentz': 'WeylFermion',\n",
       "   'Mass_basis': 'ER',\n",
       "   'update_Description': {'LaTeX': 'E_R'}},\n",
       "  'rotation': 'Ue'},\n",
       " {'Block': 'DiracSpinors',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'Down-Quarks',\n",
       "  'Field': 'Fd',\n",
       "  'Parents': 'DL',\n",
       "  'Properties': {'DiracSpinor': ['DL', 'conj[DR]'], 'Lorentz': 'DiracSpinor'},\n",
       "  'rotation': None},\n",
       " {'Block': 'DiracSpinors',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'Leptons',\n",
       "  'Field': 'Fe',\n",
       "  'Parents': 'EL',\n",
       "  'Properties': {'DiracSpinor': ['EL', 'conj[ER]'], 'Lorentz': 'DiracSpinor'},\n",
       "  'rotation': None},\n",
       " {'Block': 'DiracSpinors',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'Up-Quarks',\n",
       "  'Field': 'Fu',\n",
       "  'Parents': 'UL',\n",
       "  'Properties': {'DiracSpinor': ['UL', 'conj[UR]'], 'Lorentz': 'DiracSpinor'},\n",
       "  'rotation': None},\n",
       " {'Block': 'DiracSpinors',\n",
       "  'Definition': 'EWSB',\n",
       "  'Description': 'Neutrinos',\n",
       "  'Field': 'Fv',\n",
       "  'Parents': 'vL',\n",
       "  'Properties': {'Lorentz': 'MajoranaSpinor', 'MajoranaSpinor': ['vL', 0]},\n",
       "  'rotation': None}]"
      ]
     },
     "execution_count": 18,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "newparticles"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "OkYRh1qQ452S"
   },
   "outputs": [],
   "source": [
    "f=open('JSON/fullparticles.json','w')\n",
    "json.dump(newparticles,f)\n",
    "f.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "99MPmO97452V"
   },
   "outputs": [],
   "source": [
    "df=pd.read_json('JSON/fullparticles.json')\n",
    "df=df.where((pd.notnull(df)), None)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "zbYwcehE452c"
   },
   "outputs": [],
   "source": [
    "nprt=df#pd.DataFrame(newparticles)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 43,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 1000
    },
    "colab_type": "code",
    "id": "hDl_UwtM452f",
    "outputId": "9b5a0e11-0cb4-4c4c-da10-fac3bdf079f2"
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Field</th>\n",
       "      <th>Parents</th>\n",
       "      <th>Properties</th>\n",
       "      <th>Definition</th>\n",
       "      <th>rotation</th>\n",
       "      <th>Block</th>\n",
       "      <th>Description</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>VB</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': '   U[1]', 'Index': ' hypercharge', 'Coupling': ' g1', 'SSB': ' g1', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>B-Boson</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>gB</td>\n",
       "      <td>VB</td>\n",
       "      <td>{'Group': '   U[1]', 'Lorentz': 'Scalar'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>B-Boson Ghost</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>VWB</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': ' SU[2]', 'Index': ' left', 'Coupling': '        g2', 'SSB': '        g2', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>W-Bosons</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>gWB</td>\n",
       "      <td>VWB</td>\n",
       "      <td>{'Group': ' SU[2]', 'Lorentz': 'Scalar'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>W-Boson Ghost</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>VG</td>\n",
       "      <td>None</td>\n",
       "      <td>{'Group': '  SU[3]', 'Index': ' color', 'Coupling': '       g3', 'SSB': '       g3', 'Lorentz': 'Vector'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>Gluon</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>gG</td>\n",
       "      <td>VG</td>\n",
       "      <td>{'Group': '  SU[3]', 'Lorentz': 'Scalar'}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>Gluon Ghost</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>q</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1/6', '2', '3'], 'Lorentz': 'WeylFermion', 'multiplet': ['uL', 'dL'], 'update_Description': {'LaTeX': 'q'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>l</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['-1/2', '2', '1'], 'Lorentz': 'WeylFermion', 'multiplet': ['vL', 'eL'], 'update_Description': {'LaTeX': 'l'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>8</th>\n",
       "      <td>d</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1/3', '1', '-3'], 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'd'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>9</th>\n",
       "      <td>u</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['-2/3', '1', '-3'], 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'u'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>10</th>\n",
       "      <td>e</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '3', 'Groups': ['1', '1', '1'], 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'e'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>11</th>\n",
       "      <td>H</td>\n",
       "      <td>None</td>\n",
       "      <td>{'NF': '1', 'Groups': ['1/2', '2', '1'], 'Lorentz': 'Scalar', 'multiplet': ['Hp', 'H0'], 'update_Description': {'PDG': [0], 'Width': 0, 'Mass': 'Automatic', 'LaTeX': 'H', 'OutputName': 'H'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>12</th>\n",
       "      <td>Hp</td>\n",
       "      <td>H</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'NF': '1', 'update_Description': {'PDG': [0], 'Width': 0, 'Mass': 'Automatic', 'FeynArtsNr': 2, 'LaTeX': 'H^+', 'OutputName': 'Hp'}, 'ElectricCharge': 1}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>13</th>\n",
       "      <td>Hp</td>\n",
       "      <td>Hp</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'NF': '1', 'ElectricCharge': 1, 'update_Description': {'PDG': [0], 'PDG.IX': [0], 'Width': [0], 'Mass': [0], 'LaTeX': ['H^+', 'H^-'], 'OutputName': ['Hp', 'Hm'], 'ElectricCha...</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>Charged Higgs</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>H0</td>\n",
       "      <td>H</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'NF': '1', 'update_Description': {'PDG': [0], 'Width': 0, 'Mass': 'Automatic', 'FeynArtsNr': 1, 'LaTeX': 'H^0', 'OutputName': 'H0'}, 'ElectricCharge': 0}</td>\n",
       "      <td>GaugeES</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>15</th>\n",
       "      <td>hh</td>\n",
       "      <td>H0</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'vev': 'v', 'CP': 'Real', 'Coefficient': '1/Sqrt[2]', 'update_Description': {'PDG': [25], 'PDG.IX': [101000001]}}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>None</td>\n",
       "      <td>VEVs</td>\n",
       "      <td>Higgs</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>16</th>\n",
       "      <td>Ah</td>\n",
       "      <td>H0</td>\n",
       "      <td>{'Lorentz': 'Scalar', 'vev': 'v', 'CP': 'Imaginary', 'Coefficient': '\\[ImaginaryI]/Sqrt[2]', 'update_Description': {'PDG': [0], 'PDG.IX': [0], 'Mass': [0], 'Width': [0]}}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>None</td>\n",
       "      <td>VEVs</td>\n",
       "      <td>Pseudo-Scalar Higgs</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>17</th>\n",
       "      <td>uL</td>\n",
       "      <td>q</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3', 'update_Description': {'LaTeX': 'u_L'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>18</th>\n",
       "      <td>dL</td>\n",
       "      <td>q</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3', 'update_Description': {'LaTeX': 'd_L'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>19</th>\n",
       "      <td>vL</td>\n",
       "      <td>l</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3', 'update_Description': {'LaTeX': '\\nu_L'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>20</th>\n",
       "      <td>eL</td>\n",
       "      <td>l</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3', 'update_Description': {'LaTeX': 'e_L'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>21</th>\n",
       "      <td>dR</td>\n",
       "      <td>d</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3', 'update_Description': {'LaTeX': 'd_R'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>22</th>\n",
       "      <td>uR</td>\n",
       "      <td>u</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3', 'update_Description': {'LaTeX': 'u_R'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>23</th>\n",
       "      <td>eR</td>\n",
       "      <td>e</td>\n",
       "      <td>{'Lorentz': 'WeylFermion', 'NF': '3', 'update_Description': {'LaTeX': 'e_R'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>24</th>\n",
       "      <td>VP</td>\n",
       "      <td>VB</td>\n",
       "      <td>{'Mass_basis': ['VP', 'VZ'], 'Interaction_basis': ['VB', 'VWB[3]'], 'Lorentz': 'Vector'}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>ZZ</td>\n",
       "      <td>GaugeSector</td>\n",
       "      <td>Photon</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25</th>\n",
       "      <td>VZ</td>\n",
       "      <td>VWB[3]</td>\n",
       "      <td>{'Mass_basis': ['VP', 'VZ'], 'Interaction_basis': ['VB', 'VWB[3]'], 'Lorentz': 'Vector'}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>ZZ</td>\n",
       "      <td>GaugeSector</td>\n",
       "      <td>Z-Boson</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>26</th>\n",
       "      <td>VWp</td>\n",
       "      <td>VWB[1]</td>\n",
       "      <td>{'Mass_basis': ['VWp', 'conj[VWp]'], 'Interaction_basis': ['VWB[1]', 'VWB[2]'], 'Lorentz': 'Vector'}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>ZW</td>\n",
       "      <td>GaugeSector</td>\n",
       "      <td>W+ - Boson</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>27</th>\n",
       "      <td>DL</td>\n",
       "      <td>dL</td>\n",
       "      <td>{'Mass_basis': 'DL', 'Interaction_basis': ['dL'], 'Chirality': 'left', 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'D_L'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Vd</td>\n",
       "      <td>MatterField</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>28</th>\n",
       "      <td>UL</td>\n",
       "      <td>uL</td>\n",
       "      <td>{'Mass_basis': 'UL', 'Interaction_basis': ['uL'], 'Chirality': 'left', 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'U_L'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Vu</td>\n",
       "      <td>MatterField</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>29</th>\n",
       "      <td>EL</td>\n",
       "      <td>eL</td>\n",
       "      <td>{'Mass_basis': 'EL', 'Interaction_basis': ['eL'], 'Chirality': 'left', 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'E_L'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Ve</td>\n",
       "      <td>MatterField</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>30</th>\n",
       "      <td>DR</td>\n",
       "      <td>conj[dR]</td>\n",
       "      <td>{'Mass_basis': 'DR', 'Interaction_basis': ['conj[dR]'], 'Chirality': 'right', 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'D_R'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Ud</td>\n",
       "      <td>MatterField</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>31</th>\n",
       "      <td>UR</td>\n",
       "      <td>conj[uR]</td>\n",
       "      <td>{'Mass_basis': 'UR', 'Interaction_basis': ['conj[uR]'], 'Chirality': 'right', 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'U_R'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Uu</td>\n",
       "      <td>MatterField</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>32</th>\n",
       "      <td>ER</td>\n",
       "      <td>conj[eR]</td>\n",
       "      <td>{'Mass_basis': 'ER', 'Interaction_basis': ['conj[eR]'], 'Chirality': 'right', 'Lorentz': 'WeylFermion', 'update_Description': {'LaTeX': 'E_R'}}</td>\n",
       "      <td>WeylFermionAndIndermediate</td>\n",
       "      <td>Ue</td>\n",
       "      <td>MatterField</td>\n",
       "      <td>None</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>Fd</td>\n",
       "      <td>DL</td>\n",
       "      <td>{'Lorentz': 'DiracSpinor', 'DiracSpinor': ['DL', 'conj[DR]']}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>None</td>\n",
       "      <td>DiracSpinors</td>\n",
       "      <td>Down-Quarks</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34</th>\n",
       "      <td>Fe</td>\n",
       "      <td>EL</td>\n",
       "      <td>{'Lorentz': 'DiracSpinor', 'DiracSpinor': ['EL', 'conj[ER]']}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>None</td>\n",
       "      <td>DiracSpinors</td>\n",
       "      <td>Leptons</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>35</th>\n",
       "      <td>Fu</td>\n",
       "      <td>UL</td>\n",
       "      <td>{'Lorentz': 'DiracSpinor', 'DiracSpinor': ['UL', 'conj[UR]']}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>None</td>\n",
       "      <td>DiracSpinors</td>\n",
       "      <td>Up-Quarks</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>36</th>\n",
       "      <td>Fv</td>\n",
       "      <td>vL</td>\n",
       "      <td>{'Lorentz': 'MajoranaSpinor', 'MajoranaSpinor': ['vL', 0]}</td>\n",
       "      <td>EWSB</td>\n",
       "      <td>None</td>\n",
       "      <td>DiracSpinors</td>\n",
       "      <td>Neutrinos</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   Field   Parents  ...         Block          Description\n",
       "0     VB      None  ...          None              B-Boson\n",
       "1     gB        VB  ...          None        B-Boson Ghost\n",
       "2    VWB      None  ...          None             W-Bosons\n",
       "3    gWB       VWB  ...          None        W-Boson Ghost\n",
       "4     VG      None  ...          None                Gluon\n",
       "5     gG        VG  ...          None          Gluon Ghost\n",
       "6      q      None  ...          None                 None\n",
       "7      l      None  ...          None                 None\n",
       "8      d      None  ...          None                 None\n",
       "9      u      None  ...          None                 None\n",
       "10     e      None  ...          None                 None\n",
       "11     H      None  ...          None                 None\n",
       "12    Hp         H  ...          None                 None\n",
       "13    Hp        Hp  ...          None        Charged Higgs\n",
       "14    H0         H  ...          None                 None\n",
       "15    hh        H0  ...          VEVs                Higgs\n",
       "16    Ah        H0  ...          VEVs  Pseudo-Scalar Higgs\n",
       "17    uL         q  ...          None                 None\n",
       "18    dL         q  ...          None                 None\n",
       "19    vL         l  ...          None                 None\n",
       "20    eL         l  ...          None                 None\n",
       "21    dR         d  ...          None                 None\n",
       "22    uR         u  ...          None                 None\n",
       "23    eR         e  ...          None                 None\n",
       "24    VP        VB  ...   GaugeSector               Photon\n",
       "25    VZ    VWB[3]  ...   GaugeSector              Z-Boson\n",
       "26   VWp    VWB[1]  ...   GaugeSector           W+ - Boson\n",
       "27    DL        dL  ...   MatterField                 None\n",
       "28    UL        uL  ...   MatterField                 None\n",
       "29    EL        eL  ...   MatterField                 None\n",
       "30    DR  conj[dR]  ...   MatterField                 None\n",
       "31    UR  conj[uR]  ...   MatterField                 None\n",
       "32    ER  conj[eR]  ...   MatterField                 None\n",
       "33    Fd        DL  ...  DiracSpinors          Down-Quarks\n",
       "34    Fe        EL  ...  DiracSpinors              Leptons\n",
       "35    Fu        UL  ...  DiracSpinors            Up-Quarks\n",
       "36    Fv        vL  ...  DiracSpinors            Neutrinos\n",
       "\n",
       "[37 rows x 7 columns]"
      ]
     },
     "execution_count": 43,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "nprt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "rM232YTd452i"
   },
   "outputs": [],
   "source": [
    "ParticleDefinitions=to_defintions(newparticles,symbol='Field')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 1000
    },
    "colab_type": "code",
    "id": "RLksct5G452m",
    "outputId": "5a9f10b8-2e4b-4c8c-a990-076b06f9dbe2"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'EWSB': {'Ah': {'Description': 'Pseudo-Scalar Higgs',\n",
       "   'Mass': [0],\n",
       "   'PDG': [0],\n",
       "   'PDG.IX': [0],\n",
       "   'Width': [0]},\n",
       "  'Fd': {'Description': 'Down-Quarks'},\n",
       "  'Fe': {'Description': 'Leptons'},\n",
       "  'Fu': {'Description': 'Up-Quarks'},\n",
       "  'Fv': {'Description': 'Neutrinos'},\n",
       "  'Hp': {'Description': 'Charged Higgs',\n",
       "   'ElectricCharge': 1,\n",
       "   'LaTeX': ['H^+', 'H^-'],\n",
       "   'Mass': [0],\n",
       "   'OutputName': ['Hp', 'Hm'],\n",
       "   'PDG': [0],\n",
       "   'PDG.IX': [0],\n",
       "   'Width': [0]},\n",
       "  'VP': {'Description': 'Photon'},\n",
       "  'VWp': {'Description': 'W+ - Boson'},\n",
       "  'VZ': {'Description': 'Z-Boson'},\n",
       "  'hh': {'Description': 'Higgs', 'PDG': [25], 'PDG.IX': [101000001]}},\n",
       " 'GaugeES': {'H0': {'FeynArtsNr': 1,\n",
       "   'LaTeX': 'H^0',\n",
       "   'Mass': 'Automatic',\n",
       "   'OutputName': 'H0',\n",
       "   'PDG': [0],\n",
       "   'Width': 0},\n",
       "  'Hp': {'FeynArtsNr': 2,\n",
       "   'LaTeX': 'H^+',\n",
       "   'Mass': 'Automatic',\n",
       "   'OutputName': 'Hp',\n",
       "   'PDG': [0],\n",
       "   'Width': 0},\n",
       "  'VB': {'Description': 'B-Boson'},\n",
       "  'VG': {'Description': 'Gluon'},\n",
       "  'VWB': {'Description': 'W-Bosons'},\n",
       "  'gB': {'Description': 'B-Boson Ghost'},\n",
       "  'gG': {'Description': 'Gluon Ghost'},\n",
       "  'gWB': {'Description': 'W-Boson Ghost'}},\n",
       " 'WeylFermionAndIndermediate': {'DL': {'LaTeX': 'D_L'},\n",
       "  'DR': {'LaTeX': 'D_R'},\n",
       "  'EL': {'LaTeX': 'E_L'},\n",
       "  'ER': {'LaTeX': 'E_R'},\n",
       "  'H': {'LaTeX': 'H',\n",
       "   'Mass': 'Automatic',\n",
       "   'OutputName': 'H',\n",
       "   'PDG': [0],\n",
       "   'Width': 0},\n",
       "  'UL': {'LaTeX': 'U_L'},\n",
       "  'UR': {'LaTeX': 'U_R'},\n",
       "  'd': {'LaTeX': 'd'},\n",
       "  'dL': {'LaTeX': 'd_L'},\n",
       "  'dR': {'LaTeX': 'd_R'},\n",
       "  'e': {'LaTeX': 'e'},\n",
       "  'eL': {'LaTeX': 'e_L'},\n",
       "  'eR': {'LaTeX': 'e_R'},\n",
       "  'l': {'LaTeX': 'l'},\n",
       "  'q': {'LaTeX': 'q'},\n",
       "  'u': {'LaTeX': 'u'},\n",
       "  'uL': {'LaTeX': 'u_L'},\n",
       "  'uR': {'LaTeX': 'u_R'},\n",
       "  'vL': {'LaTeX': '\\\\nu_L'}}}"
      ]
     },
     "execution_count": 45,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ParticleDefinitions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "DV_Zguow452r"
   },
   "outputs": [],
   "source": [
    "f=open('JSON/newparticles.json','w')\n",
    "json.dump(ParticleDefinitions,f)\n",
    "f.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "yPzBLnwj452y"
   },
   "outputs": [],
   "source": [
    "dfnp=pd.read_json('JSON/newparticles.json')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "jkgbTdVV4521"
   },
   "outputs": [],
   "source": [
    "kk=to_math(dfnp,'particles.m')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 1000
    },
    "colab_type": "code",
    "id": "SgzSBTvC4527",
    "outputId": "706bf6dc-a5bd-4b88-cbb6-25973d84ed83"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "ParticleDefinitions[EWSB] = {\n",
      "    {Hp, { Description -> \"Charged Higgs\",\n",
      "           PDG -> {0},\n",
      "           PDG.IX -> {0},\n",
      "           Width -> {0},\n",
      "           Mass -> {0},\n",
      "           LaTeX -> {\"H^+\", \"H^-\"},\n",
      "           OutputName -> {\"Hp\", \"Hm\"},\n",
      "           ElectricCharge -> 1}},\n",
      "    {hh, { Description -> \"Higgs\",\n",
      "           PDG -> {25},\n",
      "           PDG.IX -> {101000001}}},\n",
      "    {Ah, { Description -> \"Pseudo-Scalar Higgs\",\n",
      "           PDG -> {0},\n",
      "           PDG.IX -> {0},\n",
      "           Mass -> {0},\n",
      "           Width -> {0}}},\n",
      "    {VP, { Description -> \"Photon\"}},\n",
      "    {VZ, { Description -> \"Z-Boson\"}},\n",
      "    {VWp, { Description -> \"W+ - Boson\"}},\n",
      "    {Fd, { Description -> \"Down-Quarks\"}},\n",
      "    {Fe, { Description -> \"Leptons\"}},\n",
      "    {Fu, { Description -> \"Up-Quarks\"}},\n",
      "    {Fv, { Description -> \"Neutrinos\"}}\n",
      "};\n",
      "\n",
      "ParticleDefinitions[GaugeES] = {\n",
      "    {Hp, { PDG -> {0},\n",
      "           Width -> 0,\n",
      "           Mass -> Automatic,\n",
      "           FeynArtsNr -> 2,\n",
      "           LaTeX -> \"H^+\",\n",
      "           OutputName -> Hp}},\n",
      "    {VB, { Description -> \"B-Boson\"}},\n",
      "    {gB, { Description -> \"B-Boson Ghost\"}},\n",
      "    {VWB, { Description -> \"W-Bosons\"}},\n",
      "    {gWB, { Description -> \"W-Boson Ghost\"}},\n",
      "    {VG, { Description -> \"Gluon\"}},\n",
      "    {gG, { Description -> \"Gluon Ghost\"}},\n",
      "    {H0, { PDG -> {0},\n",
      "           Width -> 0,\n",
      "           Mass -> Automatic,\n",
      "           FeynArtsNr -> 1,\n",
      "           LaTeX -> \"H^0\",\n",
      "           OutputName -> H0}}\n",
      "};\n",
      "\n",
      "ParticleDefinitions[WeylFermionAndIndermediate] = {\n",
      "    {q, { LaTeX -> \"q\"}},\n",
      "    {l, { LaTeX -> \"l\"}},\n",
      "    {d, { LaTeX -> \"d\"}},\n",
      "    {u, { LaTeX -> \"u\"}},\n",
      "    {e, { LaTeX -> \"e\"}},\n",
      "    {H, { PDG -> {0},\n",
      "           Width -> 0,\n",
      "           Mass -> Automatic,\n",
      "           LaTeX -> \"H\",\n",
      "           OutputName -> H}},\n",
      "    {uL, { LaTeX -> \"u_L\"}},\n",
      "    {dL, { LaTeX -> \"d_L\"}},\n",
      "    {vL, { LaTeX -> \"\\nu_L\"}},\n",
      "    {eL, { LaTeX -> \"e_L\"}},\n",
      "    {dR, { LaTeX -> \"d_R\"}},\n",
      "    {uR, { LaTeX -> \"u_R\"}},\n",
      "    {eR, { LaTeX -> \"e_R\"}},\n",
      "    {DL, { LaTeX -> \"D_L\"}},\n",
      "    {UL, { LaTeX -> \"U_L\"}},\n",
      "    {EL, { LaTeX -> \"E_L\"}},\n",
      "    {DR, { LaTeX -> \"D_R\"}},\n",
      "    {UR, { LaTeX -> \"U_R\"}},\n",
      "    {ER, { LaTeX -> \"E_R\"}}\n",
      "};\n",
      "\n"
     ]
    }
   ],
   "source": [
    "cat particles.m"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 50,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/",
     "height": 1000
    },
    "colab_type": "code",
    "id": "2AUZDRYijPy3",
    "outputId": "1ffd9964-2e17-4941-80cb-04b7e7f20efc"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Off[General::spell]\n",
      "\n",
      "Model`Name = \"SM\";\n",
      "Model`NameLaTeX =\"Standard Model\";\n",
      "Model`Authors = \"F.Staub\";\n",
      "Model`Date = \"2014-11-06\";\n",
      "\n",
      "(* 2013-01-24: changed normalization of lambda term to convention of hep-ph/0207271 *)\n",
      "(* 2013-06-24: using new name conventions (without inital \"S\" and \"F\" for scalar and matter fields) *)\n",
      "(* 2013-09-01: changing to new conventions for FermionFields/MatterFields *)\n",
      "(* 2014-11-06: Changed sign in Lagrangian to fit standard conventions *)\n",
      "(* 2016-05-03: Changed sign of Yu *)\n",
      "\n",
      "\n",
      "\n",
      "(*-------------------------------------------*)\n",
      "(*   Particle Content*)\n",
      "(*-------------------------------------------*)\n",
      "\n",
      "(* Gauge Groups *)\n",
      "\n",
      "Gauge[[1]]={B,   U[1], hypercharge, g1,False};\n",
      "Gauge[[2]]={WB, SU[2], left,        g2,True};\n",
      "Gauge[[3]]={G,  SU[3], color,       g3,False};\n",
      "\n",
      "\n",
      "(* Matter Fields *)\n",
      "\n",
      "FermionFields[[1]] = {q, 3, {uL, dL},     1/6, 2,  3};  \n",
      "FermionFields[[2]] = {l, 3, {vL, eL},    -1/2, 2,  1};\n",
      "FermionFields[[3]] = {d, 3, conj[dR],     1/3, 1, -3};\n",
      "FermionFields[[4]] = {u, 3, conj[uR],    -2/3, 1, -3};\n",
      "FermionFields[[5]] = {e, 3, conj[eR],       1, 1,  1};\n",
      "\n",
      "ScalarFields[[1]] =  {H, 1, {Hp, H0},     1/2, 2,  1};\n",
      "\n",
      "\n",
      "        \n",
      "(*----------------------------------------------*)\n",
      "(*   DEFINITION                                 *)\n",
      "(*----------------------------------------------*)\n",
      "\n",
      "NameOfStates={GaugeES, EWSB};\n",
      "\n",
      "(* ----- Before EWSB ----- *)\n",
      "\n",
      "DEFINITION[GaugeES][LagrangianInput]= {\n",
      "\t{LagHC, {AddHC->True}},\n",
      "\t{LagNoHC,{AddHC->False}}\n",
      "};\n",
      "\n",
      "\n",
      "LagNoHC = -mu2 conj[H].H - 1/2 \\[Lambda] conj[H].H.conj[H].H;\n",
      "LagHC =  -(Yd conj[H].d.q + Ye conj[H].e.l + Yu u.q.H);\n",
      "\n",
      "\n",
      "\t\t\t  \t\t  \n",
      "\n",
      "(* Gauge Sector *)\n",
      "\n",
      "DEFINITION[EWSB][GaugeSector] =\n",
      "{ \n",
      "  {{VB,VWB[3]},{VP,VZ},ZZ},\n",
      "  {{VWB[1],VWB[2]},{VWp,conj[VWp]},ZW}\n",
      "};     \n",
      "        \n",
      "        \n",
      "          \t\n",
      "\n",
      "(* ----- VEVs ---- *)\n",
      "\n",
      "DEFINITION[EWSB][VEVs]= \n",
      "{    {H0, {v, 1/Sqrt[2]}, {Ah, \\[ImaginaryI]/Sqrt[2]},{hh, 1/Sqrt[2]}}     };\n",
      " \n",
      "\n",
      "DEFINITION[EWSB][MatterSector]=   \n",
      "    {{{{dL}, {conj[dR]}}, {{DL,Vd}, {DR,Ud}}},\n",
      "     {{{uL}, {conj[uR]}}, {{UL,Vu}, {UR,Uu}}},\n",
      "     {{{eL}, {conj[eR]}}, {{EL,Ve}, {ER,Ue}}}};  \n",
      "\n",
      "\n",
      "(*------------------------------------------------------*)\n",
      "(* Dirac-Spinors *)\n",
      "(*------------------------------------------------------*)\n",
      "\n",
      "DEFINITION[EWSB][DiracSpinors]={\n",
      " Fd ->{  DL, conj[DR]},\n",
      " Fe ->{  EL, conj[ER]},\n",
      " Fu ->{  UL, conj[UR]},\n",
      " Fv ->{  vL, 0}};\n",
      "\n",
      "DEFINITION[EWSB][GaugeES]={\n",
      " Fd1 ->{  FdL, 0},\n",
      " Fd2 ->{  0, FdR},\n",
      " Fu1 ->{  Fu1, 0},\n",
      " Fu2 ->{  0, Fu2},\n",
      " Fe1 ->{  Fe1, 0},\n",
      " Fe2 ->{  0, Fe2}};\n",
      "\n",
      "\n",
      "\n",
      "\n"
     ]
    }
   ],
   "source": [
    "print(smf)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "WckXFvvzjPy6",
    "outputId": "fcd948f1-424d-43c2-a364-f5d28ac55da6"
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "2"
      ]
     },
     "execution_count": 52,
     "metadata": {
      "tags": []
     },
     "output_type": "execute_result"
    }
   ],
   "source": [
    "1+1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 0,
   "metadata": {
    "colab": {},
    "colab_type": "code",
    "id": "M126ZqmWjPy8"
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "colab": {
   "include_colab_link": true,
   "name": "particles.ipynb",
   "provenance": []
  },
  "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.5.3"
  },
  "latex_envs": {
   "LaTeX_envs_menu_present": true,
   "autoclose": false,
   "autocomplete": true,
   "bibliofile": "biblio.bib",
   "cite_by": "apalike",
   "current_citInitial": 1,
   "eqLabelWithNumbers": true,
   "eqNumInitial": 1,
   "hotkeys": {
    "equation": "Ctrl-E",
    "itemize": "Ctrl-I"
   },
   "labels_anchors": false,
   "latex_user_defs": false,
   "report_style_numbering": false,
   "user_envs_cfg": false
  }
 },
 "nbformat": 4,
 "nbformat_minor": 1
}