3.8 用户自定义字面量

类别:部分人

在C/C++程序中,程序员常常会使用结构体或者类来创造新的类型,以满足实际的需求。比如在进行科学计算时,用户可能需要用到复数(通常会包含实部和虚部两部分)。而对于颜色,用户通常会需要一个四元组(三原色及Alpha)。而对于奥运会组委会,他们则常常会需要七元组(标示来自七大洲的状况)。不过,有的时候自定义类型也会有些书写的麻烦,尤其是用户想声明一个自定义类型的“字面量”(literal)的时候。我们可以看看代码清单3-44所示的例子。

代码清单3-44


include <iostream>

using namespace std;

typedef unsigned char uint8;

struct RGBA{

uint8 r;

uint8 g;

uint8 b;

uint8 a;

RGBA(uint8 R,uint8 G,uint8 B,uint8 A=0):

r(R),g(G),b(B),a(A){}

};

std::ostream&operator<<(std::ostream&out,RGBA&col){

return out<<"r:"<<(int)col.r

<<",g:"<<(int)col.g

<<",b:"<<(int)col.b

<<",a:"<<(int)col.a<<endl;

}

void blend(RGBA&col1,RGBA&col2){

cout<<"blend"<<endl<<col1<<col2<<endl;

}

int main(){

RGBA col1(255,240,155);

RGBA col2({15,255,10,7});//C++11风格的列表初始化

blend(col1,col2);

}

//编译选项:g++ -std=c++11 3-8-1.cpp


在代码清单3-44所示的例子中,我们在main函数中想对两个确定的RGBA变量进行运算。这里我们采用了传统的方式,即先声明两个RGBA的变量,并且赋予相应初值,再将其传给函数blend。程序员在编写测试用例的时候,常会遇到需要声明较多值确定的RGBA变量。那么这样的声明变量-传值运算的方式是件非常麻烦的。如果自定义类型可以像内置类型一样向函数传递字面常量,比如向函数func传递字面常量func(2,5.0f),无疑这样的测试代码会简化很多。

C++11标准允许了这种想象,即我们可以通过定一个后缀标识的操作符,将声明了该后缀标识的字面量转化为需要的类型。我们可以看一看代码清单3-45所示的代码。

代码清单3-45


include <cstdlib>

include <iostream>

using namespace std;

typedef unsigned char uint8;

struct RGBA{

uint8 r;

uint8 g;

uint8 b;

uint8 a;

RGBA(uint8 R,uint8 G,uint8 B,uint8 A=0):

r(R),g(G),b(B),a(A){}

};

RGBA operator""_C(const char*col,size_t n){//一个长度为n的字符串col

const char*p=col;

const char*end=col+n;

const charr,g,b,a;

r=g=b=a=nullptr;

for(;p!=end;++p){

if(*p=='r')r=p;

else if(*p=='g')g=p;

else if(*p=='b')b=p;

else if(*p=='a')a=p;

}

if((r==nullptr)||(g==nullptr)||(b==nullptr))

throw;

else if(a==nullptr)

return RGBA(atoi(r+1),atoi(g+1),atoi(b+1));

else

return RGBA(atoi(r+1),atoi(g+1),atoi(b+1),atoi(b+1));

}

std::ostream&operator<<(std::ostream&out,RGBA&col){

return out<<"r:"<<(int)col.r

<<",g:"<<(int)col.g

<<",b:"<<(int)col.b

<<",a:"<<(int)col.a<<endl;

}

void blend(RGBA&&col1,RGBA&&col2){

//Some color blending action

cout<<"blend"<<endl<<col1<<col2<<endl;

}

int main(){

blend("r255 g240 b155"_C,"r15 g255 b10 a7"_C);

}

//编译选项:g++ -std=c++11 3-8-2.cpp


这里,我们声明了一个字面量操作符(literal operator)函数:RGBA operator""_C(const char*col,size_t n)函数。这个函数会解析以_C为后缀的字符串,并返回一个RGBA的临时变量。有了这样一个用户字面常量的定义,main函数中我们不再需要通过声明RGBA类型的声明变量-传值运算的方式来传递实际意义上的常量。通过声明一个字符串以及一个_C后缀,operator""_C函数会产生临时变量。blend函数就可以通过右值引用获得这些临时值并进行计算了。这样一来,用户就完成了定义自定义类型的字面常量,main函数中的代码书写显得更加清晰。

除去字符串外,后缀声明也可以作用于数值,比如,用户可能使用60W、120W的表示方式来标识功率,用50kg来表示质量,用1200N来表示力等。请看代码清单3-46所示的例子。

代码清单3-46


struct Watt{unsigned int v;};

Watt operator""_w(unsigned long long v){

return{(unsigned int)v};

}

int main(){

Watt capacity=1024_w;

}

//编译选项:g++ -std=c++11 3-8-3.cpp


这里我们用_w后缀来标识瓦特。除了整型数,用户自定义字面量还可以用于浮点型数等的字面量。不过必须注意的是,在C++11中,标准要求声明字面量操作符有一定的规则,该规则跟字面量的“类型”密切相关。C++11中具体规则如下:

❑如果字面量为整型数,那么字面量操作符函数只可接受unsigned long long或者constchar为其参数。当unsigned long long无法容纳该字面量的时候,编译器会自动将该字面量转化为以\0为结束符的字符串,并调用以const char为参数的版本进行处理。

❑如果字面量为浮点型数,则字面量操作符函数只可接受long double或者const char为参数。const char版本的调用规则同整型的一样(过长则使用const char*版本)。

❑如果字面量为字符串,则字面量操作符函数函数只可接受const char*,size_t为参数(已知长度的字符串)。

❑如果字面量为字符,则字面量操作符函数只可接受一个char为参数。

总体上来说,用户自定义字面量为代码书写带来了极大的便利。此外,在使用这个特性的时候,应该注意以下几点:

❑在字面量操作符函数的声明中,operator""与用户自定义后缀之间必须有空格。

❑后缀建议以下划线开始。不宜使用非下划线后缀的用户自定义字符串常量,否则会被编译器警告。当然,这也很好理解,因为如果重用形如201203L这样的字面量,后缀“L”无疑会引起一些混乱的状况。为了避免混乱,程序员最好只使用下划线开始的后缀名。