ISCE_INSAR/contrib/stack/stripmapStack/referenceStackCopy.py

#!/usr/bin/env python3

import isce
import isceobj
import argparse
import os
import shelve
import logging
import glob
import shutil
from osgeo import gdal

def createParser():
    parser = argparse.ArgumentParser( description='Duplicating the reference SLC')

    parser.add_argument('-i', '--input_slc', dest='input_slc', type=str, required=True,
            help = 'Directory with reference acquisition for reference')
    parser.add_argument('-o', '--output_slc', dest='output_slc', type=str, required=True,
            help='Directory with secondary acquisition')

    return parser
    
def cmdLineParse(iargs = None):
    parser = createParser()
    
    inps =  parser.parse_args(args=iargs)
    
    return inps


def main(iargs=None):
    '''
    Main driver.
    '''
    inps = cmdLineParse(iargs)

    # providing absolute paths
    inps.output_slc = os.path.abspath(inps.output_slc)
    inps.input_slc = os.path.abspath(inps.input_slc)

    # making the output direcory is non-existent
    outDir = os.path.dirname(inps.output_slc)
    inDir = os.path.dirname(inps.input_slc)
    os.makedirs(outDir, exist_ok=True)

    # copying shelf files as backup
    referenceShelveDir = os.path.join(outDir, 'referenceShelve')
    secondaryShelveDir = os.path.join(outDir, 'secondaryShelve')

    os.makedirs(referenceShelveDir, exist_ok=True)
    os.makedirs(secondaryShelveDir, exist_ok=True)

    sec_files = glob.glob(os.path.join(inDir, 'data*'))
    for file in sec_files:
        shutil.copy(file, secondaryShelveDir)
    # cmd = 'cp '+ inDir + '/data* ' + secondaryShelveDir
    # os.system(cmd) 

    ref_files = glob.glob(os.path.join(inDir, 'data*'))
    for file in ref_files:
        shutil.copy(file, referenceShelveDir)
    # cmd = 'cp '+ inDir + '/data* ' + referenceShelveDir
    # os.system(cmd)

    translate_options = gdal.TranslateOptions(format='ENVI')
    gdal.Translate(inps.output_slc, inps.input_slc, options=translate_options)
    cmd = 'gdal_translate -of ENVI ' + inps.input_slc + " " + inps.output_slc
    os.system(cmd)
    translate_options = gdal.TranslateOptions(format='VRT')
    gdal.Translate(inps.output_slc + ".vrt", inps.output_slc, options=translate_options)
    cmd = 'gdal_translate -of VRT ' + inps.output_slc + " " + inps.output_slc + ".vrt"
    os.system(cmd)

if __name__ == '__main__':
    '''
    Main driver.
    '''
    main()