""" 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,