#!/bin/env python
import numpy as np
import os,sys
import json

# Parse SRIM2008
units={
    'keV': 1e-3, 'MeV': 1, 'GeV': 1e3,
    'A': 1e-10, 'um': 1e-6, 'mm': 1e-3, 'm': 1
}
for fname in sys.argv[1:]:
    data=np.genfromtxt(fname,skip_header=24,skip_footer=13,converters=
                       {1: lambda u: units[u] or None,
                        5: lambda u: units[u] or None,
                        7: lambda u: units[u] or None,
                        9: lambda u: units[u] or None}
    )
    out=open('%s.json'%(fname),'w')
    out.write(json.dumps({
        'energy (MeV)': list(data[:,0]*data[:,1]),
        'dEdxElec (MeV-cm2/mg)': list(data[:,2]),
        'dEdxNuc (MeV-cm2/mg)': list(data[:,3]),
        'projRange (m)': list(data[:,4]*data[:,5]),
        'longStrag (m)': list(data[:,6]*data[:,7]),
        'latStrag (m)': list(data[:,8]*data[:,9])
    }))
    out.close()