multpleScattering.py

#https://www.cs.uct.ac.za/mit_notes/python/Introduction_to_GUI_Programming.html
import math, random, time
from geometry import *
from constants import *
from expDecay import *
from iterateMuon import iterateMuon
from circleIntersectLine import circleIntersectLine
from propagateMuon import propagateMuon
from chamber import chamber
import numpy as np
random.seed(1.0)


class multpleScattering:
    def __init__(self, X, Y, Phi, Dx, Dy, Dphi, Length, Accuracy, Momentum):
        ## length is a user-mutable parameter in the GUI
        self.X, self.Y, self.Phi = X, Y, Phi
        self.Dx, self.Dy, self.dPhi = Dx, Dy, Dphi
        self.pos = [self.X, self.Y, self.Phi]
        self.dPos = [self.Dx, self.Dy, self.dPhi]
        self.newPos = [
            self.pos[0] + self.dPos[0], self.pos[1] + self.dPos[1],
            self.pos[2] + self.dPos[2]
        ]
        self.aligned = False
        self.Length, self.Accuracy = Length, Accuracy

        self.momentum = Momentum
        if self.momentum == "AUTO":
            self.momentum = (self.Y + self.Dy) / 30

        self.xinit = 0  ## Figure out why xinit and xfinal have hard-coded values
        self.xfinal = 50
        self.expFit = expDecay(self.xinit, self.xfinal, self.momentum)
        self.stepSizes = self.expFit.returnStepSizes()

        self.chamber1 = chamber(1, self.Length, self.X, self.Y, self.Phi,
                                self.X + self.Dx, self.Y + self.Dy,
                                self.Phi + self.dPhi, self.Accuracy,
                                self.stepSizes)

        print("Momentum for Job: ", self.momentum)
        startTime = time.time()
        if self.pos == self.newPos:
            self.aligned = True
            #print(self.aligned)
        for i in range(400):
            if self.chamber1.isDone() or self.aligned == True:
                endTime = time.time()
                break
            shootMuons(self.chamber1)
            self.chamber1.align()
            self.chamber1.resetData()
        self.time = endTime - startTime

    def returnTime(self):
        return self.time

    def returnNumberOfIterations(self):
        return self.chamber1.returnNumberOfIterations()

    def returnMomentum(self):
        newMom = ("%.2f" % self.momentum)
        return newMom


def shootMuons(chamber1):
    for i in range(nEvents):

        if i % 1000 == 0: i * 1.0 / nEvents

        #set up inital state of muon
        angleInitial = 1
        speedInitial = 1000
        angleInitial = random.random() - .5

        charge = random.random()
        if charge > .5: charge = 1
        else: charge = -1

        xInitial = 0
        yInitial = 0

        #create muon track
        #if verbose > 5: print("")
        muonTrack, muonPath = propagateMuon(angleInitial, speedInitial, charge,
                                            xInitial, yInitial)

        chamber1.getResiduals(muonTrack, muonPath)