freesurfer-surface/freesurfer_surface/__main__.py

import argparse
import csv
import sys

from freesurfer_surface import Annotation


def annotation_labels():
    """
    List Labels Stored in Freesurfer's Annotation File
    (i.e., label/lh.aparc.annot)
    """
    argparser = argparse.ArgumentParser(
        description=annotation_labels.__doc__.strip())
    argparser.add_argument('--delimiter', default='\t',
                           help='default: %(default)r')
    argparser.add_argument('annotation_file_path')
    args = argparser.parse_args()
    annotation = Annotation.read(args.annotation_file_path)
    csv_writer = csv.writer(sys.stdout, delimiter=args.delimiter)
    csv_writer.writerow(('index', 'color', 'name'))
    labels = sorted(annotation.labels.values(), key=lambda l: l.index)
    csv_writer.writerows((l.index, l.hex_color_code, l.name,) for l in labels)