2016-12-19 2 views
1

로드 된 이미지에서 픽셀 정보를 얻을 수 있도록 이미지 뷰어 (아래 코드 참조)를 채택했습니다. '이미지로드'버튼을 사용하여 이미지를로드 한 다음 스크롤 휠을 사용하여 확대 및 축소하고 마우스 왼쪽 버튼을 클릭하고 드래그하여 이동할 수 있습니다. '픽셀 정보 입력'버튼을 누르면 드래그가 비활성화되고 (여전히 확대/축소 할 수 있음) 이미지를 클릭하면 픽셀 좌표 (정수 픽셀 인덱스)와 픽셀의 회색 음영 값이 제공됩니다.회전 된 이미지의 pyqt4 픽셀 정보

이미지를 회전하면 '이미지 회전'버튼을 눌러 픽셀 정보 버튼을 사용하면 더 이상 올바른 픽셀 정보가 제공되지 않습니다. mapToScene 메서드는 회전 된 이미지에 사용할 올바른 것이 아니라 다른 방법으로 찾을 수 없다고 생각합니다. 회전 된 pixmap에서 toImage()를 사용하고 원래 이미지를이 이미지로 바꾸는 등 다양한 작업을 시도했지만 아무 것도 작동하지 않는 것 같습니다. 이 문제를 해결하는 가장 좋은 방법은 무엇입니까?

코드 :

from PyQt4 import QtCore, QtGui 

class PhotoViewer(QtGui.QGraphicsView): 
    photoClicked = QtCore.pyqtSignal(QtCore.QPoint) 

    def __init__(self, parent): 
     super(PhotoViewer, self).__init__(parent) 
     self._zoom = 0 
     self._empty = True 
     self._scene = QtGui.QGraphicsScene(self) 
     self._photo = QtGui.QGraphicsPixmapItem() 
     self._scene.addItem(self._photo) 
     self.setScene(self._scene) 
     self.setTransformationAnchor(QtGui.QGraphicsView.AnchorUnderMouse) 
     self.setResizeAnchor(QtGui.QGraphicsView.AnchorUnderMouse) 
     self.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff) 
     self.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff) 
     self.setBackgroundBrush(QtGui.QBrush(QtGui.QColor(30, 30, 30))) 
     self.setFrameShape(QtGui.QFrame.NoFrame) 

    def fitInView(self): 
     rect = QtCore.QRectF(self._photo.pixmap().rect()) 
     if not rect.isNull(): 
      unity = self.transform().mapRect(QtCore.QRectF(0, 0, 1, 1)) 
      self.scale(1/unity.width(), 1/unity.height()) 
      viewrect = self.viewport().rect() 
      scenerect = self.transform().mapRect(rect) 
      factor = min(viewrect.width()/scenerect.width(), 
         viewrect.height()/scenerect.height()) 
      self.scale(factor, factor) 
      self.centerOn(rect.center()) 
      self._zoom = 0 

    def hasPhoto(self): 
     return not self._empty 

    def toggleDragMode(self): 
     if self.dragMode() == QtGui.QGraphicsView.ScrollHandDrag: 
      self.setDragMode(QtGui.QGraphicsView.NoDrag) 
     elif self.hasPhoto(): 
      self.setDragMode(QtGui.QGraphicsView.ScrollHandDrag) 

    def setPhoto(self, pixmap=None): 
     self._zoom = 0 
     if pixmap and not pixmap.isNull(): 
      self._empty = False 
      self.setDragMode(QtGui.QGraphicsView.ScrollHandDrag) 
      self._photo.setPixmap(pixmap) 
      self.fitInView() 
     else: 
      self._empty = True 
      self.setDragMode(QtGui.QGraphicsView.NoDrag) 
      self._photo.setPixmap(QtGui.QPixmap()) 

    def wheelEvent(self, event): 
     if not self._photo.pixmap().isNull(): 
      if event.delta() > 0: 
       factor = 1.25 
       self._zoom += 1 
      else: 
       factor = 0.8 
       self._zoom -= 1 
      if self._zoom > 0: 
       self.scale(factor, factor) 
      elif self._zoom == 0: 
       self.fitInView() 
      else: 
       self._zoom = 0 

    def mousePressEvent(self, event): 
     if (self.hasPhoto() and 
      self.dragMode() == QtGui.QGraphicsView.NoDrag and 
      self._photo.isUnderMouse()): 
      self.photoClicked.emit(QtCore.QPoint(event.pos())) 
     super(PhotoViewer, self).mousePressEvent(event) 

class Window(QtGui.QWidget): 
    def __init__(self): 
     super(Window, self).__init__() 
     self.viewer = PhotoViewer(self) 
     # 'Load image' button 
     self.btnLoad = QtGui.QToolButton(self) 
     self.btnLoad.setText('Load image') 
     self.btnLoad.clicked.connect(self.loadImage) 
     # Button to change from drag/pan to getting pixel info 
     self.btnPixInfo = QtGui.QToolButton(self) 
     self.btnPixInfo.setText('Enter pixel info mode') 
     self.btnPixInfo.clicked.connect(self.pixInfo)   
     self.editPixInfo = QtGui.QLineEdit(self) 
     self.editPixInfo.setReadOnly(True) 
     # Button to rotate image by 10 degrees 
     self.btnRotate = QtGui.QToolButton(self) 
     self.btnRotate.setText('Rotate image') 
     self.btnRotate.clicked.connect(self.rotateImage) 
     self.viewer.photoClicked.connect(self.photoClicked) 
     # Arrange layout 
     VBlayout = QtGui.QVBoxLayout(self) 
     VBlayout.addWidget(self.viewer) 
     HBlayout = QtGui.QHBoxLayout() 
     HBlayout.setAlignment(QtCore.Qt.AlignLeft) 
     HBlayout.addWidget(self.btnLoad) 
     HBlayout.addWidget(self.btnRotate) 
     HBlayout.addWidget(self.btnPixInfo) 
     HBlayout.addWidget(self.editPixInfo) 
     VBlayout.addLayout(HBlayout) 

    def loadImage(self): 
     self.viewer.setPhoto(QtGui.QPixmap('pic.jpg')) 

    def pixInfo(self): 
     self.viewer.toggleDragMode() 

    def rotateImage(self): 
     self.viewer._photo.setRotation(10) 

    def photoClicked(self, pos): 
     pos = self.viewer.mapToScene(pos) 
     # p.s. I realise the following lines are probably a very convoluted way of getting 
     # a grayscale value from RGB, but I couldn't make it work any other way I tried 
     rot_image = self.viewer._photo.pixmap().toImage().pixel(pos.x(), pos.y()) 
     colour = QtGui.QColor.fromRgb(rot_image) 
     gsval = QtGui.qGray(colour.red(), colour.green(), colour.blue()) 
     self.editPixInfo.setText('X:%d, Y:%d Grayscale: %d' % (pos.x(), pos.y(), gsval)) 

if __name__ == '__main__': 
    import sys 
    app = QtGui.QApplication(sys.argv) 
    window = Window() 
    window.setGeometry(500, 300, 800, 600) 
    window.show() 
    sys.exit(app.exec_()) 

답변

0

당신은 장면을 매핑 할 필요가 항목으로 좌표 좌표 :

pos = self.viewer._photo.mapFromScene(self.viewer.mapToScene(pos)) 
+0

은 이전과 (당신이 내 마지막 질문에 대답) 당신이 나를 위해 그것을 해결했습니다. 관심이 없으면 이런 종류의 것들을 배우기 위해 좋은 책이나 튜토리얼을 알고 있습니까 (특히 pyqt4 이미지 조작 및 그리기/애니메이션 작업). 내가 모르는 것들이 너무 많아서 내가 어둠 속에서 길을 잃고있는 것처럼 느껴진다. 나는 항상 예를 ​​통해 배우는 것이 더 쉽다고 생각한다. 어쨌든 고마워. –

+0

@AndrewBirch. 개인적인 추천은 없습니다. [PyQt Wiki] (http://wiki.python.org/moin/PyQt)에는 몇 가지 유용한 링크가 있지만, 일부는 꽤 오래된 링크 일 수 있다는 경고를받습니다. [PyQt4 예제] (http://github.com/Werkov/PyQt4/tree/master/examples/graphicsview)에는 Qt Docs에 기반한 그래픽 데모도 있습니다. – ekhumoro