""" cmb.py

Generate a cmb map from lambda-CDM power spectrum. 
"""

import healpy as hp
import pylab as pl

### Define some global parameters
lst_par = ['lmax', 'nside', 'cmb_unit', 'sim_id', 'input_cl']
lst_tag = lst_par[:-1]

nside  = seg_input.values()[0][0] ## pushed from main
sim_id = seg_input.values()[0][1] ## pushed from main
lmax   = 2*nside

### Generate a cmb map
input_cl = "lambda_best_fit.txt"
cmb_unit = "uK_CMB"
cmb_cl   = pl.loadtxt(input_cl)[0:lmax+1,0]     ## load cl
cmb_map  = hp.synfast(cmb_cl, nside, lmax=lmax) ## make a map

### Save to disk
cmb_cl_fn  = get_data_fn ('cls_cmb.txt')
pl.savetxt  (cmb_cl_fn , cmb_cl)
cmb_map_fn = get_data_fn ('map_cmb.fits')
hp.write_map(cmb_map_fn, cmb_map)

### Make a plot
cmb_map_fig = cmb_map_fn.replace('.fits', '.png')
hp.mollview(cmb_map, title="cmb in %s"%cmb_unit)
pl.savefig (cmb_map_fig)

### Set output : forward as many childs as sim ids
seg_output = [sim_id]