import numpy as np
from time import time
from itertools import product

from odeg.h_eigenvalues import getEVs



if __name__ == '__main__':
    
    cfg = {
        'r0': 1,
        'rs': 2,
        'p_max': 3,
        'pos_charge': 6,
        }
    
    # rs_grid = np.array([2., 4., 8.])
    # r0_grid = np.array([.5, 1.])
    
    rs_grid = np.array([2,4,8])
    r0_grid = np.array([.5, 1])
    pos_charge_grid = np.linspace(3, 15, 20)
    p_max_grid = np.array([3])
    
    
    for rs, r0, p_max, pos_charge in product(rs_grid, r0_grid, p_max_grid, pos_charge_grid):
        cfg['rs'] = rs
        cfg['r0'] = r0
        cfg['p_max'] = p_max
        cfg['pos_charge'] = pos_charge
        
        
        print("\n\nconfiguration:\n", cfg, "\n\n")
        start = time()
        evs = getEVs(**cfg)
        print("t[fctexp] =", time() - start)