def doeq(): import autoeq as eq eq.batch_processing(input_dir=filename, output_dir='myresult/' + filename, standardize_input=True, compensation='zero.csv', equalize=True, show_plot=showplot) print('===============================================') print('Graphic EQ generated. Check myresult/' + filename)
def main(): arg_parser = argparse.ArgumentParser() arg_parser.add_argument('--new_only', action='store_true', help='Process only new measurements?') arg_parser.add_argument('--innerfidelity', action='store_true', help='Process Innerfidelity measurements?') arg_parser.add_argument('--headphonecom', action='store_true', help='Process Headphone.com measurements?') arg_parser.add_argument('--oratory1990', action='store_true', help='Process oratory1990 measurements?') arg_parser.add_argument('--rtings', action='store_true', help='Process Rtings measurements?') arg_parser.add_argument('--referenceaudioanalyzer', action='store_true', help='Process Reference Audio Analyzer measurements?') arg_parser.add_argument('--crinacle', action='store_true', help='Process Crinacle measurements?') arg_parser.add_argument('--onear', action='store_true', help='Process on-ear measurements?') arg_parser.add_argument('--inear', action='store_true', help='Process in-ear measurements?') arg_parser.add_argument('--earbud', action='store_true', help='Process ear bud measurements?') cli_args = arg_parser.parse_args() new_only = bool(cli_args.new_only) innerfidelity = bool(cli_args.innerfidelity) headphonecom = bool(cli_args.headphonecom) oratory1990 = bool(cli_args.oratory1990) rtings = bool(cli_args.rtings) referenceaudioanalyzer = bool(cli_args.referenceaudioanalyzer) crinacle = bool(cli_args.crinacle) onear = bool(cli_args.onear) inear = bool(cli_args.inear) earbud = bool(cli_args.earbud) if not innerfidelity and not headphonecom and not oratory1990 and not rtings and not referenceaudioanalyzer and not crinacle: innerfidelity = True headphonecom = True oratory1990 = True rtings = True referenceaudioanalyzer = True crinacle = True if not onear and not inear and not earbud: onear = True inear = True earbud = True innerfidelity_overear = os.path.join(ROOT_DIR, 'measurements', 'innerfidelity', 'resources', 'innerfidelity_harman_over-ear_2018_wo_bass.csv') innerfidelity_inear = os.path.join(ROOT_DIR, 'measurements', 'innerfidelity', 'resources', 'innerfidelity_harman_in-ear_2019v2_wo_bass.csv') headphonecom_overear = os.path.join(ROOT_DIR, 'measurements', 'headphonecom', 'resources', 'headphonecom_harman_over-ear_2018_wo_bass.csv') headphonecom_inear = os.path.join(ROOT_DIR, 'measurements', 'headphonecom', 'resources', 'headphonecom_harman_in-ear_2019v2_wo_bass.csv') rtings_overear = os.path.join(ROOT_DIR, 'measurements', 'rtings', 'resources', 'rtings_harman_over-ear_2018_wo_bass.csv') rtings_inear = os.path.join(ROOT_DIR, 'measurements', 'rtings', 'resources', 'rtings_harman_in-ear_2019v2_wo_bass.csv') harman_inear = os.path.join(ROOT_DIR, 'compensation', 'harman_in-ear_2019v2_wo_bass.csv') harman_overear = os.path.join(ROOT_DIR, 'compensation', 'harman_over-ear_2018_wo_bass.csv') crinacle_overear = os.path.join(ROOT_DIR, 'measurements', 'crinacle', 'resources', 'crinacle_over-ear.csv') #measurements/referenceaudioanalyzer/resources/referenceaudioanalyzer_hdm1_harman_over-ear_2018_wo_bass.csv raa_hdmx = os.path.join(ROOT_DIR, 'measurements', 'referenceaudioanalyzer', 'resources', 'referenceaudioanalyzer_hdm-x_harman_over-ear_2018_wo_bass.csv') raa_hdm1 = os.path.join(ROOT_DIR, 'measurements', 'referenceaudioanalyzer', 'resources', 'referenceaudioanalyzer_hdm1_harman_over-ear_2018_wo_bass.csv') raa_siec = os.path.join(ROOT_DIR, 'measurements', 'referenceaudioanalyzer', 'resources', 'referenceaudioanalyzer_siec_harman_in-ear_2019v2_wo_bass.csv') eq_kwargs = { 'equalize': True, 'parametric_eq': True, 'max_filters': [5, 5], 'ten_band_eq': True, 'new_only': new_only, 'convolution_eq': True, 'fs': [44100, 48000] } onear_kwargs = eq_kwargs.copy() onear_kwargs.update({'bass_boost_gain': 4.0}) inear_kwargs = eq_kwargs.copy() inear_kwargs.update({'bass_boost_gain': 6.0}) earbud_kwargs = eq_kwargs.copy() earbud_kwargs.update({'bass_boost_gain': 0.0}) if innerfidelity: if onear: print('\nProcessing Innerfidelity on-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'innerfidelity', 'data', 'onear'), output_dir=os.path.join(ROOT_DIR, 'results', 'innerfidelity', 'innerfidelity_harman_over-ear_2018'), compensation=innerfidelity_overear, **onear_kwargs ) if inear: print('\nProcessing Innerfidelity in-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'innerfidelity', 'data', 'inear'), output_dir=os.path.join(ROOT_DIR, 'results', 'innerfidelity', 'innerfidelity_harman_in-ear_2019v2'), compensation=innerfidelity_inear, **inear_kwargs ) if earbud: print('\nProcessing Innerfidelity earbud measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'innerfidelity', 'data', 'earbud'), output_dir=os.path.join(ROOT_DIR, 'results', 'innerfidelity', 'innerfidelity_harman_in-ear_2019v2'), compensation=innerfidelity_inear, **earbud_kwargs ) if headphonecom: if onear: print('\nProcessing Headphone.com on-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'headphonecom', 'data', 'onear'), output_dir=os.path.join(ROOT_DIR, 'results', 'headphonecom', 'headphonecom_harman_over-ear_2018'), compensation=headphonecom_overear, **onear_kwargs ) if inear: print('\nProcessing Headphone.com in-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'headphonecom', 'data', 'inear'), output_dir=os.path.join(ROOT_DIR, 'results', 'headphonecom', 'headphonecom_harman_in-ear_2019v2'), compensation=headphonecom_inear, **inear_kwargs ) if earbud: print('\nProcessing Headphone.com earbud measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'headphonecom', 'data', 'earbud'), output_dir=os.path.join(ROOT_DIR, 'results', 'headphonecom', 'headphonecom_harman_in-ear_2019v2'), compensation=headphonecom_inear, **earbud_kwargs ) if oratory1990: if onear: print('\nProcessing oratory1990 on-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'oratory1990', 'data', 'onear'), output_dir=os.path.join(ROOT_DIR, 'results', 'oratory1990', 'harman_over-ear_2018'), compensation=harman_overear, **onear_kwargs ) if inear: print('\nProcessing oratory1990 in-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'oratory1990', 'data', 'inear'), output_dir=os.path.join(ROOT_DIR, 'results', 'oratory1990', 'harman_in-ear_2019v2'), compensation=harman_inear, **inear_kwargs ) if earbud: print('\nProcessing oratory1990 ear bud measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'oratory1990', 'data', 'earbud'), output_dir=os.path.join(ROOT_DIR, 'results', 'oratory1990', 'harman_in-ear_2019v2'), compensation=harman_inear, **earbud_kwargs ) if rtings: if onear: # Rtings on-ear Avg print('\nProcessing Rtings on-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'rtings', 'data', 'onear'), output_dir=os.path.join(ROOT_DIR, 'results', 'rtings', 'rtings_harman_over-ear_2018'), compensation=rtings_overear, **onear_kwargs ) if inear: print('\nProcessing Rtings in-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'rtings', 'data', 'inear'), output_dir=os.path.join(ROOT_DIR, 'results', 'rtings', 'rtings_harman_in-ear_2019v2'), compensation=rtings_inear, **inear_kwargs ) if earbud: print('\nProcessing Rtings earbud measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'rtings', 'data', 'earbud'), output_dir=os.path.join(ROOT_DIR, 'results', 'rtings', 'rtings_harman_in-ear_2019v2'), compensation=rtings_inear, **earbud_kwargs ) if referenceaudioanalyzer: if onear: print('\nProcessing Reference Audio Analyzer HDM-X on-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'referenceaudioanalyzer', 'data', 'onear', 'HDM-X'), output_dir=os.path.join(ROOT_DIR, 'results', 'referenceaudioanalyzer', 'referenceaudioanalyzer_hdm-x_harman_over-ear_2018'), compensation=raa_hdmx, **onear_kwargs ) print('\nProcessing Reference Audio Analyzer HDM1 on-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'referenceaudioanalyzer', 'data', 'onear', 'HDM1'), output_dir=os.path.join(ROOT_DIR, 'results', 'referenceaudioanalyzer', 'referenceaudioanalyzer_hdm1_harman_over-ear_2018'), compensation=raa_hdm1, **onear_kwargs ) if inear: print('\nProcessing Reference Audio Analyzer SIEC in-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'referenceaudioanalyzer', 'data', 'inear', 'SIEC'), output_dir=os.path.join(ROOT_DIR, 'results', 'referenceaudioanalyzer', 'referenceaudioanalyzer_siec_harman_in-ear_2019v2'), compensation=raa_siec, **inear_kwargs ) print('\nProcessing Reference Audio Analyzer SIEC CUSTOM in-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'referenceaudioanalyzer', 'data', 'inear', 'SIEC CUSTOM'), output_dir=os.path.join(ROOT_DIR, 'results', 'referenceaudioanalyzer', 'referenceaudioanalyzer_siec_harman_in-ear_2019v2'), compensation=raa_siec, **inear_kwargs ) if earbud: print('\nProcessing Reference Audio Analyzer SIEC earbud measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'referenceaudioanalyzer', 'data', 'earbud', 'SIEC'), output_dir=os.path.join(ROOT_DIR, 'results', 'referenceaudioanalyzer', 'referenceaudioanalyzer_siec_harman_in-ear_2019v2'), compensation=raa_siec, **earbud_kwargs ) if crinacle: if onear: print('\nProcessing Crinacle on-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'crinacle', 'data', 'onear'), output_dir=os.path.join(ROOT_DIR, 'results', 'crinacle', 'crinacle_harman_over-ear_2018'), compensation=crinacle_overear, **onear_kwargs ) if inear: print('\nProcessing Crinacle in-ear measurements...') batch_processing( input_dir=os.path.join(ROOT_DIR, 'measurements', 'crinacle', 'data', 'inear'), output_dir=os.path.join(ROOT_DIR, 'results', 'crinacle', 'harman_in-ear_2019v2'), compensation=harman_inear, **inear_kwargs )