def callBackMetadata(self, currentTime, nextTime):
""" Metadata CallBack """
try:
t = callBackMetadataThread(cmds=[
'-i', self.fileName, '-ss', currentTime, '-to', nextTime,
'-map', 'data-re', '-f', 'data', '-'
])
t.start()
t.join(1)
if t.is_alive():
t.p.terminate()
t.join()
if t.stdout == b'':
return
for packet in StreamParser(t.stdout):
try:
self.addMetadata(packet.MetadataList())
UpdateLayers(packet,
parent=self,
mosaic=self.createingMosaic)
self.iface.mapCanvas().refresh()
QApplication.processEvents()
return
except Exception as e:
None
except Exception as e:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvPlayer", "Metadata Callback Failed! : "),
str(e),
level=QGis.Info)
def SetGCPsToGeoTransform(cornerPointUL, cornerPointUR, cornerPointLR,
cornerPointLL, frameCenterLon, frameCenterLat):
''' Make Geotranform from pixel to lon lat coordinates '''
gcps = []
global gcornerPointUL
gcornerPointUL = cornerPointUL
global gcornerPointUR
gcornerPointUR = cornerPointUR
global gcornerPointLR
gcornerPointLR = cornerPointLR
global gcornerPointLL
gcornerPointLL = cornerPointLL
global gframeCenterLat
gframeCenterLat = frameCenterLat
global gframeCenterLon
gframeCenterLon = frameCenterLon
global geotransform
geotransform = gdal.GCPsToGeoTransform(gcps)
src = np.float64(
np.array([[0.0, 0.0], [xSize, 0.0], [xSize, ySize], [0.0, ySize]]))
dst = np.float64(
np.array([cornerPointUL, cornerPointUR, cornerPointLR, cornerPointLL]))
geotransform = from_points(src, dst)
if geotransform is None:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvUtils", 'Unable to extract a geotransform.'),
onlyLog=True)
return
def callBackMetadata(self, currentTime, nextTime):
""" Metadata CallBack """
try:
port = int(self.fileName.split(':')[2])
t = callBackMetadataThread(cmds=[
'-i',
self.fileName.replace(str(port), str(
port + 1)), '-ss', currentTime, '-to', nextTime, '-map',
'data-re', '-preset', 'ultrafast', '-f', 'data', '-'
])
t.start()
t.join(1)
if t.is_alive():
t.p.terminate()
t.join()
qgsu.showUserAndLogMessage("",
"callBackMetadataThread self.stdout: " +
str(t.stdout),
onlyLog=True)
if t.stdout == b'':
return
self.packetStreamParser(t.stdout)
except Exception as e:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvPlayer",
"Metadata Callback Failed! : "),
str(e))
def install_pip_requirements():
''' Install Requeriments from pip >= 10.0.1'''
package_dir = QgsApplication.qgisSettingsDirPath(
) + 'python/plugins/QGIS_FMV/'
requirements_file = os.path.join(package_dir, 'requirements.txt')
if not os.path.isfile(requirements_file):
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvInstaller",
"No requirements file found in {}").format(requirements_file),
onlyLog=True)
raise
try:
process = Popen(["python", "-m", 'pip', "install", '--upgrade', 'pip'],
shell=True,
stdout=PIPE,
stderr=PIPE)
process.wait()
process = Popen(
["python", "-m", 'pip', "install", '-U', 'pip', 'setuptools'],
shell=True,
stdout=PIPE,
stderr=PIPE)
process.wait()
process = Popen(["pip", "install", '--user', '-r', requirements_file],
shell=True,
stdout=PIPE,
stderr=PIPE)
process.wait()
except Exception:
raise
return
def saveInfoToJson(self):
""" Save video Info to json """
if not self.KillAllProcessors():
return
out_json, _ = QFileDialog.getSaveFileName(self, "Save File", "",
"Json Files (*.json)")
if out_json == "":
return
try:
self.VPProbeToJson = Converter()
self.VPTProbeToJson = QThread()
self.VPProbeToJson.moveToThread(self.VPTProbeToJson)
self.VPProbeToJson.finished.connect(self.QThreadFinished)
self.VPProbeToJson.error.connect(self.QThreadError)
self.VPProbeToJson.progress.connect(
self.progressBarProcessor.setValue)
self.VPTProbeToJson.start(QThread.LowPriority)
QMetaObject.invokeMethod(self.VPProbeToJson, 'probeToJson',
Qt.QueuedConnection,
Q_ARG(str, self.fileName),
Q_ARG(str, out_json))
except Exception as e:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvPlayer",
"Error saving Json"))
self.QThreadFinished("probeToJson", "Closing ProbeToJson")
def OpenStream(self, _):
protocol = self.cmb_protocol.currentText().lower()
host = self.ln_host.text()
port = self.ln_port.text()
v = protocol + "://" + host + ":" + port
if host != "" and port != "":
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvOpenStream",
"Checking connection!"))
QApplication.setOverrideCursor(Qt.WaitCursor)
QApplication.processEvents()
# Check if connection exist
cap = cv2.VideoCapture(v)
ret, _ = cap.read()
cap.release()
if ret:
self.parent.AddFileRowToManager(v, v)
self.close()
else:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvOpenStream",
"There is no such connection!"),
level=QGis.Warning,
)
QApplication.restoreOverrideCursor()
def install_pip_requirements():
''' Install Requeriments from pip '''
try:
import pip
except ImportError:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvUtils", "Failed Import Pip! : "),
str(e),
onlyLog=True)
raise
package_dir = QgsApplication.qgisSettingsDirPath(
) + 'python/plugins/QGIS_FMV/'
requirements_file = os.path.join(package_dir, 'requirements.txt')
if not os.path.isfile(requirements_file):
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvUtils",
'No requirements file found in {}'.format(requirements_file)),
str(e),
onlyLog=True)
raise
try:
pip.main(['install', '-r', requirements_file])
except:
raise
return
def paint(self, painter):
''' Paint Frame'''
if (self.currentFrame.map(QAbstractVideoBuffer.ReadOnly)):
oldTransform = painter.transform()
if (self.surfaceFormat().scanLineDirection() ==
QVideoSurfaceFormat.BottomToTop):
painter.scale(1, -1)
painter.translate(0, -self.widget.height())
self.image = QImage(self.currentFrame.bits(),
self.currentFrame.width(),
self.currentFrame.height(),
self.currentFrame.bytesPerLine(), self.imageFormat)
if self.widget._filterSatate.grayColorFilter:
self.image = filter.GrayFilter(self.image)
if self.widget._filterSatate.MirroredHFilter:
self.image = filter.MirrredFilter(self.image)
if self.widget._filterSatate.monoFilter:
self.image = filter.MonoFilter(self.image)
# TODO : Probar en un thread distinto
if self.widget._filterSatate.edgeDetectionFilter:
try:
self.image = filter.EdgeFilter(self.image)
except Exception:
None
# TODO : Probar en un thread distinto
if self.widget._filterSatate.contrastFilter:
try:
self.image = filter.AutoContrastFilter(self.image)
except Exception:
None
if self.widget._filterSatate.invertColorFilter:
self.image.invertPixels()
painter.drawImage(self.targetRect, self.image, self.sourceRect)
if self._interaction.objectTracking and self.widget._isinit:
frame = convertQImageToMat(self.image)
# Update tracker
ok, bbox = self.widget.tracker.update(frame)
# Draw bounding box
if ok:
# qgsu.showUserAndLogMessage(
# "bbox : ", str(bbox), level=QGis.Warning)
painter.setPen(Qt.blue)
painter.drawRect(
QRect(int(bbox[0]), int(bbox[1]), int(bbox[2]),
int(bbox[3])))
else:
qgsu.showUserAndLogMessage("Tracking failure detected ",
"",
level=QGis.Warning)
painter.setTransform(oldTransform)
self.currentFrame.unmap()
return self.painter
def RecordVideo(self, value):
''' Cut Video '''
currentTime = _seconds_to_time(self.currentInfo)
if value is False:
self.endRecord = currentTime
_, file_extension = os.path.splitext(self.fileName)
out, _ = askForFiles(self,
QCoreApplication.translate(
"QgsFmvPlayer", "Save video record"),
isSave=True,
exts=file_extension[1:])
if not out:
self.StopRecordAnimation()
return
p = _spawn([
'-i', self.fileName, '-ss', self.startRecord, '-to',
self.endRecord, '-preset', 'ultrafast', '-c', 'copy', out
])
p.communicate()
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvPlayer",
"Save file succesfully!"))
self.StopRecordAnimation()
else:
self.startRecord = currentTime
self.RecGIF.frameChanged.connect(self.ReciconUpdate)
self.RecGIF.start()
return
def showVideoInfo(self):
''' Show default probe info '''
try:
self.VPProbe = Converter()
self.VPTProbe = QThread()
self.VPProbe.moveToThread(self.VPTProbe)
self.VPProbe.finishedJson.connect(self.QThreadFinished)
self.VPProbe.error.connect(self.QThreadError)
self.VPProbe.progress.connect(self.progressBarProcessor.setValue)
self.VPTProbe.start(QThread.LowPriority)
QMetaObject.invokeMethod(self.VPProbe, 'probeShow',
Qt.QueuedConnection,
Q_ARG(str, self.fileName))
except Exception as e:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvPlayer", "Error Info Show"))
self.QThreadFinished("probeShow", "Closing Probe")
return
def __init__(self, video_path):
self.split = video_path.split(":")
self.srcProtocol = self.split[0]
self.srcHost = self.split[1]
self.srcPort = int(self.split[2])
self.destPort = self.srcPort + 10
self.connection = (self.srcProtocol + ":" + self.srcHost + ":" +
str(self.srcPort))
self.connectionDest = self.srcProtocol + "://127.0.0.1:" + str(
self.destPort)
self.splitter = Splitter([
"-i",
self.connection,
"-c",
"copy",
"-map",
"0:v?",
"-map",
"0:a?",
"-f",
"rtp_mpegts",
self.connectionDest,
"-map",
"0:d?",
"-f",
"data",
"-",
])
self.splitter.start()
qgsu.showUserAndLogMessage("", "Splitter started.", onlyLog=True)
def HasAudio(self, videoPath):
""" Check if video have Metadata or not """
try:
p = _spawn([
'-i', videoPath, '-show_streams', '-select_streams', 'a',
'-preset', 'ultrafast', '-loglevel', 'error'
],
t="probe")
stdout_data, _ = p.communicate()
if stdout_data == b'':
qgsu.showUserAndLogMessage(
QCoreApplication.translate(
"QgsFmvPlayer", "This video doesn't have Audio ! "))
self.actionAudio.setEnabled(False)
self.actionSave_Audio.setEnabled(False)
return False
return True
except Exception as e:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvPlayer",
"Audio check Failed! : "), str(e))
self.actionAudio.setEnabled(False)
self.actionSave_Audio.setEnabled(False)
def AddFileRowToManager(self, name, filename):
''' Add file Video to new Row '''
rowPosition = self.VManager.rowCount()
self.VManager.insertRow(rowPosition)
self.VManager.setItem(rowPosition, 0,
QTableWidgetItem(str(rowPosition)))
self.VManager.setItem(rowPosition, 1, QTableWidgetItem(name))
self.VManager.setItem(rowPosition, 2, QTableWidgetItem("False"))
self.VManager.setItem(rowPosition, 3, QTableWidgetItem(filename))
self.VManager.setVisible(False)
self.VManager.resizeColumnsToContents()
self.VManager.horizontalHeader().setStretchLastSection(True)
self.VManager.setVisible(True)
#init non-blocking metadata buffered reader
self.meta_reader[str(rowPosition)] = BufferedMetaReader(
filename,
pass_time=self.pass_time,
intervall=self.buffer_intervall,
min_buffer_size=self.min_buffer_size)
qgsu.showUserAndLogMessage(
"QgsManager: ",
"buffered non-blocking metadata reader initialized.",
onlyLog=True)
#init point we can center the video on
self.initialPt[str(rowPosition)] = getVideoLocationInfo(filename)
def run(self):
if self.type == "ffmpeg":
self.cmds.insert(0, ffmpeg_path)
else:
self.cmds.insert(0, ffprobe_path)
# qgsu.showUserAndLogMessage("", "starting Splitter on thread:" + str(threading.current_thread().ident), onlyLog=True)
# qgsu.showUserAndLogMessage("", "with args:" + ' '.join(self.cmds), onlyLog=True)
# Hide shell windows that pops up on windows.
if isWindows:
startupinfo = subprocess.STARTUPINFO()
startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
startupinfo.wShowWindow = subprocess.SW_HIDE
self.p = subprocess.Popen(
self.cmds,
startupinfo=startupinfo,
stdin=subprocess.DEVNULL,
stdout=subprocess.PIPE,
)
# Dont us _spawn here as it will DeadLock, and the splitter won't work
# self.p = _spawn(self.cmds)
self.nbsr = NonBlockingStreamReader(self.p)
self.nbsr._t.join()
qgsu.showUserAndLogMessage("", "Splitter thread ended.", onlyLog=True)
def getKlvStreamIndex(videoPath, islocal=False):
if islocal:
return 0
# search for klv data in 5 streams
for i in range(6):
p = _spawn([
'-i', videoPath, '-ss', '00:00:00', '-to', '00:00:01', '-map',
'0:d:' + str(i), '-f', 'data', '-'
])
stdout_data, _ = p.communicate()
if stdout_data == b'':
continue
else:
# look if stream has valid klv data
if b'\x06\x0e+4\x02\x0b\x01\x01\x0e\x01\x03\x01\x01\x00\x00\x00' in stdout_data or b'\x06\x0e+4\x02\x01\x01\x01\x0e\x01\x01\x02\x01\x01\x00\x00' in stdout_data:
return i
else:
qgsu.showUserAndLogMessage("",
"skipping stream " + str(i) +
" not a klv stream.",
onlyLog=True)
continue
qgsu.showUserAndLogMessage(
"Error interpreting klv data, metadata cannot be read.",
"the parser did not recognize KLV data",
level=QGis.Warning)
return 0
def GetLine3DIntersectionWithDEM(sensorPt, targetPt):
''' Obtain height for points,intersecting with DEM '''
pt = []
sensorLat = sensorPt[0]
sensorLon = sensorPt[1]
sensorAlt = sensorPt[2]
targetLat = targetPt[0]
targetLon = targetPt[1]
try:
targetAlt = targetPt[2]
except Exception:
targetAlt = GetFrameCenter()[2]
distance = sphere.distance([sensorLat, sensorLon], [targetLat, targetLon])
distance = sqrt(distance**2 + (targetAlt - sensorAlt)**2)
dLat = (targetLat - sensorLat) / distance
dLon = (targetLon - sensorLon) / distance
dAlt = (targetAlt - sensorAlt) / distance
xOrigin = dtm_transform[0]
yOrigin = dtm_transform[3]
pixelWidth = dtm_transform[1]
pixelHeight = -dtm_transform[5]
pixelWidthMeter = pixelWidth * (pi / 180.0) * 6378137.0
# start at k = sensor point, then test every pixel a point on the 3D line
# until we cross the dtm (diffAlt >= 0).
diffAlt = -1
for k in range(0, int(dtm_buffer * pixelWidthMeter), int(pixelWidthMeter)):
point = [
sensorLon + k * dLon, sensorLat + k * dLat, sensorAlt + k * dAlt
]
col = int((point[0] - xOrigin) / pixelWidth)
row = int((yOrigin - point[1]) / pixelHeight)
try:
diffAlt = point[2] - dtm_data[row -
dtm_rowLowerBound][col -
dtm_colLowerBound]
except Exception:
qgsu.showUserAndLogMessage(
"", "DEM point not found after all iterations.", onlyLog=True)
break
if diffAlt <= 0:
pt = [point[1], point[0], point[2]]
break
if not pt:
qgsu.showUserAndLogMessage(
"",
"DEM point not found, last computed delta high: " + str(diffAlt),
onlyLog=True)
return pt
def SetGCPsToGeoTransform(cornerPointUL, cornerPointUR, cornerPointLR,
cornerPointLL, frameCenterLon, frameCenterLat, ele):
''' Make Geotranform from pixel to lon lat coordinates '''
gcps = []
global gcornerPointUL, gcornerPointUR, gcornerPointLR, gcornerPointLL, gframeCenterLat, gframeCenterLon, geotransform_affine, geotransform
# TEMP FIX : If have elevation the geotransform is wrong
if ele:
del cornerPointUL[-1]
del cornerPointUR[-1]
del cornerPointLR[-1]
del cornerPointLL[-1]
gcornerPointUL = cornerPointUL
gcornerPointUR = cornerPointUR
gcornerPointLR = cornerPointLR
gcornerPointLL = cornerPointLL
gframeCenterLat = frameCenterLat
gframeCenterLon = frameCenterLon
Height = GetFrameCenter()[2]
gcp = gdal.GCP(cornerPointUL[1], cornerPointUL[0], Height, 0, 0,
"Corner Upper Left", "1")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointUR[1], cornerPointUR[0], Height, xSize, 0,
"Corner Upper Right", "2")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointLR[1], cornerPointLR[0], Height, xSize, ySize,
"Corner Lower Right", "3")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointLL[1], cornerPointLL[0], Height, 0, ySize,
"Corner Lower Left", "4")
gcps.append(gcp)
gcp = gdal.GCP(frameCenterLon, frameCenterLat, Height, xSize / 2,
ySize / 2, "Center", "5")
gcps.append(gcp)
geotransform_affine = gdal.GCPsToGeoTransform(gcps)
src = np.float64(
np.array([[0.0, 0.0], [xSize, 0.0], [xSize, ySize], [0.0, ySize]]))
dst = np.float64(
np.array([cornerPointUL, cornerPointUR, cornerPointLR, cornerPointLL]))
try:
geotransform = from_points(src, dst)
except Exception:
pass
if geotransform is None:
qgsu.showUserAndLogMessage("",
"Unable to extract a geotransform.",
onlyLog=True)
return
def UpdateFrameAxisData(imgSS, sensor, framecenter, ele):
""" Update Frame Axis Values """
global crtSensorSrc2, groupName, frameAxisMarker
lat = sensor[0]
lon = sensor[1]
alt = sensor[2]
fc_lat = framecenter[0]
fc_lon = framecenter[1]
fc_alt = framecenter[2]
frameaxisLyr = selectLayerByName(FrameAxis_lyr, groupName)
try:
if all(v is not None
for v in [frameaxisLyr, lat, lon, alt, fc_lat, fc_lon]):
if imgSS != crtSensorSrc2:
SetDefaultFrameAxisStyle(frameaxisLyr, imgSS)
crtSensorSrc2 = imgSS
frameaxisLyr.startEditing()
if frameaxisLyr.featureCount() == 0:
f = QgsFeature()
f.setAttributes([lon, lat, alt, fc_lon, fc_lat, fc_alt])
f.setGeometry(
QgsLineString(QgsPoint(lon, lat, alt),
QgsPoint(fc_lon, fc_lat, fc_alt)))
frameaxisLyr.addFeatures([f])
else:
frameaxisLyr.dataProvider().changeAttributeValues({
1: {
0: lon,
1: lat,
2: alt,
3: fc_lon,
4: fc_lat,
5: fc_alt
}
})
frameaxisLyr.dataProvider().changeGeometryValues({
1:
QgsGeometry(
QgsLineString(
QgsPoint(lon, lat, alt),
QgsPoint(fc_lon, fc_lat, fc_alt),
))
})
CommonLayer(frameaxisLyr)
# 3D Style
if ele:
SetDefaultFrameAxis3DStyle(frameaxisLyr)
except Exception as e:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvUtils",
"Failed Update Frame axis Layer! : "),
str(e),
)
def UpdateTrajectoryData(packet):
''' Update Trajectory Values '''
global tLastLon
global tLastLat
lat = packet.GetSensorLatitude()
lon = packet.GetSensorLongitude()
alt = packet.GetSensorTrueAltitude()
try:
if tLastLon == 0.0 and tLastLat == 0.0:
tLastLon = lon
tLastLat = lat
else:
# little check to see if telemetry data are plausible before
# drawing.
distance = sphere.distance((tLastLon, tLastLat), (lon, lat))
if distance > 1000: # 1 km is the best value for prevent draw trajectory when start video again
return
except Exception:
None
tLastLon = lon
tLastLat = lat
trajectoryLyr = qgsu.selectLayerByName(Trajectory_lyr)
try:
if all(v is not None for v in [trajectoryLyr, lat, lon, alt]):
trajectoryLyr.startEditing()
f = QgsFeature()
if trajectoryLyr.featureCount() == 0:
f.setAttributes([lon, lat, alt])
surface = QgsGeometry.fromPolylineXY(
[QgsPointXY(lon, lat),
QgsPointXY(lon, lat)])
f.setGeometry(surface)
trajectoryLyr.addFeatures([f])
else:
f_last = trajectoryLyr.getFeature(trajectoryLyr.featureCount())
f.setAttributes([lon, lat, alt])
surface = QgsGeometry.fromPolylineXY([
QgsPointXY(lon, lat),
QgsPointXY(f_last.attribute(0), f_last.attribute(1))
])
f.setGeometry(surface)
trajectoryLyr.addFeatures([f])
CommonLayer(trajectoryLyr)
except Exception as e:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvUtils",
"Failed Update Trajectory Layer! : "),
str(e))
return
def UpdateFrameAxisData(packet, ele):
''' Update Frame Axis Values '''
global crtSensorSrc2, groupName
imgSS = packet.ImageSourceSensor
lat = packet.SensorLatitude
lon = packet.SensorLongitude
alt = packet.SensorTrueAltitude
fc_lat = packet.FrameCenterLatitude
fc_lon = packet.FrameCenterLongitude
fc_alt = packet.FrameCenterElevation
frameaxisLyr = qgsu.selectLayerByName(FrameAxis_lyr, groupName)
try:
if all(v is not None
for v in [frameaxisLyr, lat, lon, alt, fc_lat, fc_lon]):
if (imgSS != crtSensorSrc2):
SetDefaultFrameAxisStyle(frameaxisLyr, imgSS)
crtSensorSrc2 = imgSS
frameaxisLyr.startEditing()
if frameaxisLyr.featureCount() == 0:
f = QgsFeature()
f.setAttributes([lon, lat, alt, fc_lon, fc_lat, fc_alt])
f.setGeometry(
QgsLineString(QgsPoint(lon, lat, alt),
QgsPoint(fc_lon, fc_lat, fc_alt)))
frameaxisLyr.addFeatures([f])
else:
frameaxisLyr.dataProvider().changeAttributeValues({
1: {
0: lon,
1: lat,
2: alt,
3: fc_lon,
4: fc_lat,
5: fc_alt
}
})
frameaxisLyr.dataProvider().changeGeometryValues({
1:
QgsGeometry(
QgsLineString(QgsPoint(lon, lat, alt),
QgsPoint(fc_lon, fc_lat, fc_alt)))
})
CommonLayer(frameaxisLyr)
# 3D Style
if ele:
SetDefaultFrameAxis3DStyle(frameaxisLyr)
except Exception as e:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvUtils",
"Failed Update Frame axis Layer! : "),
str(e))
return
def finished(e):
QApplication.restoreOverrideCursor()
if e is None:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvMetadata", "Succesfully creating PDF"))
else:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvMetadata", "Failed creating PDF : "), str(e), level=QGis.Warning)
return
def SetGCPsToGeoTransform(cornerPointUL, cornerPointUR, cornerPointLR,
cornerPointLL, frameCenterLon, frameCenterLat, ele):
''' Make Geotranform from pixel to lon lat coordinates '''
gcps = []
gv.setCornerUL(cornerPointUL)
gv.setCornerUR(cornerPointUR)
gv.setCornerLR(cornerPointLR)
gv.setCornerLL(cornerPointLL)
gv.setFrameCenter(frameCenterLat, frameCenterLon)
xSize = gv.getXSize()
ySize = gv.getYSize()
Height = GetFrameCenter()[2]
gcp = gdal.GCP(cornerPointUL[1], cornerPointUL[0], Height, 0, 0,
"Corner Upper Left", "1")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointUR[1], cornerPointUR[0], Height, xSize, 0,
"Corner Upper Right", "2")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointLR[1], cornerPointLR[0], Height, xSize, ySize,
"Corner Lower Right", "3")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointLL[1], cornerPointLL[0], Height, 0, ySize,
"Corner Lower Left", "4")
gcps.append(gcp)
gcp = gdal.GCP(frameCenterLon, frameCenterLat, Height, xSize / 2,
ySize / 2, "Center", "5")
gcps.append(gcp)
at = gdal.GCPsToGeoTransform(gcps)
gv.setAffineTransform(at)
src = np.float64(
np.array([[0.0, 0.0], [xSize, 0.0], [xSize, ySize], [0.0, ySize],
[xSize / 2.0, ySize / 2.0]]))
dst = np.float64(
np.array([[cornerPointUL[0], cornerPointUL[1]],
[cornerPointUR[0], cornerPointUR[1]],
[cornerPointLR[0], cornerPointLR[1]],
[cornerPointLL[0], cornerPointLL[1]],
[frameCenterLat, frameCenterLon]]))
try:
geotransform, _ = findHomography(src, dst)
gv.setTransform(geotransform)
except Exception:
pass
if geotransform is None:
qgsu.showUserAndLogMessage("",
"Unable to extract a geotransform.",
onlyLog=True)
return
def playFile(self, videoPath):
''' Play file from path '''
try:
RemoveVideoLayers()
RemoveGroupByName()
# settrace()
# if "udp://" in videoPath:
# host, port = videoPath.split("://")[1].split(":")
# receiver = UDPClient(host, int(port), type="udp")
# receiver.show()
# self.close()
# return
# if "tcp://" in videoPath:
# host, port = videoPath.split("://")[1].split(":")
# receiver = UDPClient(host, port, type="tcp")
# receiver.show()
# self.close()
# return
self.fileName = videoPath
self.playlist = QMediaPlaylist()
url = QUrl.fromLocalFile(videoPath)
self.playlist.addMedia(QMediaContent(url))
self.player.setPlaylist(self.playlist)
self.setWindowTitle(
QCoreApplication.translate("QgsFmvPlayer", 'Playing : ') +
os.path.basename(os.path.normpath(videoPath)))
if self.HasMetadata(videoPath):
CreateVideoLayers()
self.clearMetadata()
self.lb_cursor_coord.setText(
"<span style='font-size:10pt; font-weight:bold;'>Lon :</span>"
+
"<span style='font-size:9pt; font-weight:normal;'>Null</span>"
+
"<span style='font-size:10pt; font-weight:bold;'> Lat :</span>"
+
"<span style='font-size:9pt; font-weight:normal;'>Null</span>"
)
else:
self.btn_GeoReferencing.setEnabled(False)
self.HasFileAudio = True
if not self.HasAudio(videoPath):
self.actionAudio.setEnabled(False)
self.actionSave_Audio.setEnabled(False)
self.HasFileAudio = False
self.playClicked(True)
except Exception as e:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvPlayer", 'Open Video File : '),
str(e),
level=QGis.Warning)
def QThreadError(self, processor, e, exception_string):
""" Threads Errors"""
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvPlayer", processor),
'Failed!\n'.format(exception_string),
level=QGis.Warning)
self.QThreadFinished(processor, "Closing Processor")
return
def playFile(self, videoPath):
''' Play file from path '''
try:
RemoveVideoLayers()
RemoveGroupByName()
# if "udp://" in videoPath:
# host, port = videoPath.split("://")[1].split(":")
# receiver = UDPClient(host, int(port), type="udp")
# receiver.show()
# self.close()
# return
# if "tcp://" in videoPath:
# host, port = videoPath.split("://")[1].split(":")
# receiver = UDPClient(host, port, type="tcp")
# receiver.show()
# self.close()
# return
self.fileName = videoPath
self.playlist = QMediaPlaylist()
if self.isStreaming:
url = QUrl(videoPath)
else:
url = QUrl.fromLocalFile(videoPath)
qgsu.showUserAndLogMessage("", "Added: " + str(url), onlyLog=True)
self.playlist.addMedia(QMediaContent(url))
self.player.setPlaylist(self.playlist)
self.setWindowTitle(
QCoreApplication.translate("QgsFmvPlayer", 'Playing : ') +
os.path.basename(os.path.normpath(videoPath)))
CreateVideoLayers()
self.clearMetadata()
self.HasFileAudio = True
if not self.HasAudio(videoPath):
self.actionAudio.setEnabled(False)
self.actionSave_Audio.setEnabled(False)
self.HasFileAudio = False
# Recenter map on video initial point
if self.initialPt:
rect = QgsRectangle(self.initialPt[1], self.initialPt[0],
self.initialPt[1], self.initialPt[0])
self.iface.mapCanvas().setExtent(rect)
self.iface.mapCanvas().refresh()
self.playClicked(True)
except Exception as e:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvPlayer", 'Open Video File : '),
str(e),
level=QGis.Warning)
def dispose(self):
# qgsu.showUserAndLogMessage("", "Dispose called on StreamMetaReader.", onlyLog=True)
self.splitter.nbsr.stopped = True
# kill the process if open, releases source port
try:
self.splitter.p.kill()
qgsu.showUserAndLogMessage("",
"Splitter Popen process killed.",
onlyLog=True)
except OSError:
# can't kill a dead proc
pass
def statusChanged(self, status):
''' Signal Status video change '''
self.handleCursor(status)
if status is QMediaPlayer.LoadingMedia or status is QMediaPlayer.StalledMedia or status is QMediaPlayer.InvalidMedia:
self.videoAvailableChanged(False)
elif status == QMediaPlayer.InvalidMedia:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvPlayer", self.player.errorString()),
level=QGis.Warning)
self.videoAvailableChanged(False)
else:
self.videoAvailableChanged(True)
def SetGCPsToGeoTransform(cornerPointUL, cornerPointUR, cornerPointLR,
cornerPointLL, frameCenterLon, frameCenterLat):
''' Make Geotranform from pixel to lon lat coordinates '''
gcps = []
global gcornerPointUL
gcornerPointUL = cornerPointUL
global gcornerPointUR
gcornerPointUR = cornerPointUR
global gcornerPointLR
gcornerPointLR = cornerPointLR
global gcornerPointLL
gcornerPointLL = cornerPointLL
global gframeCenterLat
gframeCenterLat = frameCenterLat
global gframeCenterLon
gframeCenterLon = frameCenterLon
global geotransform
global geotransform_affine
Height = GetFrameCenter()[2]
gcp = gdal.GCP(cornerPointUL[1], cornerPointUL[0], Height, 0, 0,
"Corner Upper Left", "1")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointUR[1], cornerPointUR[0], Height, xSize, 0,
"Corner Upper Right", "2")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointLR[1], cornerPointLR[0], Height, xSize, ySize,
"Corner Lower Right", "3")
gcps.append(gcp)
gcp = gdal.GCP(cornerPointLL[1], cornerPointLL[0], Height, 0, ySize,
"Corner Lower Left", "4")
gcps.append(gcp)
gcp = gdal.GCP(frameCenterLon, frameCenterLat, Height, xSize / 2,
ySize / 2, "Center", "5")
gcps.append(gcp)
geotransform_affine = gdal.GCPsToGeoTransform(gcps)
src = np.float64(
np.array([[0.0, 0.0], [xSize, 0.0], [xSize, ySize], [0.0, ySize]]))
dst = np.float64(
np.array([cornerPointUL, cornerPointUR, cornerPointLR, cornerPointLL]))
geotransform = from_points(src, dst)
if geotransform is None:
qgsu.showUserAndLogMessage("",
"Unable to extract a geotransform.",
onlyLog=True)
return
def UpdatePlatformData(packet, ele):
''' Update PlatForm Values '''
global crtPltTailNum, groupName
lat = packet.SensorLatitude
lon = packet.SensorLongitude
alt = packet.SensorTrueAltitude
PlatformHeading = packet.PlatformHeadingAngle
platformTailNumber = packet.PlatformTailNumber
platformLyr = qgsu.selectLayerByName(Platform_lyr, groupName)
try:
if all(v is not None
for v in [platformLyr, lat, lon, alt, PlatformHeading]):
if platformTailNumber != crtPltTailNum:
SetDefaultPlatformStyle(platformLyr, platformTailNumber)
crtPltTailNum = platformTailNumber
platformLyr.startEditing()
platformLyr.renderer().symbol().setAngle(float(PlatformHeading))
if platformLyr.featureCount() == 0:
feature = QgsFeature()
feature.setAttributes([lon, lat, alt])
feature.setGeometry(QgsPoint(lon, lat, alt))
platformLyr.addFeatures([feature])
else:
platformLyr.dataProvider().changeAttributeValues(
{1: {
0: lon,
1: lat,
2: alt
}})
platformLyr.dataProvider().changeGeometryValues(
{1: QgsGeometry(QgsPoint(lon, lat, alt))})
CommonLayer(platformLyr)
# 3D Style
if ele:
SetDefaultPlatform3DStyle(platformLyr)
except Exception as e:
qgsu.showUserAndLogMessage(
QCoreApplication.translate("QgsFmvUtils",
"Failed Update Platform Layer! : "),
str(e))
return
def initElevationModel(frameCenterLat, frameCenterLon, dtm_path):
''' Start DEM transformation and extract data for set Z value in points '''
global dtm_data, dtm_transform, dtm_colLowerBound, dtm_rowLowerBound
# Initialize the dtm once, based on a zone arouind the target
qgsu.showUserAndLogMessage("", "Initializing DTM.", onlyLog=True)
dataset = gdal.Open(dtm_path)
if dataset is None:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvUtils", "Failed to read DTM file. "), level=QGis.Warning)
return
band = dataset.GetRasterBand(1)
dtm_transform = dataset.GetGeoTransform()
xOrigin = dtm_transform[0]
yOrigin = dtm_transform[3]
pixelWidth = dtm_transform[1]
pixelHeight = -dtm_transform[5]
cIndex = int((frameCenterLon - xOrigin) / pixelWidth)
rIndex = int((frameCenterLat - yOrigin) / (-pixelHeight))
dtm_colLowerBound = cIndex - dtm_buffer
dtm_rowLowerBound = rIndex - dtm_buffer
if dtm_colLowerBound < 0 or dtm_rowLowerBound < 0:
qgsu.showUserAndLogMessage(QCoreApplication.translate(
"QgsFmvUtils", "There is no DTM for theses bounds. Check/increase DTM_buffer_size in settings.ini"), level=QGis.Warning)
else:
# qgsu.showUserAndLogMessage("UpdateLayers: ", " dtm_colLowerBound:"+str(dtm_colLowerBound)+" dtm_rowLowerBound:"+str(dtm_rowLowerBound)+" dtm_buffer:"+str(dtm_buffer), onlyLog=True)
dtm_data = band.ReadAsArray(
dtm_colLowerBound, dtm_rowLowerBound, 2 * dtm_buffer, 2 * dtm_buffer)
if dtm_data is not None:
qgsu.showUserAndLogMessage(
"", "DTM successfully initialized, len: " + str(len(dtm_data)), onlyLog=True)