%pylab inline
from nbodykit.lab import *
# redshift=0
pcat = BigFileCatalog('constrained_1.0000/', dataset='1/', header='Header')
pcat.attrs['BoxSize'] = pcat.attrs['BoxSize'] * [1, 1, 1]
prev = pcat.to_mesh(Nmesh=128).preview()
imshow(log10(prev.sum(axis=-1)).T, origin='lower',)
fof = FOF(pcat, linking_length=0.02, nmin=100, absolute=True) # 200 kpc/h
features = fof.find_features()
massive = features[features['Length'] > 1000]
plot(massive['CMPosition'][:, 0], massive['CMPosition'][:, 1], 'x ', color='red')
imshow(log10(prev.T), origin='lower', extent=(0, 20, 0, 20))
# redshift=9
pcat9 = BigFileCatalog('constrained_0.1000/', dataset='1/', header='Header')
prev9 = pcat9.to_mesh(Nmesh=128).preview(axes=[0, 1])
imshow(prev9.T, origin='lower',)