hospitals.py

import numpy
from popdata import *
from collections import deque

"""
Comment
"""

arr = numpy.load('node.npy')
pts = None
with open('hospitals.dat', 'r') as f:
    pts = [[float(d) for d in line.split(':')] for line in f]
q = deque()
for node in arr:
    for p in pts:
        if abs(p[0] - node['lat']) < (km_in_deg/2) and abs(p[1] - node['lon']) < (km_in_deg/2):
            q.append((node['nid'], 0))
while q:
    (nid, dist) = q.popleft()
    if arr[nid]['dHosp'] > dist:
        arr[nid]['dHosp'] = dist
        for nbor in [arr[nid][x] for x in ['Nnid', 'Snid', 'Enid', 'Wnid']]:
            q.append((nbor, dist + 1))
numpy.save('node.npy', arr)
emptyval = -1000.0
exportGeoTIFF('hosp_heat.tif', arr, roi, 'dHosp', emptyval)