input_filepath = args.input_filepath
out_directory = args.output_directory
code_list_filepath = args.code_list_filepath
plant_spacing = float(args.plant_spacing)
rows, geo_images = unpickle_stage2_output(input_filepath)
if len(rows) == 0 or len(geo_images) == 0:
print "No rows or no geo images could be loaded from {}".format(input_filepath)
sys.exit(ExitReason.no_rows)
if not os.path.exists(code_list_filepath):
print "Code list file doesn't exist {}".format(code_list_filepath)
sys.exit(ExitReason.bad_arguments)
code_listings, _ = parse_code_listing_file(code_list_filepath)
code_listings = make_code_listings_unique(code_listings)
all_segments = all_segments_from_rows(rows)
groups = all_groups_from_segments(all_segments)
single_segments = []
group_segments = []
for segment in all_segments:
if segment.start_code.type == 'SingleCode':
single_segments.append(segment)
elif segment.start_code.type == 'GroupCode':
group_segments.append(segment)
single_codes = [seg.start_code for seg in single_segments]
def stage2_group_codes(**args):
'''
Group codes into rows.
args should match the names and descriptions of command line parameters,
but unlike command line, all arguments must be present.
'''
# Copy args so we can archive them to a file when function is finished.
args_copy = args.copy()
# Convert arguments to local variables of the correct type.
input_directory = args.pop('input_directory')
field_direction = float(args.pop('field_direction'))
output_directory = args.pop('output_directory')
num_rows_per_pass = int(args.pop('num_rows_per_pass'))
code_list_filepath = args.pop('code_list_filepath')
code_modifications_filepath = args.pop('code_modifications_filepath')
if len(args) > 0:
print "Unexpected arguments provided: {}".format(args)
return ExitReason.bad_arguments
geo_images, all_codes = unpickle_stage1_output(input_directory)
print 'Found {} codes in {} geo images.'.format(len(all_codes), len(geo_images))
if len(geo_images) == 0 or len(all_codes) == 0:
print "Couldn't load any geo images or codes from input directory {}".format(input_directory)
return ExitReason.no_geo_images
if code_modifications_filepath != 'none':
if not os.path.exists(code_modifications_filepath):
print "Provided code modification file {} doesn't exist".format(code_modifications_filepath)
return ExitReason.no_geo_images
modifications_out_directory = os.path.join(output_directory, 'modifications')
code_modifications = parse_code_modifications_file(code_modifications_filepath)
geo_images, all_codes = apply_code_modifications(code_modifications, geo_images, all_codes, modifications_out_directory)
# Merge items so they're unique. One code references other instances of that same code.
merged_codes = merge_items(all_codes, max_distance=500)
print '{} unique codes.'.format(len(merged_codes))
row_codes = [code for code in merged_codes if code.type.lower() == 'rowcode']
group_codes = [code for code in merged_codes if code.type.lower() == 'groupcode']
single_codes = [code for code in merged_codes if code.type.lower() == 'singlecode']
'''
if row_labeling_scheme == 0:
grouped_row_codes = group_row_codes(row_codes)
if len(grouped_row_codes) == 0:
print "No row codes found. Exiting"
return ExitReason.no_rows
display_row_info(grouped_row_codes)
rows = create_rows(grouped_row_codes, field_direction)
up_row_nums, back_row_nums = associate_row_numbers_with_up_back_rows()
assign_rows_a_direction(rows, up_row_nums, back_row_nums)
field_passes = [rows[x:x+2] for x in xrange(0, len(rows), 2)]
elif row_labeling_scheme == 1:
grouped_row_codes = group_row_codes_by_pass_name(row_codes)
rows, field_passes = create_rows_and_field_passes_by_pass_codes(grouped_row_codes, field_direction)
elif row_labeling_scheme == 2:
'''
grouped_row_codes = group_row_codes_by_row_name(row_codes)
if len(grouped_row_codes) == 0:
print "No row codes found. Exiting"
return ExitReason.no_rows
display_row_info(grouped_row_codes)
rows, field_passes = create_rows_and_field_passes_by_row_codes(grouped_row_codes, field_direction, num_rows_per_pass)
if len(rows) == 0:
print "No complete rows found. Exiting."
return ExitReason.no_rows
print sorted([r.number for r in rows], key=lambda r: r)
print "Calculating field positions."
calculate_field_positions_and_range(rows, merged_codes, all_codes, geo_images)
for code in merged_codes:
code.refresh_fields()
print "Calculating projections to nearest row"
codes_with_projections = calculate_projection_to_nearest_row(group_codes + single_codes, rows)
print "Creating segments"
group_segments, special_segments = create_segments(codes_with_projections, rows)
print "Organizing segments"
start_segments, middle_segments, end_segments, single_segments = organize_group_segments(group_segments)
if len(middle_segments) > 0:
print "Middle segments that span entire row aren't supported right now. Exiting"
return ExitReason.operation_not_supported
print "Forming groups"
groups = complete_groups(end_segments, single_segments, field_passes, num_rows_per_pass)
handle_single_segments(single_segments, groups)
# Add in information about max number of plants and optional alternate ids.
if code_list_filepath != 'none':
if not os.path.exists(code_list_filepath):
print "Code list file doesn't exist {}".format(code_list_filepath)
return ExitReason.bad_arguments
else:
code_listings, alternate_ids_included = parse_code_listing_file(code_list_filepath)
print "Applying code listings"
apply_code_listings(code_listings, groups, alternate_ids_included)
display_segment_info(group_segments, special_segments, groups)
if not os.path.exists(output_directory):
os.makedirs(output_directory)
dump_filename = "stage2_output_{}_{}.s2".format(int(geo_images[0].image_time), int(geo_images[-1].image_time))
print "Serializing {} rows and {} geo images to {}.".format(len(rows), len(geo_images), dump_filename)
pickle_results(dump_filename, output_directory, rows, geo_images)
# Write arguments out to file for archiving purposes.
args_filename = "stage2_args_{}_{}.csv".format(int(geo_images[0].image_time), int(geo_images[-1].image_time))
write_args_to_file(args_filename, output_directory, args_copy)