def get_trailing_edge_for_all_cases_at_x_m1():
    import case_dict_overall_correction as case_dict
    cases_df = case_dict.return_case_df()

    case_STE = cases_df[ 
        (cases_df.file == 'data/STE_phi0_alpha0_U20_loc00.dat_rotated.p') &\
        (cases_df.x_loc == -1) ]
    case_z00 = cases_df[ 
        (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc00.dat_rotated.p')&\
        (cases_df.x_loc == -1) ]
    case_z05 = cases_df[ 
        (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc05.dat_rotated.p')&\
        (cases_df.x_loc == -1) ]
    case_z10 = cases_df[ 
        (cases_df.file == \
         'data/Sr20R21_phi0_alpha0_U20_loc10_even_newer.dat.p')&\
        (cases_df.x_loc == -1) ]

    cases = case_STE
    cases = cases.append(case_z00, ignore_index = True)
    cases = cases.append(case_z05, ignore_index = True)
    cases = cases.append(case_z10, ignore_index = True)

    schematic = '/home/carlos/Documents/PhD/Articles/Article_2/'+\
            'Figures/measurement_locations_x0_m2.png'

    write_boundary_layers(cases)
    plot_cases( cases , plot_name = "At_x_m1", schematic = schematic)
def get_streamlined_surface(z_loc = 0):
    import article2_time_averaged_routines as tar
    import case_dict_overall_correction as case_dict
    cases_df = case_dict.return_case_df()

            
    pickle_files = [
        'data/ste_phi0_alpha0_u20_loc00.dat_rotated.p',
        'data/sr20r21_phi0_alpha0_u20_loc00.dat_rotated.p',
        'data/sr20r21_phi0_alpha0_u20_loc05.dat_rotated.p',
        'data/sr20r21_phi0_alpha0_u20_loc10_even_newer.dat.p',
    ]
    plot_names = [
        'surfaceplot_u_straight_edge.png',
        'surfaceplot_u_serrated_z00.png',
        'surfaceplot_u_serrated_z05.png',
        'surfaceplot_u_serrated_z10.png',
    ]

    for pf,pn in zip(
        pickle_files,plot_names
    ):
        df = tar.load_df_from_pickle(pf)

        height_correction = cases_df[ cases_df.file == pf ]\
                .height_correction.values[0]

        angle_correction = cases_df[ cases_df.file == pf ]\
                .rotation.values[0]
        
        streamwise_correction = cases_df[ cases_df.file == pf ]\
                .x_corr.values[0]


        tar.show_streamlined_surface_from_df(
            df                    = df,
            variable              = 'u',
            points                = [],
            mask                  = [],
            height_correction     = height_correction,
            angle_correction      = angle_correction,
            x_max                 = 40,
            x_min                 = -2,
            y_max                 = 6,
            y_min                 = -1,
            streamwise_correction = streamwise_correction,
            plot_name             = pn
        )
def get_trailing_edge_for_all_TR_cases_at_TE_m1():
    import tr_case_dict_overall_correction as case_dict
    cases_df = case_dict.return_case_df()

    case_STE = cases_df[ 
        (cases_df.file == 'data/STE_phi0_alpha0_U20_loc00_tr.dat.p') &\
        (cases_df.x_loc == -1) ]
    case_z00 = cases_df[ 
        (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc00_tr.dat.p')&\
        (cases_df.x_loc == 39) ]
    case_z05 = cases_df[ 
        (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc05_tr.dat.p')&\
        (cases_df.x_loc == 19) ]
    case_z10 = cases_df[ 
        (cases_df.file == \
         'data/Sr20R21_phi0_alpha0_U20_loc10_tr.dat.p')&\
        (cases_df.x_loc == -1) ]
    #case_z00_2 = cases_df[ 
    #    (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc00_tr.dat.p')&\
    #    (cases_df.x_loc == 34) ]
    #case_z05_2 = cases_df[ 
    #    (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc05_tr.dat.p')&\
    #    (cases_df.x_loc == 14) ]
    #case_z10_2 = cases_df[ 
    #    (cases_df.file == \
    #     'data/Sr20R21_phi0_alpha0_U20_loc10_tr.dat.p')&\
    #    (cases_df.x_loc == 3) ]
    #case_z10_2 = cases_df[ 
    #    (cases_df.file == \
    #     'data/Sr20R21_phi0_alpha0_U20_loc10_tr.dat.p')&\
    #    (cases_df.x_loc == -5) ]

    cases = case_STE
    cases = cases.append(case_z00, ignore_index = True)
    cases = cases.append(case_z05, ignore_index = True)
    cases = cases.append(case_z10, ignore_index = True)
    #cases = cases.append(case_z00_2, ignore_index = True)
    #cases = cases.append(case_z05_2, ignore_index = True)
    #cases = cases.append(case_z10_2, ignore_index = True)

    schematic = '/home/carlos/Documents/PhD/Articles/Article_2/'+\
            'Figures/measurement_locations_TE_m2.png'

    write_boundary_layers(cases)
    plot_cases( cases , plot_name = "TR_At_trailing_edge", 
               schematic = schematic, time_resolved = True)
def write_wall_normal_lines_to_csv():
    import pandas as pd
    import article2_time_averaged_routines as tar
    import case_dict_overall_correction as case_dict
    cases_df = case_dict.return_case_df()

    case_STE = cases_df[ 
        (cases_df.file == 'data/STE_phi0_alpha0_U20_loc00.dat_rotated.p') &\
        (cases_df.x_loc == 0) ]

    case_z00_x40 = cases_df[ 
        (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc00.dat_rotated.p')&\
        (cases_df.x_loc == 40) ]
    
    case_z00_x20 = cases_df[ 
        (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc00.dat_rotated.p')&\
        (cases_df.x_loc == 20) ]
    
    case_z00_x00 = cases_df[ 
        (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc00.dat_rotated.p')&\
        (cases_df.x_loc == 0) ]
    
    case_z05 = cases_df[ 
        (cases_df.file == 'data/Sr20R21_phi0_alpha0_U20_loc05.dat_rotated.p')&\
        (cases_df.x_loc == 20) ]

    case_z10 = cases_df[ 
        (cases_df.file == \
         'data/Sr20R21_phi0_alpha0_U20_loc10_even_newer.dat.p')&\
        (cases_df.x_loc == 0) ]

    cases = case_STE
    cases = cases.append(case_z00_x00)
    cases = cases.append(case_z00_x20)
    cases = cases.append(case_z00_x40)
    cases = cases.append(case_z05)
    cases = cases.append(case_z10)

    df_cases = pd.DataFrame()

    for case in cases.iterrows():

        df = tar.get_wall_normal_line(
            pickle_file               = case[1].file,
            x_loc                     = case[1].x_loc,
            variable                  = ['u','v',"u_rms","v_rms",],
            plot                      = False,
            rotation_angle            = case[1].rotation,
            trust_y_min               = case[1].y_trust_min,
            height_correction         = case[1].height_correction,
            streamwise_correction     = case[1].x_corr,
        )
        tar.write_to_csv_with_units(
            df, 
            '{0}_x{1:.2f}_z{2}'.format(
                df.trailing_edge.unique()[0],
                df.x.unique()[0],
                df.z.unique()[0],
            )
        )

        df_cases = df_cases.append(df, ignore_index = True)

    tar.write_to_csv_with_units(
        df_cases, 
        'Serrations_BoundaryLayerData_StereoPIV_v2.csv'
    )