结构运算符重载

时间:2015-07-10 16:11:26

标签: c++ struct

我有以下结构用于存储rgb值。我为rgb和标量操作重载了常用运算符+, - ,*。

struct rgb {
    float r;
    float g;
    float b;

    rgb() : r(0.f), g(0.f), b(0.f) {}
    rgb( const float& rr, const float& gg, const float& bb ): r(rr), g(gg), b(bb) {}
    rgb( const float& f ): r(f), g(f), b(f) {}

    rgb& operator +(const rgb& a) {
        rgb t;
        t.r = this->r + a.r;
        t.g = this->g + a.g;
        t.b = this->b + a.b;
        return t;
    }
    rgb& operator +(const float& a) {
        rgb t;
        t.r = this->r + a;
        t.g = this->g + a;
        t.b = this->b + a;
        return t;
    }
    rgb& operator -(const rgb& a) {
        rgb t;
        t.r = this->r - a.r;
        t.g = this->g - a.g;
        t.b = this->b - a.b;
        return t;
    }
    rgb& operator -(const float& a) {
        rgb t;
        t.r = this->r - a;
        t.g = this->g - a;
        t.b = this->b - a;
        return t;
    }
    rgb& operator *(const rgb& a) {
        rgb t;
        t.r = this->r * a.r;
        t.g = this->g * a.g;
        t.b = this->b * a.b;
        return t;
    }
    rgb& operator *(const float& a) {
        rgb t;
        t.r = this->r * a;
        t.g = this->g * a;
        t.b = this->b * a;
        return t;
    }
    float sum() {
        return r + g + b;
    }
};

当我在以下函数中使用此结构时,我得到一个意外的输出:

inline void foo(rgb &N,
                rgb &X,
                rgb &P,
                rgb &R) {
    float d = 0.0;
    rgb d_v;
    //find the dot product between N and (P-X)
    d_v = N * (P - X);  //this is always 0, 0, 0
    d = d_v.sum();
    R = P - N * d;
}

但如果我使用中间rgb存储中间操作,它可以正常工作:

inline void subspaceProjectCPU(rgb &N,
                           rgb &X,
                           rgb &P,
                           rgb &R) {
    float d = 0.0;
    rgb d_v;
    rgb PX = P - X;
    //find the dot product between N and (P-X)
    d_v = PX * N;
    d = d_v.sum();
    rgb Nd = N*d;
    R = P - Nd;
}

有人可以对此有所了解吗? (我知道我没有处理双方的标量运算,因此定义了N * d但不是d * N,其中N是rgb而d是浮点数。)

1 个答案:

答案 0 :(得分:2)

您的运算符函数返回对局部变量的引用,该函数在运算符函数返回后将不再存在。不要那样做;这是未定义的行为。只需返回rgb值。