import os, sys, fnmatch, six, subprocess, re
from PyQt4.QtSvg import *
from PyQt4.QtCore import *
from PyQt4.QtGui import *
from PyQt4.QtXml import *
from PyQt4.QtXmlPatterns import *
from PyQt4.uic.Compiler.qtproxies import QtGui
import gdal
from timeseriesviewer import DIR_UI, file_search
jp = os.path.join
def getDOMAttributes(elem):
assert isinstance(elem, QDomElement)
values = dict()
attributes = elem.attributes()
for a in range(attributes.count()):
attr = attributes.item(a)
values[str(attr.nodeName())] = attr.nodeValue()
return values
def createTestData(pathTS, subsetRectangle, crs, dirTestData):
lines = open(pathTS).readlines()
import tempfile, random
from timeseriesviewer.main import TimeSeries, TimeSeriesDatum
from qgis.core import QgsRectangle, QgsPoint, QgsPointV2, QgsCoordinateReferenceSystem
max_offset = 0 #in %
assert isinstance(subsetRectangle, QgsRectangle)
assert isinstance(crs, QgsCoordinateReferenceSystem)
TS = TimeSeries()
TS.loadFromFile(pathTS)
sw = subsetRectangle.width()
sh = subsetRectangle.height()
max_offset_x = sw / 100 * max_offset
max_offset_y = sw / 100 * max_offset
center = subsetRectangle.center()
os.mkdir(dirTestData)
pathVRT = tempfile.mkstemp(suffix='.vrt', prefix='tempVRT')
def random_xy():
offset_x = center.x() + random.randrange(-max_offset_x, max_offset_x)
offset_y = center.y() + random.randrange(-max_offset_y, max_offset_y)
return offset_x, offset_y
for TSD in TS:
assert isinstance(TSD, TimeSeriesDatum)
ox, oy = random_xy()
UL = QgsPoint(subsetRectangle.xMinimum() + ox,
subsetRectangle.yMaximum() + oy)
LR = QgsPoint(subsetRectangle.xMaximum() + ox,
subsetRectangle.yMinimum() + oy)
lyr = QgsRasterLayer(TSD.pathImg)
crsDS = lyr.dataProvider().crs()
assert isinstance(crsDS, QgsCoordinateReferenceSystem)
transform = QgsCoordinateTransform(crs, crsDS)
#transform UL and LR into CRS of source data set
UL = transform.transform(UL)
LR = transform.transform(LR)
BBOX = QgsRectangle(UL, LR)
#crop src dataset to UL-LR box
#for this we use GDAL
dsSrc = gdal.Open(TSD.pathImg)
assert isinstance(dsSrc, gdal.Dataset)
proj = dsSrc.GetProjection()
trans = dsSrc.GetGeoTransform()
trans[0] = UL.x()
trans[3] = UL.y()
ns = BBOX.width() / lyr.rasterUnitsPerPixelX()
nl = BBOX.heigth() / lyr.rasterUnitsPerPixelY()
drv = dsSrc.GetDriver()
assert isinstance(drv, gdal.Driver)
dsDst = drv.Create(ns, nl, dsSrc.RasterCount, eType = dsSrc.GetRasterBand(1).)
assert isinstance(dsDst, gdal.Dataset)
dsDst.SetProjection(proj)
dsDst.SetGeoTransform(trans)
def make(ROOT):
#find ui files
ui_files = file_search(ROOT, '*.ui', recursive=True)
qrcs = set()
doc = QDomDocument()
reg = re.compile('(?<=resource=")[^"]+\.qrc(?=")')
for ui_file in ui_files:
pathDir = os.path.dirname(ui_file)
doc.setContent(QFile(ui_file))
includeNodes = doc.elementsByTagName('include')
for i in range(includeNodes.count()):
attr = getDOMAttributes(includeNodes.item(i).toElement())
if 'location' in attr.keys():
qrcs.add((pathDir, str(attr['location'])))
#compile Qt resource files
#resourcefiles = file_search(ROOT, '*.qrc', recursive=True)
resourcefiles = list(qrcs)
assert len(resourcefiles) > 0
for root_dir, f in resourcefiles:
#dn = os.path.dirname(f)
pathQrc = os.path.normpath(jp(root_dir, f))
assert os.path.exists(pathQrc)
bn = os.path.basename(f)
bn = os.path.splitext(bn)[0]
pathPy2 = os.path.join(DIR_UI, bn+'_py2.py' )
pathPy3 = os.path.join(DIR_UI, bn+'_py3.py' )
print('Make {}'.format(pathPy2))
subprocess.call(['pyrcc4','-py2','-o',pathPy2, pathQrc])
print('Make {}'.format(pathPy3))
subprocess.call(['pyrcc4','-py3','-o',pathPy3, pathQrc])
def svg2png(pathDir, overwrite=False, mode='INKSCAPE'):
assert mode in ['INKSCAPE', 'WEBKIT', 'SVG']
from PyQt4.QtWebKit import QWebPage
svgs = file_search(pathDir, '*.svg')
app = QApplication([], True)
for pathSvg in svgs:
dn = os.path.dirname(pathSvg)
bn, _ = os.path.splitext(os.path.basename(pathSvg))
pathPng = jp(dn, bn+'.png')
if mode == 'SVG':
renderer = QSvgRenderer(pathSvg)
doc_size = renderer.defaultSize() # size in px
img = QImage(doc_size, QImage.Format_ARGB32)
#img.fill(0xaaA08080)
painter = QPainter(img)
renderer.render(painter)
painter.end()
if overwrite or not os.path.exists(pathPng):
img.save(pathPng, quality=100)
del painter, renderer
elif mode == 'WEBKIT':
page = QWebPage()
frame = page.mainFrame()
f = QFile(pathSvg)
if f.open(QFile.ReadOnly | QFile.Text):
textStream = QTextStream(f)
svgData = textStream.readAll()
f.close()
qba = QByteArray(str(svgData))
frame.setContent(qba,"image/svg+xml")
page.setViewportSize(frame.contentsSize())
palette = page.palette()
background_color = QColor(50,0,0,50)
palette.setColor(QPalette.Window, background_color)
brush = QBrush(background_color)
palette.setBrush(QPalette.Window, brush)
page.setPalette(palette)
img = QImage(page.viewportSize(), QImage.Format_ARGB32)
img.fill(background_color) #set transparent background
painter = QPainter(img)
painter.setBackgroundMode(Qt.OpaqueMode)
#print(frame.renderTreeDump())
frame.render(painter)
painter.end()
if overwrite or not os.path.exists(pathPng):
print('Save {}...'.format(pathPng))
img.save(pathPng, quality=100)
del painter, frame, img, page
s =""
elif mode == 'INKSCAPE':
dirInkscape = r'C:\Program Files\Inkscape'
assert os.path.isdir(dirInkscape)
cmd = [jp(dirInkscape,'inkscape')]
cmd.append('--file={}'.format(pathSvg))
cmd.append('--export-png={}'.format(pathPng))
subprocess.call(cmd)
s = ""
def png2qrc(icondir, pathQrc, pngprefix='timeseriesviewer/png'):
pathQrc = os.path.abspath(pathQrc)
dirQrc = os.path.dirname(pathQrc)
app = QApplication([])
assert os.path.exists(pathQrc)
doc = QDomDocument()
doc.setContent(QFile(pathQrc))
query = QXmlQuery()
#query.setQuery("doc('{}')/RCC/qresource/file".format(pathQrc))
query.setQuery("doc('{}')/RCC/qresource[@prefix=\"{}\"]/file".format(pathQrc, pngprefix))
query.setQuery("for $x in doc('{}')/RCC/qresource[@prefix=\"{}\"] return data($x)".format(pathQrc, pngprefix))
assert query.isValid()
#elem = doc.elementsByTagName('qresource')print
pngFiles = [r.strip() for r in str(query.evaluateToString()).split('\n')]
pngFiles = set([f for f in pngFiles if os.path.isfile(jp(dirQrc,f))])
for f in file_search(icondir, '*.png'):
xmlPath = os.path.relpath(f, dirQrc).replace('\\','/')
pngFiles.add(xmlPath)
pngFiles = sorted(list(pngFiles))
def getResourcePrefixNodes(prefix):
resourceNodes = doc.elementsByTagName('qresource')
nodeList = []
for i in range(resourceNodes.count()):
resourceNode = resourceNodes.item(i).toElement()
if resourceNode.hasAttribute('prefix') and resourceNode.attribute('prefix') == prefix:
nodeList.append(resourceNode)
return nodeList
resourceNode = getResourcePrefixNodes(pngprefix)
if len(resourceNode) > 0:
resourceNode = resourceNode[0]
else:
resourceNode = doc.createElement('qresource')
resourceNode.setAttribute('prefix', pngprefix)
#remove childs, as we have all stored in list pngFiles
childs = resourceNode.childNodes()
while not childs.isEmpty():
node = childs.item(0)
node.parentNode().removeChild(node)
#insert new childs
for pngFile in pngFiles:
node = doc.createElement('file')
node.appendChild(doc.createTextNode(pngFile))
resourceNode.appendChild(node)
f = open(pathQrc, "w")
f.write(doc.toString())
f.close()
s = ""
if __name__ == '__main__':
icondir = jp(DIR_UI, *['icons'])
pathQrc = jp(DIR_UI,'resources.qrc')
if True:
#convert SVG to PNG and link them into the resource file
#svg2png(icondir, overwrite=True)
#add png icons to qrc file
png2qrc(icondir, pathQrc)
if True:
make(DIR_UI)
print('Done')