#!/bin/python3

import openmc


############ MATERIALS ############ 
mat_list = []

## Matrix Graphite ##
mat_graphite_matrix = openmc.Material(name="Graphite Matrix (inside/outside)")
mat_graphite_matrix.add_element("C",1.0)
mat_graphite_matrix.add_nuclide("B10",0.3e-6)
mat_graphite_matrix.set_density("g/cc",1.806)
mat_list.append(mat_graphite_matrix)

## TRISO Pins ##
mat_triso_uco = openmc.Material(name="TRISO Pin UCO Kernel")
mat_triso_uco.add_element("U",1.0,enrichment=19.95,enrichment_type="wo")
mat_triso_uco.add_element("C",1.0)
mat_triso_uco.add_element("O",1.0)
mat_triso_uco.set_density("g/cc",10.744)
mat_triso_uco.add_s_alpha_beta("c_graphite")
mat_list.append(mat_triso_uco)

mat_triso_buffer = openmc.Material(name="TRISO Pin Buffer")
mat_triso_buffer.add_element("C",1.0)
mat_triso_buffer.add_s_alpha_beta("c_graphite")
mat_triso_buffer.set_density("g/cc",1.04)
mat_list.append(mat_triso_buffer)

mat_triso_pyc1 = openmc.Material(name="TRISO Pin PyC1")
mat_triso_pyc1.add_element("C",1.0)
mat_triso_pyc1.add_s_alpha_beta("c_graphite")
mat_list.append(mat_triso_pyc1)

mat_triso_sic = openmc.Material(name="TRISO Pin SiC")
mat_triso_sic.add_element("Si",1.0)
mat_triso_sic.add_element("C",1.0)
mat_triso_sic.add_s_alpha_beta("c_graphite")
mat_triso_sic.set_density("g/cc",3.171)
mat_list.append(mat_triso_sic)

mat_triso_pyc2 = openmc.Material(name="TRISO Pin PyC2")
mat_triso_pyc2.add_element("C",1.0)
mat_triso_pyc2.add_s_alpha_beta("c_graphite")
mat_triso_pyc2.set_density("g/cc",1.882)
mat_list.append(mat_triso_pyc2)

## Moderator Pins ##
mat_mod_helium = openmc.Material(name="Moderator Pin Helium")
mat_mod_helium.add_element("He",1.0)
mat_mod_helium.set_density("g/cc",0.18e-3)


materials = openmc.Materials(mat_list)
materials.export_to_xml()


############ GEOMETRY ############ 



############ SETTINGS ############