maped_polyline.py
3.05 KB
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
# -*- coding: utf-8 -*-
title = "Polygon"
tip = "selects a part of a map"
onein = -1
import numpy as np
try: from matplotlib.nxutils import points_inside_poly
except: from nod3tools import points_inside_poly
from guidata.qt.QtGui import QMessageBox, QApplication
from guidata.dataset.datatypes import DataSet
from guidata.dataset.dataitems import (IntItem, StringItem, ChoiceItem, FloatItem, BoolItem)
from guidata.qt.QtCore import Qt
from guiqwt.events import setup_standard_tool_filter, KeyEventMatch
from guiqwt.config import _
def nint(x):
if x > 0: return int(x+0.5)
else: return int(x-0.5)
class NOD3_App:
def __init__(self, parent):
self.parent = parent
self.parent.activateWindow()
def Info(self, msg):
QMessageBox.information(self.parent.parent(), title,
#_(u"Info:")+"\n%s" % str(msg))
str(msg))
def Error(self, msg):
QMessageBox.critical(self.parent.parent(), title,
_(u"Error:")+"\n%s" % str(msg))
def compute_app(self, **args):
class FuncParam(DataSet):
posx = IntItem('center x')
posy = IntItem('center y')
naxis1 = IntItem('shape x')
naxis2 = IntItem('shape y')
self.name = title.replace(" ", "")
if args == {}:
param = FuncParam(_(title), "Use box to select a map extraction:\n")
else:
param = self.parent.ScriptParameter(name, args)
# if no parameter needed set param to None. activate next line
param = None
msg = str("Usage of Application %s:\n \
mouse left button creates polygon lines \n \
keys <Delete> or <Backspace> delete lines one by one \n \
keys <n> or <Enter> or <Return> setup a new polygon \n \
keys <Space> or <Esc> finish the application" % self.name)
self.Info(msg)
self.parent.get_polyline(self.name,
lambda m, p, polygon: self.function(m, p, polygon),
param)
def coordinates2pixel(self, polygon):
polypix = np.copy(polygon)
for n in range(len(polypix)):
for k in range(len(polypix[n])):
l, b = polypix[n][k]
x, y = self.parent.get_pixel_coordinates(l, b)
polypix[n][k] = [nint(x), nint(y)]
return polypix
def function(self, m, p, polygon):
polypix = self.coordinates2pixel(polygon)
ny, nx = m.data.shape
x, y = np.meshgrid(np.arange(nx), np.arange(ny))
x, y = x.flatten(), y.flatten()
points = np.vstack((x,y)).T
mask = []
for n in range(len(polypix)):
mask1 = points_inside_poly(points, polypix[n])
if mask == []:
mask = np.copy(mask1)
else:
mask = np.where(mask, mask, mask1)
mask = mask.reshape((ny, nx))
#m.data = np.where(~mask, m.data, np.nan) # negativ
m.data = np.where(mask, m.data, np.nan)
return m, p