#!python
# -*- coding: utf-8 -*-

"""
convert Elegant lte file to json string file

Tong Zhang
2016-06-14 14:39:29 PM CST
"""

import argparse
import sys

parser = argparse.ArgumentParser(description="Convert Elegant lte file into json string file")

parser.add_argument('--lte', dest='ltefile', help='.lte file to read')
parser.add_argument('--json', dest='jsonfile', help='.json file to generate, if None or omitted, write to stdout')


args = parser.parse_args()

ltefile, jsonfile = args.ltefile, args.jsonfile

if args.ltefile is None:
    parser.print_help()
    sys.exit(1)

import beamline
lpins = beamline.LteParser(ltefile)
if jsonfile is None:
    print(lpins.file2json())
else:
    lpins.file2json(jsonfile)
