From c11c3f7a4844c2b7ea97674efd1c296d1cdb6dd5 Mon Sep 17 00:00:00 2001
From: Wojciech Janota <wojciech@janota.ovh>
Date: Thu, 4 Apr 2024 11:19:58 +0200
Subject: [PATCH] WIP

---
 longinus/longinus/utils/dataset.py | 31 ++++++++++++++++++++++++++++++
 longinus/longinus/utils/image.py   | 11 ++++++++---
 longinus/longinus/wikimedia        |  1 +
 3 files changed, 40 insertions(+), 3 deletions(-)
 create mode 100644 longinus/longinus/utils/dataset.py
 create mode 160000 longinus/longinus/wikimedia

diff --git a/longinus/longinus/utils/dataset.py b/longinus/longinus/utils/dataset.py
new file mode 100644
index 0000000..649b3ae
--- /dev/null
+++ b/longinus/longinus/utils/dataset.py
@@ -0,0 +1,31 @@
+from longinus.wikimedia.loaders.pytorch import WikimediaCommonsLoader
+from longinus.utils.image import Image
+
+class DatasetLoader():
+    def __init__(self, dataset_size):
+        self.dataset = list()
+        self.dataset_size = dataset_size
+        tmp_dataset = WikimediaCommonsLoader(batch_size=32, resize_to=512, crop_to=512)
+        counter = 0
+        for image in tmp_dataset:
+            self.dataset.append(image)
+            counter+=1
+            if counter >= self.dataset_size:
+                break
+    
+    
+    def get_dataset(self):
+        return self.dataset
+    
+    def get_dataset_size(self):
+        return self.dataset_size
+    
+class Dataset():
+    def __init__(self, dataset_loader: DatasetLoader, dataset_size: int):
+        self.dataset = dataset_loader(dataset_size).get_dataset()
+        self.dataset_converted = self.convert_images()
+    
+    def convert_images(self) -> list[Image]:
+        for image in self.dataset:
+            new_image = Image.load_image_from_mem(image)
+        pass
\ No newline at end of file
diff --git a/longinus/longinus/utils/image.py b/longinus/longinus/utils/image.py
index c880fc6..8a44625 100644
--- a/longinus/longinus/utils/image.py
+++ b/longinus/longinus/utils/image.py
@@ -12,10 +12,15 @@ class Image:
     def read(self):
         self.bgr_array = cv.imread(self.path, cv.IMREAD_GRAYSCALE)
         self.dct_array = cv.dct(np.float32(self.bgr_array))
-        return self.bgr_array
 
-    def write(self, image_array):
+    def write(self):
         pass
 
     def extract_coefficients(self):
-        pass
+        return self.dct_array
+    
+    def load_image_from_mem(self, image_array):
+        self.bgr_array = image_array
+
+    def override_coefficients(self, new_dct_array):
+        self.dct_array = new_dct_array
diff --git a/longinus/longinus/wikimedia b/longinus/longinus/wikimedia
new file mode 160000
index 0000000..39fa5b4
--- /dev/null
+++ b/longinus/longinus/wikimedia
@@ -0,0 +1 @@
+Subproject commit 39fa5b4ee0ff12a7c9c36b656b389f3e3a6fddef