[c++] - 实现类似printf这样的函数

来源:http://www.vimer.cn/2009/12/cc%E5%AE%9E%E7%8E%B0%E5%A4%9A%E5%8F%82%E6%95%B0%E5%87%BD%E6%95%B0%E7%BC%96%E7%A8%8B.html

 

在C/C++中,我们经常会需要实现类似printf这样的函数,即函数的参数个数是不定的,这个时候就需要用到我们这篇文章讲到的方法啦。
首先,我们要知道这种函数,如何来定义。比如我想实现一个函数能够支持 fun("%d",1);
那么这个函数的定义实际上如下:

void fun(const char *fmt, ...);

其中...的意思是说参数无法一一列出,所以用...代替,至于怎么解,我们稍后再说。
比较特殊的一点是,如果你希望将上面的函数定义成一个宏,那么这个宏可以这样写:

#define FUN(fmt, args...)    fun(fmt, ##args)

又假设你希望宏能够自动加上换行符,那么可以这样写:

#define FUN(fmt, args...)    fun(fmt"\n", ##args)

OK,那么函数定义的问题我们就解决啦,但是怎么来解呢?
C里面提供了va_start,va_arg,va_end这样几个函数,解释如下:
va_start使argp指向第一个可选参数。va_arg返回参数列表中的当前参数并使argp指向参数列表中的下一个参数。va_end把argp指针清为NULL。函数体内可以多次遍历这些参数,但是都必须以va_start开始,并以va_end结尾。
可能只是这样说并不是很清楚,我们举个例子。
1、假设我们现在想要fun实现和printf一样的功能,那么实际上,我们是不需要把所有解析出来的,我们只有把参数原样传给printf即可,代码如下:

void fun(const char* fmt,...)
{
va_list ap;
va_start(ap, fmt);//将ap指向fmt后的第一个参数
vfprintf(stderr,fmt,ap);
va_end(ap);//将ap置为NULL
}

2、假设我们现在是要将传入的参数都取出来,那么我们就要用到va_arg了,代码如下:

void fun(const char* fmt,...)
{
va_list ap;
va_start(ap, fmt);//将ap指向fmt后的第一个参数
int value = va_arg(ap,int);//前提是我们知道第一个参数是int型;指针指向下一个参数
printf("value[%d]\n",value);
va_end(ap);//将ap置为NULL
}

其实到这里大家也就不难发现,我们如何能遍历所有的参数了,只要规定好最后一个参数为一个特殊字符,比如说-1,然后判断到这个值就停止就行

void fun(const char* fmt,...)
{
va_list ap;
va_start(ap, fmt);//将ap指向fmt后的第一个参数
int value;
do{
value = va_arg(ap,int);//前提是我们知道第一个参数是int型;指针指向下一个参数
printf("value[%d]\n",value);
}while(value!=-1);
va_end(ap);//将ap置为NULL
}

OK,到此为止,可变参数的函数编写应该也就很明了啦~

posted @ 2012-03-26 08:57  炎峰森林影  阅读(2993)  评论(0编辑  收藏  举报