#!/usr/bin/env python

import argparse as arg
import os.path as op

import pdb_files as pdb

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

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

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

in_args = parser.parse_args()

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

   
   
