from utils.MSD import *

if __name__ == '__main__':
    # file_path_li = 'data/msd_li.dat'
    # final_msd = plot_and_get_final_msd(file_path_li, ion_name='Li⁺')

    num_li_ions = 144  # !! 请务必用您体系的真实值替换此示例值 !!

    # 调用新函数
    # 它会自动从 log.lammps 读取温度和体积
    results = calculate_conductivity_from_msd(
        msd_file_path='data/msd_li.dat',
        log_file_path='data/log.lammps',
        ion_name='Li⁺',
        charge=1,
        num_ions=num_li_ions,
        fit_fraction=0.5  # 可以根据图像调整此值
    )