system.hh
No OneTemporary
Actions

Subscribers

None

File Metadata

	#ifndef __SYSTEM__HH__
	#define __SYSTEM__HH__

	/* -------------------------------------------------------------------------- */
	#include "my_types.hh"
	#include "particle.hh"
	#include <memory>
	#include <vector>
	/* -------------------------------------------------------------------------- */

	//! Container for particles
	class System {
	/*
	* No need for constructor/destructor with std::unique_ptr
	*/

	// List of particle pointers
	using ParticleList = std::vector<std::shared_ptr<Particle>>;

	// Methods
	public:
	//! Remove particle from vector
	void removeParticle(UInt particle);
	//! Get particle for specific id
	Particle& getParticle(UInt i);
	//! Add a particle to the system
	void addParticle(const std::shared_ptr<Particle>& new_particle);
	//! Get number of particles
	UInt getNbParticles();


	//! Iterator class to erase the unique pointer on Particle
	struct iterator : ParticleList::iterator {
	iterator(const ParticleList::iterator& it) : ParticleList::iterator(it) {}

	//! Access the underlying particle
	Particle& operator*() {
	return ParticleList::iterator::operator();
	}
	};

	// Iterators
	public:
	auto begin() { return iterator(list_particles.begin()); }
	auto end() { return iterator(list_particles.end()); }



	private:
	ParticleList list_particles;
	};

	/* -------------------------------------------------------------------------- */
	#endif //__SYSTEM__HH__