t000_attach_ldbnames.py

import numpy as np
import dict_utils as du

import LHC_Heat_load_dict as LHD
import LHCMeasurementTools.LHC_Heatloads as HL


#thishld = main_dict['stop_squeeze']['heat_load']

def replace_single_hld_with_ldb_naming(thishld, use_dP):

    if use_dP:
        strdp = ''
    else:
        strdp = '_no_dP'

    output = {}

    # start from arc averages
    for ss in HL.sector_list():
        key = 'S%d_QBS_AVG_ARC.POSST' % ss
        output[key] = thishld['arc_averages'+strdp]['S%d'%ss]
        
    #others
    varlist_tmb = []
    for kk in HL.variable_lists_heatloads.keys():
        varlist_tmb+=HL.variable_lists_heatloads[kk]
    varlist_tmb+=HL.arcs_varnames_static


    for varname in varlist_tmb:
        special_id = varname.split('.POSST')[0][-3:]
        if special_id in('_Q1', '_D2', '_D3', '_D4'):
            cell = varname.split('_')[1]
            kkk = cell+special_id
            output[varname] = thishld['special_cells'][kkk]
        elif '_QBS9' in varname:
            firstp, lastp = tuple(varname.split('_QBS'))
            kkk = firstp.split('_')[-1]+'_'+lastp.split('.')[0]
            output[varname] = thishld['all_cells'+strdp][kkk]
        elif 'QBS_AVG_ARC' in varname or 'QBS_CALCULATED_ARC' in varname:
            continue
            
    for kk in ['all_cells', 'arc_averages', 'arc_averages_no_dP', 'all_cells_no_dP', 'special_cells']:
        del(thishld[kk])

    thishld['ldb_naming'] = output



def replace_full_hldict_with_ldb_naming(main_dict, use_dP):

    usedP = False

    replace_single_hld_with_ldb_naming(main_dict['hl_subtracted_offset'], use_dP)
    replace_single_hld_with_ldb_naming(main_dict['hl_integrated'], use_dP)

    for kk in main_dict.keys():
        if type(main_dict[kk]) is dict:
            if 'heat_load' in main_dict[kk]:
                replace_single_hld_with_ldb_naming(main_dict[kk]['heat_load'], use_dP)
            
   
maindict = LHD.get_full_heatload_dictionary().copy()


replace_full_hldict_with_ldb_naming(maindict, use_dP=True)