使用PyQt从QGIS插件拍摄照片

2024-04-24 18:44:18 发布

您现在位置:Python中文网/ 问答频道 /正文

我在插件中有一个按钮,可以拍摄照片,并将其与QGIS中的一个图层功能相关联

我的目标是选择一个功能,打开相机视图(取景器),拍摄一张照片,并将其以功能id和日期时间作为文件名保存(例如20210412-112351_223.jpg

为此,我使用下面的代码based on this post

问题是,当我在笔记本电脑中使用这个插件时,拍摄的照片是1280x720,但对于Surface Book 3,结果照片是640x320

可以在代码中设置图片的大小吗

from PyQt5.QtWidgets import *
from PyQt5.QtMultimedia import *
from PyQt5.QtMultimediaWidgets import *
from qgis.core import (QgsProject,
                       Qgis,
                       QgsExpressionContextUtils)
from qgis.utils import iface

import os
import time

class creaventana(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setGeometry(0, 0, 1280, 720)
        self.setStyleSheet("background : lightgrey;")
        self.available_cameras = QCameraInfo.availableCameras()
        
        if not self.available_cameras:
            iface.messageBar().pushMessage('No hay cámaras', 'El dispositivo no tiene ninguna cámara',
                                           level=Qgis.Warning,
                                           duration=5)

        self.status = QStatusBar()
        self.status.setStyleSheet("background : white;")
        self.setStatusBar(self.status)
        self.viewfinder = QCameraViewfinder()
        self.viewfinder.show()
        self.setCentralWidget(self.viewfinder)
        self.select_camera(0)
        toolbar = QToolBar("Cámara")
        self.addToolBar(toolbar)
        click_action = QAction("Hacer foto", self)
        click_action.setStatusTip("Esto hace la foto")
        click_action.setToolTip("Hacer foto")
        click_action.triggered.connect(self.click_photo)
        toolbar.addAction(click_action)
        camera_selector = QComboBox()
        camera_selector.setStatusTip("Cámara frontal / trasera")
        camera_selector.setToolTip("Elegir cámara")
        camera_selector.setToolTipDuration(2500)
        camera_selector.addItems([camera.description()
                                  for camera in self.available_cameras])
        camera_selector.currentIndexChanged.connect(self.select_camera)
        toolbar.addWidget(camera_selector)
        toolbar.setStyleSheet("background : white;")
        self.setWindowTitle("PyQt5 Cam")

    def select_camera(self, i):
        self.camera = QCamera(self.available_cameras[i])
        self.camera.setViewfinder(self.viewfinder)
        self.camera.setCaptureMode(QCamera.CaptureStillImage)
        self.camera.error.connect(lambda: self.alert(self.camera.errorString()))
        self.camera.start()
        self.capture = QCameraImageCapture(self.camera)
        self.capture.error.connect(lambda error_msg, error,
                                          msg: self.alert(msg))
        self.capture.imageCaptured.connect(lambda d, i: self.status.showMessage("Imagen tomada: " + str(self.save_seq)))
        self.current_camera_name = self.available_cameras[i].description()

        self.save_seq = 0

    def click_photo(self):
        individuos = QgsProject.instance().mapLayersByName('Individuos')[0]
        individuoSeleccionado = individuos.selectedFeatures()

        for individuo in individuoSeleccionado:
            idind = individuo['id_ind']

        timestamp = time.strftime('%Y%m%d-%H%M%S')
        ruta = 'D:/fotos/'

        self.capture.capture(os.path.join(ruta, "%s_%s.jpg" % (idind, timestamp)))
        self.save_seq += 1

    def alert(self, msg):
        error = QErrorMessage(self)
        error.showMessage(msg)

def abrecamara(self):
    individuos = QgsProject.instance().mapLayersByName('Individuos')[0]
    individuoSeleccionado = individuos.selectedFeatures()
    cuentaIndividuoSeleccionado = len(list(individuoSeleccionado))

    if cuentaIndividuoSeleccionado == 1:
        ventana = creaventana()
        ventana.show()
    elif cuentaIndividuoSeleccionado > 1:
        iface.messageBar().pushMessage('Selección', 'Solo puede haber un individuo seleccionado',
                                       level=Qgis.Warning,
                                       duration=5)
    elif cuentaIndividuoSeleccionado == 0:
        iface.messageBar().pushMessage('Selección', 'Tiene que haber al menos un individuo seleccionado',
                                       level=Qgis.Warning,
                                       duration=5)

Tags: fromimportselfdefconnectactionmsgerror