# db.query("""SELECT accName FROM ACCBoundary WHERE accName LIKE '""" + country + """%'""")
# r=db.store_result()
# rr=[rrr[0] for rrr in r.fetch_row(maxrows=0,how=0)]
# db.close()
rr = ['LIRR']
for zone in rr:
print "=============================================="
print " Generating traffic for zone:", zone
print "=============================================="
paras_nav = paras_strategic(zone=zone, mode='navpoints', data_version=data_version) #TODO
name_G = name_net(paras_nav, data_version)
try:
print "Trying to load network from file", _result_dir + '/networks/' + name_G
G = network_whose_name_is(_result_dir + '/networks/' + name_G)
except IOError:
print "Could not load the network, I skip it."
print
continue
with open('../libs/All_shapes_334.pic','r') as f:
all_shapes = pickle.load(f)
boundary = list(all_shapes[zone]['boundary'][0].exterior.coords)
assert boundary[0]==boundary[-1]
with open(_result_dir + '/trajectories/bounds/' + G.name + '_bound_latlon.dat', 'w') as f:
for x, y in boundary:
f.write(str(x) + '\t' + str(y) + '\n')
print "Finding best capacity factor..."
sys.path.insert(1, '..')
sys.path.insert(1, '../abm_strategic')
#from paths import result_dir
from abm_strategic.utilities import network_whose_name_is
from abm_strategic.simulationO import generate_traffic
from libs.general_tools import draw_network_and_patches
import abm_strategic
result_dir = abm_strategic.result_dir
#TODO; polish this.
#G = network_whose_name_is('../networks/D_N44_nairports22_cap_constant_C5_w_coords_Nfp2')
#G = network_whose_name_is('../networks/Real_LF_v5.8_Strong_EXTLFMM_LFMM_2010-5-6+0_d2_cut240.0_directed')
G = network_whose_name_is(result_dir + '/networks/Real_LI_v5.8_Strong_EXTLIRR_LIRR_2010-5-6+0_d2_cut240.0_directed')
name_G = 'Real_LI_v5.8_Strong_EXTLIRR_LIRR_2010-5-6+0_d2_cut240.0_directed'
save_file = result_dir + '/trajectories/M1/trajectories.dat'
#save_file = result_dir + '/trajectories/M1/test_rectificate.dat'
#file_traffic = '../networks/Real_LF_v5.8_Strong_EXTLFMM_LFMM_2010-5-6+0_d2_cut240.0_directed_flights_selected.pic'
file_traffic = result_dir + "/networks/" + name_G + '_flights_selected.pic'
#file_traffic = '../networks/Real_LF_v5.8_Strong_EXTLIRR_LIRR_2010-5-6+0_d2_cut240.0_directed_flights_selected.pic'
#save_file_capacities = '../networks/Real_LF_v5.8_Strong_EXTLFMM_LFMM_2010-5-6+0_d2_cut240.0_directed_flights_selected_capacities.dat'
save_file_capacities = result_dir + '/networks/Real_LI_v5.8_Strong_EXTLIRR_LIRR_2010-5-6+0_d2_cut240.0_directed_flights_selected_capacities.dat'
trajectories = generate_traffic(G, save_file=save_file,
file_traffic=file_traffic,
coordinates=True,
put_sectors=True,
#save_file_capacities=save_file_capacities,
