示例#1
0
def format_data(data1, data2, dim):
    '''
    format_data is used to take data from the laser bench and return specially
    formatted data for use by get_slope and get_shift. This program assumes
    that the [:,1] dimension of the data corresponds to ring radius and that
    the data contains both negative and positive angles.
    '''
    
    (OneInTwo, TwoInOne) = ADE.multi_where(data1[:,0], data2[:,0])

    data1Overlap = data1[OneInTwo]
    data2Overlap = data2[TwoInOne]

    data1GT = data1Overlap[:,dim][np.where(data1Overlap[:,0] >= 0)]
    data1LT = data1Overlap[:,dim][np.where(data1Overlap[:,0] < 0)]
    
    data2GT = data2Overlap[:,dim][np.where(data2Overlap[:,0] >= 0)]
    data2LT = data2Overlap[:,dim][np.where(data2Overlap[:,0] < 0)]

    data1GT.sort()
    data1LT.sort()
    
    data2GT.sort()
    data2LT.sort()

    return (np.array([data1GT, data2GT]), np.array([data1LT,data2LT]))