コード例 #1
0
ファイル: Cake.py プロジェクト: hwkobe/AreaDiffractionMachine
    def getImage(self, lowerBound, upperBound, logScale, colorMaps,
                 colorMapName, invert, doScaleFactor, scaleFactor, setMinMax,
                 minIntensity, maxIntensity):
        mode = "RGB"

        if self.maskedPixelInfo.doLessThanMask:
            (lessThanMaskColorR,lessThanMaskColorG, \
                    lessThanMaskColorB) = \
                    self.maskedPixelInfo.getLessThanMaskColorRGB()
        else:
            # We can just send the function a bunch
            # of junk since it won't be used
            (lessThanMaskColorR,lessThanMaskColorG, \
                    lessThanMaskColorB) = (0,0,0)

        if self.maskedPixelInfo.doGreaterThanMask:
            (greaterThanMaskColorR,greaterThanMaskColorG, \
                    greaterThanMaskColorB) = \
                    self.maskedPixelInfo.getGreaterThanMaskColorRGB()
        else:
            (greaterThanMaskColorR,greaterThanMaskColorG, \
                    greaterThanMaskColorB) = (0,0,0)

        if self.maskedPixelInfo.doPolygonMask:
            (polygonMaskColorR,polygonMaskColorG, \
                    polygonMaskColorB) = \
                    self.maskedPixelInfo.getPolygonMaskColorRGB()
        else:
            (polygonMaskColorR,polygonMaskColorG, \
                    polygonMaskColorB) = (0,0,0)


        palette = colorMaps.getPalette(colorMapName, \
                invert=invert)

        string = DrawWrap.getCakeImageString(
            self.cakeData, self.diffractionData, lowerBound, upperBound,
            logScale, palette, self.maskedPixelInfo.doLessThanMask,
            self.maskedPixelInfo.lessThanMask, lessThanMaskColorR,
            lessThanMaskColorG, lessThanMaskColorB,
            self.maskedPixelInfo.doGreaterThanMask,
            self.maskedPixelInfo.greaterThanMask, greaterThanMaskColorR,
            greaterThanMaskColorG, greaterThanMaskColorB,
            self.maskedPixelInfo.doPolygonMask, polygonMaskColorR,
            polygonMaskColorG, polygonMaskColorB, doScaleFactor, scaleFactor,
            setMinMax, minIntensity, maxIntensity)

        img = Image.fromstring(
            mode, (self.cakeData.shape[1], self.cakeData.shape[0]), string)

        return img
コード例 #2
0
    def getImage(self,lowerBound,upperBound,logScale,colorMaps,colorMapName,invert):
        mode = "RGB"

        if self.maskedPixelInfo.doLessThanMask:
            (lessThanMaskColorR,lessThanMaskColorG,lessThanMaskColorB) = \
                    self.maskedPixelInfo.getLessThanMaskColorRGB()
        else:
            # We can just send the function a bunch of junk since it won't be used
            (lessThanMaskColorR,lessThanMaskColorG,lessThanMaskColorB) = (0,0,0) 

        if self.maskedPixelInfo.doGreaterThanMask:
            (greaterThanMaskColorR,greaterThanMaskColorG,greaterThanMaskColorB) = \
                    self.maskedPixelInfo.getGreaterThanMaskColorRGB()
        else:
            (greaterThanMaskColorR,greaterThanMaskColorG,greaterThanMaskColorB) = (0,0,0)

        if self.maskedPixelInfo.doPolygonMask:
            (polygonMaskColorR,polygonMaskColorG,polygonMaskColorB) = \
                    self.maskedPixelInfo.getPolygonMaskColorRGB()
        else:
            (polygonMaskColorR,polygonMaskColorG,polygonMaskColorB) = (0,0,0) 


        palette = colorMaps.getPalette(colorMapName,invert=invert)

        string = DrawWrap.getCakeImageString(self.cakeData,
                self.diffractionData,
                lowerBound,
                upperBound,
                logScale,
                palette,
                self.maskedPixelInfo.doLessThanMask,
                self.maskedPixelInfo.lessThanMask, 
                lessThanMaskColorR, 
                lessThanMaskColorG,lessThanMaskColorB, 
                self.maskedPixelInfo.doGreaterThanMask,
                self.maskedPixelInfo.greaterThanMask, 
                greaterThanMaskColorR,
                greaterThanMaskColorG, 
                greaterThanMaskColorB,
                self.maskedPixelInfo.doPolygonMask,
                polygonMaskColorR,
                polygonMaskColorG,
                polygonMaskColorB)

        img = Image.fromstring(mode,(self.cakeData.shape[1],self.cakeData.shape[0]),string)

        return img