#ifndef AURORA_FUNCTION2D_H
#define AURORA_FUNCTION2D_H

#include "Matrix.h"
#include "Function1D.h"


namespace Aurora {
    enum FunctionDirection{
        Column,
        Row,
        All
    };
    double immse(const Matrix& aImageA, const Matrix& aImageB);
    Matrix inv(const Matrix& aMatrix);
    Matrix inv(Matrix&& aMatrix);
    Matrix interp2(const Matrix& aX, const Matrix& aY, const Matrix& aV, const Matrix& aX1, const Matrix& aY1, InterpnMethod aMethod);
    Matrix interpn(const Matrix& aX, const Matrix& aY, const Matrix& aV, const Matrix& aX1, const Matrix& aY1, InterpnMethod aMethod);
    Matrix std(const Matrix& aMatrix);

    /**
     * 求矩阵最小值，可按行、列、单元, 目前不支持三维，不支持复数
     * @param aMatrix  目标矩阵
     * @param direction 方向，Column, Row, All
     * @return
     */
    Matrix min(const Matrix& aMatrix,FunctionDirection direction = Column);

    Matrix min(const Matrix& aMatrix,FunctionDirection direction, long& rowIdx, long& colIdx);

    /**
     * 求矩阵最小值，可按行、列、单元, 目前不支持三维，不支持复数
     * @param aMatrix  目标矩阵
     * @param direction 方向，Column, Row, All
     * @return 最大值矩阵
     */
    Matrix max(const Matrix& aMatrix,FunctionDirection direction = Column);

    Matrix max(const Matrix& aMatrix,FunctionDirection direction , long& rowIdx, long& colIdx);

    /**
     * 比较两个矩阵，求对应位置的最小值，不支持三维
     * @attention 矩阵形状不一样时，如A为[MxN],则B应为标量或[1xN]的行向量
     * @param aMatrix 目标矩阵1
     * @param aOther 目标矩阵2
     * @return 最小值矩阵
     */
    Matrix min(const Matrix& aMatrix,const Matrix& aOther);

    /**
     * 求矩阵和，可按行、列、单元, 目前不支持三维，不支持复数
     * @param aMatrix  目标矩阵
     * @param direction 方向，Column, Row, All
     * @return 求和结果矩阵
     */
    Matrix sum(const Matrix& aMatrix,FunctionDirection direction = Column);

    /**
     * 求矩阵平均值，可按行、列、单元, 目前不支持三维，不支持复数
     * @param aMatrix 目标矩阵
     * @param direction 方向，Column, Row, All
     * @param aIncludeNan 是否包含nan
     * @return 平均值矩阵
     */
    Matrix mean(const Matrix& aMatrix,FunctionDirection direction = Column, bool aIncludeNan = true);

    /**
     * 矩阵排序 按列, 目前不支持三维，不支持复数
     * @param aMatrix 目标矩阵
     * @return 排序后矩阵
     */
    Matrix sort(const Matrix& aMatrix);

    /**
     * 矩阵排序 按列, 目前不支持三维，不支持复数
     * @param aMatrix 目标矩阵
     * @return 排序后矩阵
     */
    Matrix sort(Matrix&& aMatrix);

    /**
     * 矩阵排序 按行, 目前不支持三维，不支持复数
     * @param aMatrix 目标矩阵
     * @return 排序后矩阵
     */
    Matrix sortrows(const Matrix& aMatrix);

    /**
     * 矩阵排序 按行, 目前不支持三维，不支持复数
     * @param aMatrix 目标矩阵
     * @return 排序后矩阵
     */
    Matrix sortrows(Matrix&& aMatrix);

    /**
     * 对矩阵求中间值 按列, 目前不支持三维，不支持复数
     * @param aMatrix 目标矩阵
     * @return 中值矩阵
     */
    Matrix median(const Matrix& aMatrix);

    /**
     * FFT,支持到2维，输入可以是常数可以是复数，输出必是复数
     * @param aMatrix 目标矩阵
     * @return fft后的复数矩阵
     */
    Matrix fft(const Matrix& aMatrix);

    /**
     * 逆fft，支持到2维，输入必须是复数，输出必是复数
     * @attention 如有需要可使用real去除虚部
     * @param aMatrix
     * @return ifft后的复数矩阵
     */
    Matrix ifft(const Matrix& aMatrix);

    /**
     * hilbert，支持到2维，输入必须是复数，输出必是复数
     * @param aMatrix
     * @return
     */
    Matrix hilbert(const Matrix& aMatrix);
};


#endif //AURORA_FUNCTION2D_H