particle_data.hpp

Go to the documentation of this file.

#pragma once

 
#include <span>
#include <vector>
 
#include "orrery/core/aligned_allocator.hpp"
#include "orrery/core/types.hpp"
#include "orrery/core/vec3.hpp"
#include "orrery/core/vec3_span.hpp"
 
namespace orrery::core {

class ParticleData {
public:
    ParticleData() = default;

    explicit ParticleData(Index count);
 
    [[nodiscard]] Index size() const noexcept { return mass_.size(); }
 
    [[nodiscard]] bool empty() const noexcept { return mass_.empty(); }

    [[nodiscard]] Index capacity() const noexcept { return mass_.capacity(); }

    void reserve(Index count);

    void resize(Index count);

    void clear() noexcept;

    Index add(Vec3 position, Vec3 velocity, Real mass);
 
    [[nodiscard]] Vec3Span<Real> positions() noexcept {
        return {position_x_, position_y_, position_z_};
    }
 
    [[nodiscard]] Vec3Span<const Real> positions() const noexcept {
        return {position_x_, position_y_, position_z_};
    }
 
    [[nodiscard]] Vec3Span<Real> velocities() noexcept {
        return {velocity_x_, velocity_y_, velocity_z_};
    }
 
    [[nodiscard]] Vec3Span<const Real> velocities() const noexcept {
        return {velocity_x_, velocity_y_, velocity_z_};
    }
 
    [[nodiscard]] Vec3Span<Real> accelerations() noexcept {
        return {acceleration_x_, acceleration_y_, acceleration_z_};
    }
 
    [[nodiscard]] Vec3Span<const Real> accelerations() const noexcept {
        return {acceleration_x_, acceleration_y_, acceleration_z_};
    }
 
    [[nodiscard]] std::span<Real> masses() noexcept { return mass_; }
 
    [[nodiscard]] std::span<const Real> masses() const noexcept { return mass_; }
 
private:
    using ComponentArray = std::vector<Real, AlignedAllocator<Real>>;

    template<typename Operation> void for_each_array(Operation operation) {
        operation(position_x_);
        operation(position_y_);
        operation(position_z_);
        operation(velocity_x_);
        operation(velocity_y_);
        operation(velocity_z_);
        operation(acceleration_x_);
        operation(acceleration_y_);
        operation(acceleration_z_);
        operation(mass_);
    }

    static constexpr Index kInitialCapacity = 8;
 
    ComponentArray position_x_;
    ComponentArray position_y_;
    ComponentArray position_z_;
    ComponentArray velocity_x_;
    ComponentArray velocity_y_;
    ComponentArray velocity_z_;
    ComponentArray acceleration_x_;
    ComponentArray acceleration_y_;
    ComponentArray acceleration_z_;
    ComponentArray mass_;
};
 
} // namespace orrery::core