Python Initialize.getNR示例

示例#1

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])