Add galaxy sampling (#88)

* Improve calculations

* Improve flags

* Add smoothed options

* Remove some old comments

* Edit little things

* Save smoothed

* Move files

* Edit imports

* Edit imports

* Renaming imports

* Renaming imports

* Sort imports

* Sort files

* Sorting

* Optionally make copies of the field

* Add quijote backup check

* Add direct field smoothing

* Shorten stupid documentation

* Shorten stupid docs

* Update conversion

* Add particles to ASCII conversion

* Add a short comment

* Add SDSS uncorrected distance

* Adjust comment

* Add FITS index to galaxies

* Remove spare space

* Remove a stupid line

* Remove blank line

* Make space separated

* Add interpolated field path

* Add field sampling

* Sort imports

* Return density in cells

* Clear out observer velocity

* Add 170817 sampling

* Fix normalization

* Update plot

scripts/field_prop.py

@@ -23,16 +23,9 @@ from gc import collect
 
 import numpy
 from mpi4py import MPI
-
-try:
-    import csiborgtools
-except ModuleNotFoundError:
-    import sys
-    sys.path.append("../")
-    import csiborgtools
-
 from taskmaster import work_delegation
 
+import csiborgtools
 from utils import get_nsims
 
 ###############################################################################
@@ -60,6 +53,10 @@ def density_field(nsim, parser_args, to_save=True):
         radvel_field = numpy.load(paths.field(
             "radvel", parser_args.MAS, parser_args.grid, nsim, False))
 
+        if parser_args.verbose:
+            print(f"{datetime.now()}: converting density field to RSP.",
+                  flush=True)
+
         field = csiborgtools.field.field2rsp(field, radvel_field, box,
                                              parser_args.MAS)
 
@@ -187,6 +184,10 @@ def environment_field(nsim, parser_args, to_save=True):
     density_gen = csiborgtools.field.DensityField(box, parser_args.MAS)
     rho = density_gen.overdensity_field(rho)
 
+    if parser_args.smooth_scale > 0.0:
+        rho = csiborgtools.field.smoothen_field(
+            rho, parser_args.smooth_scale, box.box2mpc(1.))
+
     gen = csiborgtools.field.TidalTensorField(box, parser_args.MAS)
     field = gen(rho)
 
@@ -217,7 +218,7 @@ def environment_field(nsim, parser_args, to_save=True):
 
     if to_save:
         fout = paths.field("environment", parser_args.MAS, parser_args.grid,
-                           nsim, parser_args.in_rsp)
+                           nsim, parser_args.in_rsp, parser_args.smooth_scale)
         print(f"{datetime.now()}: saving output to `{fout}`.")
         numpy.save(fout, env)
     return env
@@ -241,6 +242,8 @@ if __name__ == "__main__":
     parser.add_argument("--grid", type=int, help="Grid resolution.")
     parser.add_argument("--in_rsp", type=lambda x: bool(strtobool(x)),
                         help="Calculate in RSP?")
+    parser.add_argument("--smooth_scale", type=float, default=0.0,
+                        help="Smoothing scale in Mpc / h. Only used for the environment field.")  # noqa
     parser.add_argument("--verbose", type=lambda x: bool(strtobool(x)),
                         help="Verbosity flag for reading in particles.")
     parser.add_argument("--simname", type=str, default="csiborg",