#!/env/python
import sys
import os

import numpy as np
import pandas as pd

from pandas.io.parsers import read_csv

BASE_PATH='/media/F_DRIVE/FSM 2014'

def interpolate_bathy(bathy_path, out_path):
    df = read_csv(bathy_path,
                  skipinitialspace=True,
                  parse_dates='mission_msecs')
    df.set_index('mission_msecs',inplace=True)
    df.replace({'altitude':{0:np.nan}},inplace=True)
    df['altitude'].interpolate().to_csv(out_path)

if __name__=='__main__':
    def bathy_path(mission_id):
        return os.path.join(BASE_PATH,mission_id,'EXPORTED/BATHY.txt')
    for mission in range(1,13):
        mission_id = 'MSN%03dA' % mission
        print 'interpolating altitude for %s' % mission_id
        try:
            interpolate_bathy(bathy_path(mission_id),
                              '%s_alt.csv' % mission_id)
        except IOError:
            pass # skip