示例#1
0
def create_rows_and_field_passes_by_row_codes(grouped_row_codes, field_direction, rows_per_pass):
    
    rows = []
    
    for pass_start_code, pass_end_code in grouped_row_codes:
        
        field_start_code, field_end_code = orient_items(pass_start_code, pass_end_code, field_direction)
    
        if pass_start_code is field_start_code:
            row_direction = 'up'
        else:
            row_direction = 'back'
            
        new_row = Row(field_start_code, field_end_code, direction=row_direction)
        rows.append(new_row)
        
    # Sort row number so can group them into passes.
    sorted_rows = sorted(rows, key=lambda row: row.number)
            
    # Current pass number to assign rows to.  Passes are indexed off 1.
    current_pass_num = 1
            
    field_passes = defaultdict(list)
    for row_idx, row in enumerate(sorted_rows):

        if row_idx == 0:
            current_direction = row.direction
            
        num_rows_in_current_pass = len(field_passes[current_pass_num])
            
        end_of_pass = (row.direction != current_direction) or (num_rows_in_current_pass >= rows_per_pass)
            
        if end_of_pass:
            
            if num_rows_in_current_pass < rows_per_pass:
                print 'WARNING - pass {} only has {} rows'.format(current_pass_num, num_rows_in_current_pass)
            
            current_pass_num += 1

        field_passes[current_pass_num].append(row)
        current_direction = row.direction
        
    # Convert field passes to a list and make sure it's sorted.
    sorted_field_passes = []
    for sorted_pass_num in sorted(field_passes.keys()):
        sorted_field_passes.append(field_passes[sorted_pass_num])

    field_passes = sorted_field_passes
    
    return rows, field_passes
示例#2
0
def create_rows(grouped_row_codes, field_direction):
    rows = []
    for row_number, codes in grouped_row_codes.iteritems():
        if len(codes) == 1:
            print "Only found 1 code for row {}".format(row_number)
        elif len(codes) > 2:
            print "Found {} codes for row {} with positions: ".format(len(codes), row_number)
            for code in codes:
                print "\t{}".format(code.position)
        else:
            # Create row objects with start/end codes.
            code1, code2 = codes
            start_code, end_code = orient_items(code1, code2, field_direction)
            if start_code and end_code:
                rows.append(Row(start_code, end_code)) 
    return rows
示例#3
0
def create_rows_and_field_passes_by_pass_codes(grouped_row_codes, field_direction):
    
    rows = []
    field_passes = defaultdict(list)
    for pass_start_code, pass_end_code in grouped_row_codes:
        
        field_start_code, field_end_code = orient_items(pass_start_code, pass_end_code, field_direction)
    
        if pass_start_code is field_start_code:
            row_direction = 'up'
        else:
            row_direction = 'back'
            
        new_row = Row(field_start_code, field_end_code, direction=row_direction)
        rows.append(new_row)
        
        # At this point row numbers should be pass numbers so can use them to group rows into passes.
        field_passes[new_row.number].append(new_row)
        
        # Convert pass number on codes to row numbers
        pass_side = pass_start_code.name[-1].lower()
        pass_start_code.assigned_row = pass_start_code.row * 2 - 1
        pass_end_code.assigned_row = pass_end_code.row * 2 - 1
        if (pass_side == 'r' and row_direction == 'up') or (pass_side == 'l' and row_direction == 'back'):
            pass_start_code.assigned_row += 1
            pass_end_code.assigned_row += 1
            
    sorted_field_passes = []
    for sorted_pass_num in sorted(field_passes.keys()):
        rows_in_pass = field_passes[sorted_pass_num]
        if len(rows_in_pass) == 1:
            print "Only 1 row found in pass {}".format(sorted_pass_num)
        elif len(rows_in_pass) > 2:
            print "Error: more than 2 rows in pass {}".format(sorted_pass_num)
            sys.exit(-1)
        
        sorted_field_passes.append(rows_in_pass)
            
    field_passes = sorted_field_passes
    
    return rows, field_passes