import sys
import fileinput
import itertools
import pickle
from locations import LocationGraph
#python build_network.py out/user_checkins.out out/austin.out out/coordinates.p out/gowalla_net
#users = pickle.load(open("out/" + sys.argv[1], "rb"))
users = {}
austin = {}
coords = pickle.load(open(sys.argv[3], "rb"))
out = sys.argv[4]
Gowalla = LocationGraph()
fmt = "%Y-%m-%d %H:%M:%S"
austin_weight = .24
epsilon = 1
if len(sys.argv) < 2:
print "Filename required"
else:
for line in fileinput.input(sys.argv[1]):
line = line.split("|")
u = line.pop(0)
users[u] = []
while len(line) > 0:
entry = line.pop(0)
entry = entry.split(">")
l = users[u]
l.append(entry[0])
#locations=pickle.load(open(sys.argv[7], 'rb'))
states = pickle.load(open(sys.argv[8], 'rb'))
#need to update with actual census data
population = {}
census = pickle.load(open(sys.argv[9], 'rb'))
total_pop = 0
for p in census:
total_pop += census[p]
for p in census:
population[p] = (census[p] / float(total_pop)) * n
#print population
network = LocationGraph()
network.load(sys.argv[10])
print "Dictionaries loaded."
params = str(beta) + "-" + str(shape) + "-" + str(scale)
matrix = open(sys.argv[11] + "_" + params + ".out", 'w')
countID = 0
infected = {}
recovered = []
def genID():
global countID
countID += 1
return countID
