natural-pdf 0.2.15__py3-none-any.whl → 0.2.16__py3-none-any.whl

natural_pdf/flows/element.py

@@ -106,6 +106,7 @@ class FlowElement:
         cross_size_absolute: Optional[float] = None,
         cross_alignment: str = "center",  # "start", "center", "end"
         until: Optional[str] = None,
+        include_source: bool = False,
         include_endpoint: bool = True,
         **kwargs,
     ) -> "FlowRegion":
@@ -178,13 +179,9 @@ class FlowElement:
             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:
@@ -206,28 +203,34 @@ class FlowElement:
                 "direction": direction,
                 "until": until,
                 "include_endpoint": include_endpoint,
+                "include_source": include_source,
                 **kwargs,
             }
 
-            # --- Cross-size logic: Default to "full" if no specific ratio or absolute is given ---
+            # --- Cross-size logic: Default based on direction ---
             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
+                # Cross dimension depends on direction, not flow arrangement
                 base_cross_dim = (
                     self.physical_object.width
-                    if is_primary_vertical
+                    if direction in ["above", "below"]
                     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"
+            else:  # Default case: neither absolute nor ratio provided
+                # Default to element size for left/right, full for above/below
+                if direction in ["left", "right"]:
+                    cross_size_for_op = self.physical_object.height
+                else:
+                    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_source"] = False
+                op_direction_params["include_source"] = include_source
 
                 source_for_op_call = op_source
                 if not isinstance(source_for_op_call, PhysicalRegion_Class):
@@ -245,7 +248,7 @@ class FlowElement:
                     "size": remaining_size if size is not None else None,
                     "cross_size": cross_size_for_op,
                     "cross_alignment": cross_alignment,  # Pass alignment
-                    "include_source": False,
+                    "include_source": include_source,
                     # 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"]},
                 }
@@ -283,7 +286,7 @@ class FlowElement:
                         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 in ["below", "above"]:
                                 if direction == "below":
                                     edge = (
                                         boundary_element_hit.bottom
@@ -300,7 +303,7 @@ class FlowElement:
                                     bottom=edge if direction == "below" else None,
                                     top=edge if direction == "above" else None,
                                 )
-                            else:
+                            else:  # direction in ["right", "left"]
                                 if direction == "right":
                                     edge = (
                                         boundary_element_hit.x1
@@ -338,7 +341,7 @@ class FlowElement:
 
                         if potential_hit:
                             boundary_element_hit = potential_hit
-                            if is_primary_vertical:
+                            if direction in ["below", "above"]:
                                 if direction == "below":
                                     edge = (
                                         boundary_element_hit.bottom
@@ -355,7 +358,7 @@ class FlowElement:
                                     bottom=edge if direction == "below" else None,
                                     top=edge if direction == "above" else None,
                                 )
-                            else:
+                            else:  # direction in ["right", "left"]
                                 if direction == "right":
                                     edge = (
                                         boundary_element_hit.x1
@@ -381,7 +384,7 @@ class FlowElement:
                 and size is not None
             ):
                 current_part_consumed_size = 0.0
-                if is_primary_vertical:
+                if direction in ["below", "above"]:
                     current_part_consumed_size = segment_contribution.height
                     if current_part_consumed_size > remaining_size:
                         new_edge = (
@@ -394,7 +397,7 @@ class FlowElement:
                             top=new_edge if not is_forward else None,
                         )
                         current_part_consumed_size = remaining_size
-                else:
+                else:  # direction in ["left", "right"]
                     current_part_consumed_size = segment_contribution.width
                     if current_part_consumed_size > remaining_size:
                         new_edge = (
@@ -451,6 +454,7 @@ class FlowElement:
         width_absolute: Optional[float] = None,
         width_alignment: str = "center",
         until: Optional[str] = None,
+        include_source: bool = False,
         include_endpoint: bool = True,
         **kwargs,
     ) -> "FlowRegion":  # Stringized
@@ -462,6 +466,7 @@ class FlowElement:
                 cross_size_absolute=width_absolute,
                 cross_alignment=width_alignment,
                 until=until,
+                include_source=include_source,
                 include_endpoint=include_endpoint,
                 **kwargs,
             )
@@ -477,6 +482,7 @@ class FlowElement:
         width_absolute: Optional[float] = None,
         width_alignment: str = "center",
         until: Optional[str] = None,
+        include_source: bool = False,
         include_endpoint: bool = True,
         **kwargs,
     ) -> "FlowRegion":  # Stringized
@@ -488,6 +494,7 @@ class FlowElement:
                 cross_size_absolute=width_absolute,
                 cross_alignment=width_alignment,
                 until=until,
+                include_source=include_source,
                 include_endpoint=include_endpoint,
                 **kwargs,
             )
@@ -503,24 +510,21 @@ class FlowElement:
         height_absolute: Optional[float] = None,
         height_alignment: str = "center",
         until: Optional[str] = None,
+        include_source: bool = False,
         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."
-            )
+        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_source=include_source,
+            include_endpoint=include_endpoint,
+            **kwargs,
+        )
 
     def right(
         self,
@@ -529,24 +533,21 @@ class FlowElement:
         height_absolute: Optional[float] = None,
         height_alignment: str = "center",
         until: Optional[str] = None,
+        include_source: bool = False,
         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."
-            )
+        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_source=include_source,
+            include_endpoint=include_endpoint,
+            **kwargs,
+        )
 
     def __repr__(self) -> str:
         return f"<FlowElement for {self.physical_object.__class__.__name__} {self.bbox} in {self.flow}>"

{natural_pdf-0.2.15.dist-info → natural_pdf-0.2.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: natural-pdf
-Version: 0.2.15
+Version: 0.2.16
 Summary: A more intuitive interface for working with PDFs
 Author-email: Jonathan Soma <jonathan.soma@gmail.com>
 License-Expression: MIT

{natural_pdf-0.2.15.dist-info → natural_pdf-0.2.16.dist-info}/RECORD

@@ -62,7 +62,7 @@ natural_pdf/extraction/mixin.py,sha256=dBcp96R8zMQqaRHiB8vpyad8GR89gv5RPXlr8Mt0a
 natural_pdf/extraction/result.py,sha256=PDaCCN2LQBbHsZy0_lrQ0ROeMsnmH1WRoXWOjk9M2o4,1825
 natural_pdf/flows/__init__.py,sha256=cUN4A8hTDLZSRr4PO2W_lR4z6hWpbNG8Seox-IIcrLU,277
 natural_pdf/flows/collections.py,sha256=ErkHWdX6W_y1SjkcA_bGM0uUYRGPWWpRkHip6LHpej0,25740
-natural_pdf/flows/element.py,sha256=T-9uXsIBe7mIim-mQQMep6Ja5dRfWaYIj8g1ak_Bv8c,24892
+natural_pdf/flows/element.py,sha256=AWXGfAo0yhHTA5h0u4teXhRaV_z3McSPGOMFQaSdkJQ,24973
 natural_pdf/flows/flow.py,sha256=BuT3DBqNvLEqYle66-nZFO91i_1s98CAat28Dg-JjGU,86149
 natural_pdf/flows/region.py,sha256=r_cFtBlmPi7ADN3k8oYA1s_vyz8GeQLCnYcv58Zt5eM,52263
 natural_pdf/ocr/__init__.py,sha256=VY8hhvDPf7Gh2lB-d2QRmghLLyTy6ydxlgo1cS4dOSk,2482
@@ -108,7 +108,7 @@ natural_pdf/vision/similarity.py,sha256=HWmXDBNLSOlRWH-_1K3FVR7tSsRuMFqXZwrVhhg2
 natural_pdf/vision/template_matching.py,sha256=91XQt5tp-vmcMX_4b2Bz-YwIAlb-hc8E5ih_qAHQuCk,7145
 natural_pdf/widgets/__init__.py,sha256=QTVaUmsw__FCweFYZebwPssQxxUFUMd0wpm_cUbGZJY,181
 natural_pdf/widgets/viewer.py,sha256=KW3JogdR2TMg2ECUMYp8hwd060hfg8EsYBWxb5IEzBY,24942
-natural_pdf-0.2.15.dist-info/licenses/LICENSE,sha256=9zfwINwJlarbDmdh6iJV4QUG54QSJlSAUcnC1YiC_Ns,1074
+natural_pdf-0.2.16.dist-info/licenses/LICENSE,sha256=9zfwINwJlarbDmdh6iJV4QUG54QSJlSAUcnC1YiC_Ns,1074
 optimization/memory_comparison.py,sha256=0i_foFSRmppj-fY069qjwH36s_zkx-1L2ASAAlepWzA,6541
 optimization/pdf_analyzer.py,sha256=HjrmTgu2qchxPeDckc5kjgxppGwd40UESrYS9Myj7pY,19352
 optimization/performance_analysis.py,sha256=JBXnR9hc7Ix7YCnt3EJPSpsyqIUgKsc7GEffQ_TDCBk,13033
@@ -145,8 +145,8 @@ tools/bad_pdf_eval/llm_enrich.py,sha256=mCh4KGi1HmIkzGjj5rrHz1Osd7sEX1IZ_FW08H1t
 tools/bad_pdf_eval/llm_enrich_with_retry.py,sha256=XUtPF1hUvqd3frDXT0wDTXoonuAivhjM5vgFdZ-tm0A,9373
 tools/bad_pdf_eval/reporter.py,sha256=e1g__mkSB4q02p3mGWOwMhvFs7F2HJosNBxup0-LkyU,400
 tools/bad_pdf_eval/utils.py,sha256=hR95XQ7qf7Cu6BdyX0L7ggGVx-ah5sK0jHWblTJUUic,4896
-natural_pdf-0.2.15.dist-info/METADATA,sha256=TfUHRd_THs6bk7Iwl8UdBjZfnWEXKg_tBiVn_Q9tv28,6960
-natural_pdf-0.2.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-natural_pdf-0.2.15.dist-info/entry_points.txt,sha256=1R_KMv7g60UBBpRqGfw7bppsMNGdayR-iJlb9ohEk_8,81
-natural_pdf-0.2.15.dist-info/top_level.txt,sha256=ZDKhxE_tg508o9BpagsjCGcI8GY4cF_8bg0e0IaLsPI,41
-natural_pdf-0.2.15.dist-info/RECORD,,
+natural_pdf-0.2.16.dist-info/METADATA,sha256=o2x-_hFHY64xJCCZM7CwEHo96vAU8XzFejobi1neE2g,6960
+natural_pdf-0.2.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+natural_pdf-0.2.16.dist-info/entry_points.txt,sha256=1R_KMv7g60UBBpRqGfw7bppsMNGdayR-iJlb9ohEk_8,81
+natural_pdf-0.2.16.dist-info/top_level.txt,sha256=ZDKhxE_tg508o9BpagsjCGcI8GY4cF_8bg0e0IaLsPI,41
+natural_pdf-0.2.16.dist-info/RECORD,,