[Examples] Update examples

Author: EdoPro98

examples/afterpulse.py

@@ -1,28 +1,30 @@
 import SiPM
 import numpy as np
 import matplotlib.pyplot as plt
+import mplhep
 
 rng = SiPM.SiPMRandom()
+prop = SiPM.SiPMProperties()
+prop.setSampling(1)
+sensor = SiPM.SiPMSensor(prop)
 
-sensor = SiPM.SiPMSensor()
-
-N = 100000
+N = 50000
 
 integral = np.empty(N, dtype=np.float64)
 
 for i in range(N):
     sensor.resetState()
-    nphe = rng.randPoisson(5)
+    nphe = rng.randPoisson(10)
     if nphe > 0:
         times = rng.randGaussian(25,0.1,nphe)
         sensor.addPhotons(times)
     sensor.runEvent()
-
     integral[i] = sensor.signal().integral(20,250,0.0)
 
+integral[integral == np.inf] = 0 
+
 fig,ax = plt.subplots()
-plt.hist(integral, 500, color="k")
+plt.hist(integral,int(2*N**0.5), color="k")
 plt.xlabel("Integral [A.U.]")
-ax.set_yscale("log")
 plt.show()
 

examples/mulptiphoton.py

@@ -6,7 +6,7 @@
 
 sensor = SiPM.SiPMSensor()
 
-sensor.setProperty("SignalLength",5000)
+sensor.setProperty("SignalLength",1000)
 sensor.setProperty("FallTimeFast",20)
 sensor.setProperty("Dcr",5e6)
 sensor.setProperty("Xt",0.5)
@@ -20,20 +20,23 @@
 
     signal = np.array(sensor.signal().waveform())
     hits = sensor.hits()
+    hitsGraph = sensor.hitsGraph()
     x = np.arange(0,sensor.properties().signalLength(),sensor.properties().sampling())
 
     fig,ax = plt.subplots()
-    ax.plot(x,signal,"k",lw=1)
+    maximum = signal.max()
 
-    for h in hits:
+
+    for i,h in enumerate(hits):
         if h.hitType() == SiPM.SiPMHit.HitType.kDarkCount:
-            plt.plot(h.time(), h.amplitude(), "or", ms=10)
-        if h.hitType() == SiPM.SiPMHit.HitType.kPhotoelectron:
-            plt.plot(h.time(), h.amplitude(), ".k", ms=10)
+            plt.vlines(h.time(),0, maximum, 'r')
+            plt.plot(h.time(),0,'^r',ms=8)
         if (h.hitType() == SiPM.SiPMHit.HitType.kDelayedOpticalCrosstalk) | (h.hitType() == SiPM.SiPMHit.HitType.kOpticalCrosstalk):
-            plt.plot(h.time(), h.amplitude(), "vg", ms=10)
+            plt.vlines(h.time(),0, maximum, 'g')
+            plt.plot(h.time(),0,'^g',ms=8)
         if (h.hitType() == SiPM.SiPMHit.HitType.kFastAfterPulse) | (h.hitType() == SiPM.SiPMHit.HitType.kSlowAfterPulse):
-            plt.plot(h.time(), h.amplitude(), "^b", ms=10)
-    ax.grid()
+            plt.vlines(h.time(),0, maximum, 'b')
+            plt.plot(h.time(),0,'^b',ms=8)
+    ax.plot(x,signal,"k",lw=2)
     plt.legend(["Signal","Dcr","Xt","Ap"])
     plt.show()

examples/noise.py

@@ -23,7 +23,7 @@
 
 sensor3.setProperty("SlowComponentFraction",0.10)
 sensor3.setProperty("FallTimeFast",5)
-sensor3.setProperty("FallTimeSlow",50)
+sensor3.setProperty("FallTimeSlow",80)
 sensor3.properties().setSlowComponentOn()
 
 sensor1.setProperty("Snr",30)

examples/signals.py

@@ -7,7 +7,7 @@
 from scipy.signal import savgol_filter
 plt.style.use(mplhep.style.ATLAS)
 
-dcr = 300
+dcr = 100
 xt = 0.10
 sensor = SiPM.SiPMSensor()
 sensor.setProperty("Dcr",dcr*1000)
@@ -26,15 +26,16 @@ def fitFun(x,
 
 start = 0
 end = 4.0
-step = 0.05
+step = 0.1
 window = 250
 N = 25000
+n = 100
 
 threshold = np.arange(start,end,step)
 nstep = threshold.size
-counts = np.zeros((nstep,25))
+counts = np.zeros((nstep,n))
 
-for j in range(25):
+for j in range(n):
     peaks = np.zeros(N)
     for i in range(N):
         sensor.resetState()
@@ -58,7 +59,7 @@ def fitFun(x,
 fig,ax = plt.subplots()
 ax.plot(threshold, rate, ".k")
 diffRate = -savgol_filter(rate,5,2,1)
-ax.plot(threshold, diffRate)
+ax.plot(threshold, diffRate,'b')
 ax.fill_between(threshold, rate+raterr, rate-raterr, color="r",alpha=0.25)
 ax.plot(np.linspace(0,4.0,1000), fitFun(np.linspace(0,4.0,1000),*par), "--g")
 ax.set_yscale("log")