gee/utils.py at main · dima4914/gee · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
from qgis.utils import iface
from qgis.core import *


def get_vector_names():
    'retrieves names of vector layers in project'
    names = []
    layers = QgsProject.instance().mapLayers().values()
    for layer in layers:
        if isinstance(layer, QgsVectorLayer):
            names.append(layer.name())
    return names


def get_gee_names():
    'retrieves names of gee layers in project that can be processed'
    names = []
    layers = QgsProject.instance().mapLayers().values()
    for layer in layers:
        if layer.customProperty('ee-object'):
            names.append(layer.name())
    return names


def pan_sharp(image, bands=['B4','B3', 'B2'], panchrom='B8'):
    hsv = image.select(bands).rgbToHsv()
    sharpened = ee.Image.cat([hsv.select('hue'), hsv.select('saturation'), image.select(panchrom)]).hsvToRgb()
    return sharpened

class GeometryNotFoundError(Exception): pass
class VectorNotFoundError(Exception): pass