Improve socket operator= with vector and array support.

aff3ct · Jun 8, 2021 · c68f71c · c68f71c
1 parent 87215f5
commit c68f71c
Show file tree

Hide file tree

Showing 4 changed files with 310 additions and 218 deletions.
diff --git a/include/Module/Socket.hpp b/include/Module/Socket.hpp
@@ -68,6 +68,18 @@ class Socket : public tools::Interface_reset
 
 	inline void operator()(Socket &s_out, const int priority = -1);
 
+	template <typename T>
+	inline void operator=(const T *array);
+
+	template <typename T>
+	inline void operator=(T *array);
+
+	template <typename T, class A = std::allocator<T>>
+	inline void operator=(const std::vector<T,A> &vector);
+
+	template <typename T, class A = std::allocator<T>>
+	inline void operator=(std::vector<T,A> &vector);
+
 	inline void operator=(Socket &s);
 
 	inline void operator=(Task &t);

diff --git a/include/Module/Socket.hxx b/include/Module/Socket.hxx
@@ -236,6 +236,86 @@ void Socket
 	bind(s_out, priority);
 }
 
+template <typename T>
+void Socket
+::operator=(const T *array)
+{
+	if (this->get_type() == socket_t::SIN)
+		this->bind(array);
+	else
+	{
+		std::stringstream message;
+		message << "Current socket have to be an input socket ("
+		        << "'datatype'"           << " = " << type_to_string[this->datatype] << ", "
+		        << "'name'"               << " = " << get_name()                     << ", "
+		        << "'task.name'"          << " = " << task.get_name()                << ", "
+		        << "'type'"               << " = " << (get_type() == socket_t::SIN ? "SIN" : "SOUT") /*<< ", "*/
+//		        << "'task.module.name'"   << " = " << task.get_module_name()
+		        << ").";
+		throw tools::runtime_error(__FILE__, __LINE__, __func__, message.str());
+	}
+}
+
+template <typename T>
+void Socket
+::operator=(T *array)
+{
+	if (this->get_type() == socket_t::SIN)
+		this->bind(array);
+	else
+	{
+		std::stringstream message;
+		message << "Current socket have to be an input socket ("
+		        << "'datatype'"           << " = " << type_to_string[this->datatype] << ", "
+		        << "'name'"               << " = " << get_name()                     << ", "
+		        << "'task.name'"          << " = " << task.get_name()                << ", "
+		        << "'type'"               << " = " << (get_type() == socket_t::SIN ? "SIN" : "SOUT") /*<< ", "*/
+//		        << "'task.module.name'"   << " = " << task.get_module_name()
+		        << ").";
+		throw tools::runtime_error(__FILE__, __LINE__, __func__, message.str());
+	}
+}
+
+template <typename T, class A>
+void Socket
+::operator=(const std::vector<T,A> &vector)
+{
+	if (this->get_type() == socket_t::SIN)
+		this->bind(vector);
+	else
+	{
+		std::stringstream message;
+		message << "Current socket have to be an input socket ("
+		        << "'datatype'"           << " = " << type_to_string[this->datatype] << ", "
+		        << "'name'"               << " = " << get_name()                     << ", "
+		        << "'task.name'"          << " = " << task.get_name()                << ", "
+		        << "'type'"               << " = " << (get_type() == socket_t::SIN ? "SIN" : "SOUT") /*<< ", "*/
+//		        << "'task.module.name'"   << " = " << task.get_module_name()
+		        << ").";
+		throw tools::runtime_error(__FILE__, __LINE__, __func__, message.str());
+	}
+}
+
+template <typename T, class A>
+void Socket
+::operator=(std::vector<T,A> &vector)
+{
+	if (this->get_type() == socket_t::SIN)
+		this->bind(vector);
+	else
+	{
+		std::stringstream message;
+		message << "Current socket have to be an input socket ("
+		        << "'datatype'"           << " = " << type_to_string[this->datatype] << ", "
+		        << "'name'"               << " = " << get_name()                     << ", "
+		        << "'task.name'"          << " = " << task.get_name()                << ", "
+		        << "'type'"               << " = " << (get_type() == socket_t::SIN ? "SIN" : "SOUT") /*<< ", "*/
+//		        << "'task.module.name'"   << " = " << task.get_module_name()
+		        << ").";
+		throw tools::runtime_error(__FILE__, __LINE__, __func__, message.str());
+	}
+}
+
 void Socket
 ::operator=(Socket &s)
 {