A generic EventList class

EventList.py

import numpy as np

class EventList(object):
  
  def __init__(self, arrival_times, energies, ra=None, dec=None):
    
    self._arrival_times = np.asarray(arrival_times)
    self._energies = np.asarray(energies)
    
    assert self._arrival_times.shape[0] == self._energies.shape[0]
    
    # ........
  
  @classmethod
  def from_TTE(cls, tte_file, response):
    
    # Open TTE, get the arrival times, read the EBOUNDS from the response and give
    # each event the appropriate energy according to the channel
    
    with pyfits.open(tte_file) as f:
      arrival_times = f['EVENTS'].TIME
      channel = f['EVENTS'].PHA
    
    with pyfits.open(response) as f:
      ebounds = f['EBOUNDS'].data
    
    # Loop over the events and give them the right energy according to their
    # channel
    
    energies =...
    
    # then:
    
    return cls(arrival_times, energies)
  
  @classmethod
  def from_FT1(cls, ft1_file):
    
    # Create event list from LAT FT1 file
    
    # Read the FT1
    with pyfits.open(ft1_file) as f:
      
      arrival_times = f['EVENTS'].data.TIME
      energies = f['EVENTS'].data.ENERGY
    
    return cls(arrival_times, energies)