避免重复/循环取消切换

Question

我有一个紧密循环运行的热点代码：

for (i = 0; i < big; i++)
{
    if (condition1) {
        do1();
    } else if (condition2) {
        do2();
    } else {
        do3();
    }
    // Shared code goes here
    // More shared code goes here
}

因为 condition1 和 condition2 是不变的，我取消了循环到

if (condition1) {
    for (i = 0; i < big; i++)
    {
        do1();
        // Shared code goes here
        // More shared code goes here
    }
} else if (condition 2) {
    for (i = 0; i < big; i++)
    {
        do2();
        // Shared code goes here
        // More shared code goes here
    }
} else {
    for (i = 0; i < big; i++)
    {
        do3();
        // Shared code goes here
        // More shared code goes here
    }
}

这运行得好多了，但我想知道是否有一种聪明的方法可以做到这一点而不重复自己？

Answer 1

我想你可以声明一个函数指针和一些函数 foo():

typedef void (*fp)(void);

void foo(int big, fp f) {
    for (int i = 0; i < big; ++i) {
        f();
        // Shared code goes here
        // More shared code goes her
    }
}

然后像这样更改您的代码：

if (condition1) {
    foo(big, do1);
} else if (condition2) {
    foo(big, do2);
} else {
    foo(big, do3);
}

Answer 2

另一个可能稍微更有效的选择是使用宏为您构建代码：

#define DO_N(name, ...) for(int i = 0; i < big; i++){name(__VA_ARGS__);/*shared code*/}

if (condition1) {
    DO_N(do1, .../*arguments here*/)
} else if (condition 2) {
    DO_N(do2, ...)
} else {
    DO_N(do3, ...)
}

#undef DO_N

它很丑，但我认为它可以满足您的需求，并且可能允许在函数指针不允许的地方进行内联。

此外，您可能会发现将共享代码放在单独的宏或函数中更具可读性。

避免重复/循环取消切换

Avoiding repetition / loop unswitching

c

optimization

dry