C++模板元编程详细教程(之九)

文章介绍了如何使用模板元编程在C++中实现一个简易版的多选一数据结构variant,讨论了union的局限性,提出了基于内存复用的实现思路,详细阐述了构造、析构、拷贝构造和拷贝赋值等关键操作的实现,并提供了完整的代码示例。
摘要由CSDN通过智能技术生成

前序文章请看:
C++模板元编程详细教程(之一)
C++模板元编程详细教程(之二)
C++模板元编程详细教程(之三)
C++模板元编程详细教程(之四)
C++模板元编程详细教程(之五)
C++模板元编程详细教程(之六)
C++模板元编程详细教程(之七)
C++模板元编程详细教程(之八)

多选一结构

这一章我们来看看如何编写一个多选一的结构,STL中提供了std::variant,我们就来实现一个简易版。(注意下面我们实现的variant与STL中的是有区别的,但大体思路是相同的,请读者不要以本节的代码参考使用std::variant)。

我们的诉求是,创建一个数据结构,「可能」存放多种类型,但同一时间只能有一种,并且可以在运行期变化为另一种。这里比较容易想到的做法就是用共合体类型存储数据,再加一个用于表示当前哪种数据是生效的index。请看代码:

template <typename T1, typename T2>
class variant {
public:
  // 针对每种情况的构造
  variant(const T1 &t1);
  variant(const T2 &t2);
private:
  union {
    T1 t1;
    T2 t2;
  } data;
  int index; // 当前生效的数据序号
};

只不过这样我们很快就会发现很多严重的问题:

  1. 对于任意个数的参数,无法映射到union结构中;
  2. 如果数据类型不含无参构造函数,union结构的构造会报错;
  3. variant的构造函数依赖数据类型的拷贝构造,对于不可拷贝类型来说无法创建。

因此,我们必须换一个思路。既然一开始我们想到用union,主要也是为了内存空间的复用,所以我们只需要自己来维护一片数据空间就好了,以参数类型中长度最大的为准,创建一个缓存空间即可。

template <typename... Types>
class variant {
 private:
  void *data = std::malloc(std::max(sizeof(Types)...)); // 计算出最长的Type
  int index; // 当前生效的数据序号
};

接下来的任务就是,构造函数。由于这里的Types是变参,所以无法穷举,与此同时,我们希望可以就地构造,跳过「拷贝构造」这个阶段,以支持不可拷贝类型的数据,因此必须有一个Index来标识,后面跟变参:

template <typename... Types>
class variant {
 public:
  template <size_t Index, typename... Args>
  variant(Args &&... args); // 大概是这个意思
 private: 
  void *data = std::malloc(std::max(sizeof(Types)...)); // 计算出最长的Type
  int index; // 当前生效的数据序号
};

然而,出于语法限制,构造函数用模板生成的话,是无法手动实例化的,例如:

variant<1, int> va; // <1, int>会识别为类型的模板参数,而不是构造函数的模板参数

因此,构造函数的模板参数只能依赖于自动推导,所以,我们就需要提供一个工具,用来「传递」这个Index

template <size_t Index>
struct in_place_index_t {}; // 单纯的静态工具,用于传递Index,没有运行期意义

template <size_t Index>
constexpr inline in_place_index_t<Index> in_place_index; // 对应in_place_index_t类型的实例

template <typename... Types>
class variant {
 public:
  template <size_t Index, typename... Args>
  variant(const in_place_index_t<Index> &, Args &&... args);
 private:
  void *data = std::malloc(std::max(sizeof(Types)...)); 
  int index;
};

然后,构造时,也同样通过in_place_index来构造,传递这个Index

 // 用于测试的类型
struct Test1 {
  Test1(int, double);
};
struct Test2 {
  Test1(char, int);
};

void Demo() {
  variant<Test1, Test2> var{in_place_index<0>, 1, 1.5}; // 用于构造Test1类型
  variant<Test1, Test2> var{in_place_index<1>, 'A', 1}; // 用于构造Test2类型
}

那么接下来的问题就是如何解析了,variant本身的参数是一组typename列表,但构造传进来的是一个Index序号,如何对应呢?相信读者到现在应该已经建立初步的感觉了,没错,递归大法好!

template <size_t Index, typename Head, typename... Args>
struct get_type_by_index : get_type_by_index<Index - 1, Args...> {};

template <typename Head, typename... Args>
struct get_type_by_index<0, Head, Args...> {
  using type = Head;
};

// 验证Demo
void Demo() {
  std::cout << std::is_same_v<typename get_type_by_index<2, int, double, char, void *>::type , char> << std::endl; // 1
}

有了这个工具,我们就可以从类型列表里通过Index取出对应的类型了,以此来完成variant的构造函数:

// 实现构造函数
template <typename... Types>
template <size_t Index, typename... Args>
variant<Types...>::variant(const in_place_index_t<Index> &, Args &&... args): index(Index) {
  // 先把需要构造的类型拿出来
  using data_type = typename get_type_by_index<Index, Args...>::type;
  // 在data处进行就地构造
  new(data) std::decay_t<data_type>(std::forward<Args>(args)...);
}

最难的构造已经完成了,我们先实现一下周边功能,析构、拷贝构造和赋值函数先放一放,给出一个阶段性的代码:

template <size_t Index>
struct in_place_index_t {}; // 单纯的静态工具,用于传递Index,没有运行期意义

template <size_t Index>
constexpr inline in_place_index_t<Index> in_place_index; // 对应in_place_index_t类型的实例

template <size_t Index, typename Head, typename... Args>
struct get_type_by_index : get_type_by_index<Index - 1, Args...> {};

template <typename Head, typename... Args>
struct get_type_by_index<0, Head, Args...> {
  using type = Head;
};

template <typename... Types>
class variant {
 public:
  template <size_t Index, typename... Args>
  variant(const in_place_index_t<Index> &, Args &&... args);
  
  // 获取当前序号
  int index() const;

  // 取出数据
  template <size_t Index>
  auto get() const -> std::add_lvalue_reference_t<std::decay_t<typename get_type_by_index<Index, Types...>::type>>;
  
 private:
  void *data_ = std::malloc(std::max(sizeof(Types)...)); // 计算出最长的Type
  int index_; // 当前生效的数据序号
};

// 实现构造函数
template <typename... Types>
template <size_t Index, typename... Args>
variant<Types...>::variant(const in_place_index_t<Index> &, Args &&... args): index_(Index) {
  // 先把需要构造的类型拿出来
  using data_type = typename get_type_by_index<Index, Args...>::type;
  // 在data处进行就地构造
  new(data_) std::decay_t<data_type>(std::forward<Args>(args)...);
}

// 获取序号
template <typename... Types>
int variant<Types...>::index() const {
  return index_;
}

// 取出数据(这个功能在std::variant中其实实现在std::get中)
template <typename... Types>
template <size_t Index>
auto variant<Types...>::get() const -> std::add_lvalue_reference_t<std::decay_t<typename get_type_by_index<Index, Types...>::type>> {
  using data_type = std::decay_t<typename get_type_by_index<Index, Types...>::type>;
  return *static_cast<data_type *>(data_);
}

接下来我们要攻克的,就是析构、拷贝构造、拷贝赋值这几个问题了,它们的难点在于,没有静态的Index参数可供使用,但还要找到现在数据的类型。就拿析构来说,析构函数并没有静态的Index参数,只有一个运行时的index成员,但我们要找到此时index_成员表示的data_所指向数据的实际类型,然后调用这个类型的析构函数。拷贝构造、拷贝赋值也同样需要先析构现有数据,所以面临同样的问题。

这种情况怎么办呢?类比前面章节介绍的元组的动态get方法,我们这里也采取「静态穷举生成所有可能情况代码」的方法。

以析构为例,在析构时,要根据index_去判断当前保存的数据是哪一种类型的,然后去调用对应类型的析构函数。因此,我们需要在静态期就提供每一种类型的析构方法,然后将它们和类型的Index关联起来,这样在运行时就可以调用了。

template <typename... Types>
class variant {
 public:
  // 无关代码暂时省略 
  ~variant(); // 析构函数

 private:
  void *data_ = std::malloc(std::max(sizeof(Types)...));
  int index_; // 当前生效的数据序号
  
  // 生成每一个Type下对应类型的析构方法
  template <typename Type>
  void destory_data();
};

template <typename... Types>
template <typename Type>
void variant<Types...>::destory_data() {
  // 用data_指针按照对应的类型调用析构函数
  static_cast<std::add_pointer_t<Type>>(data_)->~Type();
}

// 实现析构函数
template <typename... Types>
variant<Types...>::~variant() {
  // 析构函数中,把所有的index对应的destory_data保存下来(编译期完成这样一个映射表)
  std::array<void (variant<Types...>::*)(), sizeof...(Types)> destory_functions {
    &variant<Types...>::destory_data<Types>... // 按照类型参数展开,把对应的destory_data实例的函数指针保存下来
  };
  
  // 到了运行期,根据当前实际的index_值,选择对应的析构方法
  if (data_ != nullptr) {
    (this->*destory_functions.at(index_))();
    std::free(data_);
  }
}

对于拷贝构造、拷贝赋值来说,首先把原来的内容析构调,然后再根据被复制方的index_,来调用对应的构造函数。于是,我们还需要补充一个index_与构造函数的映射表,思路和析构完全相同:

template <typename... Types>
class variant {
 public:
  // 无关代码先省略
  variant(const variant &va);
 private:
  void *data_ = std::malloc(std::max(sizeof(Types)...));
  int index_; // 当前生效的数据序号
  
  // 生成每一个Type下对应类型的析构方法
  template <typename Type>
  void destory_data();
  
  // 生成每一个Type下对应类型的构造方法
  // 注意,由于我们要把生成的所有方法保存在数组中,因此函数类型需要一致,所以参数用泛型指针
  template <typename Type>
  void create_data(const void *obj);
};

template <typename... Types>
template <typename Type>
void variant<Types...>::create_data(const void *obj) {
  // 用data_指针按照对应的类型调用拷贝构造
  new(data_) Type(*static_cast<const Type *>(obj));
}

template <typename... Types>
variant<Types...>::variant(const variant &va): index_(va.index_) {
  // 静态期保存所有类型的构造函数
  std::array<void (variant<Types...>::*)(const void *), sizeof...(Types)> create_functions {
    &variant<Types...>::create_data<Types>...
  };
  // 动态时根据index_调用对应的构造
  (this->*create_functions.at(index_))(va.data_);
}

有了这样的思路,那么我们就可以完善所有的代码了,拷贝赋值函数可以按照相同的方法写出来了(只不过需要先析构,再重新构造)。下面给出整个项目的完整代码:

// 辅助工具
template <size_t Index>
struct in_place_index_t {}; // 单纯的静态工具,用于传递Index,没有运行期意义

template <size_t Index>
constexpr inline in_place_index_t<Index> in_place_index; // 对应in_place_index_t类型的实例

template <size_t Index, typename Head, typename... Args>
struct get_type_by_index : get_type_by_index<Index - 1, Args...> {};

template <typename Head, typename... Args>
struct get_type_by_index<0, Head, Args...> {
  using type = Head;
};

// variant结构的声明
template <typename... Types>
class variant {
 public:
  template <size_t Index, typename... Args>
  variant(const in_place_index_t<Index> &, Args &&... args);
  
  variant(const variant &va);
  variant(variant &&va);
  
  ~variant();
  
  variant &operator =(const variant &va);
  variant &operator =(variant &&va);
  
  // 获取当前序号
  int index() const;
  
  // 取出数据
  template <size_t Index>
  auto get() const -> std::add_lvalue_reference_t<std::decay_t<typename get_type_by_index<Index, Types...>::type>>;
  
  
 private:
  void *data_ = std::malloc(std::max(sizeof(Types)...));
  int index_; // 当前生效的数据序号
  
  // 生成每一个Type下对应类型的析构方法
  template <typename Type>
  void destory_data();
  
  // 生成每一个Type下对应类型的构造方法
  template <typename Type>
  void create_data(const void *obj);
};

// 实现构造函数
template <typename... Types>
template <size_t Index, typename... Args>
variant<Types...>::variant(const in_place_index_t<Index> &, Args &&... args): index_(Index) {
  using data_type = typename get_type_by_index<Index, Args...>::type;
  new(data_) std::decay_t<data_type>(std::forward<Args>(args)...);
}

// 实现析构函数
template <typename... Types>
variant<Types...>::~variant() {
  // 析构函数中,把所有的index对应的destory_data保存下来(编译期完成这样一个映射表)
  std::array<void (variant::* const)(), sizeof...(Types)> destory_functions {
    &variant::destory_data<Types>...
  };
  
  // 到了运行期,根据当前实际的index_值,选择对应的析构方法
  if (data_ != nullptr) {
    (this->*destory_functions.at(index_))();
    std::free(data_);
  }
}

// 实现拷贝构造函数
template <typename... Types>
variant<Types...>::variant(const variant &va): index_(va.index_) {
  // 静态期保存所有类型的构造函数
  std::array<void (variant<Types...>::*)(const void *), sizeof...(Types)> create_functions {
    &variant<Types...>::create_data<Types>...
  };
  // 动态时根据index_调用对应的构造
  (this->*create_functions.at(index_))(va.data_);
}

// 实现拷贝赋值函数
template <typename... Types>
variant<Types...> &variant<Types...>::operator =(const variant &va) {
  // 先析构现有数据
  std::array<void (variant<Types...>::*)(), sizeof...(Types)> destory_functions {
    &variant<Types...>::destory_data<Types>...
  };
  (this->*destory_functions.at(index_))();
  
  // 再重新构造
  std::array<void (variant<Types...>::*)(const void *), sizeof...(Types)> create_functions {
    &variant<Types...>::create_data<Types>...
  };
  (this->*create_functions.at(index_))(va.data_);
  
  return *this;
}

// 实现移动构造函数
template <typename... Types>
variant<Types...>::variant(variant &&va): index_(va.index_), data_(va.index_) {
  // 上面直接浅拷贝即可,然后把被移动的data_置空
  va.data_ = nullptr;
  va.index_ = -1; // 标记为不合法值
}

// 实现移动赋值函数
template <typename... Types>
variant<Types...> &variant<Types...>::operator =(variant &&va) {
  // 先析构现有数据
  std::array<void (variant<Types...>::*)(), sizeof...(Types)> destory_functions {
    &variant<Types...>::destory_data<Types>...
  };
  (this->*destory_functions.at(index_))();
  
  // 再浅拷贝
  data_ = va.data_;
  index_ = va.index_;
  
  // 把被移动的data_置空
  va.data_ = nullptr;
  va.index_ = -1; // 标记为不合法值
  
  return *this;
}

// 私有方法的实现
template <typename... Types>
template <typename Type>
void variant<Types...>::create_data(const void *obj) {
  // 用data_指针按照对应的类型调用拷贝构造
  new(data_) Type(*static_cast<const Type *>(obj));
}

template <typename... Types>
template <typename Type>
void variant<Types...>::destory_data() {
  // 用data_指针按照对应的类型调用析构函数
  static_cast<std::add_pointer_t<Type>>(data_)->~Type();
}

// 公有方法的实现
// 获取序号
template <typename... Types>
int variant<Types...>::index() const {
  return index_;
}

// 取出数据(这个功能在std::variant中其实实现在std::get中)
template <typename... Types>
template <size_t Index>
auto variant<Types...>::get() const -> std::add_lvalue_reference_t<std::decay_t<typename get_type_by_index<Index, Types...>::type>> {
  using data_type = std::decay_t<typename get_type_by_index<Index, Types...>::type>;
  return *static_cast<data_type *>(data_);
}

小结

这一篇以一个「多选一」结构为例,带大家体验了一下模板元编程的实际使用方式,当然,这还没完,下一篇我们会针对这个多选一结构写一个「成员访问器」,会用到更多模板元编程的技巧。

C++模板元编程详细教程(之十)

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

borehole打洞哥

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值