Python create_fitsheader Examples

Example #1

File: test_mpifits.py Project: pchanial/tamasis-pacs

def test_fitsheader():
    def func(n, hdr):
        hdr_local = scatter_fitsheader(hdr)
        assert_equal(hdr_local['NAXIS2'], distribute(n))
        assert gather_fitsheader(hdr_local) == hdr
    for n in range(10):
        yield func, n, create_fitsheader((5,n), crval=(1,1), cdelt=1.)

Example #2

def test_fitsheader():
    def func(n, hdr):
        hdr_local = scatter_fitsheader(hdr)
        assert_equal(hdr_local['NAXIS2'], distribute(n))
        assert gather_fitsheader(hdr_local) == hdr

    for n in range(10):
        yield func, n, create_fitsheader((5, n), crval=(1, 1), cdelt=1.)

Example #3

File: test_mpirls.py Project: pchanial/tamasis-pacs

#solver = scipy.sparse.linalg.bicgstab
solver = tamasis.solvers.cg
tol = 1.e-2
mtol = 1.e-10
maxiter = 100
hyper = 1

rank = MPI.COMM_WORLD.rank
tamasis.var.verbose = True
profile = None#'test_rls.png'
data_dir = os.path.dirname(__file__) + '/data/'

# reference map (no communication)
header_ref_global = create_fitsheader((97,108), cdelt=3.2/3600,
                                      crval=(245.998427916727,61.5147650744551))
comm_tod = MPI.COMM_SELF
comm_map = MPI.COMM_SELF
obs_ref = PacsObservation(data_dir + 'frames_blue.fits', comm=comm_tod)
obs_ref.pointing.chop = 0
tod_ref = obs_ref.get_tod()
model_ref = MaskOperator(tod_ref.mask) * obs_ref.get_projection_operator(
            downsampling=True, npixels_per_sample=6, commin=comm_map,
            header=header_ref_global)
map_ref_global = mapper_rls(tod_ref, model_ref, tol=tol, maxiter=maxiter,
                            solver=solver, hyper=hyper)
header_ref_global = map_ref_global.header
cov_ref_global = map_ref_global.coverage
mask_ref_global = map_ref_global.coverage == 0
tolocal = MPIDistributionGlobalOperator(map_ref_global.shape,
                                        attrin={'header':header_ref_global})

Example #4

File: test_mpirls.py Project: pchanial/tamasis-pacs

from tamasis.utils import assert_all_eq

#solver = scipy.sparse.linalg.bicgstab
solver = tamasis.solvers.cg
tol = 1.e-2
mtol = 1.e-10
maxiter = 100
hyper = 1

rank = MPI.COMM_WORLD.rank
tamasis.var.verbose = True
profile = None  #'test_rls.png'
data_dir = os.path.dirname(__file__) + '/data/'

# reference map (no communication)
header_ref_global = create_fitsheader(
    (97, 108), cdelt=3.2 / 3600, crval=(245.998427916727, 61.5147650744551))
comm_tod = MPI.COMM_SELF
comm_map = MPI.COMM_SELF
obs_ref = PacsObservation(data_dir + 'frames_blue.fits', comm=comm_tod)
obs_ref.pointing.chop = 0
tod_ref = obs_ref.get_tod()
model_ref = MaskOperator(tod_ref.mask) * obs_ref.get_projection_operator(
    downsampling=True,
    npixels_per_sample=6,
    commin=comm_map,
    header=header_ref_global)
map_ref_global = mapper_rls(tod_ref,
                            model_ref,
                            tol=tol,
                            maxiter=maxiter,
                            solver=solver,