Вычислить сумму главной диагонали квадратной матрицы

public class Main {
    public static void main(String[] args) {
        int[][] a = {{ 1, 2, 3 }, 
                     { 4, 5, 6 },
                     { 7, 8, 9 }};
        
        int s = 0;
        
        for(int i=0; i<a.length; ++i){
            for(int j=0; j<a[i].length; ++j){
                if(i == j){
                    s += a[i][j];
                }
            }
        }
        
        System.out.println(s);
    }
}

package lab;
 
public class SquareMatrix {
 
    private float [][]matrixData;
    private int size;
 
    public SquareMatrix(int size) {
        this.size = size;
        this.matrixData = new float[size][];
        for (int i = 0; i < size; i++) {
            this.matrixData[i] = new float[size];
            for (int j = 0; j < size; j++) {
                this.matrixData[i][j] = 0;
            }
        }
    }
 
    public void setValue(int row, int col, float val) {
        this.matrixData[row][col] = val;
    }
 
    public float getSumm() {
        float summ = 0;
        for (int i = 0; i < size; i++) {
            summ += this.matrixData[i][i];
        }
        return summ;
    }
 
    public static void main(String[] args) {
        SquareMatrix m = new SquareMatrix(4);
        m.setValue(0,0,1);
        m.setValue(1,1,2);
        m.setValue(2,2,3);
        m.setValue(3,3,4);
 
        System.out.println(m.getSumm());
    }
}