# configs/infer_1photon.yaml
experiment:
  name: "SiemenStarLowerRight"  # options: SiemenStarLowerLeft, SiemenStarLowerRight
  task: "2Photon" 
  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/2Photon_CS6_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: 6
    SiemenStarLowerRight:
      file_pattern: "SiemenStarLowerRight/2Photon_CS6_chunk{}.h5" ### 15 keV
      NX: 400
      NY: 400
      file_range: [0, 8] # [0, 16)
      roi: [235, 345, 110, 220]
      nSize: 6
    KnifeEdge_3filters_pos0:
      file_pattern: "2603MaxIV_Edge3Filters_pos0_12keV/2Photon_CS6_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
      nSize: 6
    FlatField_3filters_pos0:
      file_pattern: "2603MaxIV_FlatField3Filters_pos0_12keV/2Photon_CS6_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
      nSize: 6
    KnifeEdge_2filters:
      file_pattern: "2603MaxIV_Edge2Filters_12keV/2Photon_CS6_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
      nSize: 6
    FlatField_2filters:
      file_pattern: "2603MaxIV_Flat2Filters_12keV/2Photon_CS6_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
      nSize: 6
    KnifeEdge_1filters:
      file_pattern: "2603MaxIV_Edge1Filters_12keV/2Photon_CS6_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
      nSize: 6
    FlatField_1filters:
      file_pattern: "2603MaxIV_Flat1Filters_12keV/2Photon_CS6_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 16] # [0, 16)
      roi: [0, 101, 0, 101]
      nSize: 6
    Uniformity_test:
      file_pattern: "2603MaxIV_Flat2Filters_12keV/2Photon_CS6_chunk{}.h5" ### 12 keV
      NX: 101
      NY: 101
      file_range: [0, 1]
      roi: [0, 101, 0, 101]
      nSize: 6
  
  energy: 12  # keV
  normalize: false # for inference dataloaders
  batch_size: 8192
  num_workers: 16

model:
  base_dir: "/home/xie_x1/MLXID/DeepLearning/Results/2ph"
  ### 12 keV
  # experiment_name: "260610_12keV_v260610_01"
  # name: "doublePhoton260610_12keV_Noise0.13keV_E500.pth"
  ### 15 keV
  experiment_name: "260622_15keV_v260610_00"
  name: "doublePhoton260610_15keV_Noise0.13keV_E1000.pth"

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