Add bitops macro

Author: Erfan-Ahmadi

include/nbl/asset/IDescriptorSetLayout.h

@@ -86,7 +86,7 @@ class NBL_API IDescriptorSetLayout : public virtual core::IReferenceCounted
 			uint32_t binding;
 			E_DESCRIPTOR_TYPE type;
 			uint32_t count;
-			IShader::E_SHADER_STAGE stageFlags;
+			IShader::E_SHADER_STAGE stageFlags; // TODO: make it bitflag
 			// Use this if you want an immutable sampler that is baked into the DS layout itself.
 			// If its `nullptr` then the sampler used is mutable and can be specified while writing the image descriptor to a binding while updating the DS.
 			const core::smart_refctd_ptr<sampler_type>* samplers;

include/nbl/asset/IShader.h

@@ -75,6 +75,8 @@ class NBL_API IShader : public virtual core::IReferenceCounted
 	std::string m_filepathHint;
 };
 
+NBL_ENUM_ADD_BITWISE_OPERATORS(IShader::E_SHADER_STAGE)
+
 }
 
 #endif

include/nbl/macros.h

@@ -253,5 +253,46 @@
 
 #define NBL_FOREACH(WHAT, ... ) NBL_EVAL(NBL_CONCATENATE(NBL_FOREACH_,NBL_VA_ARGS_COUNT(__VA_ARGS__))(WHAT, __VA_ARGS__))
 
+#define NBL_ENUM_ADD_BITWISE_OPERATORS(EnumType)												\
+	inline constexpr EnumType operator&(const EnumType& lhs, const EnumType& rhs) noexcept		\
+	{																							\
+		using T = typename std::underlying_type_t<EnumType>;									\
+		return static_cast<EnumType>(static_cast<T>(lhs) & static_cast<T>(rhs));				\
+	}																							\
+																								\
+	inline constexpr EnumType& operator&=(EnumType& lhs, const EnumType& rhs) noexcept			\
+	{																							\
+	return lhs = lhs & rhs;																		\
+	}																							\
+																								\
+	inline constexpr EnumType operator|(const EnumType& lhs, const EnumType& rhs) noexcept		\
+	{																							\
+	using T = typename std::underlying_type_t<EnumType>;										\
+	return static_cast<EnumType>(static_cast<T>(lhs) | static_cast<T>(rhs));					\
+	}																							\
+																								\
+	inline constexpr EnumType& operator|=(EnumType& lhs, const EnumType& rhs) noexcept			\
+	{																							\
+	return lhs = lhs | rhs;																		\
+	}																							\
+																								\
+	inline constexpr EnumType operator^(const EnumType& lhs, const EnumType& rhs) noexcept		\
+	{																							\
+	using T = typename std::underlying_type_t<EnumType>;										\
+	return static_cast<EnumType>(static_cast<T>(lhs) ^ static_cast<T>(rhs));					\
+	}																							\
+																								\
+	inline constexpr EnumType& operator^=(EnumType& lhs, const EnumType& rhs) noexcept			\
+	{																							\
+	return lhs = lhs ^ rhs;																		\
+	}																							\
+																								\
+	inline constexpr EnumType& operator~(EnumType& e) noexcept									\
+	{																							\
+	using T = typename std::underlying_type_t<EnumType>;										\
+	return e = static_cast<EnumType>(~static_cast<T>(e));										\
+	}																							\
+	/**/
+
 
 #endif