Exemple #1
0
def get_hdf5_file(shotnum,isprint=True):
    """
    从HDF5文件读取 HMode_Ha 信号
    :param shotnum:     炮号
    :param isprint:     打印错误信息
    :return:            HMode_Ha 信号
    
    """
    try:
        _, signal = mh.read_channel(shotnum, channel="HMode_Ha")
        if max(signal) <= Signal_th:
            if isprint:
                print("{} Maximum < Signal_th ({}), pass".format(shotnum,Signal_th))
            return None
        freq = mh.get_attrs("T_Freq", shot_number=shotnum, channel="HMode_Ha") / 1000
        if not freq == default_Freq:
            if isprint:
                print("Frequency should be 10/ms,but it is : {}/ms, pass".format(freq))
            return None
        # 部分信号长度不足,进行扩充
        signal = np.append(signal, [0] * signal_max_len)
        return signal
    except:
        if isprint:
            print("Something is wrong,pass")
        return None
Exemple #2
0
def label_data(shots):
    """
    通过查看各通道信号,给数据打标签,并通过 save_label 存储
    :param shots: 
    :return: 
    """
    data = {}
    for shot in shots:
        if shot in pdc.bad_shots or not if_all_channel_exist(shot):
            continue
        print("shot is: ", shot)

        i = 1
        figure, ax = plt.subplots()
        figure.suptitle('炮号:{} '.format(shot), size=20)
        for channel in pdc.Channels:
            plt.subplot(pdc.fig_height, pdc.fig_width, i)
            plt.subplots_adjust(left=None,
                                bottom=None,
                                right=None,
                                top=None,
                                wspace=None,
                                hspace=0.5)
            plt.title(channel, size=15)
            plt.xlabel("                                         Time (ms)",
                       size=12)

            Time, value = mh.read_channel(shot_number=shot, channel=channel)
            plt.plot(Time, value)
            if channel is "Ha_MidPlane_1":
                plt.text(200,
                         3,
                         "击穿时间: {:} ms.".format(np.where(value > 0.5)[0][0]),
                         size=16)
                plt.plot(40 * [10], np.arange(0, 4, 0.1), linestyle="-.")
            i += 1
        # 打标签
        if pdc.maxshow:
            plt.get_current_fig_manager().window.showMaximized()
        plt.show()

        if not pdc.is_save_label:
            continue

        try:
            data[shot] = int(input("please input label: "))
        except ValueError:

            continue
        if len(data) == pdc.save_numbers:
            save_label(data)
            data = {}
Exemple #3
0
def padding_len_to_same(shot):
    """
    将数据扩充为统一长度
    :param shot: 炮号 
    :return:     包含5个通道的字典
    """
    _, gas = mh.read_channel(shot, "GasPuffing_FeedForward")
    _, m_d_h = mh.read_channel(shot, "Mag_detla_H")
    data = {
        "IP": ip_subtract_target_ip(shot),
        "HX": max_of_hx(shot),
        "Ha_MidPlane_1": re_sampling_ha(shot),
        "Mag_detla_H": m_d_h,
        "GasPuffing_"
        "FeedForward": gas[500:],
    }
    del gas, m_d_h
    for channel, value in data.items():
        channel_len = min(tc.Channels_len[channel], len(value))
        data[channel] = np.append(value[0:channel_len],
                                  np.zeros(tc.Signal_max_len - channel_len))
    return data
Exemple #4
0
def re_sampling_ha(shot):
    # 采样率不一样,只能先这样将就了
    _, ha = mh.read_channel(shot, channel="Mag_detla_H")
    return ha[::10]
Exemple #5
0
def max_of_hx(shot):
    _, hx1 = mh.read_channel(shot, channel="HX_1")
    _, hx2 = mh.read_channel(shot, channel="HX_2")
    # hx_len = min(len(hx1), len(hx2))
    return np.maximum(hx1, hx2)
Exemple #6
0
def ip_subtract_target_ip(shot):
    _, ip = mh.read_channel(shot, channel="IP")
    _, target_ip = mh.read_channel(shot, channel="Target_IP")
    ip_len = min(len(ip), len(target_ip), tc.Channels_len["IP"])
    return ip[0:ip_len] - target_ip[0:ip_len]