import h5py import numpy as np import sys sys.path.append('..') from GreensFunction import GreensFunction import Initialize import SMPI import smul import time smul.initialize('genimg.conf', inputRealImage=False) if SMPI.is_root(): gf = smul.getGreensFunction() nr = Initialize.getNR() # Generate input vector E = 10.0 # Electric field Z = 5.0 # Effective charge # Linearly decaying radial profile b = np.linspace(1, 0, nr) #b = np.ones((nr,)) # Momentum space parameters remain the same at all radii a = (E / Z) * np.ones(b.shape) c = (17*np.sqrt(Z+5)) * np.ones(b.shape) # Combine to form input vector v = np.concatenate([a,b,c])