Повторное использование вложенных циклов без копирования и вставки - PullRequest
Новая
       25

Повторное использование вложенных циклов без копирования и вставки

7 голосов
/ 23 августа 2011

Предположим, у меня есть этот вложенный цикл 

for (int a=1; a<MAX_A; ++a)
  for (int b=1; b<MAX_B; ++b)
    for (int c=1; c<MAX_C; ++c)
    {
       do_something(a, b ,c);
    }

и я повторно использую этот цикл в различных частях моего кода, меняя функцию do_something. Довольно скучно каждый раз переписывать первые три строки. В Python, например, я бы создал генератор для возврата итератора (1, 1, 1), (1, 1, 2), ... или что-то вроде itertools.product.

В c ++ единственное решение, которое я имею в виду, - определить макрос. Что-то лучше? E

Ответы [ 2 ]

11 голосов
/ 23 августа 2011

Использовать шаблоны: 

template<typename Func>
inline void do_something_loop(Func f)
{
    for (int a=1; a<MAX_A; ++a) 
      for (int b=1; b<MAX_B; ++b) 
        for (int c=1; c<MAX_C; ++c) 
        { 
           f(a, b ,c); 
        } 
}

Это можно вызвать с любым указателем функции или функциональным объектом, соответствующим сигнатуре, например: 

void do_something(int a, int b, int c) { /* stuff */ }

do_something_loop(do_something);

или с объектом функции: 

struct do_something
{
    void operator()(int a, int b, int c) { /* stuff */ }
};

do_something_loop(do_something());

Или, если ваш компилятор поддерживает C ++ 11, даже с лямбда-выражением: 

do_something_loop([](int a, int b, int c) { /* stuff */ });

Обратите внимание, что вы также можете объявить параметр f как указатель на функцию с подписью void(*f)(int,int,int) вместо использования шаблона, но он менее гибкий (он не будет работать с функциональными объектами (включая результат std :: bind) или лямбда-выражениями).

3 голосов
/ 23 августа 2011

Сделать функцию вспомогательного цикла: 

template<typename Work>
void loop(Work work) {
  for (int a=1; a<MAX_A; ++a)
    for (int b=1; b<MAX_B; ++b)
      for (int c=1; c<MAX_C; ++c)
      {
        work(a, b ,c);
      }
}

Определите работу, которую нужно сделать: 

void some_work(int a, int b, int c) {
}

Тогда используйте: 

loop(some_work);

Обратите внимание, что теперь вам не нужно использовать только функцию, но вы можете использовать функциональный объект (объект, для которого перегружен operator()).

...