Benchmark

# [DocSampleImports]
import os
import sys, getopt
import time
import PyIPSDK
import PyIPSDK.IPSDKIPLStats as stats
import PyIPSDK.IPSDKIPLUtility as util
# [DocSampleImports]

def readCmdArguments(argv):
    
    # Default parameters values
    imagesSamplePath = PyIPSDK.getIPSDKDirectory(PyIPSDK.eInternalDirectory.eID_Images)
    inputImgPath = os.path.join(imagesSamplePath,  "Lena_510x509_UInt8.tif")
    tmpPath = PyIPSDK.getIPSDKDefaultDirectory(PyIPSDK.eDefaultExternalDirectory.eDED_Tmp)
    outputImgPath = os.path.join(tmpPath, "benchmark.tif")
    inHalfKnlSizeX = 3
    inHalfKnlSizeY = 3
    inOptNbClasses = 16
    
    try:
        # Read the command line, the second argument is a string with all the possible options (short version)
        # with a ':' if the option requires an argument and the third argument is the set of trings corresponding
        # to the long versions of the program options
        opts, args = getopt.getopt(argv,"hi:o:x:y:n:",["inputImgFilePath=","outputImgFilePath=","inHalfKnlSizeX=","inHalfKnlSizeY=", "inOptNbClasses="])
    except getopt.GetoptError:
        print('<application_script_filename> [--inputImgFilePath <input_image_file_path>] [--outputImgFilePath <output_image_file_path>] [--inHalfKnlSizeX <input_half_kernel_x_size_value>] [--inHalfKnlSizeY <input_half_kernel_y_size_value>] [--inOptNbClasses <input_number_of_classes>]')
        sys.exit(2)
    
    # Parse the program options
    for opt, arg in opts:
        if opt == '-h':
            print('<application_script_filename> [--inputImgFilePath <input_image_file_path>] [--outputImgFilePath <output_image_file_path>] [--inHalfKnlSizeX <input_half_kernel_x_size_value>] [--inHalfKnlSizeY <input_half_kernel_y_size_value>] [--inOptNbClasses <input_number_of_classes>]')
            sys.exit(0)
        elif opt in ("-i", "--inputImgFilePath"):
            inputImgPath = arg
        elif opt in ("-o", "--outputImgFilePath"):
            outputImgPath = arg
        elif opt in ("-x", "--inHalfKnlSizeX"):
            inHalfKnlSizeX = int(arg)
        elif opt in ("-y", "--inHalfKnlSizeY"):
            inHalfKnlSizeY = int(arg)
        elif opt in ("-n", "--inOptNbClasses"):
            inOptNbClasses = int(arg)
    
    return inputImgPath, outputImgPath, inHalfKnlSizeX, inHalfKnlSizeY, inOptNbClasses

def main(argv):
# [DocSampleDefineArguments]
    # retrieve program parameters
    inputImgPath, outputImgPath, inHalfKnlSizeX, inHalfKnlSizeY, inOptNbClasses = readCmdArguments(argv)
# [DocSampleDefineArguments]

# [DocSampleLoadImage]
    # opening of input image
    inImg = PyIPSDK.loadTiffImageFile(inputImgPath)
# [DocSampleLoadImage]

# [DocSampleInitOutput]
    # initialization of output image
    # This step allows to avoid to mix processing time associated to
    # output image allocation with real image processing duration
    outImg = util.convertImg(inImg, PyIPSDK.eImageBufferType.eIBT_Real32)
# [DocSampleInitOutput]

# [DocSampleApplyOp]
    # boundary 2d image computation
    tStart = time.perf_counter()
    outImg = stats.localEntropy2dImg(inImg, inHalfKnlSizeX, inHalfKnlSizeY, inOptNbClasses)
    tEnd = time.perf_counter()
    print("The application needed " + str(tEnd-tStart) + " seconds.")
# [DocSampleApplyOp]

# [DocSampleSaveImage]
    # save generated image
    print("Writing result in file : " + outputImgPath)
    PyIPSDK.saveTiffImageFile(outputImgPath, outImg)
# [DocSampleSaveImage]
    
if __name__ == "__main__":
   main(sys.argv[1:])