import re
import json
import os

def parse_inp_file(file_path):
    commands_dict = []
    comments_dict = {}
    
    with open(file_path, 'r') as f:
        lines = f.readlines()
        
    for line in lines:
        parts = line.strip().split(' ')
        if len(parts) == 2:  # 只处理有一个参数的指令
            command, value = parts
            # 检查参数是否为数字（包括科学计数法）
            if re.match(r'^-?\d+\.?\d*([eE][-+]?\d+)?$', value):
                commands_dict.append((command, value))  # 记录指令和参数的元组

    # 读取注释文件
    comments_file_path = 'visa-scpi-example/comments.json'
    if not os.path.exists(comments_file_path):
        with open(comments_file_path, 'w') as f:
            comments_dict = {key: " " for key, _ in commands_dict}
            json.dump(comments_dict, f)
    else:
        with open(comments_file_path, 'r') as f:
            comments_dict = json.load(f)

    # 保存为JSON文件
    with open('visa-scpi-example/commands.json', 'w') as f:
        json.dump([{ "command": cmd, "value": val } for cmd, val in commands_dict], f, indent=2)  # 保存所有指令

    return commands_dict, comments_dict

def save_comments(comments_dict):
    with open('visa-scpi-example/comments.json', 'w') as f:
        json.dump(comments_dict, f, indent=2)

def update_inp_file(file_path, command, new_values):
    with open(file_path, 'r') as f:
        lines = f.readlines()
    
    updated_lines = []
    new_values_set = set(new_values)  # 使用集合以避免重复值
    for line in lines:
        parts = line.strip().split(' ')
        if len(parts) == 2 and parts[0] == command:
            # 如果当前行的指令在新值中，则更新为新值
            if parts[1] in new_values_set:
                updated_lines.append(f"{command} {parts[1]}\n")
            else:
                updated_lines.append(line)  # 保留原行
        else:
            updated_lines.append(line)
    
    with open(file_path, 'w') as f:
        f.writelines(updated_lines)