from .node import BatchedMaskedReferenceGen,ReferenceConditionCombine,MasksToPatches,ConditioningMatchMask,GenSubject,PreviewSubjectMask,PreRun,CrossSimMaskExtract,GenConsistent
NODE_CLASS_MAPPINGS = {

    "BatchedMaskedReferenceGen": BatchedMaskedReferenceGen,
    "MasksToPatches": MasksToPatches,
    "ReferenceConditionCombine": ReferenceConditionCombine,
    "ConditioningMatchMask": ConditioningMatchMask,
    "GenSubject": GenSubject,
    "PreviewSubjectMask" : PreviewSubjectMask,
    "PreRun": PreRun,
    "CrossSimMaskExtract": CrossSimMaskExtract,
    "GenConsistent": GenConsistent
}
NODE_DISPLAY_NAME_MAPPINGS = {
    "ReferenceConditionCombine": "Reference Condition Combine",
    "MasksToPatches": "Masks to Patches",
    "ConditioningMatchMask": "Conditioning Match Mask",
    "BatchedMaskedReferenceGen": "Batched Masked Reference Gen",
    "GenSubject": "Gen Subject",
    "PreviewSubjectMask": "Preview Subject Mask",
    "PreRun": "PreRun",
    "CrossSimMaskExtract": "CrossSim Mask Extract",
    "GenConsistent": "GenConsistent"
}

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']