#!/usr/bin/env python
"""
Plot diagonalization results of full speedtest, which are generated by
either cpp or python version of speedtest.
"""
from plot_diag import main as plot
from utils import print_color


dat_files = (
    "graphene_bands.dat", "graphene_dos.dat", "graphene_local_dos.dat",
    "kane_mele_spin_sigma_x.dat","bismuth_spin_sigma_x.dat", "kane_mele_z2.dat",
    "haldane_kubo_berry_omega_xy.dat", "haldane_wilson_berry_omega_xy.dat",
    "graphene_dyn_pol.dat", "graphene_epsilon.dat", "graphene_epsilon_q0.dat",
    "graphene_ac_cond.dat")

for f in dat_files:
    try:
        plot(f)
    except (FileNotFoundError, NameError):  # NameError for missing HDF5 support
        print_color(f"WARNING: {f} and found and skipped!", "r")
        continue
