ActuatorContainer.H

Go to the documentation of this file.

#ifndef ACTUATORCONTAINER_H
#define ACTUATORCONTAINER_H
 
#include "src/core/vs/vector_space.H"
 
#include "AMReX_AmrParticles.H"
 
namespace kynema_sgf {
 
class Field;
 
namespace actuator {
 
class Actuator;

struct ActuatorCloud
{
    amrex::Vector<int> num_pts;

    amrex::Vector<int> global_id;

    amrex::Vector<vs::Vector> position;

    amrex::Vector<vs::Vector> velocity;

    amrex::Vector<amrex::Real> density;

    int num_objects;
 
    explicit ActuatorCloud(int nobjects);
};

static constexpr int NumPStructReal = AMREX_SPACEDIM + 1;
static constexpr int NumPStructInt = 1;
static constexpr int NumPArrayReal = 0;
static constexpr int NumPArrayInt = 0;

class ActuatorContainer
    : public amrex::AmrParticleContainer<
          NumPStructReal,
          NumPStructInt,
          NumPArrayReal,
          NumPArrayInt>
{
public:
    friend class Actuator;
 
    explicit ActuatorContainer(amrex::AmrCore& mesh, int num_objects);
 
    void post_regrid_actions();
 
    void initialize_container();
 
    void reset_container();
 
    void update_positions();
 
    void sample_fields(const Field& vel, const Field& density);
 
    int num_actuator_points() const
    {
        return static_cast<int>(m_data.position.size());
    }
 
    // public for CUDA, not safe for general access
 
    void interpolate_fields(const Field& vel, const Field& density);
 
    void populate_field_buffers();
 
    void initialize_particles(int total_pts);
 
protected:
    void compute_local_coordinates();
 
    // Accessor to allow unit testing
    ActuatorCloud& point_data() { return m_data; }
 
private:
    amrex::AmrCore& m_mesh;
 
    // Object that holds the position and velocity information
    ActuatorCloud m_data;

    amrex::Vector<vs::Vector> m_proc_pos;

    amrex::Gpu::DeviceVector<vs::Vector> m_pos_device;
 
    amrex::Vector<int> m_proc_offsets;
    amrex::Gpu::DeviceVector<int> m_proc_offsets_device;

    bool m_container_initialized{false};

    bool m_is_scattered{false};
};
 
} // namespace actuator
} // namespace kynema_sgf
 
#endif /* ACTUATORCONTAINER_H */