Multifile Density

density.py

import os
import matplotlib.pyplot as plt
from pathlib import Path
from export_draft import Multifile
DATA_DIRECTORY = Path("/nsls2/data/chx/legacy/Compressed_Data")

densities = []

def multifile_density(multifile):
   total_pixels = 0
   frame_size = multifile.md['nrows'] * multifile.md['ncols']
   for frame_num in range(20, 30):
        pixel_count = multifile.rdrawframe(frame_num)[0].shape[0]
        total_pixels += pixel_count
   print(multifile.filename.name, total_pixels/(frame_size*10))
   return total_pixels/(frame_size*10)

for file in os.listdir(DATA_DIRECTORY):
    if file.endswith(".cmp"):
        try:
            multifile = Multifile(DATA_DIRECTORY / file, 0 ,200)
            density = multifile_density(multifile)
        except Exception as e:
            pass
        else:
            densities.append(density)
            
len(densities)
plt.hist(densities, bins=20)
plt.show()