def listSkimmedSampleDirectories(input_directory):
for _, sample in listSampleDirectories(input_directory, 'ntuples_skimmed'):
yield sample
def listSkimmedSampleDirectories(input_directory):
# list all directories containing unmerged skimmed ntuples
# note: naming convention depends on skimTuples.py / skimTuplesFromList.py
for _, sample in listSampleDirectories(input_directory, 'ntuples_skimmed'):
yield sample
# print sample names to use
print('Extracted {} sample names from the sample list:'.format(
len(samples_to_use)))
for sample_to_use in samples_to_use:
print(' - {}'.format(sample_to_use))
# make a list of samples (the main directories)
# and an equally long list of subdirectories with the latest version of the ntuples
# (one for each main directory)
print('Finding samples in provided input directory...')
sample_directories = []
sample_sub_directories = []
# case 1: version name is given as fixed argument for all samples
if version_name is not None:
for sample_directory, subdirectory in listSampleDirectories(
input_directory, version_name):
# extra selection using samplelist:
sample_name = sample_directory.rstrip(os.path.sep).split(
os.path.sep)[-1]
if (useall or sample_name in samples_to_use):
sample_directories.append(sample_directory)
sample_sub_directories.append(subdirectory)
# case 2: version name is extracted from the sample list
else:
for sample in samples_dict:
fullpath = os.path.join(input_directory, sample['sample_name'],
sample['version_name'])
if os.path.exists(fullpath):
sample_directories.append(
os.path.join(input_directory, sample['sample_name']))
sample_sub_directories.append(sample['version_name'])