MHN.delete_if_exists(update_link_csv) MHN.delete_if_exists(flag_node_csv) # ----------------------------------------------------------------------------- # Write data relevant to specified year and pass to SAS for processing. # ----------------------------------------------------------------------------- arcpy.AddMessage('\nPreparing {0} network attributes...'.format(build_year)) # Export coding for highway projects completed by scenario year. hwyproj_id_field = MHN.route_systems[MHN.hwyproj][1] year_attr = [hwyproj_id_field,'COMPLETION_YEAR'] year_query = '"COMPLETION_YEAR" <= {0}'.format(build_year) year_view = MHN.make_skinny_table_view(MHN.hwyproj, 'year_view', year_attr, year_query) MHN.write_attribute_csv(year_view, year_csv, year_attr) hwy_projects = MHN.make_attribute_dict(year_view, hwyproj_id_field, attr_list=[]) arcpy.Delete_management(year_view) transact_attr = [hwyproj_id_field,'ACTION_CODE','NEW_DIRECTIONS','NEW_TYPE1','NEW_TYPE2','NEW_AMPM1','NEW_AMPM2','NEW_POSTEDSPEED1', 'NEW_POSTEDSPEED2','NEW_THRULANES1','NEW_THRULANES2','NEW_THRULANEWIDTH1','NEW_THRULANEWIDTH2','ADD_PARKLANES1', 'ADD_PARKLANES2','ADD_SIGIC','ADD_CLTL','ADD_RRGRADECROSS','NEW_TOLLDOLLARS','NEW_MODES','TOD','ABB','REP_ANODE','REP_BNODE'] transact_query = '"{0}" IN (\'{1}\')'.format(hwyproj_id_field, "','".join((hwyproj_id for hwyproj_id in hwy_projects))) transact_view = MHN.make_skinny_table_view(MHN.route_systems[MHN.hwyproj][0], 'transact_view', transact_attr, transact_query) MHN.write_attribute_csv(transact_view, transact_csv, transact_attr) hwy_abb = MHN.make_attribute_dict(transact_view, 'ABB', attr_list=[]) arcpy.Delete_management(transact_view) # Export arc & node attributes of all baselinks and skeletons used in # projects completed by scenario year. network_attr = ['ANODE','BNODE','ABB','DIRECTIONS','TYPE1','TYPE2','AMPM1','AMPM2','POSTEDSPEED1','POSTEDSPEED2', 'THRULANES1','THRULANES2','THRULANEWIDTH1','THRULANEWIDTH2','PARKLANES1','PARKLANES2','BASELINK',
for bus_fc in bus_fc_dict: arcpy.AddMessage('\nIdentifying representative runs from {0}...'.format(bus_fc)) which_bus = bus_fc_dict[bus_fc] rep_runs_dict[which_bus] = {} for tod in out_tod_periods: arcpy.AddMessage('-- TOD {0}...'.format(tod.upper())) # Export header info of bus routes in current TOD. bus_id_field = MHN.route_systems[bus_fc][1] bus_route_attr = [bus_id_field, 'DESCRIPTION', 'MODE', 'VEHICLE_TYPE', 'HEADWAY', 'SPEED', 'ROUTE_ID', 'START'] bus_route_query = MHN.tod_periods[tod][1] bus_route_view = MHN.make_skinny_table_view(bus_fc, 'bus_route_view', bus_route_attr, bus_route_query) MHN.write_attribute_csv(bus_route_view, bus_route_csv, bus_route_attr) selected_bus_routes = MHN.make_attribute_dict(bus_route_view, bus_id_field, attr_list=[]) arcpy.Delete_management(bus_route_view) # Export itineraries for selected runs. bus_order_field = MHN.route_systems[bus_fc][2] bus_itin_attr = [bus_id_field, 'ITIN_A', 'ITIN_B', bus_order_field, 'LAYOVER', 'DWELL_CODE', 'ZONE_FARE', 'LINE_SERV_TIME', 'TTF'] bus_itin_query = ''' "{0}" IN ('{1}') '''.format(bus_id_field, "','".join((bus_id for bus_id in selected_bus_routes))) bus_itin_view = MHN.make_skinny_table_view(MHN.route_systems[bus_fc][0], 'bus_itin_view', bus_itin_attr, bus_itin_query) MHN.write_attribute_csv(bus_itin_view, bus_itin_csv, bus_itin_attr) arcpy.Delete_management(bus_itin_view) # Process exported route & itin tables with gtfs_reformat_feed.sas. sas1_sas = os.path.join(MHN.prog_dir, '{0}.sas'.format(sas1_name)) sas1_output = os.path.join(MHN.temp_dir, 'bus_{0}_runs_{1}.csv'.format(which_bus, tod)) sas1_args = [MHN.prog_dir, bus_route_csv, bus_itin_csv, oneline_itin_txt, feed_groups_txt, sas1_output, tod] MHN.delete_if_exists(sas1_output)
) arcpy.MakeFeatureLayer_management(temp_arcs, bad_arcs_lyr, bad_arcs_query) bad_arcs_count = int(arcpy.GetCount_management(bad_arcs_lyr).getOutput(0)) if bad_arcs_count > 0: arcpy.CopyFeatures_management(bad_arcs_lyr, bad_arcs_shp) MHN.die('Some arcs are missing required attributes. Check {0} for specific arcs.'.format(bad_arcs_shp)) raise arcpy.ExecuteError else: arcpy.Delete_management(bad_arcs_lyr) arcpy.AddMessage('-- All arcs have all required attributes') # ----------------------------------------------------------------------------- # Copy node attributes to memory, for fast access. # ----------------------------------------------------------------------------- current_nodes_dict = MHN.make_attribute_dict(MHN.node, 'NODE', ['POINT_X','POINT_Y']) # ----------------------------------------------------------------------------- # Generate nodes from arcs, to check for changes and errors. # ----------------------------------------------------------------------------- # Generate ANODES, including a copy with no BNODE field. anodes = os.path.join(MHN.mem, 'anodes') anodes_copy = os.path.join(MHN.mem, 'anodes_copy') arcpy.FeatureVerticesToPoints_management(temp_arcs, anodes, 'START') arcpy.CopyFeatures_management(anodes, anodes_copy) arcpy.DeleteField_management(anodes_copy, ['BNODE']) # Generate BNODES, including a copy with no ANODE field. bnodes = os.path.join(MHN.mem, 'bnodes') bnodes_copy = os.path.join(MHN.mem, 'bnodes_copy')
MHN.die( 'The following highway projects have no completion year: {0}'.format( ', '.join(invalid_hwyproj))) # ----------------------------------------------------------------------------- # Export highway project coding info to determine future arc availability. # ----------------------------------------------------------------------------- hwyproj_id_field = MHN.route_systems[MHN.hwyproj][1] # Export projects with valid completion years. year_attr = (hwyproj_id_field, 'COMPLETION_YEAR') year_query = '{0} <= {1}'.format("COMPLETION_YEAR", MHN.max_year) year_view = MHN.make_skinny_table_view(MHN.hwyproj, 'year_view', year_attr, year_query) MHN.write_attribute_csv(year_view, year_csv, year_attr) projects = MHN.make_attribute_dict(year_view, hwyproj_id_field, attr_list=[]) arcpy.Delete_management(year_view) # Export coding for valid projects. transact_attr = (hwyproj_id_field, 'ABB', 'ACTION_CODE', 'NEW_DIRECTIONS') transact_query = ''' "{0}" IN ('{1}') '''.format( hwyproj_id_field, "','".join((hwyproj_id for hwyproj_id in projects))) transact_view = MHN.make_skinny_table_view(MHN.route_systems[MHN.hwyproj][0], 'transact_view', transact_attr, transact_query) MHN.write_attribute_csv(transact_view, transact_csv, transact_attr) project_arcs = MHN.make_attribute_dict(transact_view, 'ABB', attr_list=[]) arcpy.Delete_management(transact_view) # Export base year arc attributes. network_attr = ('ANODE', 'BNODE', 'BASELINK', 'ABB', 'DIRECTIONS', 'TYPE1',
invalid_hwyproj = MHN.get_yearless_hwyproj() if invalid_hwyproj: MHN.die("The following highway projects have no completion year: {0}".format(", ".join(invalid_hwyproj))) # ----------------------------------------------------------------------------- # Export highway project coding info to determine future arc availability. # ----------------------------------------------------------------------------- hwyproj_id_field = MHN.route_systems[MHN.hwyproj][1] # Export projects with valid completion years. year_attr = (hwyproj_id_field, "COMPLETION_YEAR") year_query = "{0} <= {1}".format("COMPLETION_YEAR", MHN.max_year) year_view = MHN.make_skinny_table_view(MHN.hwyproj, "year_view", year_attr, year_query) MHN.write_attribute_csv(year_view, year_csv, year_attr) projects = MHN.make_attribute_dict(year_view, hwyproj_id_field, attr_list=[]) arcpy.Delete_management(year_view) # Export coding for valid projects. transact_attr = (hwyproj_id_field, "ABB", "ACTION_CODE", "NEW_DIRECTIONS") transact_query = """ "{0}" IN ('{1}') """.format(hwyproj_id_field, "','".join((hwyproj_id for hwyproj_id in projects))) transact_view = MHN.make_skinny_table_view( MHN.route_systems[MHN.hwyproj][0], "transact_view", transact_attr, transact_query ) MHN.write_attribute_csv(transact_view, transact_csv, transact_attr) project_arcs = MHN.make_attribute_dict(transact_view, "ABB", attr_list=[]) arcpy.Delete_management(transact_view) # Export base year arc attributes. network_attr = ( "ANODE",
elif node > MHN.max_poe and zone == 0: zoned_node[1] = 9999 zoned_node[3] = 11 zoned_nodes_cursor.updateRow(zoned_node) elif node < MHN.min_poe and node != zone: arcpy.AddWarning('-- WARNING: Zone {} centroid is in zone {}! Please verify that this is intentional.'.format(node, zone)) else: pass arcpy.AddMessage('-- Node {}, {}, {} & {} fields recalculated'.format(MHN.zone_attr, MHN.subzone_attr, MHN.capzone_attr, MHN.imarea_attr)) # Calculate arc ANODE and BNODE values. anodes_id = os.path.join(MHN.mem, 'anodes_id') bnodes_id = os.path.join(MHN.mem, 'bnodes_id') arcpy.Identity_analysis(anodes, new_nodes, anodes_id) arcpy.Identity_analysis(bnodes, new_nodes, bnodes_id) anodes_id_dict = MHN.make_attribute_dict(anodes_id, 'ORIG_FID', ['NODE']) bnodes_id_dict = MHN.make_attribute_dict(bnodes_id, 'ORIG_FID', ['NODE']) with arcpy.da.UpdateCursor(temp_arcs, ['OID@', 'ANODE', 'BNODE']) as arcs_cursor: for arc in arcs_cursor: OID = arc[0] old_a = arc[1] old_b = arc[2] new_a = anodes_id_dict[OID]['NODE'] new_b = bnodes_id_dict[OID]['NODE'] if new_a != old_a or new_b != old_b: arc[1] = new_a arc[2] = new_b arcs_cursor.updateRow(arc) arcpy.Delete_management(anodes) arcpy.Delete_management(anodes_id) arcpy.Delete_management(bnodes)
arcpy.MakeFeatureLayer_management(temp_arcs, bad_arcs_lyr, bad_arcs_query) bad_arcs_count = int(arcpy.GetCount_management(bad_arcs_lyr).getOutput(0)) if bad_arcs_count > 0: arcpy.CopyFeatures_management(bad_arcs_lyr, bad_arcs_shp) MHN.die( 'Some arcs are missing required attributes. Check {0} for specific arcs.' .format(bad_arcs_shp)) raise arcpy.ExecuteError else: arcpy.Delete_management(bad_arcs_lyr) arcpy.AddMessage('-- All arcs have all required attributes') # ----------------------------------------------------------------------------- # Copy node attributes to memory, for fast access. # ----------------------------------------------------------------------------- current_nodes_dict = MHN.make_attribute_dict(MHN.node, 'NODE', ['POINT_X', 'POINT_Y']) # ----------------------------------------------------------------------------- # Generate nodes from arcs, to check for changes and errors. # ----------------------------------------------------------------------------- # Generate ANODES, including a copy with no BNODE field. anodes = os.path.join(MHN.mem, 'anodes') anodes_copy = os.path.join(MHN.mem, 'anodes_copy') arcpy.FeatureVerticesToPoints_management(temp_arcs, anodes, 'START') arcpy.CopyFeatures_management(anodes, anodes_copy) arcpy.DeleteField_management(anodes_copy, ['BNODE']) # Generate BNODES, including a copy with no ANODE field. bnodes = os.path.join(MHN.mem, 'bnodes') bnodes_copy = os.path.join(MHN.mem, 'bnodes_copy') arcpy.FeatureVerticesToPoints_management(temp_arcs, bnodes, 'END')