import multiprocessing
from G4Calo import GeometryDescriptor, run_batch, display_event
import sys

if __name__ == '__main__':
    
    gd = GeometryDescriptor()
    
    for _ in range(25):
        gd.addLayer(0.5, "G4_Pb", False)
        gd.addLayer(1.,"G4_POLYSTYRENE",True,1)
    
    df = run_batch(gd, 1000, 'gamma', 1)
    gd = GeometryDescriptor()
    
    for _ in range(5):
        gd.addLayer(0.5, "G4_Pb", False)
        gd.addLayer(1.,"G4_POLYSTYRENE",True,1)
    
    df = run_batch(gd,10, 'gamma', 1)
    #display_event(gd,"gamma", 2)