import numpy as np
import matplotlib.pyplot as plt
from time import time

import odeg as od

plt.rcParams['figure.dpi'] = 300
plt.rcParams['savefig.dpi'] = 300


if __name__ == '__main__':
    
    cfg = {
        'r0': 1,
        'rs': 2,
        'p_max': 5,
        'pos_charge': 10,
        'theta': .5,
        'mu': -.5,
        }
    
    # start = time()
    h_exp, n_exp, n2_exp = od.fctExp(od.energyFct, od.numberFct, od.number2Fct, **cfg)
    # print("t[fctexp] =", time() - start)
    free_energy = od.freeEnergy(**cfg)
    entropy = od.entropy(**cfg)