maubin/sbmy_control
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

COCA parameters

Browse source
This commit is contained in:
Mayeul Aubin 2025-06-02 18:11:19 +02:00
parent e189ca9a7b
commit e328e1250b
1 changed files with 22 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

22
sbmy_control/parameters_card.py
View file

@ -47,6 +47,10 @@ def register_arguments_card(parser:ArgumentParser):
parser.add_argument("--OutputFCsDensity", type=str, default=None, help="Output FCs density file.")
parser.add_argument("--OutputFCsSnapshot", type=str, default=None, help="Output FCs snapshot file.")
parser.add_argument("--OutputRngStateLPT", type=str, default=None, help="Output RNG state file.")
parser.add_argument("--WriteReferenceFrame", type=bool, default=False, help="Write reference frame (COCA).")
parser.add_argument("--OutputMomentaBase", type=str, default=None, help="Output momenta base (COCA).")
parser.add_argument("--ReadReferenceFrame", type=bool, default=False, help="Read reference frame (COCA).")
parser.add_argument("--InputMomentaBase", type=str, default=None, help="Read momenta base (COCA).")
def register_arguments_card_for_ICs(parser:ArgumentParser):
@ -112,6 +116,10 @@ def parse_arguments_card(parsed_args):
GenerateLightcone=parsed_args.GenerateLightcone,
OutputAlsoFCs=parsed_args.OutputAlsoFCs,
Observer=parsed_args.Observer,
WriteReferenceFrame=parsed_args.WriteReferenceFrame,
OutputMomentaBase=parsed_args.OutputMomentaBase,
ReadReferenceFrame=parsed_args.ReadReferenceFrame,
InputMomentaBase=parsed_args.InputMomentaBase,
N_tiles=parsed_args.N_tiles,
Np_buffer=parsed_args.Np_buffer,
Np_lpt_buffer=parsed_args.Np_lpt_buffer,
@ -203,6 +211,10 @@ def parse_arguments_card(parsed_args):
card_dict["OutputFCsSnapshot"] = main_dict["resultdir"]+"final_particles_"+main_dict["simname"]+".gadget3"
if card_dict["OutputRngStateLPT"] is None:
card_dict["OutputRngStateLPT"] = main_dict["workdir"]+"rng_state.h5"
if card_dict["OutputMomentaBase"] is None:
card_dict["OutputMomentaBase"] = main_dict["workdir"]+"momenta_"+main_dict["simname"]+"_"
if card_dict["InputMomentaBase"] is None:
card_dict["InputMomentaBase"] = main_dict["workdir"]+"momenta_"+main_dict["simname"]+"_"
return card_dict
@ -321,6 +333,12 @@ def create_parameter_card_dict(
OutputFCsDensity:str = 'fcs_density.h5',
OutputFCsSnapshot:str = 'fcs_particles.gadget3',
## COCA parameters
WriteReferenceFrame:bool = False,
OutputMomentaBase:str = 'momenta_',
ReadReferenceFrame:bool = False,
InputMomentaBase:str = 'momenta_',
## Cosmological parameters
h:float = 0.6732,
Omega_m:float = 0.302,
@ -375,6 +393,10 @@ def create_parameter_card_dict(
Observer2=Observer[2],
OutputFCsDensity=OutputFCsDensity,
OutputFCsSnapshot=OutputFCsSnapshot,
WriteReferenceFrame=int(WriteReferenceFrame),
OutputMomentaBase=OutputMomentaBase,
ReadReferenceFrame=int(ReadReferenceFrame),
InputMomentaBase=InputMomentaBase,
NumberOfTilesPerDimension=N_tiles,
NumberOfParticlesInBuffer=Np_buffer,
NumberOfParticlesInLPTBuffer=Np_lpt_buffer,