Skip to content

Commit 1e144ec

Browse files
committed
remove explicit compute have in a later PR
Signed-off-by: Joe Isaacs <joe.isaacs@live.co.uk>
1 parent f6cbb3c commit 1e144ec

File tree

10 files changed

+61
-449
lines changed

10 files changed

+61
-449
lines changed

vortex-array/src/arrays/dict/execute.rs

Lines changed: 0 additions & 117 deletions
This file was deleted.

vortex-array/src/arrays/dict/mod.rs

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,6 @@ mod array;
1010
pub use array::*;
1111

1212
mod compute;
13-
mod execute;
14-
15-
pub use execute::take_canonical;
1613

1714
pub mod vtable;
1815
pub use vtable::*;

vortex-array/src/arrays/dict/vtable/mod.rs

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ use vortex_error::vortex_err;
1111

1212
use super::DictArray;
1313
use super::DictMetadata;
14-
use super::take_canonical;
1514
use crate::ArrayRef;
1615
use crate::Canonical;
1716
use crate::DeserializeMetadata;
@@ -20,6 +19,7 @@ use crate::SerializeMetadata;
2019
use crate::VectorExecutor;
2120
use crate::arrays::vtable::rules::PARENT_RULES;
2221
use crate::buffer::BufferHandle;
22+
use crate::compute::take;
2323
use crate::executor::ExecutionCtx;
2424
use crate::serde::ArrayChildren;
2525
use crate::vtable;
@@ -132,9 +132,8 @@ impl VTable for DictVTable {
132132
}
133133

134134
fn execute(array: &Self::Array, ctx: &mut ExecutionCtx) -> VortexResult<Canonical> {
135-
let values = array.values().execute(ctx)?;
136-
let codes = array.codes().execute(ctx)?.into_primitive();
137-
Ok(take_canonical(values, &codes))
135+
// TODO(joe): remove take compute fn
136+
take(array.values(), array.codes()).and_then(|a| a.execute(ctx))
138137
}
139138

140139
fn reduce_parent(

vortex-array/src/arrays/filter/execute.rs

Lines changed: 0 additions & 162 deletions
This file was deleted.

vortex-array/src/arrays/filter/mod.rs

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,8 @@
22
// SPDX-FileCopyrightText: Copyright the Vortex contributors
33

44
mod array;
5-
mod execute;
65
mod rules;
76
mod vtable;
87

98
pub use array::*;
10-
pub use execute::FILTER_SLICES_SELECTIVITY_THRESHOLD;
11-
pub use execute::filter_canonical;
12-
pub use execute::filter_slice;
139
pub use vtable::*;

vortex-array/src/arrays/filter/vtable.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ use vortex_error::vortex_ensure;
1515
use vortex_mask::Mask;
1616
use vortex_scalar::Scalar;
1717

18-
use super::execute::filter_canonical;
1918
use crate::Array;
2019
use crate::ArrayBufferVisitor;
2120
use crate::ArrayChildVisitor;
@@ -31,6 +30,7 @@ use crate::VortexSessionExecute;
3130
use crate::arrays::filter::array::FilterArray;
3231
use crate::arrays::filter::rules::PARENT_RULES;
3332
use crate::buffer::BufferHandle;
33+
use crate::compute::filter;
3434
use crate::executor::ExecutionCtx;
3535
use crate::serde::ArrayChildren;
3636
use crate::stats::StatsSetRef;
@@ -114,8 +114,7 @@ impl VTable for FilterVTable {
114114
}
115115

116116
fn execute(array: &Self::Array, ctx: &mut ExecutionCtx) -> VortexResult<Canonical> {
117-
let child = array.child.execute(ctx)?;
118-
Ok(filter_canonical(child, &array.mask))
117+
filter(array.child.as_ref(), &array.mask).and_then(|a| a.execute(ctx))
119118
}
120119

121120
fn reduce_parent(

0 commit comments

Comments
 (0)