Array2D<T> in Swift

array2d<T>.swift

class Array2D<T> {

   let columns: Int
   let rows: Int
   var array: Array<T?>

   init(columns: Int, rows: Int) {
      self.columns = columns
      self.rows = rows
      array = Array<T?>(count: rows*columns, repeatedValue: nil)
   }

   subscript(column: Int, row: Int) -> T? {
      get {
         return array[row*columns + column]
      }
      set {
         array[row*columns + column] = newValue
      }
   }
}
 
// using
let foo = Foo()
let arr = Array2D<Foo>(columns: 3, rows: 5)
abc[2, 1] = foo