74-2d-thermal…

workspace.py
View Options

			__copyright__ = (
			"Copyright (©) 2021-2023 EPFL (Ecole Polytechnique Fédérale de Lausanne)"
			"Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)"
			)
			__license__ = "LGPLv3"


			import os


			class Workspace:
			def __init__(self, include_paths, suffixes=['.c', '.cpp', '.cc', '.cxx']):
			self._include_paths = include_paths
			self._suffixes = suffixes

			@property
			def files(self):
			paths = self._walk()
			return [path for path in paths
			if self._should_include(path)]

			@property
			def include_paths(self):
			paths = self._walk()
			return [path for path in paths if os.path.isdir(path)]

			@property
			def paths(self):
			return [path for path in self._include_paths
			if self._should_include(path) or os.path.isdir(path)]

			def _should_include(self, name):
			_, ext = os.path.splitext(name)
			return ext in self._suffixes

			def _walk(self):
			paths = []

			for path in self._include_paths:
			if os.path.isdir(path):
			for root, dirs, files in os.walk(path):
			paths.extend(
			[os.path.join(root, dir_) for dir_ in dirs])
			paths.extend(
			[os.path.join(root, file_) for file_ in files])
			else:
			paths.append(path)

			return [os.path.abspath(path) for path in paths]