Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[framework] vector zip destroy_empty + allow return values in lambdas #20745

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
+33 −16
Open

[framework] vector zip destroy_empty + allow return values in lambdas #20745

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
5fb3456
vector and option macro improvements
damirka Dec 27, 2024
7f31efc
adds _ to macro lambdas
damirka Dec 27, 2024
f40b77f
Update crates/sui-framework/packages/move-stdlib/sources/macros.move
damirka Jan 17, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
vector and option macro improvements
  • Loading branch information
@damirka
damirka committed Dec 27, 2024
commit 5fb34563cda21425618ed7d8445ade21a30828a5
8 changes: 4 additions & 4 deletions crates/sui-framework/packages/move-stdlib/sources/option.move
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -144,26 +144,26 @@ public fun to_vec<Element>(t: Option<Element>): vector<Element> {
// === Macro Functions ===

/// Destroy `Option<T>` and call the closure `f` on the value inside if it holds one.
public macro fun destroy<$T>($o: Option<$T>, $f: |$T|) {
public macro fun destroy<$T>($o: Option<$T>, $f: |$T| -> _) {
let o = $o;
o.do!($f);
}

/// Destroy `Option<T>` and call the closure `f` on the value inside if it holds one.
public macro fun do<$T>($o: Option<$T>, $f: |$T|) {
public macro fun do<$T>($o: Option<$T>, $f: |$T| -> _) {
let o = $o;
if (o.is_some()) $f(o.destroy_some())
else o.destroy_none()
}

/// Execute a closure on the value inside `t` if it holds one.
public macro fun do_ref<$T>($o: &Option<$T>, $f: |&$T|) {
public macro fun do_ref<$T>($o: &Option<$T>, $f: |&$T| -> _) {
let o = $o;
if (o.is_some()) $f(o.borrow());
}

/// Execute a closure on the mutable reference to the value inside `t` if it holds one.
public macro fun do_mut<$T>($o: &mut Option<$T>, $f: |&mut $T|) {
public macro fun do_mut<$T>($o: &mut Option<$T>, $f: |&mut $T| -> _) {
let o = $o;
if (o.is_some()) $f(o.borrow_mut());
}
Expand Down
17 changes: 9 additions & 8 deletions crates/sui-framework/packages/move-stdlib/sources/vector.move
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -172,30 +172,30 @@ public macro fun tabulate<$T>($n: u64, $f: |u64| -> $T): vector<$T> {

/// Destroy the vector `v` by calling `f` on each element and then destroying the vector.
/// Does not preserve the order of elements in the vector (starts from the end of the vector).
public macro fun destroy<$T>($v: vector<$T>, $f: |$T|) {
public macro fun destroy<$T>($v: vector<$T>, $f: |$T| -> _) {
let mut v = $v;
while (v.length() != 0) $f(v.pop_back());
v.destroy_empty();
}

/// Destroy the vector `v` by calling `f` on each element and then destroying the vector.
/// Preserves the order of elements in the vector.
public macro fun do<$T>($v: vector<$T>, $f: |$T|) {
public macro fun do<$T>($v: vector<$T>, $f: |$T| -> _) {
let mut v = $v;
v.reverse();
v.length().do!(|_| $f(v.pop_back()));
v.destroy_empty();
}

/// Perform an action `f` on each element of the vector `v`. The vector is not modified.
public macro fun do_ref<$T>($v: &vector<$T>, $f: |&$T|) {
public macro fun do_ref<$T>($v: &vector<$T>, $f: |&$T| -> _) {
let v = $v;
v.length().do!(|i| $f(&v[i]))
}

/// Perform an action `f` on each element of the vector `v`.
/// The function `f` takes a mutable reference to the element.
public macro fun do_mut<$T>($v: &mut vector<$T>, $f: |&mut $T|) {
public macro fun do_mut<$T>($v: &mut vector<$T>, $f: |&mut $T| -> _) {
let v = $v;
v.length().do!(|i| $f(&mut v[i]))
}
Expand Down Expand Up @@ -295,19 +295,20 @@ public macro fun all<$T>($v: &vector<$T>, $f: |&$T| -> bool): bool {
/// Destroys two vectors `v1` and `v2` by calling `f` to each pair of elements.
/// Aborts if the vectors are not of the same length.
/// The order of elements in the vectors is preserved.
public macro fun zip_do<$T1, $T2>($v1: vector<$T1>, $v2: vector<$T2>, $f: |$T1, $T2|) {
public macro fun zip_do<$T1, $T2>($v1: vector<$T1>, $v2: vector<$T2>, $f: |$T1, $T2| -> _) {
let v1 = $v1;
let mut v2 = $v2;
v2.reverse();
let len = v1.length();
assert!(len == v2.length());
v1.do!(|el1| $f(el1, v2.pop_back()));
v2.destroy_empty();
}

/// Destroys two vectors `v1` and `v2` by calling `f` to each pair of elements.
/// Aborts if the vectors are not of the same length.
/// Starts from the end of the vectors.
public macro fun zip_do_reverse<$T1, $T2>($v1: vector<$T1>, $v2: vector<$T2>, $f: |$T1, $T2|) {
public macro fun zip_do_reverse<$T1, $T2>($v1: vector<$T1>, $v2: vector<$T2>, $f: |$T1, $T2| -> _) {
let v1 = $v1;
let mut v2 = $v2;
let len = v1.length();
Expand All @@ -319,7 +320,7 @@ public macro fun zip_do_reverse<$T1, $T2>($v1: vector<$T1>, $v2: vector<$T2>, $f
/// elements. The vectors are not modified.
/// Aborts if the vectors are not of the same length.
/// The order of elements in the vectors is preserved.
public macro fun zip_do_ref<$T1, $T2>($v1: &vector<$T1>, $v2: &vector<$T2>, $f: |&$T1, &$T2|) {
public macro fun zip_do_ref<$T1, $T2>($v1: &vector<$T1>, $v2: &vector<$T2>, $f: |&$T1, &$T2| -> _) {
let v1 = $v1;
let v2 = $v2;
let len = v1.length();
Expand All @@ -334,7 +335,7 @@ public macro fun zip_do_ref<$T1, $T2>($v1: &vector<$T1>, $v2: &vector<$T2>, $f:
public macro fun zip_do_mut<$T1, $T2>(
$v1: &mut vector<$T1>,
$v2: &mut vector<$T2>,
$f: |&mut $T1, &mut $T2|,
$f: |&mut $T1, &mut $T2| -> _,
) {
let v1 = $v1;
let v2 = $v2;
Expand Down