#!/usr/bin/env python3
############################################################
# Program is part of PySAR                                 #
# Copyright(c) 2013-2018, Zhang Yunjun, Heresh Fattahi     #
# Author:  Zhang Yunjun, Heresh Fattahi                    #
############################################################


import os
import argparse
import time
import h5py
from numpy import std
from pysar.utils import readfile, ptime
from pysar.objects import (geometry, 
                           giantIfgramStack, 
                           giantTimeseries, 
                           ifgramStack, 
                           timeseries, 
                           HDFEOS)

output = ""

############################################################
EXAMPLE = """example:
  info.py timeseries.h5
  info.py velocity.h5
  info.py ifgramStack.h5

  # Time / Date Info
  info.py ifgramStack.h5 --date                             # print master/slave date pairs info of interferograms.
  info.py ifgramStack.h5 --date --nodrop > date12_list.txt  # save master/slave date pairs info of interferograms.
  info.py timeseries.h5  --date --num                       # print date list of timeseries with its number
  info.py LS-PARAMS.h5   --date > date_list.txt             # print date list of timeseries and save it to txt file.
  info.py S1_IW12_128_0593_0597_20141213_20180619.h5 --date

  # Slice / Dataset Info
  info.py timeseries.h5                              --slice
  info.py timeseries.h5                              --slice  --num
  info.py INPUTS/ifgramStack.h5                      --slice
  info.py S1_IW12_128_0593_0597_20141213_20180619.h5 --slice
  info.py LS-PARAMS.h5                               --slice
"""


def create_parser():
    """Create command line parser."""
    parser = argparse.ArgumentParser(description='Display Metadata / Structure information of ANY File',
                                     formatter_class=argparse.RawTextHelpFormatter,
                                     epilog=EXAMPLE)
    parser.add_argument('file', type=str, help='File to check')
    parser.add_argument('--date', dest='disp_date', action='store_true',
                        help='Show date/date12 info of input file')
    parser.add_argument('--num', dest='disp_num', action='store_true',
                        help='Show date/date12 number')
    parser.add_argument('--nodrop', dest='drop_ifgram', action='store_true',
                        help='Do not display dropped interferograms info.')
    parser.add_argument('--slice', dest='disp_slice', action='store_true',
                        help='Print slice list of the file')
    return parser


def cmd_line_parse(iargs=None):
    """Command line parser."""
    parser = create_parser()
    inps = parser.parse_args(args=iargs)
    return inps


############################################################
def attributes_string(atr, string=str(), sorting=True):
    ## Print Dictionary of Attributes
    digits = max([len(key) for key in list(atr.keys())] + [0])
    for key, value in sorted(atr.items(), key=lambda x: x[0]):
        try:
            value = value.decode('utf8')
        except:
            pass
        string += '  {k:<{d}}    {v}\n'.format(k=key,
                                               d=digits,
                                               v=value)
    return string


def print_attributes(atr, string=str(), sorting=True):
    print((attributes_string(atr, string, sorting)))


############################################################
def hdf5_structure_string(file):
    global output, maxDigit

    def print_hdf5_structure_obj(name, obj):
        global output, maxDigit
        if isinstance(obj, h5py.Group):
            output += 'HDF5 group   "/{n}"\n'.format(n=name)
        elif isinstance(obj, h5py.Dataset):
            output += ('HDF5 dataset "/{n:<{w}}": shape {s:<20}, '
                       'dtype <{t}>\n').format(n=name,
                                               w=maxDigit,
                                               s=str(obj.shape),
                                               t=obj.dtype)
        atr = dict(obj.attrs)
        if len(atr) > 0:
            output = attributes_string(atr, output)+"\n"

    f = h5py.File(file, 'r')
    # metadata in root level
    atr = dict(f.attrs)
    if len(atr) > 0:
        output += 'Attributes in / level:\n'
        output = attributes_string(atr, output)+"\n"

    # max length of dataset name
    maxDigit = max([len(i) for i in f.keys()])
    maxDigit = max(20, maxDigit+1)
    if atr.get('FILE_TYPE', 'timeseries') == 'HDFEOS':
        maxDigit += 35

    f.visititems(print_hdf5_structure_obj)
    f.close()

    local_output = output
    output = ""
    return local_output


## By andrewcollette at https://github.com/h5py/h5py/issues/406
def print_hdf5_structure(file):
    string = hdf5_structure_string(file)
    print(string)


def print_timseries_date_stat(dateList):
    datevector = ptime.date_list2vector(dateList)[1]
    print('Start Date: '+dateList[0])
    print('End   Date: '+dateList[-1])
    print('Number of acquisitions    : %d' % len(dateList))
    print('Std. of acquisition times : %.2f yeras' % std(datevector))
    print('----------------------')
    print('List of dates:')
    print(dateList)
    print('----------------------')
    print('List of dates in years')
    print(datevector)
    return


def print_date_list(fname, disp_num=False, drop_ifgram=False, print_msg=False):
    """Print time/date info of file"""
    atr = readfile.read_attribute(fname)
    k = atr['FILE_TYPE']
    dateList = None
    if k in ['timeseries']:
        dateList = timeseries(fname).get_date_list()
    elif k == 'HDFEOS':
        obj = HDFEOS(fname)
        obj.open(print_msg=False)
        dateList = obj.dateList
    elif k == 'giantTimeseries':
        obj = giantTimeseries(fname)
        obj.open(print_msg=False)
        dateList = obj.dateList
    elif k in ['ifgramStack']:
        dateList = ifgramStack(fname).get_date12_list(dropIfgram=drop_ifgram)
    elif k in ['giantIfgramStack']:
        obj = giantIfgramStack(fname)
        obj.open(print_msg=False)
        dateList = obj.date12List
    else:
        print('--date option can not be applied to {} file, ignore it.'.format(k))

    if print_msg and dateList is not None:
        for i in range(len(dateList)):
            if disp_num:
                print('{}\t{}'.format(dateList[i], i))
            else:
                print(dateList[i])
    return dateList


def print_slice_list(fname, disp_num=False, print_msg=False):
    """Print slice info of file"""
    slice_list = readfile.get_slice_list(fname)
    if print_msg:
        for i in range(len(slice_list)):
            if disp_num:
                print('{}\t{}'.format(slice_list[i], i))
            else:
                print(slice_list[i])
    return slice_list


def print_pysar_info(fname):
    try:
        atr = readfile.read_attribute(fname)
        k = atr['FILE_TYPE']
        print('{} {:*<40}'.format('*'*20, 'Basic File Info '))
        print('file name: '+atr['FILE_PATH'])
        print('file type: '+atr['FILE_TYPE'])
        if 'Y_FIRST' in atr.keys():
            print('coordinates : GEO')
        else:
            print('coordinates : RADAR')
        if k in ['timeseries']:
            dateList = print_date_list(fname)
            print('\n{} {:*<40}'.format('*'*20, 'Date Stat Info '))
            print_timseries_date_stat(dateList)
    except:
        pass
    return


############################################################
def main(iargs=None):
    inps = cmd_line_parse(iargs)
    if not os.path.isfile(inps.file):
        print('ERROR: input file does not exists: {}'.format(inps.file))
        return
    ext = os.path.splitext(inps.file)[1].lower()

    # --date option
    if inps.disp_date:
        print_date_list(inps.file, disp_num=inps.disp_num, drop_ifgram=inps.drop_ifgram, print_msg=True)
        return

    # --slice option
    if inps.disp_slice:
        print_slice_list(inps.file, disp_num=inps.disp_num, print_msg=True)
        return

    # Basic info from PySAR reader
    print_pysar_info(inps.file)

    # Generic Attribute/Structure of all files
    if ext in ['.h5', '.he5']:
        print('\n{} {:*<40}'.format('*'*20, 'HDF5 File Structure '))
        print_hdf5_structure(inps.file)
    else:
        print('\n{} {:*<40}'.format('*'*20, 'Binary File Attributes '))
        atr = readfile.read_attribute(inps.file)
        print_attributes(atr)

    return


############################################################
if __name__ == '__main__':
    main()