STL源码剖析之ROUND_UP函数实现原理及其拓展

最新推荐文章于 2024-11-22 21:22:05 发布

weixin_33843409

最新推荐文章于 2024-11-22 21:22:05 发布

阅读量296

点赞数

CC 4.0 BY-SA版权

文章标签： c/c++

原文链接：http://www.cnblogs.com/muyangshaonian/p/9650474.html

本文详细解析了STL源码中allocator使用的ROUND_UP函数实现原理，介绍了如何通过位运算快速调整内存分配大小至8的倍数，并探讨了不同情况下的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

STL源码allocator的实现中有一个ROUND_UP函数，作用是将非8倍数的整数上调到8的倍数。

// instead of enum { x = N }, but fewcompilers accept the former.
# ifndef__SUNPRO_CC
    enum {__ALIGN = 8};
    enum {__MAX_BYTES = 128};
    enum {__NFREELISTS = __MAX_BYTES/__ALIGN};
# endif
  static size_t ROUND_UP(size_t bytes) {
        return (((bytes) + __ALIGN-1) &~(__ALIGN - 1));
  }

这里记录一下ROUND_UP函数的实现原理，并考虑相关的拓展情形。

1 原理

int a;
    int size = 8;        <----> 1000(bin)

    计算a以size为倍数的下界数:
    就让这个数(要计算的这个数)表示成二进制时，最后三位为0就可以达到这个目标。只要下面这个数与a进行"与运算"就可以了:
    11111111 11111111 11111111 11111000
    而上面这个数实际下就是 ~(size - 1)，可以将该数称为size的对齐掩码size_mask.

    计算a以size为倍数的上下界数:

#define alignment_down(a, size) (a & (~(size-1)) )
#define alignment_up(a, size)   ((a+size-1) & (~ (size-1)))

注: 上界数的计算方法，如果要求出比a大的是不是需要加上8就可以了?可是如果a本身就是8的倍数,这样加8不就错了吗,所以在a基础上加上(size - 1), 然后与size的对齐掩码进行与运算.

    例如:
    a=0, size=8, 则alignment_down(a,size)=0, alignment_up(a,size)=0.
    a=6, size=8, 则alignment_down(a,size)=0, alignment_up(a,size)=8.
    a=8, size=8, 则alignment_down(a,size)=8, alignment_up(a,size)=8.
    a=14, size=8,则alignment_down(a,size)=8, alignment_up(a,size)=16.
    注:size应当为2的n次方, 即2, 4, 8, 16, 32, 64, 128, 256, 1024, 2048, 4096 ...

2 拓展

如果size不是2的幂，则不能采用位运算的方法，可以采用普通的十进制方法。

size_t round_low(size_t x, size_t N) {
	return x - (x% N);
}
size_t round_up(size_t x,size_t N) {
	return (x + N - 1) - ((x + N - 1) % N);
}

转载于:https://www.cnblogs.com/muyangshaonian/p/9650474.html