本文介绍在C++ 语言中,使用一个函数,并返回两个及以上、同类型或不同类型的返回值的具体方法。
对于C++ 语言而言,其不能像Python等语言一样在一个函数中返回多个返回值;但是我们也会经常遇到需要返回两个甚至更多个值的需求。针对这种情况,我们可以通过pair
、tuple
(元组)等数据结构,实现C++ 函数返回两个或多个返回值的需求。本文就以pair
为例,介绍二者的具体用法。
首先,我们需要将C++ 函数的类型定义为pair<>
,其中<>
内为两个返回值各自的数据类型。例如,如以下代码所示,我们定义了一个函数raster_to_series
,函数类型为pair<double**, double*>
,表示这一函数的返回值有两个,且两个返回值的数据类型分别为double**
与double*
。
pair<double**, double*> raster_to_series(double** pafScanline)
{
// 这里是函数的代码部分
// ...
return make_pair(pixel_paf, miss_rate_paf);
}
其中,在函数的返回值部分,我们需要基于make_pair()
函数,将数据类型分别为double**
与double*
的两个变量放在一起,形成pair
这一结构,再进行返回。
其次,以上pair
、make_pair()
等都是定义在std
这一命名空间内的,因此需要首先声明一下这一命名空间;再次,在调用这一函数前,我们首先需要定义一个pair <double**, double*>
类型的变量,并将函数的返回值赋给它。
using namespace std;
int main() {
pair <double**, double*> pair_paf = raster_to_series(pafScanline);
double** pixel_paf = pair_paf.first;
double* miss_rate_paf = pair_paf.second;
return 0;
}
随后,通过.first
与.second
命令,分别获取pair<double**, double*>
中的两个变量。
至此,我们即实现了通过1
个C++ 函数返回2
个返回值的方法。
同样的思路,如果需要通过1
个C++ 函数返回3
个或更多的返回值,则可以基于tuple
(元组)这一数据结构,用类似于上述操作的方法来实现。
举一个例子,假如我们想通过1
个函数返回3
个返回值,就需要将前述代码中函数的类型定义为tuple<int, double, int>
,将make_pair()
函数更改为make_tuple()
(或者我们就不用这个make_tuple()
了,直接自己手动构建一个tuple
,然后返回就好,如下面的例子);且在调用函数时首先将返回值赋给一个tuple<int, double, int>
类型的数据,再依次从这一元组类型的数据中提取出三个返回值的具体数值。
这里给出一个具体的实现代码,如以下代码所示。其中,函数get_tif_information()
的返回值一共有5
个,且还是不同的数据类型——如果将这个函数分开,写成多个函数的话,那也未免太麻烦了点。
tuple<float*, double*, int, int, const char*> get_tif_information(string path, int band_idx = 1) {
GDALDataset* dataset = (GDALDataset*)GDALOpen(path.c_str(), GA_ReadOnly);
double* geotransform = new double[6];
dataset->GetGeoTransform(geotransform);
const char* projection = dataset->GetProjectionRef();
int width = dataset->GetRasterXSize();
int height = dataset->GetRasterYSize();
GDALRasterBand* band = dataset->GetRasterBand(band_idx);
float* array_tif = new float[width * height];
band->RasterIO(GF_Read, 0, 0, width, height, array_tif, width, height, GDT_Float32, 0, 0);
GDALClose(dataset);
return { array_tif, geotransform, width, height, projection };
}
auto tuple_gf_blue = get_tif_information(file_path_gf, 3);
float* array_gf_blue = get<0>(tuple_gf_blue);
double* geotransform_gf = get<1>(tuple_gf_blue);
const int width_gf = get<2>(tuple_gf_blue);
const int height_gf = get<3>(tuple_gf_blue);
const char* projection_gf = get<4>(tuple_gf_blue);
本文最前面的代码中,我们是通过.first
与.second
命令,分别获取pair<double**, double*>
中的两个变量;而在上述这个代码中,因为是要获取tuple
中的元素,所以我们可以通过get<0>()
方法,来获取其中的元素(下标从0
开始算起)。
至此,大功告成。