https://github.com/minorua/Qgis2threejs
Tip revision: 68114ecb413adc395354168c9c446f15c7684dec authored by Minoru Akagi on 23 January 2020, 05:13:49 UTC
version 2.4.2
version 2.4.2
Tip revision: 68114ec
qgis2threejscore.py
# -*- coding: utf-8 -*-
"""
/***************************************************************************
Qgis2threejs
A QGIS plugin
export terrain data, map canvas image and vector data to web browser
-------------------
begin : 2014-01-16
copyright : (C) 2014 Minoru Akagi
email : akaginch@gmail.com
***************************************************************************/
/***************************************************************************
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
***************************************************************************/
"""
import struct
from math import floor
from osgeo import gdal
from PyQt5.QtCore import QSize
from .geometry import GridGeometry
from .mapextent import MapExtent
from .qgis2threejstools import logMessage
class MapTo3D:
def __init__(self, mapSettings, planeWidth=100, verticalExaggeration=1, verticalShift=0):
# map canvas
self.rotation = mapSettings.rotation()
self.mapExtent = MapExtent.fromMapSettings(mapSettings)
rect = self.mapExtent.unrotatedRect()
self._xmin, self._ymin = (rect.xMinimum(), rect.yMinimum())
self._width, self._height = (rect.width(), rect.height())
# 3d
canvas_size = mapSettings.outputSize()
self.planeWidth = planeWidth
self.planeHeight = planeWidth * canvas_size.height() / canvas_size.width()
self.verticalExaggeration = verticalExaggeration
self.verticalShift = verticalShift
self.multiplier = planeWidth / self.mapExtent.width()
self.multiplierZ = self.multiplier * verticalExaggeration
def transform(self, x, y, z=0):
x, y = self.mapExtent.normalizePoint(x, y)
return [(x - 0.5) * self.planeWidth,
(y - 0.5) * self.planeHeight,
(z + self.verticalShift) * self.multiplierZ]
def transformXY(self, x, y, z=0):
x, y = self.mapExtent.normalizePoint(x, y)
return [(x - 0.5) * self.planeWidth,
(y - 0.5) * self.planeHeight,
z]
def transformRotated(self, x, y, z=0):
"""transform coordinates of a point of rotated geometry"""
x, y = ((x - self._xmin) / self._width,
(y - self._ymin) / self._height)
return [(x - 0.5) * self.planeWidth,
(y - 0.5) * self.planeHeight,
(z + self.verticalShift) * self.multiplierZ]
def transformRotatedXY(self, x, y, z=0):
x, y = ((x - self._xmin) / self._width,
(y - self._ymin) / self._height)
return [(x - 0.5) * self.planeWidth,
(y - 0.5) * self.planeHeight,
z]
class GDALDEMProvider:
def __init__(self, filename, dest_wkt, source_wkt=None):
self.filename = filename
self.dest_wkt = dest_wkt
self.source_wkt = source_wkt
self.mem_driver = gdal.GetDriverByName("MEM")
filename_utf8 = filename.encode("utf-8") if isinstance(filename, str) else filename
self.ds = gdal.Open(filename_utf8, gdal.GA_ReadOnly)
if self.ds is None:
logMessage("Cannot open file: " + filename)
self.ds = self.mem_driver.Create("", 1, 1, 1, gdal.GDT_Float32)
self.width = self.ds.RasterXSize
self.height = self.ds.RasterYSize
def _read(self, width, height, geotransform):
# create a memory dataset
warped_ds = self.mem_driver.Create("", width, height, 1, gdal.GDT_Float32)
warped_ds.SetProjection(self.dest_wkt)
warped_ds.SetGeoTransform(geotransform)
# reproject image
gdal.ReprojectImage(self.ds, warped_ds, self.source_wkt, None, gdal.GRA_Bilinear)
band = warped_ds.GetRasterBand(1)
return band.ReadRaster(0, 0, width, height, buf_type=gdal.GDT_Float32)
def read(self, width, height, extent):
"""read data into a byte array"""
return self._read(width, height, extent.geotransform(width, height))
def readValues(self, width, height, extent):
"""read data into a list"""
return struct.unpack("f" * width * height, self.read(width, height, extent))
def readAsGridGeometry(self, width, height, extent):
return GridGeometry(extent,
width - 1, height - 1,
self.readValues(width, height, extent))
def readValue(self, x, y):
"""get value at specified position using 1px * 1px memory raster"""
res = 0.1
geotransform = [x - res / 2, res, 0, y + res / 2, 0, -res]
return struct.unpack("f", self._read(1, 1, geotransform))[0]
def readValueOnTriangles(self, x, y, xmin, ymin, xres, yres):
mx0 = floor((x - xmin) / xres)
my0 = floor((y - ymin) / yres)
px0 = xmin + xres * mx0
py0 = ymin + yres * my0
geotransform = [px0, xres, 0, py0 + yres, 0, -yres]
z = struct.unpack("f" * 4, self._read(2, 2, geotransform))
sdx = (x - px0) / xres
sdy = (y - py0) / yres
if sdx <= sdy:
return z[0] + (z[1] - z[0]) * sdx + (z[2] - z[0]) * (1 - sdy)
return z[3] + (z[2] - z[3]) * (1 - sdx) + (z[1] - z[3]) * sdy
class FlatDEMProvider:
def __init__(self, value=0):
self.value = value
def name(self):
return "Flat Plane"
def read(self, width, height, extent):
return struct.pack("{0}f".format(width * height), *([self.value] * width * height))
def readValues(self, width, height, extent):
return [self.value] * width * height
def readAsGridGeometry(self, width, height, extent):
return GridGeometry(extent,
width - 1, height - 1,
[self.value] * width * height)
def readValue(self, x, y):
return self.value
def calculateDEMSize(canvasSize, sizeLevel, roughness=0):
width, height = canvasSize.width(), canvasSize.height()
size = 100 * sizeLevel
s = (size * size / (width * height)) ** 0.5
if s < 1:
width = int(width * s)
height = int(height * s)
if roughness:
if width % roughness != 0:
width = int(width / roughness + 0.9) * roughness
if height % roughness != 0:
height = int(height / roughness + 0.9) * roughness
return QSize(width + 1, height + 1)