# -*- coding: utf-8 -*-
# -------------------------------------------------------------
# Analysis from File
# -------------------------------------------------------------
path = 'silicon_pn_junction.hdf5'
configuration = nlread(path, object_id='DeviceConfiguration_0')[0]

# -------------------------------------------------------------
# Photocurrent
# -------------------------------------------------------------
kpoints = MonkhorstPackGrid(
    na=9,
    nb=9,
    )

photocurrent = Photocurrent(
    configuration=configuration,
    energies=Automatic,
    photon_energies=numpy.linspace(0, 5, 31)*eV,
    kpoints=kpoints,
    photon_polarization=[0+0j, 0+0j, 1+0j],
    energy_resolution=0.05*eV,
    )
nlsave('photocurrent.hdf5', photocurrent)