[c++] - 实现类似printf这样的函数
在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,到此为止,可变参数的函数编写应该也就很明了啦~
个性签名:做要做好,做到不三不四不如不做。