# configs/infer_1photon.yaml
experiment:
  name: "SiemenStarLowerRight"  # options: SiemenStarLowerLeft, SiemenStarLowerRight
  task: "3Photon" 
  output_base: "./InferenceResults"

data:
  # sample_folder: "/home/xie_x1/MLXID/DataProcess/Samples"
  sample_folder: "/mnt/sls_det_storage/moench_data/MLXID/Samples/Measurement" ### for SiemenStarLowerRight
  names:
    SiemenStarLowerLeft:
      file_pattern: "SiemenStarLowerLeft/3Photon_CS9_chunk{}.h5" ### 15 keV
      NX: 400
      NY: 400
      file_range: [0, 16]  # [0, 16)
      roi: [140, 230, 120, 210]  # [x_min, x_max, y_min, y_max]
      nSize: 9
    SiemenStarLowerRight:
      file_pattern: "SiemenStarLowerRight/3Photon_CS9_chunk{}.h5" ### 15 keV
      NX: 400
      NY: 400
      file_range: [0, 8]  # [0, 16)
      roi: [235, 345, 110, 220]  # [x_min, x_max, y_min, y_max]
      nSize: 9
    KnifeEdge_3filters_pos0:
      file_pattern: "2603MaxIV_Edge3Filters_pos0_12keV/3Photon_CS9_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
    FlatField_3filters_pos0:
      file_pattern: "2603MaxIV_FlatField3Filters_pos0_12keV/3Photon_CS9_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
    KnifeEdge_2filters:
      file_pattern: "2603MaxIV_Edge2Filters_12keV/3Photon_CS9_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
    FlatField_2filters:
      file_pattern: "2603MaxIV_Flat2Filters_12keV/3Photon_CS9_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
    KnifeEdge_1filters:
      file_pattern: "2603MaxIV_Edge1Filters_12keV/3Photon_CS9_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
    FlatField_1filters:
      file_pattern: "2603MaxIV_Flat1Filters_12keV/3Photon_CS9_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
    Uniformity_test:
      file_pattern: "2603MaxIV_Flat2Filters_12keV/3Photon_CS9_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 1]
      roi: [0, 101, 0, 101]
  
  normalize: false # for inference dataloaders
  batch_size: 8192
  num_workers: 16

model:
  base_dir: "/home/xie_x1/MLXID/DeepLearning/Results/3ph"
  
  ### 12 keV
  # experiment_name: "260611_12keV_v260611_00"
  # name: "triplePhoton260611_12keV_E500.pth"
  ### 15 keV
  experiment_name: "260623_15keV_v260611_00"
  name: "triplePhoton260611_15keV_E500.pth"

inference:
  binning_factor: 10
  chunk_size: 2  # 
  num_aug_ops: 1  # augX8 not implemented for 3-photon inference yet