os: implement Barrier

libraries/libstratosphere/include/stratosphere/os.hpp

@@ -48,4 +48,5 @@
 #include <stratosphere/os/os_light_event.hpp>
 #include <stratosphere/os/os_light_message_queue.hpp>
 #include <stratosphere/os/os_light_semaphore.hpp>
+#include <stratosphere/os/os_barrier.hpp>
 #include <stratosphere/os/os_waitable.hpp>

libraries/libstratosphere/include/stratosphere/os/os_barrier.hpp

@@ -0,0 +1,55 @@
+/*
+ * Copyright (c) 2018-2020 Atmosphère-NX
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms and conditions of the GNU General Public License,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope 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 this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#pragma once
+#include <vapours.hpp>
+#include <stratosphere/os/os_barrier_types.hpp>
+#include <stratosphere/os/os_barrier_api.hpp>
+
+namespace ams::os {
+
+    class Barrier {
+        NON_COPYABLE(Barrier);
+        NON_MOVEABLE(Barrier);
+        private:
+            BarrierType m_barrier;
+        public:
+            explicit Barrier(int num_threads) {
+                InitializeBarrier(std::addressof(m_barrier), num_threads);
+            }
+
+            ~Barrier() {
+                FinalizeBarrier(std::addressof(m_barrier));
+            }
+
+            void Await() {
+                return AwaitBarrier(std::addressof(m_barrier));
+            }
+
+            operator BarrierType &() {
+                return m_barrier;
+            }
+
+            operator const BarrierType &() const {
+                return m_barrier;
+            }
+
+            BarrierType *GetBase() {
+                return std::addressof(m_barrier);
+            }
+    };
+
+}

libraries/libstratosphere/include/stratosphere/os/os_barrier_api.hpp

@@ -0,0 +1,29 @@
+/*
+ * Copyright (c) 2018-2020 Atmosphère-NX
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms and conditions of the GNU General Public License,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope 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 this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#pragma once
+#include <vapours.hpp>
+
+namespace ams::os {
+
+    struct BarrierType;
+
+    void InitializeBarrier(BarrierType *barrier, int num_threads);
+    void FinalizeBarrier(BarrierType *barrier);
+
+    void AwaitBarrier(BarrierType *barrier);
+
+}

libraries/libstratosphere/include/stratosphere/os/os_barrier_types.hpp

@@ -0,0 +1,35 @@
+/*
+ * Copyright (c) 2018-2020 Atmosphère-NX
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms and conditions of the GNU General Public License,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope 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 this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#pragma once
+#include <vapours.hpp>
+#include <stratosphere/os/impl/os_internal_critical_section.hpp>
+#include <stratosphere/os/impl/os_internal_condition_variable.hpp>
+
+namespace ams::os {
+
+    struct BarrierType {
+        u16 max_threads;
+        u16 waiting_threads;
+        u32 base_counter_lower;
+        u32 base_counter_upper;
+
+        impl::InternalCriticalSectionStorage cs_barrier;
+        impl::InternalConditionVariableStorage cv_gathered;
+    };
+    static_assert(std::is_trivial<BarrierType>::value);
+
+}