import RunTntMaker

sys.path.append('%s/RunHelpers/' % os.environ['BASE_WORK_DIR'])
import RunHelpers

# ==============================================================================
if __name__ == '__main__':
    base_work_dir = os.environ['BASE_WORK_DIR']
    print 'getting file list'

    data_set_dicts = []
    RunHelpers.addSamplesToList( sample_dict = {'label':'202641.MadGraphPythia_AUET2B_CTEQ6L1_SM_TT_directBL_1000' , 'num_jobs':1}
                               , data_set_dicts = data_set_dicts
                               # , file_list_path = '%s/EosFileLists/local/raw_d3pd.202641.MadGraphPythia_AUET2B_CTEQ6L1_SM_TT_directBL_1000.txt' % os.environ['BASE_WORK_DIR']
                               , file_list_path = 'EosFileLists/local/raw_d3pd.202641.MadGraphPythia_AUET2B_CTEQ6L1_SM_TT_directBL_1000.txt'
                               , is_data = False
                               , is_egamma_stream = False
                               , is_full_sim = False
                               , dsid = 202641
                               , out_dir = 'local_tnt'
                               )

    print ''
    print data_set_dicts
    print ''
    RunTntMaker.runTntMaker( file_list = data_set_dicts[0]['file_list']
                           , is_data = False
                           , is_full_sim = False
                           )

    # # file_list = RunTntMaker.getFileListFromFile('%s/EosFileLists/local/raw_d3pd.202641.MadGraphPythia_AUET2B_CTEQ6L1_SM_TT_directBL_1000.txt' % os.environ['BASE_WORK_DIR'])
    # # file_list = file_list[:1]
                          , 110825:{'label':'110825.AlpgenPythia_P2011C_ZtautaubbNp0'   , 'num_jobs':1}
                          , 110826:{'label':'110826.AlpgenPythia_P2011C_ZtautaubbNp1'   , 'num_jobs':1}
                          , 110827:{'label':'110827.AlpgenPythia_P2011C_ZtautaubbNp2'   , 'num_jobs':1}
                          , 110828:{'label':'110828.AlpgenPythia_P2011C_ZtautaubbNp3'   , 'num_jobs':1}
                          }

    fast_sim_mc_samples = {}

    # - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    # add data samples
    for dsid in data_samples:
        RunHelpers.addSamplesToList( sample_dict    = data_samples[dsid]
                                   , data_set_dicts = data_set_dicts
                                   , file_list_path = 'EosFileLists/tnt_102/tnt_102.%s.txt' % data_samples[dsid]['label']
                                   , is_data        = True
                                   , is_full_sim    = False
                                   , dsid           = dsid
                                   , out_dir        = out_dir
                                   )

    # add full sim samples
    for dsid in full_sim_mc_samples:
        RunHelpers.addSamplesToList( sample_dict    = full_sim_mc_samples[dsid]
                                   , data_set_dicts = data_set_dicts
                                   , file_list_path = 'EosFileLists/tnt_102/tnt_102.%s.txt' % full_sim_mc_samples[dsid]['label']
                                   , is_data        = False
                                   , is_full_sim    = True
                                   , dsid           = dsid
                                   , out_dir        = out_dir
                                   )