TDCRPy 2.8.0__py3-none-any.whl → 2.11.0__py3-none-any.whl

tdcrpy/TDCR_model_lib.py

@@ -61,8 +61,12 @@ def readParameters(disp=False):
     fAq = config["Inputs"].getfloat("fAq")
     micCorr = config["Inputs"].getboolean("micCorr")
     alphaDir = config["Inputs"].getfloat("alphaDir")
-    effQuantic = config["Inputs"].getfloat("effQuantum")
-    optionModel = config["Inputs"].getfloat("optionModel")
+    effQuantic = config["Inputs"].get("effQuantum")
+    effQuantic = effQuantic.split(',')
+    for i, iS in enumerate(effQuantic):
+        iS=iS.replace(" ","")
+        if iS != 'None':  effQuantic[i]=float(iS)
+    optionModel = config["Inputs"].get("optionModel")
     
     if disp:
         print(f"number of integration bins for electrons = {nE_electron}")
@@ -3067,11 +3071,6 @@ def detectProbabilitiesMC(L, e_quenching, e_quenching2, t1, evenement, extDT, me
         detection probability of coincidences in a C/N system.
 
     """
-    if isinstance(L, (tuple, list)):
-        symm = False
-    else:
-        symm = True
-    
     mu = effQuantic
     
     if dir_param < 1000:
@@ -3081,129 +3080,91 @@ def detectProbabilitiesMC(L, e_quenching, e_quenching2, t1, evenement, extDT, me
         dirichTD = [1/3, 1/3, 1/3]
         dirichCN = [1/2, 1/2]
     
+    if type(L) == float:
+        L = [L, L, L]
+    
+    def PMBmodel(e_q, dirichTD, dirichCN, L, mu):
+        n_e=np.zeros(3); n_eCN=np.zeros(2)
+        n_ph = np.random.poisson(sum(np.asarray(e_q))*np.mean(L)/np.mean(mu))
+        # TDCR
+        n_phPMT = np.random.multinomial(n_ph, dirichTD)
+        n_e[0]=np.random.binomial(n_phPMT[0],mu[0])
+        n_e[1]=np.random.binomial(n_phPMT[1],mu[0])
+        n_e[2]=np.random.binomial(n_phPMT[2],mu[0])
+        # C/N
+        n_phPMTCN = np.random.multinomial(n_ph, dirichCN)
+        n_eCN[0]=np.random.binomial(n_phPMTCN[0],mu[0])
+        n_eCN[1]=np.random.binomial(n_phPMTCN[1],mu[0])
+        return n_e, n_eCN
+    
+    def Pmodel(e_q, dirichTD, dirichCN, L, mu):
+        n_e=np.zeros(3); n_eCN=np.zeros(2)
+        n_e[0] = np.random.poisson(sum(np.asarray(e_q))*L[0]*mu[0]*dirichTD[0])
+        n_e[1] = np.random.poisson(sum(np.asarray(e_q))*L[1]*mu[1]*dirichTD[1])
+        n_e[2] = np.random.poisson(sum(np.asarray(e_q))*L[2]*mu[2]*dirichTD[2])
+        n_eCN[0] = np.random.poisson(sum(np.asarray(e_q))*L[0]*mu[0]*dirichCN[0])
+        n_eCN[1] = np.random.poisson(sum(np.asarray(e_q))*L[1]*mu[1]*dirichCN[1])
+        return n_e, n_eCN
+    
+    def EPmodel(e_q, dirichTD, dirichCN, L, mu):
+        n_e=np.zeros(3); n_eCN=np.zeros(2)
+        n_e[0] = sum(np.asarray(e_q))*L[0]*mu[0]*dirichTD[0]
+        n_e[1] = sum(np.asarray(e_q))*L[1]*mu[1]*dirichTD[1]
+        n_e[2] = sum(np.asarray(e_q))*L[2]*mu[2]*dirichTD[2]
+        n_eCN[0] = sum(np.asarray(e_q))*L[0]*mu[0]*dirichCN[0]
+        n_eCN[1] = sum(np.asarray(e_q))*L[1]*mu[1]*dirichCN[1]
+        return n_e, n_eCN
+    
+    # def Amodel(e_q, dirichTD, dirichCN, L, mu):
+    #     n_e=np.zeros(3); n_eCN=np.zeros(2)
+    #     n_e[0] = 1-np.exp(-L[0]*np.sum(np.asarray(e_q))*dirichTD[0])
+    #     n_e[1] = 1-np.exp(-L[0]*np.sum(np.asarray(e_q))*dirichTD[0])
+    #     n_e[2] = 1-np.exp(-L[0]*np.sum(np.asarray(e_q))*dirichTD[0])
+    #     n_eCN[0] = 1-np.exp(-L[0]*np.sum(np.asarray(e_q))*dirichTD[0])
+    #     n_eCN[1] = 1-np.exp(-L[0]*np.sum(np.asarray(e_q))*dirichTD[0])
+    #     return n_e, n_eCN  
+    
+    
     efficiency0_S = 0;    efficiency0_T = 0;    efficiency0_D = 0
     efficiency0_AB = 0;    efficiency0_BC = 0;    efficiency0_AC = 0
     efficiency0_D2 = 0;
-    n_e = np.zeros(3); n_eCN = np.zeros(2); n_e2 = np.zeros(3); n_e2CN = np.zeros(2)
-    
-    if symm:
-        if optionModel == "poisson-multinomial-binomial":
-            n_ph = np.random.poisson(sum(np.asarray(e_quenching))*L/mu)
-            # TDCR
-            n_phPMT = np.random.multinomial(n_ph, dirichTD)
-            n_e[0]=np.random.binomial(n_phPMT[0],mu)
-            n_e[1]=np.random.binomial(n_phPMT[1],mu)
-            n_e[2]=np.random.binomial(n_phPMT[2],mu)
-            # C/N
-            n_phPMTCN = np.random.multinomial(n_ph, dirichCN)
-            n_eCN[0]=np.random.binomial(n_phPMTCN[0],mu)
-            n_eCN[1]=np.random.binomial(n_phPMTCN[1],mu)
-        elif optionModel == "poisson":
-            n_e[0] = np.random.poisson(sum(np.asarray(e_quenching))*L*mu*dirichTD[0])
-            n_e[1] = np.random.poisson(sum(np.asarray(e_quenching))*L*mu*dirichTD[1])
-            n_e[2] = np.random.poisson(sum(np.asarray(e_quenching))*L*mu*dirichTD[2])
-            n_eCN[0] = np.random.poisson(sum(np.asarray(e_quenching))*L*mu*dirichCN[0])
-            n_eCN[1] = np.random.poisson(sum(np.asarray(e_quenching))*L*mu*dirichCN[1])           
-        else:
-            print("unknown model")        
-                
-        if sum(n_e>1)>0: efficiency0_S =1
-        if sum(n_e>1)>1: efficiency0_D =1
-        if sum(n_e>1)>2: efficiency0_T =1
-        if n_e[0]>1 and n_e[1]>1: efficiency0_AB =1 
-        if n_e[1]>1 and n_e[2]>1: efficiency0_BC =1 
-        if n_e[0]>1 and n_e[2]>1: efficiency0_AC =1
-        if sum(n_eCN>1)>1: efficiency0_D2 =1
-        
-        if evenement !=1 and t1 > extDT*1e-6 and t1 < measTime*60:
-            if optionModel == "poisson-multinomial-binomial":
-                n_ph2 = np.random.poisson(sum(np.asarray(e_quenching2))*L/mu)
-                # TDCR
-                n_phPMT2 = np.random.multinomial(n_ph2, dirichTD)
-                n_e2[0]=np.random.binomial(n_phPMT2[0],mu)
-                n_e2[1]=np.random.binomial(n_phPMT2[1],mu)
-                n_e2[2]=np.random.binomial(n_phPMT2[2],mu)            
-                # C/N
-                n_phPMT2CN = np.random.multinomial(n_ph2, dirichCN)
-                n_e2CN[0]=np.random.binomial(n_phPMT2CN[0],mu)
-                n_e2CN[1]=np.random.binomial(n_phPMT2CN[1],mu)
-            elif optionModel == "poisson":
-                n_e2[0] = np.random.poisson(sum(np.asarray(e_quenching2))*L*mu*dirichTD[0])
-                n_e2[1] = np.random.poisson(sum(np.asarray(e_quenching2))*L*mu*dirichTD[1])
-                n_e2[2] = np.random.poisson(sum(np.asarray(e_quenching2))*L*mu*dirichTD[2])
-                n_e2CN[0] = np.random.poisson(sum(np.asarray(e_quenching2))*L*mu*dirichCN[0])
-                n_e2CN[1] = np.random.poisson(sum(np.asarray(e_quenching2))*L*mu*dirichCN[1])           
-            else:
-                print("unknown model")        
-            
-                
-            if sum(n_e2>1)>0: efficiency0_S +=1
-            if sum(n_e2>1)>1: efficiency0_D +=1
-            if sum(n_e2>1)>2: efficiency0_T +=1
-            if n_e2[0]>1 and n_e2[1]>1: efficiency0_AB +=1 
-            if n_e2[1]>1 and n_e2[2]>1: efficiency0_BC +=1 
-            if n_e2[0]>1 and n_e2[2]>1: efficiency0_AC +=1
-            if sum(n_e2CN>1)>1: efficiency0_D2 +=1           
+    # n_e = np.zeros(3); n_eCN = np.zeros(2); n_e2 = np.zeros(3); n_e2CN = np.zeros(2)
+
+    if optionModel == "poisson-multinomial-binomial":
+        n_e, n_eCN = PMBmodel(e_quenching, dirichTD, dirichCN, L, mu)
+    elif optionModel == "poisson":
+        n_e, n_eCN = Pmodel(e_quenching, dirichTD, dirichCN, L, mu)
+    elif optionModel == "expectation":
+        n_e, n_eCN =  EPmodel(e_quenching, dirichTD, dirichCN, L, mu)
+    else:
+        print("unknown model")        
             
-    else: # asym
+    if sum(n_e>0)>0: efficiency0_S =1
+    if sum(n_e>0)>1: efficiency0_D =1
+    if sum(n_e>0)>2: efficiency0_T =1
+    if n_e[0]>0 and n_e[1]>0: efficiency0_AB =1 
+    if n_e[1]>0 and n_e[2]>0: efficiency0_BC =1 
+    if n_e[0]>0 and n_e[2]>0: efficiency0_AC =1
+    if sum(n_eCN>1)>1: efficiency0_D2 =1
+    
+    if evenement !=1 and t1 > extDT*1e-6 and t1 < measTime*60:
         if optionModel == "poisson-multinomial-binomial":
-            Lm = np.mean(L)
-            n_ph = np.random.poisson(sum(np.asarray((e_quenching))*Lm/mu))
-            # TDCR
-            n_phPMT = np.random.multinomial(n_ph, [L[0]*dirichTD[0]/Lm, L[1]*dirichTD[1]/Lm, L[2]*dirichTD[2]/Lm])
-            n_e[0]=np.random.binomial(n_phPMT[0],mu)
-            n_e[1]=np.random.binomial(n_phPMT[1],mu)
-            n_e[2]=np.random.binomial(n_phPMT[2],mu)
-            # C/N
-            n_phPMTCN = np.random.multinomial(n_ph, [L[0]/(2*Lm), L[1]/(2*Lm)])
-            n_eCN[0]=np.random.binomial(n_phPMTCN[0],mu)
-            n_eCN[1]=np.random.binomial(n_phPMTCN[1],mu)
+            n_e2, n_e2CN = PMBmodel(e_quenching2, dirichTD, dirichCN, L, mu)
         elif optionModel == "poisson":
-            n_e[0] = np.random.poisson(sum(np.asarray(e_quenching))*L[0]*mu*dirichTD[0])
-            n_e[1] = np.random.poisson(sum(np.asarray(e_quenching))*L[1]*mu*dirichTD[1])
-            n_e[2] = np.random.poisson(sum(np.asarray(e_quenching))*L[2]*mu*dirichTD[2])
-            n_eCN[0] = np.random.poisson(sum(np.asarray(e_quenching))*L[0]*mu*dirichCN[0])
-            n_eCN[1] = np.random.poisson(sum(np.asarray(e_quenching))*L[1]*mu*dirichCN[1])           
+            n_e2, n_e2CN = Pmodel(e_quenching2, dirichTD, dirichCN, L, mu) 
+        elif optionModel == "expectation":
+            n_e2, n_e2CN =  EPmodel(e_quenching2, dirichTD, dirichCN, L, mu)
         else:
             print("unknown model")        
-            
-        if sum(n_e>1)>0: efficiency0_S =1
-        if sum(n_e>1)>1: efficiency0_D =1
-        if sum(n_e>1)>2: efficiency0_T =1
-        if n_e[0]>1 and n_e[1]>1: efficiency0_AB =1 
-        if n_e[1]>1 and n_e[2]>1: efficiency0_BC =1 
-        if n_e[0]>1 and n_e[2]>1: efficiency0_AC =1
-        if sum(n_eCN>1)>1: efficiency0_D2 =1       
-        
-        if evenement !=1 and t1 > extDT*1e-6 and t1 < measTime*60:
-            if optionModel == "poisson-multinomial-binomial":
-                n_ph2 = np.random.poisson(sum(np.asarray(e_quenching2))*Lm/mu)
-                # TDCR
-                n_phPMT2 = np.random.multinomial(n_ph2, [L[0]*dirichCN[0]/Lm, L[1]*dirichCN[1]/Lm, L[2]*dirichCN[2]/Lm])
-                n_e2[0]=np.random.binomial(n_phPMT2[0],mu)
-                n_e2[1]=np.random.binomial(n_phPMT2[1],mu)
-                n_e2[2]=np.random.binomial(n_phPMT2[2],mu)
-                # C/N
-                n_phPMT2CN = np.random.multinomial(n_ph2, [L[0]/(2*Lm), L[1]/(2*Lm)])
-                n_e2CN[0]=np.random.binomial(n_phPMT2CN[0],mu)
-                n_e2CN[1]=np.random.binomial(n_phPMT2CN[1],mu)
-            elif optionModel == "poisson":
-                n_e2[0] = np.random.poisson(sum(np.asarray(e_quenching2))*L[0]*mu*dirichTD[0])
-                n_e2[1] = np.random.poisson(sum(np.asarray(e_quenching2))*L[1]*mu*dirichTD[1])
-                n_e2[2] = np.random.poisson(sum(np.asarray(e_quenching2))*L[2]*mu*dirichTD[2])
-                n_e2CN[0] = np.random.poisson(sum(np.asarray(e_quenching2))*L[0]*mu*dirichCN[0])
-                n_e2CN[1] = np.random.poisson(sum(np.asarray(e_quenching2))*L[1]*mu*dirichCN[1])           
-            else:
-                print("unknown model")   
-                
-            if sum(n_e2>1)>0: efficiency0_S +=1
-            if sum(n_e2>1)>1: efficiency0_D +=1
-            if sum(n_e2>1)>2: efficiency0_T +=1
-            if n_e2[0]>1 and n_e2[1]>1: efficiency0_AB +=1 
-            if n_e2[1]>1 and n_e2[2]>1: efficiency0_BC +=1 
-            if n_e2[0]>1 and n_e2[2]>1: efficiency0_AC +=1
-            if sum(n_e2CN>1)>1: efficiency0_D2 +=1  
         
+        if sum(n_e2>0)>0: efficiency0_S +=1
+        if sum(n_e2>0)>1: efficiency0_D +=1
+        if sum(n_e2>0)>2: efficiency0_T +=1
+        if n_e2[0]>0 and n_e2[1]>0: efficiency0_AB +=1 
+        if n_e2[1]>0 and n_e2[2]>0: efficiency0_BC +=1 
+        if n_e2[0]>0 and n_e2[2]>0: efficiency0_AC +=1
+        if sum(n_e2CN>1)>1: efficiency0_D2 +=1           
+                    
     return efficiency0_S, efficiency0_D, efficiency0_T, efficiency0_AB, efficiency0_BC, efficiency0_AC, efficiency0_D2         
 
 

tdcrpy/config.toml

@@ -35,7 +35,7 @@ fAq = 0.1
 # Dirichlet parameter
 alphaDir = 100000
 # Quantum efficiency
-effQuantum = 0.25
+effQuantum = 0.25, 0.25, 0.25
 # Optical MC model
 optionModel = poisson-multinomial-binomial
 

{tdcrpy-2.8.0.dist-info → tdcrpy-2.11.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: TDCRPy
-Version: 2.8.0
+Version: 2.11.0
 Summary: TDCR model
 Home-page: https://pypi.org/project/TDCRPy/
 Author: RomainCoulon (Romain Coulon)

{tdcrpy-2.8.0.dist-info → tdcrpy-2.11.0.dist-info}/RECORD

@@ -1,9 +1,9 @@
 tdcrpy/TDCRPy.py,sha256=VIwZJHtpjBt5e87KUqatCC6eTMr2sFVRl52LAc3PZq8,67850
 tdcrpy/TDCRPy1.py,sha256=QTBZh5B5JWnGB0BQfD-cFmwA9W080OD4sG-aj50-ejo,38106
-tdcrpy/TDCR_model_lib.py,sha256=A48vTVqIVaAslYqE41b15Q_hsMpN5rNQZOf3JVNK6e0,136503
+tdcrpy/TDCR_model_lib.py,sha256=SIC_NRJuZHPW_kXsKksicXyooHula3y-Yd2LKlBDpig,133717
 tdcrpy/TDCRoptimize.py,sha256=c2XIGveeLdVYYek4Rg6dygMvVA2xIrIkMb3L-_jUucM,6496
 tdcrpy/__init__.py,sha256=9Djir8dPNchcJVQvhl-oRHEOsoDkiZlkOhWT-eHR7wQ,95
-tdcrpy/config.toml,sha256=Yg-MBlzxOYiovgVFWq6VtivPkzwyNeWBzwNTc7ADYkU,1634
+tdcrpy/config.toml,sha256=zwVBgi2xtHri9WVcipksO7jHaeuJXGXks4tYhzoRHFY,1646
 tdcrpy/test2.py,sha256=poLLXJyIaCeqh1VSkwgbi-udvY7lQjxz_YStKjJXGhU,501
 tdcrpy/MCNP-MATRIX/Spectra_for_analytical_model/dep_spectrum_C-14.txt,sha256=Eh3KaNbfYHakk_uStLu8K1aFciO6_i_rS2yKxGGppDE,8416
 tdcrpy/MCNP-MATRIX/Spectra_for_analytical_model/dep_spectrum_Ca-45.txt,sha256=TymodcK4ttoO1duZuW3RGOwHFwPPzw2ESPc_H_QQN8k,8830
@@ -74,8 +74,8 @@ tdcrpy/docs/_build/html/source/modules.html,sha256=Jf-qxVBId0UgpwyvYuyjtMNG-ezPO
 tdcrpy/docs/_build/html/source/tdcrpy.html,sha256=-38lHMNFB22p1tWJEeN3yDqfDiCYE304vxDamO1-iRc,3779
 tdcrpy/test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tdcrpy/test/test_tdcrpy.py,sha256=JINqSEMFoNpptE4f3h6ZzTYW1rBx90KkaoQzltSg-No,4692
-tdcrpy-2.8.0.dist-info/licenses/LICENCE.md,sha256=ZTpWyGU3qv_iwEpgvCijoCuCYpOPpyzJCgOk46WpUKU,1066
-tdcrpy-2.8.0.dist-info/METADATA,sha256=DC9TjSh3dw67Jh_F9xMOjXUZAOgF5pyL8Rp7ClBeOzo,45298
-tdcrpy-2.8.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-tdcrpy-2.8.0.dist-info/top_level.txt,sha256=f4vzFFcKSEnonAACs0ZXuRczmroLLqtPTqXFymU_VU0,14
-tdcrpy-2.8.0.dist-info/RECORD,,
+tdcrpy-2.11.0.dist-info/licenses/LICENCE.md,sha256=ZTpWyGU3qv_iwEpgvCijoCuCYpOPpyzJCgOk46WpUKU,1066
+tdcrpy-2.11.0.dist-info/METADATA,sha256=PqaOFwuqYUdhziM8kGpVOMSRxCUvF5zY-U3R1QWGjic,45299
+tdcrpy-2.11.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+tdcrpy-2.11.0.dist-info/top_level.txt,sha256=f4vzFFcKSEnonAACs0ZXuRczmroLLqtPTqXFymU_VU0,14
+tdcrpy-2.11.0.dist-info/RECORD,,