Workflow_preprocess.py

# -*- coding: utf-8 -*-

from WorkflowFunctions import *


#%% Input parameters


###########################################
###  I N P U T     P A R A M E T E R S  ###
###########################################

# NOTE: the file structure must be as follows:
# the parent_directory should contain all the individual chunk folders and no others

parser = argparse.ArgumentParser()
# required args
parser.add_argument('parent_directory', help="parent directory of the image directories")
parser.add_argument('output_directory', help="output directory")

# optional args
parser.add_argument('--bbox', default=None, nargs="+", type=float, help="bounding box used in preprocessing with coordinates given as: ULX ULY LRX LRY")
parser.add_argument('--altmin', default=100, type=int, help="Minmal altitude for image to be considered (in metadata units)")
parser.add_argument('--no_lwir', action='store_false', help="do not include long wave IR (thermal) band in output")
parser.add_argument('--no_sbw', action='store_false', help="do not sort bands by wavelength")
parser.add_argument('--vc', action='store_true', help="perform vignetting correction in preprocessing")
parser.add_argument('--spec_irr', action='store_true', help="Use spectral irradiance instead of horizontal")
# although the arguments with the 'no' prefix indicate NOT to do something,
# a True value means it WILL do it and a false value means it WILL NOT.
# the default values are True, meaning if you don't add the flag it WILL
# perform those things
# the vc/spec_irr argument is the converse (i.e. using the flag DOES perform it)
parser.set_defaults(no_lwir=True)
parser.set_defaults(no_sbw=True)
parser.set_defaults(vc=False)
parser.set_defaults(spec_irr=False)
args = parser.parse_args()

# set variables from parser
parent_dir = args.parent_directory
out_dir = args.output_directory
altmin = args.altmin  # the minimum altitude for images to be included in processing
# a bounding box with the coordinates ordered as: ULX, ULY, LRX, LRY
bbox = args.bbox
lwir = args.no_lwir  # whether or not to include thermal band in processing
sort_by_wavelength = args.no_sbw  # whether or not to include thermal band in processing
vignette_correct = args.vc  # whether to apply vignetting correction
spectral_irr = args.spec_irr

#################################################################
#################################################################



#%%


def main():
    
    csv_path = preprocess(parent_dir,
                      out_dir,
                      bbox=bbox,
                      altmin=altmin,
                      lwir=lwir,
                      sort_by_wavelength=sort_by_wavelength,
                      vignette_correct=vignette_correct,
                      spectral_irr=spectral_irr)
    
    print("The CSV path to be used as input for the 'Workflow_agisoft script is:")
    print(csv_path)

                                    
if __name__ == '__main__':
    main()