import os

from lib.test.evaluation.environment import EnvSettings


def local_env_settings():
    settings = EnvSettings()
    project_dir = os.environ.get(
        'ATCTRACK_VLM_PROJECT_DIR',
        os.path.abspath(os.path.join(os.path.dirname(__file__), '../../..')),
    )
    dataset_root = os.environ.get(
        'DATASET_ROOT',
        os.path.abspath(os.path.join(project_dir, '../Dataset')),
    )
    output_root = os.environ.get('SAVE_DIR', os.path.join(project_dir, 'output'))

    settings.davis_dir = os.environ.get('DAVIS_DIR', '')
    settings.got10k_lmdb_path = os.environ.get('GOT10K_LMDB_DIR', os.path.join(dataset_root, 'got10k_lmdb'))
    settings.got10k_path = os.environ.get('GOT10K_DIR', os.path.join(dataset_root, 'got10k'))
    settings.got_packed_results_path = os.environ.get('GOT_PACKED_RESULTS_PATH', '')
    settings.got_reports_path = os.environ.get('GOT_REPORTS_PATH', '')
    settings.lasot_extension_subset_path = os.environ.get('LASOT_EXTENSION_SUBSET_DIR', os.path.join(dataset_root, 'lasot_extension_subset'))
    settings.lasot_lmdb_path = os.environ.get('LASOT_LMDB_DIR', os.path.join(dataset_root, 'lasot_lmdb'))
    settings.lasot_path = os.environ.get('LASOT_TEST_DIR', os.path.join(dataset_root, 'lasot'))
    settings.lasotlang_path = os.environ.get('LASOT_LANG_DIR', os.path.join(dataset_root, 'lasot'))
    settings.network_path = os.environ.get('NETWORK_PATH', os.path.join(output_root, 'test/networks'))
    settings.nfs_path = os.environ.get('NFS_DIR', os.path.join(dataset_root, 'nfs'))
    settings.otb_path = os.environ.get('OTB_DIR', os.path.join(dataset_root, 'OTB2015'))
    settings.otblang_path = os.environ.get('OTB_LANG_DIR', os.path.join(dataset_root, 'otb_lang'))
    settings.prj_dir = project_dir
    settings.result_plot_path = os.environ.get('RESULT_PLOT_PATH', os.path.join(output_root, 'test/result_plots'))
    settings.results_path = os.environ.get('RESULTS_PATH', os.path.join(output_root, 'test/tracking_results'))
    settings.save_dir = output_root
    settings.segmentation_path = os.environ.get('SEGMENTATION_PATH', os.path.join(output_root, 'test/segmentation_results'))
    settings.tc128_path = os.environ.get('TC128_DIR', os.path.join(dataset_root, 'TC128'))
    settings.tn_packed_results_path = os.environ.get('TN_PACKED_RESULTS_PATH', '')
    settings.tnl2k_path = os.environ.get('TNL2K_TEST_DIR', os.path.join(dataset_root, 'tnl2k/test'))
    settings.tpl_path = os.environ.get('TPL_PATH', '')
    settings.trackingnet_path = os.environ.get('TRACKINGNET_DIR', os.path.join(dataset_root, 'trackingnet'))
    settings.uav_path = os.environ.get('UAV_DIR', os.path.join(dataset_root, 'UAV123'))
    settings.vot_path = os.environ.get('VOT_DIR', os.path.join(dataset_root, 'VOT2019'))
    settings.youtubevos_dir = os.environ.get('YOUTUBEVOS_DIR', '')

    return settings