import os

# Set to your instrument directory.
inst_dir = r"C:\Users\stv04463\Development\Mantid\Source\WIP\Code\Mantid\instrument"

OLD_SANS_RUN = "SANS2D00022300.nxs"
NEW_SANS_RUN = "SANS2D00023519.nxs"

IDF_SINCE_9333 = "SANS2D_Tubes_Definition.xml"
PROPOSED_IDF_IN_9361 = "SANS2D_Definition_Tubes.xml"

OLD_IDF = "SANS2D_Definition.xml"
NEW_IDF = IDF_SINCE_9333

def new_idf_was_loaded(inst):
        # A component that existed in the old SANS2D IDF.
        OLD_COMP = "SANS2D/rear-detector/rear-detector(x=184)"
        # A component that exists in the new SANS2D IDF.
        NEW_COMP = "SANS2D/rear-detector/left32"
        
        if inst.getComponentByName(OLD_COMP) != None and inst.getComponentByName(NEW_COMP) == None:
                return False
        if inst.getComponentByName(OLD_COMP) == None and inst.getComponentByName(NEW_COMP) != None:
                return True

def new_param_file_was_loaded(inst):
        # The name of a parameter which should be different in the old and new versions of the param files.
        COL_PARAM = "low-angle-detector-num-columns"
        
        if len(inst.getNumberParameter(COL_PARAM)) == 0:
                return False
        if 192 == inst.getNumberParameter(COL_PARAM)[0]:
                return False
        if 512 == inst.getNumberParameter(COL_PARAM)[0]:
                return True

# a) Load an old and a new run, and see if correct IDFs / param files are being pulled in.
ws = Load(Filename=OLD_SANS_RUN)
inst = ws.getInstrument()
assert not new_idf_was_loaded(inst)
assert not new_param_file_was_loaded(inst)
ws = Load(Filename=NEW_SANS_RUN)
inst = ws.getInstrument()
assert new_idf_was_loaded(inst) # Fails since #9333.
assert new_param_file_was_loaded(inst) # Fails since #9333.

# b) Load the old and new IDFs manually and see if the correct param files are being pulled in.
inst_ws = LoadEmptyInstrument(Filename=os.path.join(inst_dir, OLD_IDF))
inst = inst_ws.getInstrument()
assert not new_idf_was_loaded(inst)
assert not new_param_file_was_loaded(inst)
inst_ws = LoadEmptyInstrument(Filename=os.path.join(inst_dir, NEW_IDF))
inst = inst_ws.getInstrument()
assert new_idf_was_loaded(inst) # Fails since #9333.
assert new_param_file_was_loaded(inst) # Fails since #9333.