natural-pdf 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

natural_pdf/flows/element.py

@@ -0,0 +1,382 @@
+import logging
+from typing import TYPE_CHECKING, Any, List, Optional, Tuple, Union, cast
+
+if TYPE_CHECKING:
+    from natural_pdf.elements.base import Element as PhysicalElement
+    from natural_pdf.elements.region import Region as PhysicalRegion
+    from natural_pdf.core.page import Page as PhysicalPage # For type checking physical_object.page
+    from .flow import Flow
+    from .region import FlowRegion
+
+logger = logging.getLogger(__name__)
+
+
+class FlowElement:
+    """
+    Represents a physical PDF Element or Region that is anchored within a Flow.
+    This class provides methods for flow-aware directional navigation (e.g., below, above)
+    that operate across the segments defined in its associated Flow.
+    """
+
+    def __init__(self, physical_object: Union["PhysicalElement", "PhysicalRegion"], flow: "Flow"):
+        """
+        Initializes a FlowElement.
+
+        Args:
+            physical_object: The actual natural_pdf.elements.base.Element or
+                             natural_pdf.elements.region.Region object.
+            flow: The Flow instance this element is part of.
+        """
+        if not (hasattr(physical_object, 'bbox') and hasattr(physical_object, 'page')):
+            raise TypeError(
+                f"physical_object must be a valid PDF element-like object with 'bbox' and 'page' attributes. Got {type(physical_object)}"
+            )
+        self.physical_object: Union["PhysicalElement", "PhysicalRegion"] = physical_object
+        self.flow: "Flow" = flow
+
+    # --- Properties to delegate to the physical_object --- 
+    @property
+    def bbox(self) -> Tuple[float, float, float, float]:
+        return self.physical_object.bbox
+
+    @property
+    def x0(self) -> float:
+        return self.physical_object.x0
+
+    @property
+    def top(self) -> float:
+        return self.physical_object.top
+
+    @property
+    def x1(self) -> float:
+        return self.physical_object.x1
+
+    @property
+    def bottom(self) -> float:
+        return self.physical_object.bottom
+
+    @property
+    def width(self) -> float:
+        return self.physical_object.width
+
+    @property
+    def height(self) -> float:
+        return self.physical_object.height
+
+    @property
+    def text(self) -> Optional[str]:
+        return getattr(self.physical_object, 'text', None)
+    
+    @property
+    def page(self) -> Optional["PhysicalPage"]:
+        """Returns the physical page of the underlying element."""
+        return getattr(self.physical_object, 'page', None)
+
+    def _flow_direction(
+        self,
+        direction: str,  # "above", "below", "left", "right"
+        size: Optional[float] = None,
+        cross_size_ratio: Optional[float] = None, # Default to None for full flow width
+        cross_size_absolute: Optional[float] = None, 
+        cross_alignment: str = "center", # "start", "center", "end"
+        until: Optional[str] = None,
+        include_endpoint: bool = True,
+        **kwargs,
+    ) -> "FlowRegion":
+        from .region import FlowRegion # Runtime import for return if not stringized, but stringizing is safer
+        # Ensure correct import for creating new PhysicalRegion instances if needed
+        from natural_pdf.elements.region import Region as PhysicalRegion_Class # Runtime import
+
+        collected_constituent_regions: List[PhysicalRegion_Class] = [] # PhysicalRegion_Class is runtime
+        boundary_element_hit: Optional["PhysicalElement"] = None # Stringized
+        # Ensure remaining_size is float, even if size is int.
+        remaining_size = float(size) if size is not None else float('inf')
+
+
+        # 1. Identify Starting Segment and its index
+        start_segment_index = -1
+        for i, segment_in_flow in enumerate(self.flow.segments):
+            if self.physical_object.page != segment_in_flow.page:
+                continue
+
+            obj_center_x = (self.physical_object.x0 + self.physical_object.x1) / 2
+            obj_center_y = (self.physical_object.top + self.physical_object.bottom) / 2
+
+            if segment_in_flow.is_point_inside(obj_center_x, obj_center_y):
+                start_segment_index = i
+                break
+            obj_bbox = self.physical_object.bbox
+            seg_bbox = segment_in_flow.bbox
+            if not (obj_bbox[2] < seg_bbox[0] or obj_bbox[0] > seg_bbox[2] or \
+                    obj_bbox[3] < seg_bbox[1] or obj_bbox[1] > seg_bbox[3]):
+                 if start_segment_index == -1: 
+                    start_segment_index = i 
+        
+        if start_segment_index == -1:
+            page_num_str = str(self.physical_object.page.page_number) if self.physical_object.page else 'N/A'
+            logger.warning(
+                f"FlowElement's physical object {self.physical_object.bbox} on page {page_num_str} "
+                f"not found within any flow segment. Cannot perform directional operation '{direction}'."
+            )
+            # Need FlowRegion for the return type, ensure it's available or stringized
+            from .region import FlowRegion as RuntimeFlowRegion 
+            return RuntimeFlowRegion(
+                flow=self.flow,
+                constituent_regions=[],
+                source_flow_element=self,
+                boundary_element_found=None
+            )
+
+        is_primary_vertical = self.flow.arrangement == "vertical"
+        segment_iterator: range
+
+        if direction == "below":
+            if not is_primary_vertical: raise NotImplementedError("'below' is for vertical flows.")
+            is_forward = True 
+            segment_iterator = range(start_segment_index, len(self.flow.segments))
+        elif direction == "above":
+            if not is_primary_vertical: raise NotImplementedError("'above' is for vertical flows.")
+            is_forward = False
+            segment_iterator = range(start_segment_index, -1, -1)
+        elif direction == "right":
+            if is_primary_vertical: raise NotImplementedError("'right' is for horizontal flows.")
+            is_forward = True
+            segment_iterator = range(start_segment_index, len(self.flow.segments))
+        elif direction == "left":
+            if is_primary_vertical: raise NotImplementedError("'left' is for horizontal flows.")
+            is_forward = False
+            segment_iterator = range(start_segment_index, -1, -1)
+        else:
+            raise ValueError(f"Internal error: Invalid direction '{direction}' for _flow_direction.")
+
+        for current_segment_idx in segment_iterator:
+            if remaining_size <= 0 and size is not None: break
+            if boundary_element_hit: break
+
+            current_segment: PhysicalRegion_Class = self.flow.segments[current_segment_idx]
+            segment_contribution: Optional[PhysicalRegion_Class] = None
+            
+            op_source: Union["PhysicalElement", PhysicalRegion_Class] # Stringized PhysicalElement
+            op_direction_params: dict = {
+                "direction": direction, "until": until, "include_endpoint": include_endpoint, **kwargs
+            }
+            
+            # --- Cross-size logic: Default to "full" if no specific ratio or absolute is given ---
+            cross_size_for_op: Union[str, float]
+            if cross_size_absolute is not None:
+                cross_size_for_op = cross_size_absolute
+            elif cross_size_ratio is not None: # User explicitly provided a ratio
+                base_cross_dim = self.physical_object.width if is_primary_vertical else self.physical_object.height
+                cross_size_for_op = base_cross_dim * cross_size_ratio
+            else: # Default case: neither absolute nor ratio provided, so use "full"
+                cross_size_for_op = "full"
+            op_direction_params["cross_size"] = cross_size_for_op
+
+            if current_segment_idx == start_segment_index:
+                op_source = self.physical_object 
+                op_direction_params["size"] = remaining_size if size is not None else None
+                op_direction_params["include_element"] = False 
+
+                source_for_op_call = op_source
+                if not isinstance(source_for_op_call, PhysicalRegion_Class):
+                    if hasattr(source_for_op_call, 'to_region'): 
+                        source_for_op_call = source_for_op_call.to_region()
+                    else: 
+                        logger.error(f"FlowElement: Cannot convert op_source {type(op_source)} to region.")
+                        continue
+                
+                # 1. Perform directional operation *without* 'until' initially to get basic shape.
+                initial_op_params = {
+                    "direction": direction,
+                    "size": remaining_size if size is not None else None,
+                    "cross_size": cross_size_for_op,
+                    "cross_alignment": cross_alignment, # Pass alignment
+                    "include_element": False,
+                    # Pass other relevant kwargs if Region._direction uses them (e.g. strict_type)
+                    **{k: v for k, v in kwargs.items() if k in ['strict_type', 'first_match_only']} 
+                }
+                initial_region_from_op = source_for_op_call._direction(**initial_op_params)
+
+                # 2. Clip this initial region to the current flow segment's boundaries.
+                clipped_search_area = current_segment.clip(initial_region_from_op)
+                segment_contribution = clipped_search_area # Default contribution
+
+                # 3. If 'until' is specified, search for it *within* the clipped_search_area.
+                if until and clipped_search_area and clipped_search_area.width > 0 and clipped_search_area.height > 0:
+                    # kwargs for find_all are the general kwargs passed to _flow_direction
+                    until_matches = clipped_search_area.find_all(until, **kwargs)
+                    
+                    if until_matches:
+                        potential_hit: Optional["PhysicalElement"] = None
+                        if direction == "below": potential_hit = until_matches.sort(key=lambda m: m.top).first
+                        elif direction == "above": potential_hit = until_matches.sort(key=lambda m: m.bottom, reverse=True).first
+                        elif direction == "right": potential_hit = until_matches.sort(key=lambda m: m.x0).first
+                        elif direction == "left": potential_hit = until_matches.sort(key=lambda m: m.x1, reverse=True).first
+                        
+                        if potential_hit:
+                            boundary_element_hit = potential_hit # Set the overall boundary flag
+                            # Adjust segment_contribution to stop at this boundary_element_hit.
+                            if is_primary_vertical:
+                                if direction == "below":
+                                    edge = boundary_element_hit.bottom if include_endpoint else (boundary_element_hit.top - 1)
+                                else:  # direction == "above"
+                                    edge = boundary_element_hit.top if include_endpoint else (boundary_element_hit.bottom + 1)
+                                segment_contribution = segment_contribution.clip(
+                                    bottom=edge if direction == "below" else None,
+                                    top=edge if direction == "above" else None
+                                )
+                            else: 
+                                if direction == "right":
+                                    edge = boundary_element_hit.x1 if include_endpoint else (boundary_element_hit.x0 - 1)
+                                else:  # direction == "left"
+                                    edge = boundary_element_hit.x0 if include_endpoint else (boundary_element_hit.x1 + 1)
+                                segment_contribution = segment_contribution.clip(
+                                    right=edge if direction == "right" else None,
+                                    left=edge if direction == "left" else None
+                                )
+            else: 
+                candidate_region_in_segment = current_segment
+                if until and not boundary_element_hit:
+                    until_matches = candidate_region_in_segment.find_all(until, **kwargs)
+                    if until_matches:
+                        potential_hit = None
+                        if direction == "below": potential_hit = until_matches.sort(key=lambda m: m.top).first
+                        elif direction == "above": potential_hit = until_matches.sort(key=lambda m: m.bottom, reverse=True).first
+                        elif direction == "right": potential_hit = until_matches.sort(key=lambda m: m.x0).first
+                        elif direction == "left": potential_hit = until_matches.sort(key=lambda m: m.x1, reverse=True).first
+                        
+                        if potential_hit:
+                            boundary_element_hit = potential_hit
+                            if is_primary_vertical:
+                                if direction == "below":
+                                    edge = boundary_element_hit.bottom if include_endpoint else (boundary_element_hit.top - 1)
+                                else:  # direction == "above"
+                                    edge = boundary_element_hit.top if include_endpoint else (boundary_element_hit.bottom + 1)
+                                candidate_region_in_segment = candidate_region_in_segment.clip(bottom=edge if direction == "below" else None, top=edge if direction == "above" else None)
+                            else: 
+                                if direction == "right":
+                                    edge = boundary_element_hit.x1 if include_endpoint else (boundary_element_hit.x0 - 1)
+                                else:  # direction == "left"
+                                    edge = boundary_element_hit.x0 if include_endpoint else (boundary_element_hit.x1 + 1)
+                                candidate_region_in_segment = candidate_region_in_segment.clip(right=edge if direction == "right" else None, left=edge if direction == "left" else None)
+                segment_contribution = candidate_region_in_segment
+
+            if segment_contribution and segment_contribution.width > 0 and segment_contribution.height > 0 and size is not None:
+                current_part_consumed_size = 0.0
+                if is_primary_vertical: 
+                    current_part_consumed_size = segment_contribution.height
+                    if current_part_consumed_size > remaining_size:
+                        new_edge = (segment_contribution.top + remaining_size) if is_forward else (segment_contribution.bottom - remaining_size)
+                        segment_contribution = segment_contribution.clip(bottom=new_edge if is_forward else None, top=new_edge if not is_forward else None)
+                        current_part_consumed_size = remaining_size 
+                else: 
+                    current_part_consumed_size = segment_contribution.width
+                    if current_part_consumed_size > remaining_size:
+                        new_edge = (segment_contribution.x0 + remaining_size) if is_forward else (segment_contribution.x1 - remaining_size)
+                        segment_contribution = segment_contribution.clip(right=new_edge if is_forward else None, left=new_edge if not is_forward else None)
+                        current_part_consumed_size = remaining_size 
+                remaining_size -= current_part_consumed_size
+            
+            if segment_contribution and segment_contribution.width > 0 and segment_contribution.height > 0:
+                collected_constituent_regions.append(segment_contribution)
+
+            # If boundary was hit in this segment, and we are not on the start segment (where we might still collect part of it)
+            # or if we are on the start segment AND the contribution became zero (e.g. until was immediate)
+            if boundary_element_hit and (current_segment_idx != start_segment_index or not segment_contribution or (segment_contribution.width <= 0 or segment_contribution.height <= 0)):
+                 break # Stop iterating through more segments
+
+            is_logically_last_segment = (is_forward and current_segment_idx == len(self.flow.segments) - 1) or \
+                                      (not is_forward and current_segment_idx == 0)
+            if not is_logically_last_segment and self.flow.segment_gap > 0 and size is not None:
+                if remaining_size > 0 : 
+                     remaining_size -= self.flow.segment_gap
+        
+        from .region import FlowRegion as RuntimeFlowRegion # Ensure it's available for return
+        return RuntimeFlowRegion(
+            flow=self.flow,
+            constituent_regions=collected_constituent_regions,
+            source_flow_element=self,
+            boundary_element_found=boundary_element_hit
+        )
+
+    # --- Public Directional Methods --- 
+    # These will largely mirror DirectionalMixin but call _flow_direction.
+
+    def above(
+        self,
+        height: Optional[float] = None, 
+        width_ratio: Optional[float] = None,      
+        width_absolute: Optional[float] = None, 
+        width_alignment: str = "center", 
+        until: Optional[str] = None,
+        include_endpoint: bool = True,
+        **kwargs,
+    ) -> "FlowRegion": # Stringized
+        if self.flow.arrangement == "vertical":
+            return self._flow_direction(
+                direction="above", size=height, cross_size_ratio=width_ratio,
+                cross_size_absolute=width_absolute, cross_alignment=width_alignment, 
+                until=until, include_endpoint=include_endpoint, **kwargs,
+            )
+        else: 
+            raise NotImplementedError("'above' in a horizontal flow is ambiguous with current 1D flow logic and not yet implemented.")
+
+    def below(
+        self,
+        height: Optional[float] = None, 
+        width_ratio: Optional[float] = None,
+        width_absolute: Optional[float] = None,
+        width_alignment: str = "center",
+        until: Optional[str] = None,
+        include_endpoint: bool = True,
+        **kwargs,
+    ) -> "FlowRegion": # Stringized
+        if self.flow.arrangement == "vertical":
+            return self._flow_direction(
+                direction="below", size=height, cross_size_ratio=width_ratio,
+                cross_size_absolute=width_absolute, cross_alignment=width_alignment,
+                until=until, include_endpoint=include_endpoint, **kwargs,
+            )
+        else: 
+            raise NotImplementedError("'below' in a horizontal flow is ambiguous with current 1D flow logic and not yet implemented.")
+
+    def left(
+        self,
+        width: Optional[float] = None, 
+        height_ratio: Optional[float] = None,     
+        height_absolute: Optional[float] = None, 
+        height_alignment: str = "center", 
+        until: Optional[str] = None,
+        include_endpoint: bool = True,
+        **kwargs,
+    ) -> "FlowRegion": # Stringized
+        if self.flow.arrangement == "horizontal":
+            return self._flow_direction(
+                direction="left", size=width, cross_size_ratio=height_ratio,
+                cross_size_absolute=height_absolute, cross_alignment=height_alignment, 
+                until=until, include_endpoint=include_endpoint, **kwargs,
+            )
+        else: 
+            raise NotImplementedError("'left' in a vertical flow is ambiguous with current 1D flow logic and not yet implemented.")
+
+    def right(
+        self,
+        width: Optional[float] = None, 
+        height_ratio: Optional[float] = None,
+        height_absolute: Optional[float] = None,
+        height_alignment: str = "center",
+        until: Optional[str] = None,
+        include_endpoint: bool = True,
+        **kwargs,
+    ) -> "FlowRegion": # Stringized
+        if self.flow.arrangement == "horizontal":
+            return self._flow_direction(
+                direction="right", size=width, cross_size_ratio=height_ratio,
+                cross_size_absolute=height_absolute, cross_alignment=height_alignment,
+                until=until, include_endpoint=include_endpoint, **kwargs,
+            )
+        else: 
+            raise NotImplementedError("'right' in a vertical flow is ambiguous with current 1D flow logic and not yet implemented.")
+
+    def __repr__(self) -> str:
+        return f"<FlowElement for {self.physical_object.__class__.__name__} {self.bbox} in {self.flow}>" 

natural_pdf/flows/flow.py

@@ -0,0 +1,216 @@
+import logging
+from typing import TYPE_CHECKING, Any, List, Literal, Optional, Union
+
+if TYPE_CHECKING:
+    from natural_pdf.core.page import Page
+    from natural_pdf.elements.region import Region as PhysicalRegion
+    from natural_pdf.elements.base import Element as PhysicalElement
+    from natural_pdf.elements.collections import ElementCollection as PhysicalElementCollection
+    from .element import FlowElement
+    from .collections import FlowElementCollection
+
+logger = logging.getLogger(__name__)
+
+
+class Flow:
+    """
+    Defines a logical flow or sequence of physical Page or Region objects,
+    specifying their arrangement and alignment to enable operations that
+    span across these segments as if they were a continuous area.
+    """
+
+    def __init__(
+        self,
+        segments: List[Union["Page", "PhysicalRegion"]],
+        arrangement: Literal["vertical", "horizontal"],
+        alignment: Literal["start", "center", "end", "top", "left", "bottom", "right"] = "start",
+        segment_gap: float = 0.0,
+    ):
+        """
+        Initializes a Flow object.
+
+        Args:
+            segments: An ordered list of natural_pdf.core.page.Page or
+                      natural_pdf.elements.region.Region objects that constitute the flow.
+            arrangement: The primary direction of the flow.
+                         - "vertical": Segments are stacked top-to-bottom.
+                         - "horizontal": Segments are arranged left-to-right.
+            alignment: How segments are aligned on their cross-axis if they have
+                       differing dimensions. For a "vertical" arrangement:
+                       - "left" (or "start"): Align left edges.
+                       - "center": Align centers.
+                       - "right" (or "end"): Align right edges.
+                       For a "horizontal" arrangement:
+                       - "top" (or "start"): Align top edges.
+                       - "center": Align centers.
+                       - "bottom" (or "end"): Align bottom edges.
+            segment_gap: The virtual gap (in PDF points) between segments.
+        """
+        if not segments:
+            raise ValueError("Flow segments cannot be empty.")
+        if arrangement not in ["vertical", "horizontal"]:
+            raise ValueError("Arrangement must be 'vertical' or 'horizontal'.")
+
+        self.segments: List["PhysicalRegion"] = self._normalize_segments(segments)
+        self.arrangement: Literal["vertical", "horizontal"] = arrangement
+        self.alignment: Literal["start", "center", "end", "top", "left", "bottom", "right"] = alignment
+        self.segment_gap: float = segment_gap
+
+        self._validate_alignment()
+
+        # TODO: Pre-calculate segment offsets for faster lookups if needed
+
+    def _normalize_segments(self, segments: List[Union["Page", "PhysicalRegion"]]) -> List["PhysicalRegion"]:
+        """Converts all Page segments to full-page Region objects for uniform processing."""
+        normalized = []
+        from natural_pdf.core.page import Page as CorePage
+        from natural_pdf.elements.region import Region as ElementsRegion
+
+        for i, segment in enumerate(segments):
+            if isinstance(segment, CorePage):
+                normalized.append(segment.region(0, 0, segment.width, segment.height))
+            elif isinstance(segment, ElementsRegion):
+                normalized.append(segment)
+            elif hasattr(segment, 'object_type') and segment.object_type == "page":
+                if not isinstance(segment, CorePage):
+                    raise TypeError(f"Segment {i} has object_type 'page' but is not an instance of natural_pdf.core.page.Page. Got {type(segment)}")
+                normalized.append(segment.region(0, 0, segment.width, segment.height))
+            elif hasattr(segment, 'object_type') and segment.object_type == "region":
+                if not isinstance(segment, ElementsRegion):
+                    raise TypeError(f"Segment {i} has object_type 'region' but is not an instance of natural_pdf.elements.region.Region. Got {type(segment)}")
+                normalized.append(segment)
+            else:
+                raise TypeError(
+                    f"Segment {i} is not a valid Page or Region object. Got {type(segment)}."
+                )
+        return normalized
+
+    def _validate_alignment(self) -> None:
+        """Validates the alignment based on the arrangement."""
+        valid_alignments = {
+            "vertical": ["start", "center", "end", "left", "right"],
+            "horizontal": ["start", "center", "end", "top", "bottom"],
+        }
+        if self.alignment not in valid_alignments[self.arrangement]:
+            raise ValueError(
+                f"Invalid alignment '{self.alignment}' for '{self.arrangement}' arrangement. "
+                f"Valid options are: {valid_alignments[self.arrangement]}"
+            )
+
+    def find(
+        self,
+        selector: Optional[str] = None,
+        *,
+        text: Optional[str] = None,
+        apply_exclusions: bool = True,
+        regex: bool = False,
+        case: bool = True,
+        **kwargs,
+    ) -> Optional["FlowElement"]:
+        """
+        Finds the first element within the flow that matches the given selector or text criteria.
+
+        Elements found are wrapped as FlowElement objects, anchored to this Flow.
+
+        Args:
+            selector: CSS-like selector string.
+            text: Text content to search for.
+            apply_exclusions: Whether to respect exclusion zones on the original pages/regions.
+            regex: Whether the text search uses regex.
+            case: Whether the text search is case-sensitive.
+            **kwargs: Additional filter parameters for the underlying find operation.
+
+        Returns:
+            A FlowElement if a match is found, otherwise None.
+        """
+        results = self.find_all(
+            selector=selector,
+            text=text,
+            apply_exclusions=apply_exclusions,
+            regex=regex,
+            case=case,
+            **kwargs
+        )
+        return results.first if results else None
+
+    def find_all(
+        self,
+        selector: Optional[str] = None,
+        *,
+        text: Optional[str] = None,
+        apply_exclusions: bool = True,
+        regex: bool = False,
+        case: bool = True,
+        **kwargs,
+    ) -> "FlowElementCollection":
+        """
+        Finds all elements within the flow that match the given selector or text criteria.
+        Elements are collected segment by segment, preserving the flow order.
+
+        Elements found are wrapped as FlowElement objects, anchored to this Flow,
+        and returned in a FlowElementCollection.
+        """
+        from .collections import FlowElementCollection
+        from .element import FlowElement
+
+        all_flow_elements: List["FlowElement"] = []
+
+        # Iterate through segments in their defined flow order
+        for physical_segment in self.segments:
+            # Find all matching physical elements within the current segment
+            # Region.find_all() should return elements in local reading order.
+            matches_in_segment: "PhysicalElementCollection" = physical_segment.find_all(
+                selector=selector,
+                text=text,
+                apply_exclusions=apply_exclusions,
+                regex=regex,
+                case=case,
+                **kwargs,
+            )
+            if matches_in_segment:
+                # Wrap each found physical element as a FlowElement and add to the list
+                # This preserves the order from matches_in_segment.elements
+                for phys_elem in matches_in_segment.elements:
+                    all_flow_elements.append(FlowElement(physical_object=phys_elem, flow=self))
+        
+        # The global sort that was here previously has been removed.
+        # The order is now determined by segment sequence, then by local order within each segment.
+
+        return FlowElementCollection(all_flow_elements)
+
+    def __repr__(self) -> str:
+        return (
+            f"<Flow segments={len(self.segments)}, "
+            f"arrangement='{self.arrangement}', alignment='{self.alignment}', gap={self.segment_gap}>"
+        )
+
+    # --- Helper methods for coordinate transformations and segment iteration --- 
+    # These will be crucial for FlowElement's directional methods.
+
+    def get_segment_bounding_box_in_flow(self, segment_index: int) -> Optional[tuple[float, float, float, float]]:
+        """
+        Calculates the conceptual bounding box of a segment within the flow's coordinate system.
+        This considers arrangement, alignment, and segment gaps.
+        (This is a placeholder for more complex logic if a true virtual coordinate system is needed)
+        For now, it might just return the physical segment's bbox if gaps are 0 and alignment is simple.
+        """
+        if segment_index < 0 or segment_index >= len(self.segments):
+            return None
+        
+        # This is a simplified version. A full implementation would calculate offsets.
+        # For now, we assume FlowElement directional logic handles segment traversal and uses physical coords.
+        # If we were to *draw* the flow or get a FlowRegion bbox that spans gaps, this would be critical.
+        # physical_segment = self.segments[segment_index]
+        # return physical_segment.bbox
+        raise NotImplementedError("Calculating a segment's bbox *within the flow's virtual coordinate system* is not yet fully implemented.")
+
+    def get_element_flow_coordinates(self, physical_element: "PhysicalElement") -> Optional[tuple[float, float, float, float]]:
+        """
+        Translates a physical element's coordinates into the flow's virtual coordinate system.
+        (Placeholder - very complex if segment_gap > 0 or complex alignments)
+        """
+        # For now, elements operate in their own physical coordinates. This method would be needed
+        # if FlowRegion.bbox or other operations needed to present a unified coordinate space.
+        # As per our discussion, elements *within* a FlowRegion retain original physical coordinates.
+        # So, this might not be strictly necessary for the current design's core functionality.
+        raise NotImplementedError("Translating element coordinates to a unified flow coordinate system is not yet implemented.")