c++ - 如何在对基类和派生类进行操作时返回派生类型？

Question

我有两个类，my_matrix 和 my_vector 类，其中 my_vector 派生自 my_matrix。

当我执行 my_matrix tmp = vec * mat 时，它返回一个矩阵(mat 是 my_matrix 和 vec 的实例 是 my_vector 的实例)。我在这里很好。

但是当我执行 my_vector tmp = mat * vec 时，我在 clang++ 中得到这个错误:

no viable conversion from 'my_matrix' to 'my_vector'

g++ 中的这个错误:

conversion from 'my_matrix' to non-scalar type 'my_vector' requested

我想我知道问题出在哪里了。 mat 和 vec 相乘后，返回类型为 my_matrix 并且不能将基类转换为派生类.

您认为解决这个问题的方法是什么？

我尝试定义 void operator*(my_vector&, my_matrix&)(不是派生类的成员函数)，它在派生类中调用 my_vector operator*(my_matrix&)类，但到目前为止还没有用。

class my_matrix{
public:
    my_matrix operator*(my_matrix& other_mat){
        my_matrix tmp(row, other_mat.col);
        // for loop to matrix multiply that fill up the tmp
        return tmp;
    }
};

class my_vector:public my_matrix{
public:
    // not sure if I need operator* here 
    // Somehow I want to bring the my_vector tmp = mat * vec;  here so 
    // it would return vector instead of matrix
    // I have tried : but not successful 
    // my_vector operator*(my_matrix&)
    // void operator*(my_vector&, my_matrix&); (non member)
};

int main(){
    my_matrix mat = {{1,2,3},{4,5,6}}; // 2x3
    my_vector vec = {1,2,3}; // 3x1

    my_vector tmp = mat * vec; // 2x3 * 3*1 gives 2x1 vector
}

Answer 1

尝试这样的事情:

class my_matrix {
public:
    my_matrix operator*(const my_matrix& other_mat) const {
        my_matrix tmp(row, other_mat.col);
        // for loop to matrix multiply that fill up the tmp
        return tmp;
    }
};

class my_vector : public my_matrix {
public:
    using my_matrix::operator*;

    my_vector operator*(const my_vector& other_vec) const {
        my_vector tmp(...);
        // for loop to vector multiply that fill up the tmp
        return tmp;
    }
};

my_vector operator*(const my_matrix &lhs, const my_vector &rhs)
{
    my_vector tmp(...);
    // for loop to vector multiply that fill up the tmp
    return tmp;
}

int main() {
    my_matrix mat = {{1,2,3},{4,5,6}}; // 2x3
    my_vector vec = {1,2,3}; // 3x1

    my_vector tmp = mat * vec; // 2x3 * 3x1 gives 2x1 vector
}