def getMappingDict(imgList,varList, MODEL , const): mappingDict = {} for index in range(len(imgList)): i, j, imgBrightNess, type = imgList[index] _, _, srcX, srcY = deflection.getDeflection(i+0.5, j+0.5, MODEL, const=const) mappingDict[(i,j)]= (srcX, srcY, imgBrightNess, type, varList[index][2]) #commons.plotMappingDict(mappingDict, const) return mappingDict
def testWholeGridMapping(const): imgXm, imgYm = commons.getGrid(-50, 50, 2, -50, 50,2, 0.05) deflXm = [] deflYm = [] for i in range(srcDimX): deflx, defly, _, _ = deflection.getDeflection(imgXm[i], imgYm[i],'SIE',const) deflXm.append(deflx) deflYm.append(defly) new_deflXm = np.asarray(deflXm) new_deflYm = np.asarray(deflYm) srcXm = imgXm - new_deflXm srcYm = imgYm - new_deflYm commons.plotGrid(srcXm, srcYm)