Source code for gammapy.scripts.cube_bin

# Licensed under a 3-clause BSD style license - see LICENSE.rst
from __future__ import absolute_import, division, print_function, unicode_literals
import logging
from astropy.table import Table
from ..utils.scripts import get_parser
from ..cube import SkyCube

__all__ = ['make_counts_cube']

log = logging.getLogger(__name__)


def cube_bin_main(args=None):
    parser = get_parser(make_counts_cube)
    parser.add_argument('event_file', type=str,
                        help='Input FITS event file name')
    parser.add_argument('reference_file', type=str,
                        help='Input FITS reference cube file name')
    parser.add_argument('out_file', type=str,
                        help='Output FITS counts cube file name')
    parser.add_argument('--overwrite', action='store_true',
                        help='Overwrite existing output file?')
    args = parser.parse_args(args)
    make_counts_cube(**vars(args))


def make_counts_cube(event_file,
                     reference_file,
                     out_file,
                     overwrite):
    """Bin events into a LON-LAT-Energy cube.

    """
    events = Table.read(event_file)
    refcube = SkyCube.read(reference_file)
    cube = SkyCube.empty_like(refcube)
    cube.fill(events)
    cube.writeto(out_file, clobber=overwrite)