From 97c4ec8ca6e07b9b156147f1f0576ee716b4008b Mon Sep 17 00:00:00 2001
From: Filippo Pesavento <pesaventofilippo@gmail.com>
Date: Mon, 9 Oct 2023 21:32:46 +0200
Subject: [PATCH] ENH: Add pick_inlaps() and pick_outlaps() functions (#454)

---
 fastf1/core.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/fastf1/core.py b/fastf1/core.py
index 0d845dd0e..582b6b495 100644
--- a/fastf1/core.py
+++ b/fastf1/core.py
@@ -2921,6 +2921,34 @@ def pick_wo_box(self) -> "Laps":
         """
         return self[pd.isnull(self['PitInTime']) & pd.isnull(self['PitOutTime'])]
 
+    def pick_box_laps(self, which: str = 'both') -> "Laps":
+        """Return all laps which are either in-laps, out-laps, or both.
+        Note: a lap could be an in-lap and an out-lap at the same time.
+        In that case, it will get returned regardless of the 'which'
+        parameter.
+
+        Args:
+            which (str): one of 'in'/'out'/'both'
+
+                - which='in': only laps in which the driver entered
+                  the pit lane are returned
+                - which='out': only laps in which the driver exited
+                  the pit lane are returned
+                - which='both': both in-laps and out-laps are returned
+
+        Returns:
+            instance of :class:`Laps`
+        """
+        if which == 'in':
+            return self[~pd.isnull(self['PitInTime'])]
+        elif which == 'out':
+            return self[~pd.isnull(self['PitOutTime'])]
+        elif which == 'both':
+            return self[~pd.isnull(self['PitInTime'])
+                        | ~pd.isnull(self['PitOutTime'])]
+        else:
+            raise ValueError(f"Invalid value '{which}' for kwarg 'which'")
+
     def pick_not_deleted(self) -> "Laps":
         """Return all laps whose lap times are NOT deleted.