From 767e617a8b48a0d2a406289954652e9f89d9a203 Mon Sep 17 00:00:00 2001
From: "(no author)" <(no author)@8a9318a1-56ba-4d59-b755-99d26321be01>
Date: Thu, 20 Jun 2013 16:40:16 +0000
Subject: [PATCH] adding code to support plotting of geotiffs in the gui
git-svn-id: https://svn.ssec.wisc.edu/repos/glance/trunk@229 8a9318a1-56ba-4d59-b755-99d26321be01
---
pyglance/glance/gui_model.py | 68 +++++++++++++++++++++++++++++++-----
1 file changed, 59 insertions(+), 9 deletions(-)
diff --git a/pyglance/glance/gui_model.py b/pyglance/glance/gui_model.py
index 5120522..4174a22 100644
--- a/pyglance/glance/gui_model.py
+++ b/pyglance/glance/gui_model.py
@@ -9,6 +9,7 @@ Copyright (c) 2011 University of Wisconsin SSEC. All rights reserved.
import logging
import numpy as np
+from os import path
import glance.data as dataobjects
import glance.io as io
@@ -94,6 +95,7 @@ class GlanceGUIModel (object) :
self.dataForm - the form the data should be considered to be
self.useSharedRange - True if images for the original data should be displayed in a range
that includeds the data from both files, False if not
+ self.plotGeoTiffAsRGB - True if multi-channel geotiff images should be treated as RGB or RGBA images
self.hideMismatchNav - True if data corresponding to navigation that's more different
than self.llEpsilon should be hidden when plotting, False if not
@@ -125,14 +127,15 @@ class GlanceGUIModel (object) :
self.fileData[B_CONST] = _FileModelData( )
# general settings
- self.epsilon = 0.0
- self.epsilonPercent = None
- self.llEpsilon = None
- self.imageType = IMAGE_TYPES[0]
- self.colormap = COLORMAP_NAMES[0]
- self.dataForm = SIMPLE_2D
- self.useSharedRange = False
- self.hideMismatchNav = False
+ self.epsilon = 0.0
+ self.epsilonPercent = None
+ self.llEpsilon = None
+ self.imageType = IMAGE_TYPES[0]
+ self.colormap = COLORMAP_NAMES[0]
+ self.dataForm = SIMPLE_2D
+ self.useSharedRange = False
+ self.plotGeoTiffAsRGB = False
+ self.hideMismatchNav = False
# This is obviously only going to work for these two prefixes, would need
# to add a fully formed sub-class to make this more general
@@ -298,6 +301,7 @@ class GlanceGUIModel (object) :
dataListener.updateColormaps(self.colormap, list=COLORMAP_NAMES)
dataListener.updateDataForms(self.dataForm, list=DATA_FORMS)
dataListener.updateUseSharedRange(self.useSharedRange)
+ dataListener.updatePlotGeoTiffAsRGB(self.plotGeoTiffAsRGB)
dataListener.updateHideMismatchNav(self.hideMismatchNav)
self.sendFileSettings(A_CONST)
@@ -371,7 +375,8 @@ class GlanceGUIModel (object) :
def updateSettingsDataSelection (self, newEpsilonValue=np.nan, newImageType=None, newDataForm=None,
newEpsilonPercent=np.nan, newllEpsilon=np.nan,
useSharedRangeForOriginals=None, newColormap=None,
- doHideDataFromMismatchedNav=None,) :
+ doHideDataFromMismatchedNav=None,
+ doPlotGeoTiffAsRGB=None) :
"""
someone has changed one or more of the general settings related data values
@@ -427,6 +432,13 @@ class GlanceGUIModel (object) :
self.useSharedRange = useSharedRangeForOriginals
didUpdate = True
+ # update the geotiff plotting settings if needed
+ if (doPlotGeoTiffAsRGB is not None) and (doPlotGeoTiffAsRGB != self.plotGeoTiffAsRGB) :
+ if doPlotGeoTiffAsRGB is True or doPlotGeoTiffAsRGB is False :
+ LOG.debug("Setting plot geoTiff data as RGB images to: " + str(doPlotGeoTiffAsRGB))
+ self.plotGeoTiffAsRGB = doPlotGeoTiffAsRGB
+ didUpdate = True
+
# update whether or not we'll hide data based on the lon/lat comparison
if (doHideDataFromMismatchedNav is not None) and (doHideDataFromMismatchedNav != self.hideMismatchNav) :
if doHideDataFromMismatchedNav is True or doHideDataFromMismatchedNav is False :
@@ -444,6 +456,7 @@ class GlanceGUIModel (object) :
listener.updateColormaps(self.colormap)
listener.updateDataForms(self.dataForm)
listener.updateUseSharedRange(self.useSharedRange)
+ listener.updatePlotGeoTiffAsRGB(self.plotGeoTiffAsRGB)
listener.updateHideMismatchNav(self.hideMismatchNav)
def updateFileSettings (self, file_prefix, doRestrictRange=None,
@@ -673,6 +686,23 @@ class GlanceGUIModel (object) :
return self.useSharedRange
+ def getDoPlotAsRGB (self, filePrefix) :
+ """
+ get whether or not multi-channel geotiffs should be treated as RGB or RGBA
+ """
+
+ isGeoTiff = False
+ if ( self.fileData[filePrefix].file is not None ) :
+ extension_temp = path.splitext(self.fileData[filePrefix].file.path)[-1]
+ isGeoTiff = (extension_temp == '.tiff') or (extension_temp == '.tif') or (extension_temp == '.tifa')
+
+ LOG.debug ("Checking for file " + str(filePrefix) + " geoTiff status... ")
+ toReturn = self.plotGeoTiffAsRGB and isGeoTiff
+ if toReturn :
+ LOG.debug ("file " + str(filePrefix) + " is a geoTiff and should be plotted as an RGB image.")
+
+ return toReturn
+
def getDoHideDataBasedOnMismatchedNavigation (self) :
"""
get whether or not mapped plots should hide data in places where the
@@ -681,6 +711,26 @@ class GlanceGUIModel (object) :
return self.hideMismatchNav
+ # FUTURE, use this to do more last minute loading?
+ def makeSureVariablesAreAvailable (self, filePrefix, listOfVariableNames) :
+ """
+ given a list of variable names, make sure that they're all loaded
+ and available for use.
+
+ If the model is able to load all the requested variables (or they're
+ already loaded) it will return True, otherwise it will load whichever
+ ones it can and return False.
+ """
+
+ couldLoadAll = True
+
+ for varName in listOfVariableNames :
+ varDataObj = self._load_variable_data(filePrefix, varName) if varName in self.fileData[filePrefix].ALL_VARIABLES else None
+
+ couldLoadAll = couldLoadAll and (varDataObj is not None)
+
+ return couldLoadAll
+
def registerDataListener (self, objectToRegister) :
"""
add the given object to our list of data listeners
--
GitLab