# import mantid algorithms, numpy and matplotlib
from mantid.simpleapi import *
import matplotlib.pyplot as plt
import numpy as np

LoadDiffCal(
    InstrumentName="PG3",
    Filename="/SNS/PG3/shared/CALIBRATION/2026-1_11A_CAL/2026A_PAC_d62140_2026-03-05.h5",
    WorkspaceName="zyp"
)
LoadDiffCal(
    InstrumentName="PG3",
    Filename="/SNS/PG3/shared/CALIBRATION/2026-1_11A_CAL/2026A_PAC_d62140_2026-03-05_correct.h5",
    WorkspaceName="zq"
)

all_difc_zyp = []
all_difc_zq = []
all_difc_diff = []

all_mask_zyp = []
all_mask_zq = []
all_mask_diff = []

for i in range(43120):
    difc_zyp = mtd["zyp_cal"].row(i)["difc"]
    difc_zq = mtd["zq_cal"].row(i)["difc"]
    
    all_difc_zyp.append(difc_zyp)
    all_difc_zq.append(difc_zq)
    all_difc_diff.append(difc_zyp - difc_zq)
    
    all_mask_zyp.append(int(mtd["zyp_mask"].readY(i)[0]))
    all_mask_zq.append(int(mtd["zq_mask"].readY(i)[0]))
    all_mask_diff.append(int(mtd["zyp_mask"].readY(i)[0]) - int(mtd["zq_mask"].readY(i)[0]))

with open("/SNS/PG3/shared/Yuanpeng/PG3_calib_test/PAC_difc_compare.dat", "w") as f:
    for i in range(43120):
        f.write(
            "{0:10.2f}{1:10.2f}{2:10.2f}{3:10d}{4:10d}{5:10d}\n".format(
                all_difc_zyp[i], all_difc_zq[i], all_difc_diff[i],
                all_mask_zyp[i], all_mask_zq[i], all_mask_diff[i]
            )
        )