From c16fc07b988aac8217fa16c912e460e994ae2363 Mon Sep 17 00:00:00 2001
From: rsiddharth <s@ricketyspace.net>
Date: Sun, 6 Sep 2020 14:53:53 -0400
Subject: lib: add TransposeBlocks

* lib/blocks.go (TransposeBlocks): New function.
---
 lib/blocks.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'lib/blocks.go')

diff --git a/lib/blocks.go b/lib/blocks.go
index 3805d86..e817e6f 100644
--- a/lib/blocks.go
+++ b/lib/blocks.go
@@ -30,3 +30,20 @@ func BreakIntoBlocks(cb []byte, keysize int) [][]byte {
 	}
 	return blocks
 }
+
+func TransposeBlocks(blocks [][]byte, keysize int) [][]byte {
+	if len(blocks) < 1 {
+		return make([][]byte, 0)
+	}
+
+	tblocks := make([][]byte, keysize)
+	for i := 0; i < len(blocks); i++ {
+		for j := 0; j < len(blocks[i]); j++ {
+			if len(tblocks[j]) == 0 {
+				tblocks[j] = make([]byte, len(blocks))
+			}
+			tblocks[j][i] = blocks[i][j]
+		}
+	}
+	return tblocks
+}
-- 
cgit v1.2.3