ParticleChain.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstddef>
#include <vector>
 
#include <opencv2/core.hpp>
 
#include "vc/segmentation/stps/ForceChain.hpp"
#include "vc/segmentation/stps/Particle.hpp"
 
namespace volcart::segmentation
{
class ParticleChain
{
public:
    using Chain = std::vector<Particle>;
 
    ParticleChain() = default;
 
    explicit ParticleChain(Chain c) : data_{std::move(c)} {}
 
    auto operator+=(const ForceChain& rhs) -> ParticleChain&;
 
    auto operator*=(const double& rhs) -> ParticleChain&;
 
    auto operator[](std::size_t i) { return data_[i]; }
 
    auto operator[](std::size_t i) const { return data_[i]; }
 
    auto begin() { return data_.begin(); }
 
    auto begin() const { return data_.begin(); }
 
    auto end() { return data_.end(); }
 
    auto end() const { return data_.end(); }
 
    template <class... Args>
    void emplace_back(Args&&... args)
    {
        data_.emplace_back(std::forward<Args>(args)...);
    }
 
    void push_back(const Particle& val) { data_.push_back(val); }
 
    auto size() -> std::size_t { return data_.size(); }
 
    auto size() const -> std::size_t { return data_.size(); }
 
    void clear() { data_.clear(); }
 
private:
    Chain data_;
};
 
auto operator+(ParticleChain lhs, const ForceChain& rhs) -> ParticleChain;
auto operator+(const ForceChain& rhs, ParticleChain lhs) -> ParticleChain;
auto operator*(ParticleChain lhs, const double& rhs) -> ParticleChain;
auto operator*(const double& rhs, ParticleChain lhs) -> ParticleChain;
}  // namespace volcart::segmentation