From 8da447ac5a2dd2c987b5e1cbacc4d693d5617ef4 Mon Sep 17 00:00:00 2001
From: Csaba Kiraly <csaba.kiraly@gmail.com>
Date: Wed, 7 Dec 2022 15:10:35 +0100
Subject: [PATCH] add column and row accessors

Signed-off-by: Csaba Kiraly <csaba.kiraly@gmail.com>
---
 DAS/block.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/DAS/block.py b/DAS/block.py
index 5539865..5c09ca1 100644
--- a/DAS/block.py
+++ b/DAS/block.py
@@ -15,6 +15,18 @@ class Block:
         for i in range(self.blockSize*self.blockSize):
             self.data[i] = random.randint(1, 9)
 
+    def getColumn(self, columnID):
+        column = [0] * self.blockSize
+        for i in range(self.blockSize):
+            column[i] = self.data[(i*self.blockSize)+columnID]
+        return column
+
+    def getRow(self, rowID):
+        row = [0] * self.blockSize
+        for i in range(self.blockSize):
+            row[i] = self.data[(rowID*self.blockSize)+i]
+        return row
+
     def print(self):
         dash = "-" * (self.blockSize+2)
         print(dash)