Fix MatrixSlice scalar mode bug.

src/Matrix.cpp

@@ -335,8 +335,8 @@ namespace Aurora {
         double *startPointer = getData() + (rowStride * rowOffset
                                             + colStride * colOffset
                                             + sliceStride * sliceOffset) * getValueType();
-        int size1 = getDimSize(allDimIndex[0]);
-        int stride1 = strides[allDimIndex[0]];
+        int size1 = allDimIndex.empty()?1:getDimSize(allDimIndex[0]);
+        int stride1 = allDimIndex.empty()?1:strides[allDimIndex[0]];
         switch (mode) {
             //matrix mode
             case 2:{
@@ -479,6 +479,49 @@ namespace Aurora {
         return *this;
     }
 
+    Matrix::MatrixSlice &Matrix::MatrixSlice::operator=(double value) {
+        if(!mData){
+            std::cerr <<"Assign value fail!Des data pointer is null!";
+            return *this;
+        }
+        if (mSliceMode!=0) {
+            std::cerr <<"Assign value fail!Des slicemode is"<<mSliceMode<<", not scalar mode!";
+            return *this;
+        }
+        if (mSize!=1) {
+            std::cerr <<"Assign value fail!Des  size:"<<mSize<<", not scalar mode!";
+            return *this;
+        }
+        if (mType!=Normal) {
+            std::cerr <<"Assign value fail!Des type is complex!";
+            return *this;
+        }
+        mData[0]=value;
+        return *this;
+    }
+
+    Matrix::MatrixSlice &Matrix::MatrixSlice::operator=(std::complex<double> value) {
+        if(!mData){
+            std::cerr <<"Assign value fail!Des data pointer is null!";
+            return *this;
+        }
+        if (mSliceMode!=0) {
+            std::cerr <<"Assign value fail!Des slicemode is"<<mSliceMode<<", not scalar mode!";
+            return *this;
+        }
+        if (mSize!=1) {
+            std::cerr <<"Assign value fail!Des  size:"<<mSize<<", not scalar mode!";
+            return *this;
+        }
+        if (mType!=Complex) {
+            std::cerr <<"Assign value fail!Des type is not complex!";
+            return *this;
+        }
+        mData[0]=value.real();
+        mData[1]=value.imag();
+        return *this;
+    }
+
     Matrix Matrix::MatrixSlice::toMatrix() const {
         double * data = (double *) mkl_malloc(mSize*(mSize2>0?mSize2:1) * sizeof(double)*mType, 64);