Most possible combinations of disparity (Updated)

import cv2
import numpy as np
import re
import math

testMode = False
file1 = open("log.txt", "w")
# Using http://vision.middlebury.edu/stereo/data/scenes2006/FullSize/zip-2views/Aloe-2views.zip
# Using http://vision.middlebury.edu/stereo/data/scenes2006/FullSize/zip-2views/Bowling1-2views.zip

def parseInputToUnderstandShiftingLogic(shiftingDescription):
    # example input:
    #   "shift leftStereoImg to the right using leftDisparityMap to get rightStereoImg"
    regexExpressions = {
        "imageToShift": r"(?![(shift)\s])\w+(?=( to the))",
        "action": r"(?![(to the)\s])\w+(?=( using))",
        "disparityMapToUse": r"(?![(using)\s])\w+(?=( to get))",
        "expectedResultingImage": r"(?![(to get)\s])\w+$"
    }
    imageToShift = re.search(regexExpressions["imageToShift"], shiftingDescription).group()
    action = re.search(regexExpressions["action"], shiftingDescription).group()
    expectedResultingImage = re.search(regexExpressions["expectedResultingImage"], shiftingDescription).group()
    disparityMapToUse = re.search(regexExpressions["disparityMapToUse"], shiftingDescription).group()
    return [imageToShift, action, expectedResultingImage, disparityMapToUse]

if testMode :
    # test1
    shiftingDescription = "shift leftStereoImg to the right using leftDisparityMap to get rightStereoImg"
    [imageToShift, action, expectedResultingImage, disparityMapToUse] = parseInputToUnderstandShiftingLogic(shiftingDescription)
    description = "Using " + disparityMapToUse + " to shift " + imageToShift + " to the " + action + " to get " + expectedResultingImage
    assert description == "Using leftDisparityMap to shift leftStereoImg to the right to get rightStereoImg"
    shiftingDescription = "shift leftStereoImg to the right using leftDisparityMap to get rightStereoImg"
    # test2
    shiftingDescription = "shift rightStereoImg to the left using leftDisparityMap to get leftStereoImg"
    [imageToShift, action, expectedResultingImage, disparityMapToUse] = parseInputToUnderstandShiftingLogic(shiftingDescription)
    description = "Using " + disparityMapToUse + " to shift " + imageToShift + " to the " + action + " to get " + expectedResultingImage
    assert description == "Using leftDisparityMap to shift rightStereoImg to the left to get leftStereoImg"

def generateShiftedImage(shiftingDescription, dataset, folderName):
    leftStereoImg = cv2.imread(dataset["leftStereoImg"])
    rightStereoImg = cv2.imread(dataset["rightStereoImg"])

    [rows, columns, channels] = leftStereoImg.shape # TODO: better image shape comparison logic
    [imageToShift, action, expectedResultingImage, disparityMapToUse] = parseInputToUnderstandShiftingLogic(shiftingDescription)
    description = "Using " + disparityMapToUse + " to shift " + imageToShift + " to the " + action + " to get " + expectedResultingImage
    print(description)

    disparityMap = cv2.imread(dataset[disparityMapToUse], cv2.IMREAD_GRAYSCALE)

    shiftedImage = np.zeros_like(leftStereoImg)

    imageToShiftSource = leftStereoImg if imageToShift == "leftStereoImg" else rightStereoImg

    for row in range(rows):
        for column in range(columns):
            disparityMap_point = disparityMap[row, column]
            if disparityMap_point != 0 :
                # Since we want to shift to the right, we need to move the pixels in the negative axis
                disparityValue = math.floor(disparityMap[row,column]/1)
                t = column - disparityValue if action == "right" else column + disparityValue
                if(t>=0 and t<columns):
                    for ch in range(channels):
                        shiftedImage[row,column,ch] = imageToShiftSource[row,t,ch]


    cv2.imwrite("./" + folderName + "/" + shiftingDescription + ".png", shiftedImage)

    scoreImg1 = cv2.cvtColor(cv2.imread(dataset[expectedResultingImage]), cv2.COLOR_BGR2GRAY).reshape((-1, 1))
    scoreImg2 = cv2.cvtColor(shiftedImage, cv2.COLOR_BGR2GRAY).reshape((-1, 1))
    score = np.mean(scoreImg1 - scoreImg2)
    print("Mean pixel difference score is " + str(score) + " for " + shiftingDescription)
    file1.write("Mean pixel difference score is " + str(score) + " for " + shiftingDescription + "\n\n")

def getDataset(folderName):
    return {
            "leftStereoImg": "./"+folderName+"/view1.png",
            "rightStereoImg": "./"+folderName+"/view5.png",
            "leftDisparityMap": "./"+folderName+"/disp1.png",
            "rightDisparityMap": "./"+folderName+"/disp5.png"
    }

def main():
    datasetFolders = ["Aloe", "Bowling1", "Cloth1", "Cloth3", "Flowerpots", "Wood1"]
    for folderName in datasetFolders:
        print("")
        print("")
        print("Starting folder ", folderName)
        file1.write("\n\nStarting folder " + folderName)
        dataset = getDataset(folderName)
        shiftingDescriptions = [
            "shift leftStereoImg to the left using leftDisparityMap to get rightStereoImg",
            "shift leftStereoImg to the left using rightDisparityMap to get rightStereoImg",
            "shift rightStereoImg to the right using leftDisparityMap to get leftStereoImg",
            "shift rightStereoImg to the right using rightDisparityMap to get leftStereoImg"

            # "shift leftStereoImg to the right using leftDisparityMap to get rightStereoImg",
            # "shift leftStereoImg to the right using rightDisparityMap to get rightStereoImg",
            # "shift rightStereoImg to the left using leftDisparityMap to get leftStereoImg",
            # "shift rightStereoImg to the left using rightDisparityMap to get leftStereoImg"
        ]
        for shiftingDescription in shiftingDescriptions:
            generateShiftedImage(shiftingDescription, dataset, folderName)
            print("done with " + shiftingDescription)
            print("")

    file1.close()

if __name__ == "__main__":
    main()