import numpy as np
from sys import argv
import baryonification as bfc
#import corforb

if ((len(argv)-1) != 14):
    print("Usage: python PSplot.py <partfile_in> <halofile_in> <partfile_out> <Mc> <mu> <nu> <thej> <ga> <de> <eta> <deta> <redshift> <Lbox> <N_chunk>")
    exit()

partfile_in = argv[1]
halofile_in = argv[2]
partfile_out = argv[3]
Mc      = float(argv[6])
mu      = float(argv[7])
nu      = float(argv[8])
thej    = float(argv[9])
ga      = float(argv[10])
de      = float(argv[11])
eta     = float(argv[12])
deta    = float(argv[13])

redshift = float(argv[14])
Lbox     = float(argv[15])
N_chunk  = int(argv[16])

print("Mc,mu,nu,thej,ga,de,eta,deta")
print(Mc,mu,nu,thej,ga,de,eta,deta)

#initialise parameters
par = bfc.par()

par.code.beta_model = 1

par.cosmo.z  = redshift

par.baryon.Mc      = Mc
par.baryon.mu      = mu
par.baryon.nu      = nu
par.baryon.thej    = thej
par.baryon.gamma   = ga
par.baryon.delta   = de
par.baryon.eta     = eta
par.baryon.deta    = deta


par.files.transfct     = "TF/CDM_PLANCK_tk.dat"
par.files.cosmofct     = "cosmofct.dat"
par.files.displfct     = "displfct.dat"
par.files.partfile_in  = partfile_in
par.files.partfile_out = partfile_out
par.files.halofile_in  = halofile_in

par.sim.Lbox = Lbox
par.sim.Nmin_per_halo = 100
par.sim.N_chunk = N_chunk

#calculate 2-halo term
bfc.cosmo(par)

#displace particles
bfc.displace(par)