# Configuration for Petri Net Recognition

# --- Image Loading & Preprocessing ---
image_processing:
  min_dimension_threshold: 800
  upscale_factor: 2

# --- Text Detection (doctr) ---
text_detection:
  # doctr model parameters (defaults used in notebook, add if changed)
  # arch: "db_resnet50"
  # assume_straight_pages: True
  # symmetric_pad: True
  # preserve_aspect_ratio: True
  # batch_size: 1
  bin_thresh: 0.3
  box_thresh: 0.1

# --- Shape Detection (Nodes) ---
shape_detection:
  # Parameters for fill_empty_nodes
  fill_circle_enclosing_threshold: 0.8
  fill_rect_enclosing_threshold: 0.95 ##0.78 #### 0.7 was here
  # Parameters for get_nodes_mask
  erosion_kernel_size: [3, 3]
  min_stable_length: 3
  max_erosion_iterations: 30
  # dilation_iterations_match_erosion: True # This can be implicitly handled or added if needed
  # Parameters for detect_shapes (classification)
  classify_circle_overlap_threshold: 0.8 # Place detection
  classify_rect_overlap_threshold: 0.85 # Transition detection
  # Parameters for remove_nodes
  remove_nodes_dilation_kernel_size: [3, 3]
  remove_nodes_dilation_iterations: 3

# --- Connection Processing (Arcs) ---
connection_processing:
  # Skeletonization ( skimage.morphology.skeletonize )
  # No specific params mentioned, uses defaults

  # Hough Line Transform (cv2.HoughLinesP)
  hough_rho: 1
  hough_theta_degrees: 1 # Corresponds to np.pi/180
  hough_threshold: 15
  hough_min_line_length: 10
  hough_max_line_gap: 25

  # Hough Bundler
  hough_bundler_min_distance: 10
  hough_bundler_min_angle: 5

  # Arrowhead Detection (Roboflow API)
  # Note: project_id and version are hardcoded in the pipeline for consistency
  # API key is loaded from ROBOFLOW_API_KEY environment variable
  arrowhead_confidence_threshold_percent: 10.0 # Confidence is 0-100 for Roboflow API

  # Linking nodes to lines
  proximity_thres_place: 1.5 # Multiplied by place radius (heuristic)
  proximity_thres_trans_height: 1.4
  proximity_thres_trans_width: 3

  # Path Finding (find_line_paths)
  path_finding:
    proximity_threshold: 30.0
    dot_product_weight: 0.6
    distance_to_line_weight: 0.2
    endpoint_distance_weight: 0.2

  # Linking arrowheads to path endpoints
  arrowhead_proximity_threshold: 40 # Multiplied by place radius (heuristic)
  text_linking_threshold: 25.0

# --- Clustering (Optional - if DBSCAN part is used) ---
# clustering:
#   dbscan_eps: 0.02
#   dbscan_min_samples: 3