Skip to content

cosmodesi/pyrecon

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pyrecon - Python reconstruction code

Introduction

pyrecon is a package to perform reconstruction within Python, using different algorithms, so far:

With Python, a typical reconstruction run is (e.g. for MultiGridReconstruction; the same works for other algorithms):

from pyrecon import MultiGridReconstruction

# line-of-sight "los" can be local (None, default) or an axis, 'x', 'y', 'z', or a 3-vector
# Instead of boxsize and boxcenter, one can provide a (N, 3) array of Cartesian positions: positions=
recon = MultiGridReconstruction(f=0.8, bias=2.0, los=None, nmesh=512, boxsize=1000., boxcenter=2000.)
recon.assign_data(data_positions, data_weights) # data_positions is a (N, 3) array of Cartesian positions, data_weights a (N,) array
# You can skip the following line if you assume uniform selection function (randoms)
recon.assign_randoms(randoms_positions, randoms_weights)
recon.set_density_contrast(smoothing_radius=15.)
recon.run()
# A shortcut of the above is:
# recon = MultiGridReconstruction(f=0.8, bias=2.0, data_positions=data_positions, data_weights=data_weights, randoms_positions=randoms_positions, randoms_weights=randoms_weights, los=None, nmesh=512, boxsize=1000., boxcenter=2000.)
# If you are using IterativeFFTParticleReconstruction, displacements are to be taken at the reconstructed data real-space positions;
# in this case, do: data_positions_rec = recon.read_shifted_positions('data')
data_positions_rec = recon.read_shifted_positions(data_positions)
# RecSym = remove large scale RSD from randoms
randoms_positions_rec = recon.read_shifted_positions(randoms_positions)
# or RecIso
# randoms_positions_rec = recon.read_shifted_positions(randoms_positions, field='disp')

Also provided a script to run reconstruction as a standalone:

pyrecon [-h] config-fn [--data-fn [<fits, hdf5 file>]] [--randoms-fn [<fits, hdf5 file>]] [--output-data-fn [<fits, hdf5 file>]] [--output-randoms-fn [<fits, hdf5file>]]

An example of configuration file is provided in config. data-fn, randoms-fn are input data and random file names to override those in configuration file. The same holds for output files output-data-fn, output-randoms-fn.

In progress

Check algorithm details (see notes in docstrings).

Documentation

Documentation is hosted on Read the Docs, pyrecon docs.

Requirements

Only strict requirements are:

  • numpy
  • scipy
  • pmesh

Extra requirements are:

  • mpytools, fitsio, h5py to run pyrecon as a standalone
  • pypower to evaluate reconstruction metrics (correlation, transfer function and propagator)

Installation

See pyrecon docs.

License

pyrecon is free software distributed under a BSD3 license. For details see the LICENSE.

Credits