From 3b894d24bfaa5db4851d6b6f3529c86944adcd73 Mon Sep 17 00:00:00 2001
From: kradchen <chenjuijun@realcan.cn>
Date: Fri, 19 May 2023 13:43:43 +0800
Subject: [PATCH] Add new size function

---
 src/Function3D.cpp | 5 +++++
 src/Function3D.h   | 1 +
 2 files changed, 6 insertions(+)

diff --git a/src/Function3D.cpp b/src/Function3D.cpp
index cfb24e4..ddb5fc8 100644
--- a/src/Function3D.cpp
+++ b/src/Function3D.cpp
@@ -117,3 +117,8 @@ Matrix Aurora::size(const Matrix &aMatrix)
         return Matrix::New(output,2,1,1);
     }
 }
+
+int Aurora::size(const Matrix &aMatrix,int dims)
+{
+    return aMatrix.getDimSize(dims-1);
+}
diff --git a/src/Function3D.h b/src/Function3D.h
index bf63753..17bc282 100644
--- a/src/Function3D.h
+++ b/src/Function3D.h
@@ -42,6 +42,7 @@ namespace Aurora {
     Matrix interpn(const Matrix& aX, const Matrix& aY, const Matrix& aZ, const Matrix& aV, const Matrix& aX1, const Matrix& aY1, const Matrix& aZ1,InterpnMethod aMethod);
 
     Matrix size(const Matrix &aMatrix);
+    int size(const Matrix &aMatrix,int dims);
 };