#!/usr/bin/env python
# -*- coding: utf-8 -*-
import seutils
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('rootfile', type=str, help='Rootfile to be drawn from')
parser.add_argument('branch', type=str, help='Branch name to be drawn')
parser.add_argument('-t', '--tree', type=str, help='Name of the tree to count', default='auto')
parser.add_argument('--pdf', action='store_true', help='Make a pdf instead of a (default) png')
args = parser.parse_args()

def main():
    ext = '.pdf' if args.pdf else '.png'
    seutils.root.draw_branch(args.rootfile, args.branch, tree=args.tree, ext=ext)

if __name__ == '__main__':
    main()