c++ - Rcpp 枚举支持

Question

我正在使用 Rcpp 模块导出类方法。其中一些方法的返回类型是枚举。例如:

#include "Enum.h"
#include <Rcpp.h>
using namespace Rcpp;

class EnumTest{
public:
  EnumTest(){}
  void setP(Polarization pol){p = pol;}
  Polarization getP(){return p;}
private:
  Polarization p;
};

RCPP_EXPOSED_CLASS(EnumTest)
RCPP_MODULE(EnumTest){
  class_<EnumTest>("EnumTest")
  .property("p", &EnumTest::getP, &EnumTest::setP)
  ;
}

Polarization 是一个枚举，定义如下:

enum Polarization{
  HORIZONTAL_POL = 0,
  VERTICAL_POL   = 1
};

当我尝试构建代码时，出现以下错误。

cannot convert 'SEXP' to 'Polarization' in initialization

是否有向 Rcpp 公开枚举的方法，类似于公开类的方式？我在 Rcpp 模块小插图中注意到枚举类型列在未来的扩展中。这是否意味着没有办法做到这一点？如果是这样，是否有任何可能的解决方法？

Answer 1

Rcpp 中对枚举的支持有限，由宏 here 给出。 .

所以你可以这样做:

#include <Rcpp.h>
using namespace Rcpp; 

enum Polarization{
  HORIZONTAL_POL = 0,
  VERTICAL_POL   = 1
};

RCPP_EXPOSED_ENUM_NODECL(Polarization)

class EnumTest{
  public:
    EnumTest(){}
  void setP(Polarization pol){p = pol;}
  Polarization getP(){return p;}
  private:
    Polarization p;
};


RCPP_MODULE(Bla){
  class_<EnumTest>("EnumTest")
    .constructor()
    .property("p", &EnumTest::getP, &EnumTest::setP)
  ;
}

支持非常基本，本质上允许您将枚举作为整数传递，从而丢失类和所有内容。有时这就足够了。我建议在 R 端有一个简单的命名列表以使代码更清晰，例如

Polarization <- list( HORIZONTAL_POL = 0L, VERTICAL_POL = 1L )

这样你就可以做，例如

test <- new( EnumTest )
test$p <- Polarization$HORIZONTAL_POL
names(which( test$p == Polarization ))
## [1] "HORIZONTAL_POL"