Question

我正在尝试编写一些简单的Rcpp代码示例。使用Rcpp和inline包非常容易。

但我对如何测试两个字符元素是否相等感到困惑。以下示例比较了两个字符向量的第一个元素。但我无法编译。

诀窍是什么？

library(Rcpp)
library(inline)

cCode <- '
    Rcpp::CharacterVector cx(x);
    Rcpp::CharacterVector cy(y);
    Rcpp::LogicalVector r(1);
    r[0] = (cx[0] == cy[0]);
    return(r);
    '

cCharCompare <- cxxfunction(signature(x="character", y="character"), 
                            plugin="Rcpp", body=cCode)
cCharCompare("a", "b")

-

如果两个元素中的一个是常量，则使用==进行的比较完全正常。以下代码编译并给出预期结果：

cCode <- '
    Rcpp::CharacterVector cx(x);
    Rcpp::LogicalVector r(1);
    r[0] = (cx[0] == "a");
    return(r);
    '

cCharCompareA <- cxxfunction(signature(x="character"), plugin="Rcpp", body=cCode)

cCharCompareA("a")
[1] TRUE

cCharCompareA("b")
[1] FALSE

Answer 1

在Rcpp 0.10.4中引入了相等运算符。 string_proxy类中的实现如下所示：

bool operator==( const string_proxy& other){
    return strcmp( begin(), other.begin() ) == 0 ;
}

现在我们可以写：

#include <Rcpp.h>
using namespace Rcpp ;

// [[Rcpp::export]]
LogicalVector test( CharacterVector x, CharacterVector y){
    Rcpp::LogicalVector r(x.size());
    for( int i=0; i<x.size(); i++){
        r[i] = (x[i] == y[i]);
    }
    return(r);
}

在我们的单元测试中使用了类似的东西：

> test(letters, letters)
 [1] TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE
[16] TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE TRUE

Answer 2

试试这个：

  // r[0] = (cx[0] == cy[0]);
  // r[0] = ((char*)cx[0] == (char*)cy[0]); <- this is wrong
  r[0] = (*(char*)cx[0] == *(char*)cy[0]); // this is correct.

这不容易解释，但

CharacterVector不是char[]。
operator []返回StringProxy。
StringProxy不是char的类型。
StringProxy有一个成员运算符函数char*，可将StringProxy转换为char*。

所以，也许(char*)cx[0]是一个指针。现在我忘了很多关于C ++语法的事情......

编译失败的原因是==的运算符重载StringProxy中的类型推断失败。

Answer 3

@kohske非常好（技术性）回答，但这里有更多C ++ - ish：只是比较字符串！

library(inline)      ## implies library(Rcpp) when we use the plugin

cCode <- '
    std::string cx = Rcpp::as<std::string>(x);
    std::string cy = Rcpp::as<std::string>(y);
    bool res = (cx == cy);
    return(Rcpp::wrap(res));
    '

cCharCompare <- cxxfunction(signature(x="character", y="character"),
                            plugin="Rcpp", body=cCode)
cCharCompare("a", "b")

如果你真的只想比较字符串的第一个字符，那么你可以从x转到x.c_str()并索引其初始元素，或者只是取消引用指向第一个字符的指针。

更多的R-ish答案可能会扫过实际的字符串向量......

如何测试Rcpp :: CharacterVector元素是否相等？

3 个答案: