Final-Project/object_bm.py at master · KatieButler/Final-Project · 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
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
from mpl_toolkits.basemap import Basemap
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from matplotlib.patches import Polygon
from matplotlib.pyplot import figure, show


class ZoomMap:
    def __init__(self):
        """ Creates main figure where subplots will reside """
        self.fig = plt.figure() # creates a matplotlib figure

        self.ax = self.fig.add_subplot(211) # creates main map subplot
        self.ax.set_title("Click to zoom")
        self.my_map = Basemap(projection='merc',
                         lat_0=50,
                         lon_0=-100,
                         resolution = 'l',
                         area_thresh = 100000.0,
                         llcrnrlon=-125,
                         llcrnrlat=24,
                         urcrnrlon=-67,
                         urcrnrlat=50)
        self.my_map.drawcoastlines()
        self.my_map.drawcountries()
        self.my_map.drawmapboundary()
        self.my_map.drawstates()

        self.fig.canvas.draw() # draws the figure

    def panda_to_list(self, file_name, title1):
        """
        The opens a saved csv file and convert it to a panda file. The
        panda file is separated into columns and the columns are made into
        lists. The lists are the output of the funciton.
        """
        datafile = pd.read_csv(file_name) #opens a data file

        # puts the data in a column
        for col in datafile.columns:
            datafile[col] = datafile[col].astype(str)
        column1 = datafile[title1]

        # creates a list from column data in the csv
        list1 = []
        for col_value in column1:
            list1.append(float(col_value))
        return list1


    def in_box(self, event):
        """
        If the event values of x and y are within a reasonable range, this function prints
        the values and a w value of zero. If they're out of range, it just prints their
        values.
        """
        if event.x >= 80 and event.y <= 200:
            print "x=%d, y=%d"%(event.x, event.y)
        else:
            print "x=%d, y=%d w=%d"%(event.x, event.y, 0)

    def on_click(self, event):
        """
        This function clears the figure completely and redraws the main map as a subplot.
        If main map is clicked, a subplot showing a zoomed in version will pop up beneath it.
        If the main map is clicked again, it will zoom in on the newly clicked area.
        """
        self.fig.clf() # Clears the entire figure

        print "CLICKED"

        """ The function only continues if button has been pressed once """
        if event.button != 1:
            return

        self.ax = self.fig.add_subplot(211) # replots the large map subplot
        self.ax.set_title("Click to zoom")
        self.my_map = Basemap(projection='merc',
                         lat_0=50,
                         lon_0=-100,
                         resolution = 'l',
                         area_thresh = 100000.0,
                         llcrnrlon=-125,
                         llcrnrlat=24,
                         urcrnrlon=-67,
                         urcrnrlat=50)
        self.my_map.drawcoastlines()
        self.my_map.drawcountries()
        self.my_map.drawmapboundary()
        self.my_map.drawstates()

        """
        Transfers xdata and ydata from the mouseclick to longitude and latitude coordinates
        """
        x, y = float(event.xdata), float(event.ydata)
        print x, y
        lon = 0
        lat = 0
        lon, lat = self.my_map(x,y,inverse=True)

        ax = self.fig.add_subplot(212) # creates a zoomed subplot using lon,lat coordinates
        ax.set_title("Zoom Map")
        self.zoom_map = Basemap(projection='merc',
                         lat_0=50,
                         lon_0=-100,
                         area_thresh = 10000.0,
                         llcrnrlat=lat-1.5,
                         llcrnrlon=lon-2.75,
                         urcrnrlat=lat+1.5,
                         urcrnrlon=lon+2.75,
                         resolution='i')
        self.zoom_map.drawcoastlines()
        self.zoom_map.drawcountries()
        self.zoom_map.drawmapboundary()
        self.zoom_map.drawstates()

        self.fig.canvas.draw() # draws the figure

    def run(self):
        """
        This functions runs the other functions in the class. When the mouse is clicked,
        first in_box runs then on_click to make the experience interactive
        """
        cid_up = self.fig.canvas.mpl_connect('button_press_event', self.in_box)
        self.fig.canvas.mpl_connect('button_press_event', self.on_click)

        plt.show()

""" Runs the ZoomMap class and its methods """
zm = ZoomMap()
zm.run()