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'])