Как правильно написать макрос для такого выражения?

Question

[iihaarr]

Хочу написать макрос вида:
name![type; {el, ..., el}, ... , {el, ... , el}]

Написал такое:

[$type:ty, ($row:block ($el:expr)+)+] => {
}

Но выдает ошибку при
name![i32; {1}]: missing tokens in macro arguments
Собственно вопрос в заголовке, а также можно ли как-то узнать, сколько раз повторялись элементы в скобках и само количество скобок?

Answer 1

[Василий Банников]

Вот эта ошибка:

missing tokens in macro arguments

Означает, что раст не может сматчить макрос с таким аргументом. Дело в том, что "1" не является BlockExpression, который ты ожидаешь в начале (https://doc.rust-lang.org/reference/expressions/bl...)

Вот примерно рабочий макрос, который ничего не делает.

macro_rules! name {
    ($t: ty, $({$($el:expr ),*}),*) => {42};
    
}
fn main() {
    name!(i32, {1, 2, 3 ,4}, {1,2});
}

Comments

[iihaarr]

А можно ли как-то подсчитать сколько таких блоков {} и сколько в них элементов?
Что-то типо такого
($t: ty, $({$brackets?($el:expr ),*}),*)

[Василий Банников]

iihaarr, всм "посчитать"?
Могу ошибаться, но для всяких сложных не-декларативных макросов нужно писать процедурный макрос, а это уже сложнее немного.

[iihaarr]

Василий Банников, То есть для такого входа name!(i32, {1, 2, 3 ,4}, {1,2}) в результате будет вектор, где кол-во элементов в векторе равно кол-ву {}, а vec[i] = кол-во элементов в i-тых {}

[Василий Банников]

iihaarr, не уверен, но мне кажется, что да такого нужны процедурные макросы.
Либо я не достаточно хорошо изучил возможности декларативных.
А какую задачу ты хочешь так решить?

[iihaarr]

Василий Банников, Создание матрицы по макросу. Количество скобочек - количество строк, количество элементов в скобочках - количество столбцов. Если хоть в одних скобочках отличается количество элементов, то ошибка.

[Василий Банников]

iihaarr, тогда и правда легче через процедурный макрос.
А почему бы не сделать обычный двумерный массив?

[iihaarr]

Василий Банников, Ради производительности. А если даже делать через двумерный массив, то как будет выглядеть цикл добавления в вектор каждую строку? Ведь по сути после каждой строки его надо обнулять. Как вообще в макросе выглядет цикл?

[Василий Банников]

iihaarr, не, можно же вроде именно как массив, а не вектор, без циклов.

[iihaarr]

Василий Банников, Просто чтобы создать надо знать размер его, а как посчитать количество скобочек тогда? У нас же только по одной скобочке известно

[Василий Банников]

iihaarr, раст же автоматом считает размер, когда создаёшь массив и не указываешь размер

[iihaarr]

Василий Банников, ($t: ty, $({$($el:expr ),*}),*)
В этом же макросе неизвестно же количество строк и столбцов.

[Василий Банников]

iihaarr, если использовать процедурный макрос - вполне можно посчитать, сколько их там.
А в декларативном, компилятор сам узнает, сколько их там, когда оно развернётся в код.

[iihaarr]

Василий Банников, Буду разбираться. Благодарю за ответы

[Василий Банников]

iihaarr, https://veykril.github.io/tlborm

[Дмитрий Беляев]

Василий Банников, iihaarr, можно вот такое извращение написать для подсчета:
https://github.com/tokio-rs/tokio/blob/master/toki...

Но как по мне, через proc-macro такое сделать все же проще...

Хотя если вопрос стоит только в подсчете количества в компайл-тайм, то можно так попробовать:

macro_rules! name {
    ($t:ty, $({$($el:expr),*}),*) => {
        vec![$(
            name!(@chunk {$($el),*}) as $t
        ),*]
    };
    (@chunk {}) => { 0 };
    (@chunk {$($el:expr),+}) => {
        std::mem::size_of_val(&[$(
            name!(@byte $el)
        ),+])
    };
    (@byte $el:expr) => { 0u8 };
}

fn main() {
    let v = name!(i32, {1, 2, 3, 4}, {1, 2}, {});
    println!("{v:?}")
}

std::mem::size_of_val - константная, для sized типов гарантировано считается в compile-time, и из условногоstd::mem::size_of_val(&[0u8, 0, 0, 0])генерируется movq $4, (%rsp) без создания массива