96 lines
2.0 KiB
Python
96 lines
2.0 KiB
Python
import os, sys
|
|
from pathlib import Path
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
#from creader import ClusterFileReader
|
|
import creader as cr
|
|
import clustersFunctions as cf
|
|
|
|
fname = "/mnt/myData/230914_30s_star_100um_nofi/star_"
|
|
fnameff = "/mnt/myData/230914_30s_flat_100um_nofi/flat_"
|
|
xmin=161+20
|
|
xmax=xmin+40
|
|
ymin=161+20
|
|
ymax=ymin+40
|
|
emin=0
|
|
emax=30
|
|
ecutmin=8
|
|
ecutmax=12
|
|
etabins=251
|
|
csize=3
|
|
gain=150
|
|
nbins=100
|
|
indmin=1
|
|
indmax=20
|
|
|
|
|
|
subpix=5
|
|
|
|
|
|
im=None
|
|
intim=None
|
|
etas=None
|
|
sp=None
|
|
ietax=None
|
|
ietay=None
|
|
|
|
for i in range(indmin,indmax+1):
|
|
#ff=fname
|
|
ff=fnameff+str(i)+".clust"
|
|
print(ff)
|
|
r = cr.ClusterFileReader(ff)
|
|
im, sp, ebins, etas, etabinsx, etabinsy=cf.analyze_clusters(r, emin, emax, ecutmin, ecutmax, xmin, xmax, ymin, ymax, ietax, ietay, im, sp, etas, intim,csize, gain, nbins, etabins)
|
|
print(np.sum(im))
|
|
|
|
ietax, ietay=cf.prepare_interpolation(etas)
|
|
|
|
|
|
|
|
|
|
im=None
|
|
intim=None
|
|
etas=None
|
|
sp=None
|
|
|
|
|
|
for i in range(indmin,indmax+1):
|
|
#ff=fname
|
|
ff=fname+str(i)+".clust"
|
|
print(ff)
|
|
r = cr.ClusterFileReader(ff)
|
|
im, intim, sp, ebins, etas, etabinsx, etabinsy=cf.analyze_clusters(r, emin, emax, ecutmin, ecutmax, xmin, xmax, ymin, ymax, ietax, ietay, im, sp, etas, intim, csize, gain, nbins, etabins, subpix)
|
|
|
|
|
|
|
|
imff=None
|
|
intimff=None
|
|
etasff=None
|
|
spff=None
|
|
|
|
for i in range(indmin,indmax):
|
|
ff=fnameff+str(i)+".clust"
|
|
print(ff)
|
|
r = cr.ClusterFileReader(ff)
|
|
imff, intimff, spff, ebins, etasff, etabinsx, etabinsy=cf.analyze_clusters(r, emin, emax, ecutmin, ecutmax, xmin, xmax, ymin, ymax,ietax, ietay, imff, spff, etasff, intimff, csize, gain, nbins, etabins, subpix)
|
|
|
|
|
|
|
|
|
|
fig, ax = plt.subplots()
|
|
ax.plot(ebins[:-1],sp)
|
|
ax.plot(ebins[:-1],spff)
|
|
#ax.set_yscale('log')
|
|
fig.show()
|
|
"""
|
|
fig1, axs1 = plt.subplots()
|
|
vv=axs1.imshow(intim/intimff,vmax=1.,origin='upper',cmap=plt.cm.jet)
|
|
fig1.colorbar(vv, ax=axs1)
|
|
fig1.show()
|
|
"""
|
|
cf.plot_colz(im/imff,1.1)
|
|
cf.plot_colz(intim/intimff,1.1)
|
|
|
|
cf.plot_colz(etas,np.max(etas))
|
|
cf.plot_colz(ietax,1.1)
|
|
cf.plot_colz(ietay,1.1)
|