#!/usr/bin/env python

import argparse as arg
import os.path as op

import pdb_files as pdb

parser = arg.ArgumentParser('Convert trk files to pdb files')

parser.add_argument('in_file', action='store', metavar='File', 
                    help='File containing tractography results in trk format')

parser.add_argument('--out_file', action='store', metavar='File', 
                    help='File in trk format to save (default: same name as trk input)',
		    default=False)

in_args = parser.parse_args()

if __name__=="__main__":
    trk_file = in_args.in_file
    pdb_file = in_args.out_file
    if not pdb_file:
        pdb_file = op.splitext(trk_file)[0] + '.pdb'
    pdb.trk2pdb(trk_file, pdb_file)

   
   
