Question

所以我在使用RCPP创建的函数中返回值时遇到了一些麻烦。它只返回NumericVector的第一个值。问题是当我在自身内部调用函数并将NumericVector传递回out变量时。任何关于我正在做错的事情的输入都会非常感激。

NumericVector advantage(double tRos, double tTat, double cRos, double cTat){
double tRosAd = 0;
double tTatAd = 0;
double ja = 0;
double rosAtt = 0;
double tatAtt = 0;
double ttlAd = 0;
NumericVector out(1);

if ((tRos >= 0 || cRos >=0) && cRos >= tRos && cTat <= tTat){
    //case 1.1//
    tRosAd = (cRos - tRos) * cTat * -1;
    tTatAd = (tTat - cTat) * tRos;
    out[0] = tRosAd;
    out[1] = tTatAd;
}
else if((tRos >= 0 && cRos >= 0) && cRos <= tRos && cTat >= tTat){
    //case 1.2//
    out = advantage(cRos, cTat, tRos, tTat);
    out[0] = out[0] * -1;
    out[1] = out[1] * -1;
}
else if(tRos >= 0 && cRos <= 0 && tTat >= cTat){
    //case 2.1//
    tRosAd = (tRos - cRos) * cTat;
    tTatAd = (tTat - cTat) * tRos;
    out[0] = tRosAd;
    out[1] = tTatAd;
}
else if(tRos <= 0 && cRos >= 0 && tTat <= cTat){
    //case 2.2//
    out = advantage(cRos, cTat, tRos, tTat);
    out[0] = out[0] * -1;
    out[1] = out[1] * -1;
}
else if(tRos <= 0 && cRos <= 0 && tTat >= cTat & tRos >= cRos){
    //case 3.1//
    tRosAd = (tRos - cRos) * cTat;
    tTatAd = (tTat - cTat) * tRos;
    out[0] = tRosAd;
    out[1] = tTatAd;
}
else if(tRos <= 0 && cRos <= 0 && tTat <= cTat && tRos <= cRos){
    //case 3.2//
    out = advantage(cRos, cTat, tRos, tTat);
    out[0] = out[0] * -1;
    out[1] = out[1] * -1;
}
else if(tRos >= 0 && cRos >=  0 && tTat >= cTat && tRos >= cRos){
    //case 4.1//
    //calculate two attribution advantages//
    tRosAd = (tRos - cRos) * cTat;
    tTatAd = (tTat - cTat) * cRos;
    //calculate total & joint advantage//
    ttlAd = (tTat * tRos) - (cTat * cRos);
    ja = ttlAd - (tRosAd + tTatAd);
    rosAtt = tRosAd / (tRosAd / tTatAd);
    tatAtt = tTatAd / (tRosAd / tTatAd);
    //add joint advantage//
    tRosAd = ja * rosAtt + tRosAd;
    tTatAd = ja * tatAtt + tTatAd;
    out[0] = tRosAd;
    out[1] = tTatAd;
}
else if(tRos >= 0 && cRos >= 0 && tTat <= cTat && tRos <= cRos){
    //case 4.2//
    out = advantage(cRos, cTat, tRos, tTat);
    out[0] = out[0] * -1;
    out[1] = out[1] * -1;
}
else if(tRos <= 0 && cRos <= 0 && tTat <= cTat && tRos >= cRos){
    //case 5.1//
    tRosAd = (tRos - cRos) * tTat;
    tTatAd = (cTat - tTat) * tRos * -1;
    //calculate total and joint advantage//
    ttlAd = (tTat * tRos) - (cTat * cRos);
    ja = ttlAd - (tRosAd + tTatAd);
    rosAtt = tRosAd / (tRosAd + tTatAd);
    tatAtt = tTatAd / (tRosAd + tTatAd);
    tRosAd = ja * rosAtt + tRosAd;
    tTatAd = ja * tatAtt + tTatAd;
    out[0] = tRosAd;
    out[1] = tTatAd;
}
else if(tRos <= 0 && cRos <= 0 && tTat >= cTat && tRos <= cRos){
    //case 5.2//
    out = advantage(cRos, cTat, tRos, tTat);
    out[0] = out[0] * -1;
    out[1] = out[1] * -1;
}
return out;
}

Answer 1

正如Hadley的评论所指出的那样NumericVector out(1)创建了一个长度为1的新实例。

当你通过out[0]和out[1]访问重复的两个元素时，你很幸运，代码没有爆炸。

如果您知道矢量的长度n，请使用它：NumericVector out(n)。否则稍后会调整大小，但可能会丢失您的内容。

如果您不知道矢量的长度，请动态增长std::vector并在返回R时进行转换。这两种方法都有很多示例。

Rcpp数值向量输出仅返回一个值

1 个答案: