def getDiffractionImage(data, lowerBound, upperBound, logScale, colorMaps, colorMapName, maskedPixelInfo, invert=None): mode = "RGB" if maskedPixelInfo.doLessThanMask: (lessThanMaskColorR,lessThanMaskColorG,lessThanMaskColorB) = \ maskedPixelInfo.getLessThanMaskColorRGB() lessThanMask = maskedPixelInfo.lessThanMask else: # We can just send the function a bunch of junk since it won't be used (lessThanMaskColorR, lessThanMaskColorG, lessThanMaskColorB) = (0, 0, 0) lessThanMask = -1 if maskedPixelInfo.doGreaterThanMask: (greaterThanMaskColorR,greaterThanMaskColorG,greaterThanMaskColorB) = \ maskedPixelInfo.getGreaterThanMaskColorRGB() greaterThanMask = maskedPixelInfo.greaterThanMask else: (greaterThanMaskColorR, greaterThanMaskColorG, greaterThanMaskColorB) = (0, 0, 0) greaterThanMask = -1 if maskedPixelInfo.doPolygonMask: (polygonMaskColorR,polygonMaskColorG,polygonMaskColorB) = \ maskedPixelInfo.getPolygonMaskColorRGB() polygonsX = maskedPixelInfo.polygonsX polygonsY = maskedPixelInfo.polygonsY polygonBeginningsIndex = maskedPixelInfo.polygonBeginningsIndex polygonNumberOfItems = maskedPixelInfo.polygonNumberOfItems else: (polygonMaskColorR, polygonMaskColorG, polygonMaskColorB) = (0, 0, 0) polygonsX = Numeric.array([]) polygonsY = Numeric.array([]) polygonBeginningsIndex = Numeric.array([]) polygonNumberOfItems = Numeric.array([]) palette = colorMaps.getPalette(colorMapName, invert=invert) string = DrawWrap.getDiffractionImageString( data, lowerBound, upperBound, logScale, palette, maskedPixelInfo.doLessThanMask, lessThanMask, lessThanMaskColorR, lessThanMaskColorG, lessThanMaskColorB, maskedPixelInfo.doGreaterThanMask, greaterThanMask, greaterThanMaskColorR, greaterThanMaskColorG, greaterThanMaskColorB, maskedPixelInfo.doPolygonMask, polygonsX, polygonsY, polygonBeginningsIndex, polygonNumberOfItems, polygonMaskColorR, polygonMaskColorG, polygonMaskColorB) img = Image.fromstring(mode, (data.shape[0], data.shape[1]), string) img = img.rotate(90) img = img.transpose(Image.FLIP_TOP_BOTTOM) return img
def getDiffractionImage(data,lowerBound,upperBound,logScale, colorMaps,colorMapName,maskedPixelInfo,invert=None): mode = "RGB" if maskedPixelInfo.doLessThanMask: (lessThanMaskColorR,lessThanMaskColorG,lessThanMaskColorB) = \ maskedPixelInfo.getLessThanMaskColorRGB() lessThanMask = maskedPixelInfo.lessThanMask else: # We can just send the function a bunch of junk since it won't be used (lessThanMaskColorR,lessThanMaskColorG,lessThanMaskColorB) = (0,0,0) lessThanMask = -1 if maskedPixelInfo.doGreaterThanMask: (greaterThanMaskColorR,greaterThanMaskColorG,greaterThanMaskColorB) = \ maskedPixelInfo.getGreaterThanMaskColorRGB() greaterThanMask = maskedPixelInfo.greaterThanMask else: (greaterThanMaskColorR,greaterThanMaskColorG,greaterThanMaskColorB) = (0,0,0) greaterThanMask = -1 if maskedPixelInfo.doPolygonMask: (polygonMaskColorR,polygonMaskColorG,polygonMaskColorB) = \ maskedPixelInfo.getPolygonMaskColorRGB() polygonsX = maskedPixelInfo.polygonsX polygonsY = maskedPixelInfo.polygonsY polygonBeginningsIndex = maskedPixelInfo.polygonBeginningsIndex polygonNumberOfItems = maskedPixelInfo.polygonNumberOfItems else: (polygonMaskColorR,polygonMaskColorG,polygonMaskColorB) = (0,0,0) polygonsX = Numeric.array([]) polygonsY = Numeric.array([]) polygonBeginningsIndex = Numeric.array([]) polygonNumberOfItems = Numeric.array([]) palette = colorMaps.getPalette(colorMapName,invert=invert) string = DrawWrap.getDiffractionImageString(data, lowerBound, upperBound, logScale, palette, maskedPixelInfo.doLessThanMask, lessThanMask, lessThanMaskColorR, lessThanMaskColorG, lessThanMaskColorB, maskedPixelInfo.doGreaterThanMask, greaterThanMask, greaterThanMaskColorR, greaterThanMaskColorG, greaterThanMaskColorB, maskedPixelInfo.doPolygonMask, polygonsX, polygonsY, polygonBeginningsIndex, polygonNumberOfItems, polygonMaskColorR, polygonMaskColorG, polygonMaskColorB) img = Image.fromstring(mode,(data.shape[0],data.shape[1]),string) img = img.rotate(90) img = img.transpose(Image.FLIP_TOP_BOTTOM) return img