From 6c3104dd3e974fa6cfbfd2484be9888d2cda7ebb Mon Sep 17 00:00:00 2001
From: Beliy Nikita <beliy.nikita@uliege.be>
Date: Thu, 12 Sep 2024 10:50:43 +0200
Subject: [PATCH] Moved selector into a class in tools

---
 bidsme/tools/__init__.py |   4 ++
 bidsme/tools/selector.py | 108 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 112 insertions(+)
 create mode 100644 bidsme/tools/selector.py

diff --git a/bidsme/tools/__init__.py b/bidsme/tools/__init__.py
index a57580f..ae98a10 100644
--- a/bidsme/tools/__init__.py
+++ b/bidsme/tools/__init__.py
@@ -1 +1,5 @@
+from .selector import type_selector
+
 name = "tools"
+
+__all__ = ["type_selector"]
diff --git a/bidsme/tools/selector.py b/bidsme/tools/selector.py
new file mode 100644
index 0000000..2ea79c9
--- /dev/null
+++ b/bidsme/tools/selector.py
@@ -0,0 +1,108 @@
+###############################################################################
+# selector.py provides the function for identify recordings and select
+# the recording class accordingly
+###############################################################################
+# Copyright (c) 2019-2020, University of Liège
+# Author: Nikita Beliy
+# Owner: Liege University https://www.uliege.be
+# Credits: [Marcel Zwiers]
+# Maintainer: Nikita Beliy
+# Email: Nikita.Beliy@uliege.be
+# Status: developpement
+###############################################################################
+# This file is part of BIDSme
+# BIDSme is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 2 of the License, or
+# (at your option) any later version.
+# eegBidsCreator is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# You should have received a copy of the GNU General Public License
+# along with BIDSme.  If not, see <https://www.gnu.org/licenses/>.
+##############################################################################
+from types import ModuleType
+
+from .. import Modules
+
+
+class type_selector(object):
+    __slots__ = ["types_list"]
+
+    def __init__(self):
+
+        self.types_list = {}
+        for i in dir(Modules):
+            obj = getattr(Modules, i)
+            if type(obj) is not ModuleType:
+                continue
+            if not hasattr(obj, "__formats"):
+                continue
+            self.types_list[i] = getattr(obj, "__formats")
+
+    def select(self, folder: str, module: str = ""):
+        """
+        Returns first class for wich given folder is correct
+
+        Parameters
+        ----------
+        folder: str
+            folder to scan
+        module: str
+            restrict type of class
+        """
+        if module == "":
+            for m in self.types_list:
+                for cls in self.types_list[m]:
+                    if cls.isValidRecording(folder):
+                        return cls
+        else:
+            for cls in self.types_list[module]:
+                if cls.isValidRecording(folder):
+                    return cls
+        return None
+
+    def selectFile(self, file: str, module: str = ""):
+        """
+        Returns first class for wich given file is correct
+
+        Parameters
+        ----------
+        folder: str
+            file to scan
+        module: str
+            restrict type of class
+        """
+        if module == "":
+            for m in self.types_list:
+                for cls in self.types_list[m]:
+                    if cls.isValidFile(file):
+                        return cls
+        else:
+            for cls in self.types_list[module]:
+                if cls.isValidFile(file):
+                    return cls
+        return None
+
+    def selectByName(self, name: str, module: str = ""):
+        """
+        Returns first class with given name
+
+        Parameters
+        ----------
+        name: str
+            name of class
+        module: str
+            restrict type of class
+        """
+        if module == "":
+            for m in self.types_list:
+                for cls in self.types_list[m]:
+                    if cls.Type() == name:
+                        return cls
+        else:
+            for cls in self.types_list[module]:
+                if cls.Type() == name:
+                    return cls
+        return None