将边缘包裹的条件逻辑转为数学表达式并修正边缘值映射问题

问题背景 #

你需要实现一个无分支的浮点映射函数，替代原有的条件判断函数wrap_ternary：

float wrap_ternary(float x) {
    if (x < 0.0f) return x + 1.0f;
    else if (x > 1.0f) return x - 1.0f;
    else return x;
}

该函数将[-1.0, 2.0]范围内的输入x映射到[0.0, 1.0]，其中x=1.0和x=2.0时返回1.0。

原模运算实现wrap_mod在边缘值上不符合预期：

float wrap_mod(float x) {
    return (((x + 1.0f) % 2.0f + 1.0f) % 1.0f);
}

当x=1.0或x=2.0时，该函数返回0.0，而非预期的1.0。

问题原因 #

C语言中（包括浮点模运算fmod），当被除数是除数的整数倍时，模运算结果为0。例如：

• x=1.0时，(1.0+1.0)%2.0=0.0，后续运算得到(0.0+1.0)%1.0=0.0
• x=2.0时，(2.0+1.0)%2.0=1.0，后续运算得到(1.0+1.0)%1.0=0.0

这种行为导致边缘值的映射结果不符合原函数要求。

修正方案 #

可以通过偏移后取模再还原的方式，实现无分支的正确映射，表达式如下：

#include <math.h> // 需要包含math.h以使用fmod函数

float wrap_mod_fixed(float x) {
    return fmod(x - 0.5f, 1.0f) + 0.5f;
}

验证所有边界及中间值 #

• x=-1.0：fmod(-1.0-0.5,1.0)=-0.5，加0.5后返回0.0 ✔️
• x=-0.5：fmod(-0.5-0.5,1.0)=0.0，加0.5后返回0.5 ✔️
• x=0.0：fmod(0.0-0.5,1.0)=-0.5，加0.5后返回0.0 ✔️
• x=0.5：fmod(0.5-0.5,1.0)=0.0，加0.5后返回0.5 ✔️
• x=1.0：fmod(1.0-0.5,1.0)=0.5，加0.5后返回1.0 ✔️
• x=1.5：fmod(1.5-0.5,1.0)=0.0，加0.5后返回0.5 ✔️
• x=2.0：fmod(2.0-0.5,1.0)=0.5，加0.5后返回1.0 ✔️

该实现完全符合原函数的映射逻辑，且无任何条件判断。

内容的提问来源于stack exchange，提问作者RoyNeary