Beispiel #1
0
class LaueImage(QtGui.QWidget):
    def __init__(self,  parent,  _name):
        QtGui.QWidget.__init__(self,  parent)
        self.name=os.path.split(_name)[1]
        self.fullImg=Image.open(_name)
        self.VRGB_BezierParams=[[], [], [], []]
        initParams=[[(0, 0),  (1, 1)], 
                 [(0, 0),  (1, 1)], 
                 [(0, 0),  (1, 1)], 
                 [(0, 0),  (1, 1)]]
        
        
        self.transfer=ImageTransfer()
        self.resizeImg(self.size())
        self.setTransferCurves(initParams)
        sp=self.sizePolicy()
        #sp.setHeightForWidth(True)
        sp.setHorizontalPolicy(QtGui.QSizePolicy.Expanding)
        sp.setVerticalPolicy(QtGui.QSizePolicy.Expanding)
        self.setSizePolicy(sp)
        
    def resizeEvent(self,  e):
        #QtGui.QWidget.resizeEvent(self,  e)
        self.resizeImg(e.size())
            
    def resizeImg(self,  size):        
        s=QtCore.QSize(self.fullImg.width,  self.fullImg.height)
        s.scale(size,  QtCore.Qt.KeepAspectRatio)
        print 'Size',self.width(),  self.height() ,  s.width(),  s.height() 

        self.scaledImg=self.fullImg.resize((s.width(),  s.height()))
        mode=None
        if self.scaledImg.mode=='RGB':
            mode=1
        elif self.scaledImg.mode=='F':
            mode=0
        if mode!=None:
            self.transfer.setData(s.width(), s.height(), mode, self.scaledImg.tostring())
    
    
    def paintEvent(self, e):
        p=QtGui.QPainter(self)
        
        s=self.transfer.qImg().size()
        x=(self.width()-s.width())/2
        y=(self.height()-s.height())/2
        p.drawImage(x, y, self.transfer.qImg())
        
    def setTransferCurves(self, newParams):
        curvesChanged=False
        for i, pa in enumerate(zip(newParams,  self.VRGB_BezierParams)):
            pn,po=pa
            if pn!=po:
                self.VRGB_BezierParams[i]=pn
                curvesChanged=True

        if curvesChanged:
            for i in range(4):
                c=BezierCurve(*zip(*self.VRGB_BezierParams[i]))
                p1=c.x[1:]
                p2=reduce(lambda x, y:x+y, c.D)
                self.transfer.setTransferCurve(i, p1,  p2)
            self.update()