Exemplo n.º 1
0
    print("Input data = "+species+" from " + data_dir)
    print("Total no. of particles =" +str(N_part_total))
    print("No. of particles used =" +str(N_part))

#Partition the total input file with N particles into mpi_size parts:
mpi_borders = np.linspace(0, N_part, mpi_size + 1).astype('int')

my_border_low = mpi_borders[mpi_rank]
my_border_high = mpi_borders[mpi_rank+1]

#Redefining N_part as number of particles analysed per processor
N_part = my_border_high - my_border_low


#Read Trajectory Data
time, U, E, B, Energy = Read_Data.Read_Trajectory_Data(my_border_low, my_border_high, data_dir, species)
#Comment: U is used to denote the proper relativistic velocity, which is U = gamma*V

N = len(time)

#Process Trajectory Data using chosen wavelet and edge mode
if mpi_rank==0:
    print('Processing Data for particles...')

#Transfrom the data and compute new quantities:
time, V, E, b, V_par, V_perp, vpar, vperp, Acc_mag, Acc_par, Acc_perp1, Acc_perp2 = Process_Data.Process_raw(time, U, E, B, Omega_c, q, m)
N_pad = len(vpar[0,:])

ref=int(np.ceil(N/4))

#vpar_Global=Diffusion(vpar,ref,N)