Thomas Fitoussi / simulation-analysis

Blame view

Modules/Read.py 2.25 KB
Edit Raw Normal View History
7030f150   Thomas Fitoussi   Full reorganisati... 
1
2
3
4
5
6
 
from numpy import loadtxt

resultDirectory="Results/"
positionFile="/position"
momentumFile="/momentum"
extraFile="/extra"
8011cc96   Thomas Fitoussi   Add profile readi... 
7
 
profileFile="/profile"
8ffbee93   Thomas Fitoussi   Analytic expressi... 
8
 
ElmagFile="/Elmag_spectrum"
7030f150   Thomas Fitoussi   Full reorganisati... 
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 

def ReadTime(fileName):
   return loadtxt(resultDirectory+fileName+positionFile,unpack=True,usecols=[0])

def ReadPosition(fileName):
   return loadtxt(resultDirectory+fileName+positionFile,unpack=True,usecols=[1,2,3])

def ReadPositionAngle(fileName):
   return loadtxt(resultDirectory+fileName+positionFile,unpack=True,usecols=[4,5])

def ReadEnergy(fileName):
   return loadtxt(resultDirectory+fileName+momentumFile,unpack=True,usecols=[0])

def ReadMomentum(fileName):
   return loadtxt(resultDirectory+fileName+momentumFile,unpack=True,usecols=[1,2,3])

def ReadMomentumAngle(fileName):
   return loadtxt(resultDirectory+fileName+momentumFile,unpack=True,usecols=[4,5])

def ReadCharge(fileName):
   return loadtxt(resultDirectory+fileName+extraFile,unpack=True,usecols=[0])

def ReadRedshift(fileName):
   return loadtxt(resultDirectory+fileName+extraFile,unpack=True,usecols=[1])

def ReadWeight(fileName):
   return loadtxt(resultDirectory+fileName+extraFile,unpack=True,usecols=[2])

def ReadGeneration(fileName):
   return loadtxt(resultDirectory+fileName+extraFile,unpack=True,usecols=[3])

def ReadElmag(fileName):
   Elmag=resultDirectory+fileName+ElmagFile
   energy,fluxGamma,fluxLepton = loadtxt(Elmag,unpack=True,usecols=[0,1,2])
374f3d52   Thomas Fitoussi   add Weight distri... 
43
44
 
   energy *= 10**(-9) #GeV
   flux=(fluxGamma+fluxLepton)*1e-9 #GeV normalized to 1 initial photon
7030f150   Thomas Fitoussi   Full reorganisati... 
45
 
   return energy,flux
8011cc96   Thomas Fitoussi   Add profile readi... 
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
 

def ReadE_source(fileName): # Emax (GeV), redshift, distance (Mpc), Nb photons 
   return loadtxt(resultDirectory+fileName+profileFile,unpack=True,usecols=[0])

def Readz_source(fileName): # redshift
   return loadtxt(resultDirectory+fileName+profileFile,unpack=True,usecols=[1])

def ReadD_source(fileName): # distance (Mpc)
   return loadtxt(resultDirectory+fileName+profileFile,unpack=True,usecols=[2])

def ReadNphot_source(fileName): # Nb photons emitted
   return loadtxt(resultDirectory+fileName+profileFile,unpack=True,usecols=[3])

def ReadEGMF(fileName):
   return loadtxt(resultDirectory+fileName+profileFile,unpack=True,usecols=[4])

def ReadNbLeptonsProd(fileName):
   return loadtxt(resultDirectory+fileName+profileFile,unpack=True,usecols=[5])