Public interface for layout.recursively_apply and broadcast_and_apply.

src/awkward/_v2/contents/bitmaskedarray.py

@@ -6,7 +6,7 @@
 
 import awkward as ak
 from awkward._v2.index import Index
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.contents.bytemaskedarray import ByteMaskedArray
 from awkward._v2.forms.bitmaskedform import BitMaskedForm
 from awkward._v2.forms.form import _parameters_equal
@@ -18,6 +18,28 @@
 class BitMaskedArray(Content):
     is_OptionType = True
 
+    def copy(
+        self,
+        mask=unset,
+        content=unset,
+        valid_when=unset,
+        length=unset,
+        lsb_order=unset,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return BitMaskedArray(
+            self._mask if mask is unset else mask,
+            self._content if content is unset else content,
+            self._valid_when if valid_when is unset else valid_when,
+            self._length if length is unset else length,
+            self._lsb_order if lsb_order is unset else lsb_order,
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(
         self,
         mask,
@@ -635,6 +657,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )
 

src/awkward/_v2/contents/bytemaskedarray.py

@@ -6,7 +6,7 @@
 
 import awkward as ak
 from awkward._v2.index import Index
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.forms.bytemaskedform import ByteMaskedForm
 from awkward._v2.forms.form import _parameters_equal
 
@@ -17,6 +17,24 @@
 class ByteMaskedArray(Content):
     is_OptionType = True
 
+    def copy(
+        self,
+        mask=unset,
+        content=unset,
+        valid_when=unset,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return ByteMaskedArray(
+            self._mask if mask is unset else mask,
+            self._content if content is unset else content,
+            self._valid_when if valid_when is unset else valid_when,
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(
         self, mask, content, valid_when, identifier=None, parameters=None, nplike=None
     ):
@@ -1052,6 +1070,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )
 

src/awkward/_v2/contents/content.py

@@ -12,6 +12,8 @@
 np = ak.nplike.NumpyMetadata.instance()
 numpy = ak.nplike.Numpy.instance()
 
+unset = object()
+
 
 class Content:
     is_NumpyType = False
@@ -1346,7 +1348,7 @@ def recursively_apply(
             action,
             behavior,
             1,
-            depth_context,
+            copy.copy(depth_context),
             lateral_context,
             {
                 "allow_records": allow_records,

src/awkward/_v2/contents/emptyarray.py

@@ -1,7 +1,7 @@
 # BSD 3-Clause License; see https://github.com/scikit-hep/awkward-1.0/blob/main/LICENSE
 
 import awkward as ak
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.forms.emptyform import EmptyForm
 from awkward._v2.forms.form import _parameters_equal
 
@@ -13,6 +13,18 @@ class EmptyArray(Content):
     is_NumpyType = True
     is_UnknownType = True
 
+    def copy(
+        self,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return EmptyArray(
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(self, identifier=None, parameters=None, nplike=None):
         if nplike is None:
             nplike = numpy
@@ -334,6 +346,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )
 

src/awkward/_v2/contents/indexedarray.py

@@ -4,7 +4,7 @@
 
 import awkward as ak
 from awkward._v2.index import Index
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.forms.indexedform import IndexedForm
 from awkward._v2.forms.form import _parameters_equal
 
@@ -15,6 +15,22 @@
 class IndexedArray(Content):
     is_IndexedType = True
 
+    def copy(
+        self,
+        index=unset,
+        content=unset,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return IndexedArray(
+            self._index if index is unset else index,
+            self._content if content is unset else content,
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(self, index, content, identifier=None, parameters=None, nplike=None):
         if not (
             isinstance(index, Index)
@@ -1226,6 +1242,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )
 

src/awkward/_v2/contents/indexedoptionarray.py

@@ -4,7 +4,7 @@
 
 import awkward as ak
 from awkward._v2.index import Index
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.forms.indexedoptionform import IndexedOptionForm
 from awkward._v2.forms.form import _parameters_equal
 
@@ -16,6 +16,22 @@ class IndexedOptionArray(Content):
     is_OptionType = True
     is_IndexedType = True
 
+    def copy(
+        self,
+        index=unset,
+        content=unset,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return IndexedOptionArray(
+            self._index if index is unset else index,
+            self._content if content is unset else content,
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(self, index, content, identifier=None, parameters=None, nplike=None):
         if not (
             isinstance(index, Index)
@@ -1665,6 +1681,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )
 

src/awkward/_v2/contents/listarray.py

@@ -4,7 +4,7 @@
 
 import awkward as ak
 from awkward._v2.index import Index
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.contents.listoffsetarray import ListOffsetArray
 from awkward._v2.forms.listform import ListForm
 from awkward._v2.forms.form import _parameters_equal
@@ -15,6 +15,24 @@
 class ListArray(Content):
     is_ListType = True
 
+    def copy(
+        self,
+        starts=unset,
+        stops=unset,
+        content=unset,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return ListArray(
+            self._starts if starts is unset else starts,
+            self._stops if stops is unset else stops,
+            self._content if content is unset else content,
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(
         self, starts, stops, content, identifier=None, parameters=None, nplike=None
     ):
@@ -1458,6 +1476,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )
 

src/awkward/_v2/contents/listoffsetarray.py

@@ -4,7 +4,7 @@
 
 import awkward as ak
 from awkward._v2.index import Index
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.forms.listoffsetform import ListOffsetForm
 from awkward._v2.forms.form import _parameters_equal
 
@@ -15,6 +15,22 @@
 class ListOffsetArray(Content):
     is_ListType = True
 
+    def copy(
+        self,
+        offsets=unset,
+        content=unset,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return ListOffsetArray(
+            self._offsets if offsets is unset else offsets,
+            self._content if content is unset else content,
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(self, offsets, content, identifier=None, parameters=None, nplike=None):
         if not isinstance(offsets, Index) and offsets.dtype in (
             np.dtype(np.int32),
@@ -2082,6 +2098,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )
 

src/awkward/_v2/contents/numpyarray.py

@@ -2,7 +2,7 @@
 
 import copy
 import awkward as ak
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.forms.numpyform import NumpyForm
 from awkward._v2.forms.form import _parameters_equal
 from awkward._v2.types.numpytype import primitive_to_dtype
@@ -14,6 +14,20 @@
 class NumpyArray(Content):
     is_NumpyType = True
 
+    def copy(
+        self,
+        data=unset,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return NumpyArray(
+            self._data if data is unset else data,
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(self, data, identifier=None, parameters=None, nplike=None):
         if nplike is None:
             nplike = ak.nplike.of(data)
@@ -1312,6 +1326,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )
 

src/awkward/_v2/contents/recordarray.py

@@ -7,7 +7,7 @@
 
 import awkward as ak
 from awkward._v2.record import Record
-from awkward._v2.contents.content import Content
+from awkward._v2.contents.content import Content, unset
 from awkward._v2.forms.recordform import RecordForm
 from awkward._v2.forms.form import _parameters_equal
 
@@ -18,6 +18,24 @@
 class RecordArray(Content):
     is_RecordType = True
 
+    def copy(
+        self,
+        contents=unset,
+        fields=unset,
+        length=unset,
+        identifier=unset,
+        parameters=unset,
+        nplike=unset,
+    ):
+        return RecordArray(
+            self._contents if contents is unset else contents,
+            self._fields if fields is unset else fields,
+            self._length if length is unset else length,
+            self._identifier if identifier is unset else identifier,
+            self._parameters if parameters is unset else parameters,
+            self._nplike if nplike is unset else nplike,
+        )
+
     def __init__(
         self,
         contents,
@@ -1007,6 +1025,8 @@ def continuation():
             depth_context=depth_context,
             lateral_context=lateral_context,
             continuation=continuation,
+            behavior=behavior,
+            nplike=self._nplike,
             options=options,
         )