2 |
# Authors: |
# Authors: |
3 |
# Bernhard Herzog <[email protected]> |
# Bernhard Herzog <[email protected]> |
4 |
# Jonathan Coles <[email protected]> |
# Jonathan Coles <[email protected]> |
5 |
|
# Silke Reimer <[email protected]> |
6 |
# |
# |
7 |
# This program is free software under the GPL (>=v2) |
# This program is free software under the GPL (>=v2) |
8 |
# Read the file COPYING coming with Thuban for details. |
# Read the file COPYING coming with Thuban for details. |
25 |
|
|
26 |
import resource |
import resource |
27 |
|
|
28 |
|
from color import Color |
29 |
|
|
30 |
shapetype_names = {SHAPETYPE_POINT: "Point", |
shapetype_names = {SHAPETYPE_POINT: "Point", |
31 |
SHAPETYPE_ARC: "Arc", |
SHAPETYPE_ARC: "Arc", |
56 |
self.issue(LAYER_VISIBILITY_CHANGED, self) |
self.issue(LAYER_VISIBILITY_CHANGED, self) |
57 |
|
|
58 |
def HasClassification(self): |
def HasClassification(self): |
59 |
"""Determine if this layer support classifications.""" |
"""Determine if this layer supports classifications.""" |
60 |
return False |
return False |
61 |
|
|
62 |
def HasShapes(self): |
def HasShapes(self): |
68 |
return self.projection |
return self.projection |
69 |
|
|
70 |
def SetProjection(self, projection): |
def SetProjection(self, projection): |
71 |
"""Set the layer's projection""" |
"""Set the layer's projection.""" |
72 |
self.projection = projection |
self.projection = projection |
73 |
self.changed(LAYER_PROJECTION_CHANGED, self) |
self.changed(LAYER_PROJECTION_CHANGED, self) |
74 |
|
|
75 |
|
def Type(self): |
76 |
|
return "Unknown" |
77 |
|
|
78 |
class Layer(BaseLayer): |
class Layer(BaseLayer): |
79 |
|
|
80 |
"""Represent the information of one geodata file (currently a shapefile) |
"""Represent the information of one geodata file (currently a shapefile) |
175 |
bbox = self.projection.InverseBBox(bbox) |
bbox = self.projection.InverseBBox(bbox) |
176 |
return bbox |
return bbox |
177 |
|
|
178 |
|
def Type(self): |
179 |
|
return self.ShapeType(); |
180 |
|
|
181 |
def ShapesBoundingBox(self, shapes): |
def ShapesBoundingBox(self, shapes): |
182 |
"""Return a bounding box in lat/long coordinates for the given |
"""Return a bounding box in lat/long coordinates for the given |
183 |
list of shape ids. |
list of shape ids. |
234 |
form (minx, miny, maxx, maxy) in unprojected coordinates. |
form (minx, miny, maxx, maxy) in unprojected coordinates. |
235 |
""" |
""" |
236 |
if self.projection is not None: |
if self.projection is not None: |
237 |
bbox = self.projection.ForwardBBox(bbox) |
# Ensure that region lies within the layer's bounding box |
238 |
|
# Otherwise projection of the region would lead to incorrect |
239 |
|
# values. |
240 |
|
clipbbox = self.ClipBoundingBox(bbox) |
241 |
|
bbox = self.projection.ForwardBBox(clipbbox) |
242 |
return self.store.ShapesInRegion(bbox) |
return self.store.ShapesInRegion(bbox) |
243 |
|
|
244 |
def GetClassificationColumn(self): |
def GetClassificationColumn(self): |
314 |
|
|
315 |
return (_("Layer '%s'") % self.Title(), items) |
return (_("Layer '%s'") % self.Title(), items) |
316 |
|
|
317 |
|
def ClipBoundingBox(self, bbox): |
318 |
|
""" Clip bbox to layer's bounding box. |
319 |
|
|
320 |
|
Returns that part of bbox that lies within the layers bounding box. |
321 |
|
If bbox is completely outside of the layers bounding box, bbox is |
322 |
|
returned. It is assumed that bbox has sensible values, i.e. bminx |
323 |
|
< bmaxx and bminy < bmaxy. |
324 |
|
""" |
325 |
|
bminx, bminy, bmaxx, bmaxy = bbox |
326 |
|
lminx, lminy, lmaxx, lmaxy = self.LatLongBoundingBox() |
327 |
|
if bminx > lmaxx or bmaxx < lminx: |
328 |
|
left, right = bminx, bmaxx |
329 |
|
else: |
330 |
|
left = max(lminx, bminx) |
331 |
|
right = min(lmaxx, bmaxx) |
332 |
|
if bminy > lmaxy or bmaxy < lminy: |
333 |
|
bottom, top = bminy, bmaxy |
334 |
|
else: |
335 |
|
bottom = max(lminy, bminy) |
336 |
|
top = min(lmaxy, bmaxy) |
337 |
|
|
338 |
|
return (left, bottom, right, top) |
339 |
|
|
340 |
|
|
341 |
if resource.has_gdal_support(): |
if resource.has_gdal_support(): |
342 |
import gdal |
import gdal |
367 |
|
|
368 |
self.bbox = -1 |
self.bbox = -1 |
369 |
|
|
370 |
|
self.use_mask = True |
371 |
|
|
372 |
|
self.image_info = None |
373 |
|
|
374 |
if resource.has_gdal_support(): |
if resource.has_gdal_support(): |
375 |
# |
# |
376 |
# temporarily open the file so that GDAL can test if it's valid. |
# temporarily open the file so that GDAL can test if it's valid. |
380 |
if dataset is None: |
if dataset is None: |
381 |
raise IOError() |
raise IOError() |
382 |
|
|
383 |
|
# |
384 |
|
# while we have the file, extract some basic information |
385 |
|
# that we can display later |
386 |
|
# |
387 |
|
self.image_info = {} |
388 |
|
|
389 |
|
self.image_info["nBands"] = dataset.RasterCount |
390 |
|
self.image_info["Size"] = (dataset.RasterXSize, dataset.RasterYSize) |
391 |
|
self.image_info["Driver"] = dataset.GetDriver().ShortName |
392 |
|
|
393 |
|
# store some information about the individual bands |
394 |
|
# [min_value, max_value] |
395 |
|
a = self.image_info["BandData"] = [] |
396 |
|
|
397 |
|
for i in range(1, dataset.RasterCount+1): |
398 |
|
band = dataset.GetRasterBand(i) |
399 |
|
a.append(band.ComputeRasterMinMax()) |
400 |
|
|
401 |
self.UnsetModified() |
self.UnsetModified() |
402 |
|
|
403 |
def BoundingBox(self): |
def BoundingBox(self): |
452 |
|
|
453 |
return bbox |
return bbox |
454 |
|
|
455 |
|
def Type(self): |
456 |
|
return "Image" |
457 |
|
|
458 |
def GetImageFilename(self): |
def GetImageFilename(self): |
459 |
return self.filename |
return self.filename |
460 |
|
|
461 |
|
def UseMask(self): |
462 |
|
"""Return True if the mask should be used when rendering the layer.""" |
463 |
|
return self.use_mask |
464 |
|
|
465 |
|
def SetUseMask(self, use): |
466 |
|
"""Set whether to use a mask when render the image. |
467 |
|
|
468 |
|
If the state changes, a LAYER_CHANGED message is sent. |
469 |
|
""" |
470 |
|
if use != self.use_mask: |
471 |
|
self.use_mask = use |
472 |
|
self.changed(LAYER_CHANGED, self) |
473 |
|
|
474 |
|
def ImageInfo(self): |
475 |
|
return self.image_info |
476 |
|
|
477 |
def TreeInfo(self): |
def TreeInfo(self): |
478 |
items = [] |
items = [] |
479 |
|
|