#!/usr/bin/env python3

import isce
from isceobj.Sensor import createSensor
import shelve
import argparse
import glob
from isceobj.Util import Poly1D
from isceobj.Planet.AstronomicalHandbook import Const
import os
from mroipac.dopiq.DopIQ import DopIQ
#from isceobj.Util.decorators import use_api

import copy

def cmdLineParse():
    '''
    Command line parser.
    '''

    parser = argparse.ArgumentParser(description='Unpack CSK SLC data and store metadata in pickle file.')
    parser.add_argument('-i','--input', dest='h5dir', type=str,
            required=True, help='Input CSK directory')
    parser.add_argument('-o', '--output', dest='slcdir', type=str,
            required=True, help='Output SLC directory')
    parser.add_argument('-f', '--resample_flag', dest='resampFlag', type=str,
            default=None, help='fbd2fbs or fbs2fbd resampling flag')
    parser.add_argument('-p', '--polarization', dest='polarization', type=str,
            default='HH', help='polarization in case if quad or full pol data exists. Deafult: HH')
    parser.add_argument('-m', '--mult', dest='multiple',
            action='store_true', default=False,
            help='Use multiple frames')

    return parser.parse_args()

#@use_api
def unpack(hdf5, slcname, multiple=False):
    '''
    Unpack HDF5 to binary SLC file.
    '''
    os.makedirs(slcname, exist_ok=True)

    date = os.path.basename(slcname)
    obj = createSensor('ALOS')
    obj.configure()

    if multiple:
        print('Trying multiple subdirs...')
        obj._imageFileList = glob.glob(os.path.join(hdf5, '*', 'IMG-' + inps.polarization + '*'))
        obj._leaderFileList = glob.glob(os.path.join(hdf5, '*', 'LED*'))

        if (len(obj._imageFileList) == 0) or (len(obj._leaderFileList) == 0):
            print('No imagefiles / leaderfiles found in sub-dirs. Trying same directory ...')
            obj._imageFileList = glob.glob(os.path.join(hdf5, 'IMG-' + inps.polarization + '*'))
            obj._leaderFileList = glob.glob(os.path.join(hdf5, 'LED*'))

    else:
        imgname = glob.glob(os.path.join(hdf5, '*' , 'IMG-' + inps.polarization + '*'))[0]
        ldrname = glob.glob(os.path.join(hdf5, '*' , 'LED*'))[0]



        obj._leaderFileList = [ldrname]
        obj._imageFileList = [imgname]

    obj.output = os.path.join(slcname, date+'.raw')

    print(obj._leaderFileList)
    print(obj._imageFileList)
    print(obj.output)

    #if inps.fbd2fbs:
   #    print('fbd2fbs flag activated')
   #    obj._resampleFlag = 'dual2single'
    
    if inps.resampFlag == 'fbd2fbs':
       print('fbd2fbs flag activated')
       obj._resampleFlag = 'dual2single'
    elif inps.resampFlag == 'fbs2fbd':
       print('fbs2fbd flag activated')
       obj._resampleFlag = 'single2dual'


    
    obj.extractImage()
    obj.frame.getImage().renderHdr()


        #####Estimate doppler
    dop = DopIQ()
    dop.configure()

    img = copy.deepcopy(obj.frame.getImage())
    img.setAccessMode('READ')

    dop.wireInputPort('frame', object=obj.frame)
    dop.wireInputPort('instrument', object=obj.frame.instrument)
    dop.wireInputPort('image', object=img)

    dop.calculateDoppler()
    dop.fitDoppler()
    fit = dop.quadratic
    coef = [fit['a'], fit['b'], fit['c']]

    obj.frame._dopplerVsPixel = [x*obj.frame.PRF for x in coef]

    pickName = os.path.join(slcname, 'raw')
    with shelve.open(pickName) as db:
        db['frame'] = obj.frame

if __name__ == '__main__':
    '''
    Main driver.
    '''

    inps = cmdLineParse()
    if inps.slcdir.endswith('/'):
        inps.slcdir = inps.slcdir[:-1]

    if inps.h5dir.endswith('/'):
        inps.h5dir = inps.h5dir[:-1]

    unpack(inps.h5dir, inps.slcdir, 
            multiple=inps.multiple)