Add base compiled code from GRChombo

Author: Masquerade0097

Examples/BinaryBH/BinaryBHLevel.cpp

@@ -0,0 +1,121 @@
+/* GRChombo
+ * Copyright 2012 The GRChombo collaboration.
+ * Please refer to LICENSE in GRChombo's root directory.
+ */
+
+#include "BinaryBHLevel.hpp"
+#include "BinaryBH.hpp"
+#include "BoxLoops.hpp"
+#include "CCZ4.hpp"
+#include "ChiExtractionTaggingCriterion.hpp"
+#include "ComputePack.hpp"
+#include "Constraints.hpp"
+#include "NanCheck.hpp"
+#include "PositiveChiAndAlpha.hpp"
+#include "SetValue.hpp"
+#include "TraceARemoval.hpp"
+#include "Weyl4.hpp"
+#include "WeylExtraction.hpp"
+
+void BinaryBHLevel::specificAdvance()
+{
+    // Enforce the trace free A_ij condition and positive chi and alpha
+    BoxLoops::loop(make_compute_pack(TraceARemoval(), PositiveChiAndAlpha()),
+                   m_state_new, m_state_new, INCLUDE_GHOST_CELLS);
+
+    // Check for nan's
+    if (m_p.nan_check)
+        BoxLoops::loop(NanCheck("NaNCheck in specific Advance: "), m_state_new,
+                       m_state_new, EXCLUDE_GHOST_CELLS, disable_simd());
+}
+
+void BinaryBHLevel::initialData()
+{
+    CH_TIME("BinaryBHLevel::initialData");
+    if (m_verbosity)
+        pout() << "BinaryBHLevel::initialData " << m_level << endl;
+
+    // Set up the compute class for the BinaryBH initial data
+    BinaryBH binary(m_p.bh1_params, m_p.bh2_params, m_dx);
+
+    // First set everything to zero (to avoid undefinded values in constraints)
+    // then calculate initial data
+    BoxLoops::loop(make_compute_pack(SetValue(0.), binary), m_state_new,
+                   m_state_new, INCLUDE_GHOST_CELLS);
+}
+
+void BinaryBHLevel::preCheckpointLevel()
+{
+    fillAllGhosts();
+    BoxLoops::loop(Constraints(m_dx), m_state_new, m_state_new,
+                   EXCLUDE_GHOST_CELLS);
+}
+
+void BinaryBHLevel::specificEvalRHS(GRLevelData &a_soln, GRLevelData &a_rhs,
+                                    const double a_time)
+{
+    // Enforce positive chi and alpha and trace free A
+    BoxLoops::loop(make_compute_pack(TraceARemoval(), PositiveChiAndAlpha()),
+                   a_soln, a_soln, INCLUDE_GHOST_CELLS);
+
+    // Calculate CCZ4 right hand side and set constraints to zero to avoid
+    // undefined values
+    BoxLoops::loop(
+        make_compute_pack(CCZ4(m_p.ccz4_params, m_dx, m_p.sigma),
+                          SetValue(0, Interval(c_Ham, NUM_VARS - 1))),
+        a_soln, a_rhs, EXCLUDE_GHOST_CELLS);
+}
+
+void BinaryBHLevel::specificUpdateODE(GRLevelData &a_soln,
+                                      const GRLevelData &a_rhs, Real a_dt)
+{
+    // Enforce the trace free A_ij condition
+    BoxLoops::loop(TraceARemoval(), a_soln, a_soln, INCLUDE_GHOST_CELLS);
+}
+
+void BinaryBHLevel::computeTaggingCriterion(FArrayBox &tagging_criterion,
+                                            const FArrayBox &current_state)
+{
+    BoxLoops::loop(
+        ChiExtractionTaggingCriterion(m_dx, m_level, m_p.extraction_params),
+        current_state, tagging_criterion);
+}
+
+void BinaryBHLevel::specificPostTimeStep()
+{
+    CH_TIME("BinaryBHLevel::specificPostTimeStep");
+    if (m_p.activate_extraction == 1)
+    {
+        // Populate the Weyl Scalar values on the grid
+        fillAllGhosts();
+        BoxLoops::loop(Weyl4(m_p.extraction_params.extraction_center, m_dx),
+                       m_state_new, m_state_new, EXCLUDE_GHOST_CELLS);
+
+        // Do the extraction on the min extraction level
+        if (m_level == m_p.extraction_params.min_extraction_level)
+        {
+            // Now refresh the interpolator and do the interpolation
+            m_gr_amr.m_interpolator->refresh();
+            WeylExtraction my_extraction(m_p.extraction_params, m_dt, m_time,
+                                         m_restart_time);
+            my_extraction.execute_query(m_gr_amr.m_interpolator);
+        }
+    }
+}
+
+// Things to do before a plot level - need to calculate the Weyl scalars
+void BinaryBHLevel::prePlotLevel()
+{
+    fillAllGhosts();
+    if (m_p.activate_extraction == 1)
+    {
+        BoxLoops::loop(Weyl4(m_p.extraction_params.extraction_center, m_dx),
+                       m_state_new, m_state_new, EXCLUDE_GHOST_CELLS);
+    }
+}
+
+// Specify if you want any plot files to be written, with which vars
+void BinaryBHLevel::specificWritePlotHeader(std::vector<int> &plot_states) const
+{
+    plot_states = {c_chi, c_Weyl4_Re, c_Weyl4_Im};
+}

Examples/BinaryBH/BinaryBHLevel.hpp

@@ -0,0 +1,52 @@
+/* GRChombo
+ * Copyright 2012 The GRChombo collaboration.
+ * Please refer to LICENSE in GRChombo's root directory.
+ */
+
+#ifndef BINARYBHLEVEL_HPP_
+#define BINARYBHLEVEL_HPP_
+
+#include "DefaultLevelFactory.hpp"
+#include "GRAMRLevel.hpp"
+
+class BinaryBHLevel : public GRAMRLevel
+{
+    friend class DefaultLevelFactory<BinaryBHLevel>;
+    // Inherit the contructors from GRAMRLevel
+    using GRAMRLevel::GRAMRLevel;
+
+    /// Things to do at every full timestep
+    ///(might include several substeps, e.g. in RK4)
+    virtual void specificAdvance() override;
+
+    /// Initial data calculation
+    virtual void initialData() override;
+
+    /// Any actions that should happen just before checkpointing
+    virtual void preCheckpointLevel() override;
+
+    /// Calculation of the right hand side for the time stepping
+    virtual void specificEvalRHS(GRLevelData &a_soln, GRLevelData &a_rhs,
+                                 const double a_time) override;
+
+    /// Things to do after dt*rhs has been added to the solution
+    virtual void specificUpdateODE(GRLevelData &a_soln,
+                                   const GRLevelData &a_rhs,
+                                   Real a_dt) override;
+
+    /// Identify and tag the cells that need higher resolution
+    virtual void
+    computeTaggingCriterion(FArrayBox &tagging_criterion,
+                            const FArrayBox &current_state) override;
+
+    // to do post each time step on every level
+    virtual void specificPostTimeStep() override;
+
+    /// Any actions that should happen just before plot files output
+    virtual void prePlotLevel() override;
+
+    //! Specify which variables to write at plot intervals
+    virtual void specificWritePlotHeader(std::vector<int> &plot_states) const;
+};
+
+#endif /* BINARYBHLEVEL_HPP_ */