Thomas Fitoussi / simulation-analysis

Blame view

Modules/Read.py 2.4 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
65135318   Thomas Fitoussi   Reset modules 
10
11
12
13
14
15
16
17
18
19
20
21
22
 
def ReadExtraFile(fileName,cols=[0,1,2,3,4,5]):
   '''
      charge, arrival redshift, weight, generation, arrival angle, emitting energy (source)
   '''
   return loadtxt(resultDirectory+fileName+extraFile,unpack=True,usecols=cols)

def ReadProfile(fileName,cols=[0,1,2,3,4,5]): 
   '''
      Source: Emin, redshift, dist (Mpc), total number of photons emitted, EGMF amplitude, 
              number of leptons producted
   '''
   return loadtxt(resultDirectory+fileName+profileFile,unpack=True,usecols=cols)
7030f150   Thomas Fitoussi   Full reorganisati... 
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
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])
7030f150   Thomas Fitoussi   Full reorganisati... 
41
42
43
 
def ReadElmag(fileName):
   Elmag=resultDirectory+fileName+ElmagFile
   energy,fluxGamma,fluxLepton = loadtxt(Elmag,unpack=True,usecols=[0,1,2])
374f3d52   Thomas Fitoussi   add Weight distri... 
44
45
 
   energy *= 10**(-9) #GeV
   flux=(fluxGamma+fluxLepton)*1e-9 #GeV normalized to 1 initial photon
7030f150   Thomas Fitoussi   Full reorganisati... 
46
 
   return energy,flux
074c1729   Thomas Fitoussi   Analysis.py used ... 
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 

def ReadSpectrum(fileName,cols=[0,1,2,3,4,5,6,7,8]): 
   return loadtxt(resultDirectory+fileName+"/Spectrum.txt",unpack=True,usecols=cols)

def ReadSourceSpectrum(fileName,cols=[0,1,2,3,4]): 
   return loadtxt(resultDirectory+fileName+"/Source_spectrum.txt",unpack=True,usecols=cols)

def ReadAngleVsEnergy(fileName,cols=[0,1,2,3,4]): 
   return loadtxt(resultDirectory+fileName+"/Angle_vs_Energy.txt",unpack=True,usecols=cols)

def ReadRadial(fileName,cols=[0,1,2,3,4]): 
   return loadtxt(resultDirectory+fileName+"/Radial_distribution.txt",unpack=True,usecols=cols)

def ReadTiming(fileName,cols=[0,1,2,3,4]): 
   return loadtxt(resultDirectory+fileName+"/Timing.txt",unpack=True,usecols=cols)
631c85a9   Thomas Fitoussi   Ad generation com... 
62
63
64
 

def ReadGeneration(fileName,cols=[0,1,2,3,4]):
   return loadtxt(resultDirectory+fileName+"/Generation.txt",unpack=True,usecols=cols)