C:/work/RevealedSystems/EyeSpy-RS/eyespy/src/objects/gui/mainFrameMap.py

Go to the documentation of this file.

# Copyright (C) 2008-2009  Mark Hanegraaff
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import matplotlib
from objects.logic.common.logger import logger
from objects.logic.common.exception import exception
from matplotlib.toolkits.basemap import Basemap


import pylab as p

matplotlib.interactive(False)
matplotlib.use('WXAgg')



## matplotlib Basemap container. Contains all relevant code needed to create and draw the actual map
class mainFrameMap():
    def __init__(self, figAxesWrapper):
        self.fig_axes = figAxesWrapper
    
    
    ## Creates a new map object
    # @param mapDetail Map Detail Level 
    # @param drawCSBorders True if the map should display country and state borders
    # @param drawRivers True if the map should display rivers  
    def createMap(self, mapDetail, drawCSBorders, drawRivers):
        
        if mapDetail == True:
            res = 'l'
        else:
            res = 'c'
                                    
        try: 
            self.main_map = Basemap(-180.,-80.,180.,80.,\
                                    resolution=res,area_thresh=10000.,projection='merc',\
                                    lon_0=0.,lat_ts=20.,\
                                    ax=self.fig_axes.axes)


            
        except IOError, mapCreationError:
            e = exception(str(mapCreationError ), True)
            raise e
  
        # draw coastlines, country boundaries, fill continents.
        self.main_map.drawcoastlines(linewidth=0.5)
        if drawCSBorders == True:            
            self.main_map.drawcountries()
            self.main_map.drawstates(linewidth=0.1)
                    
        if drawRivers == True:
            self.main_map.drawrivers(color='b',linewidth=0.05)
            
        self.main_map.fillcontinents(color='beige')        
        self.main_map.drawmapboundary()


    ## Clear all Endpoints from the map. Essentially clears all circles and lines.
    def clearAllMapElements(self):
        # erase map points
        self.fig_axes.axes.lines = []            
        # erase text
        self.fig_axes.axes.texts = []
    
    
    ## Map Draw Cycles. Essentially updates all the graphical Endpoints onto the map.
    # @param entityDict Dictionary of Endpoints to be displayed onto the Map
    # @param displayer entityDisplayer derived object which knows how to draw the Endpoints onto the screen
    # @param showLines True if this redraw cycle should include lines    
    # @param showLabels True if this redraw cycle should include labels 
    # @param homeLon Longitude of the home location marker
    # @param homeLat Latitude of the home location marker
    def draw(self, entityDict, displayer, 
                 showLines, showLabels,
                 homeLon, homeLat):
        
        lonList = list()
        latList = list()
        
        markerList = list()
        lineList   = list()

        self.clearAllMapElements()
        
        # this is a little hoakey... but it works.                   
        textYOffsetMagicNum = int(self.fig_axes.getZoomFactor() / 100)
        
        displayedItems = 0
        
        for key, entList in entityDict.iteritems():                            
            
            # markers
            if len(entList) > 1:
                displayedItems += len(entList)
                displayAttributes = displayer.getAttrFromList(entList)
            else:               
                displayedItems += 1     
                displayAttributes = displayer.getAttrFromEntity(entList[0])
            
            longitude = key[0]
            latitude  = key[1]
            
            lonList.append(longitude)                
            latList.append(latitude)
            
            xMap,yMap = self.main_map((lonList), (latList))
            x,y = self.main_map((longitude), (latitude))
            
            markerList.append(xMap)
            markerList.append(yMap)
            
            markerList.append(displayAttributes["marker"])                

            lonList = list()
            latList = list()
            
            if showLines == True:                                                      
                #draw lines
                line_x,line_y = self.main_map.gcpoints(homeLon, homeLat,longitude,latitude,2)
                
                self.main_map.plot(line_x,line_y, "k" + displayAttributes["lineStyle"], linewidth=displayAttributes["lineWidth"],antialiased=True, alpha=displayAttributes["lineAlpha"])                
                
            #draw text
            if showLabels == True: 
                self.fig_axes.axes.text(x, y + textYOffsetMagicNum, displayAttributes["label"], fontsize=10, 
                                         horizontalalignment='center')

            self.main_map.plot(xMap, yMap, displayAttributes["marker"], markersize=displayAttributes["markerSize"], color=displayAttributes["markerColor"] )
        
        #draw markers
        #self.main_map.plot(*markerList)        
                
        # draw self -- surely this can be done better
        selfX, selfY = self.main_map(homeLon, homeLat)
        
        selfXLst= []
        selfYLst = []
        selfXLst.append(selfX)
        selfYLst.append(selfY)
        self.main_map.plot(selfXLst,selfYLst,'b^',label='', markersize=8)
                
        
        self.fig_axes.resetCurrentZoom()
        
        return displayedItems

    
    
    

---