import numpy as np
import matplotlib.pyplot as plt
import plotly.graph_objects as go
from CADMium import Psgrid
from CADMium import Kohnsham
#Distance of the nucley from grid center
a = 10
for k in [3,]:
for i in [30]:
for j in [8]:
#Grid Options
NP = k#Number of points per block
NM = [i,j] #Number of blocks [angular, radial]
L = np.arccosh(15./a)#Maximum radial coordinate value
loc = np.array(range(-4,5)) #Non inclusive on upper bound2
#Create and initialize grid object
grid = Psgrid(NP, NM, a, L, loc)
grid.initialize()
ps_x = np.vstack((-np.flip(grid.Z), grid.Z))
ps_y = np.vstack((-np.flip(grid.Y), grid.Y))
fig, ax = plt.subplots(dpi=200)
ax.scatter(ps_x, ps_y, s=10, alpha=1, c='#4DBF7A', label=f"{k,i,j}")
ax.set_facecolor("#E5DBC0")
# ax.text(-5, 0.8, 'Pycadmium', family="monospace", fontsize=40)
fig.legend()
fig.show()