"""
from __future__ import division, absolute_import, print_function
import math
import operator
import pickle
import numpy as np
import matplotlib.pyplot as plt
from km3pipe import Module
from km3pipe.dataclasses import Position
from km3pipe.tools import unit_vector, angle_between
from km3pipe import constants
from km3pipe.logger import logging
log = logging.getLogger(__name__) # pylint: disable=C0103
from .minimiser import QualityFunction
import iminuit as minuit
n = 1.3797
c = constants.c / 1e9
class ZTPlotter(Module):
"""A z-t-plotter"""
#TODO: rewrite me, I'm hard coded!
def process(self, blob):
mc_track = blob['TrackIns'][0]
self.plot_hyperbola(mc_track)
from PIL import Image
from rainbowalga.tools import Clock, Camera, draw_text_2d
from rainbowalga.physics import Particle, ParticleFit, Neutrino, Hit
from rainbowalga.gui import Colourist
from rainbowalga import constants
from rainbowalga import version
from km3pipe.dataclasses import Position
from km3pipe.hardware import Detector
from km3pipe.pumps import EvtPump
from km3pipe.tools import pdg2name, angle_between
from km3pipe import constants
from km3pipe.logger import logging
log = logging.getLogger('rainbowalga') # pylint: disable=C0103
class RainbowAlga(object):
def __init__(self, detector_file=None, event_file=None, min_tot=None,
skip_to_blob=0,
width=1000, height=700, x=50, y=50):
self.camera = Camera()
self.camera.is_rotating = True
self.colourist = Colourist()
current_path = os.path.dirname(os.path.abspath(__file__))
if not detector_file:
detector_file = os.path.join(current_path,
