if slice_entries <= file_entries[file_curr] - entry_curr:

return [file_curr, slice_entries + entry_curr]

elif file_curr == len(file_entries) - 1:

return [file_curr, file_entries[-1]]

else:

slice_entries = math.ceil(sum(file_entries) / n_processes)

slices = []

file_start = 0

entry_start = 0

for i in range(n_processes):

slices.append([file_start, entry_start] + partition_helper(slice_entries, file_entries, file_start, entry_start))

file_start = slices[-1][-2]

entry_start = slices[-1][-1]

data_slice = []

for i in range(slices[index][0], slices[index][2] + 1):

with uproot.open(paths[i], object_cache=None, array_cache=None) as file:

data_slice.append(file.arrays("candidate_vMass",

"(candidate_charge == 0)\

& (candidate_vMass > 5.33) & (candidate_vMass < 5.4)",

entry_start=slices[index][1] if i == slices[index][0] else None,

entry_stop=slices[index][3] if i == slices[index][2] else None,

array_cache=None,

library="np")["candidate_vMass"])

if n_files == 0: return 0

if n_processes == 0: return runtime_measure(path, n_files)

start = time.time()

paths = [path + filename + ":rootuple/CandidateTree" for filename in sorted(os.listdir(path))[:n_files]]

file_entries = [n[2] for n in uproot.num_entries([p for p in paths])]

slices = partition(file_entries, n_processes)

result = multiprocessing.Manager().list()

processes = []

for i in range(n_processes):

p = multiprocessing.Process(target=read_slice, args=[paths, slices, i, result])

p.start()

processes.append(p)

for p in processes:

p.join()

np.concatenate(tuple(result))

if n_files == 0: return 0

start = time.time()

paths = [path + filename + ":rootuple/CandidateTree" for filename in sorted(os.listdir(path))[:n_files]]

data = []

for p in paths:

with uproot.open(p, object_cache=None, array_cache=None) as file:

data.append(file.arrays("candidate_vMass",

"(candidate_charge == 0)\

& (candidate_vMass > 5.33) & (candidate_vMass < 5.4)",

array_cache=None,

library="np")["candidate_vMass"])

np.concatenate(tuple(data))

if not os.path.exists(target_dir):

os.makedirs(target_dir)

result_path = ("%s/runtime_vs_%s_%d_%d_%d_%d.csv" % (target_dir, variable, constant, var_max, step, n_loops)) if constant else ("%s/runtime_vs_%s_%d_%d_%d.csv" % (target_dir, variable, var_max, step, n_loops))

x = [a for a in range(0, var_max + step, step)]

if not os.path.exists(result_path):

with open(result_path, "w", newline="") as f:

csv.writer(f).writerow(x)

for n in range(n_loops):

with open(result_path, "r") as f:

if sum(1 for row in csv.reader(f)) == n_loops + 1: break

y = [measure_function(*(path, i if "size" in variable else constant, constant if "size" in variable else i) if constant else (path, i)) for i in range(0, var_max + step, step)]

with open(result_path, "a", newline="") as f: