RelativeThermostat.cpp File Reference

#include "RelativeThermostat.h"
#include <cmath>
#include "utils/ArrayUtils.h"

Include dependency graph for RelativeThermostat.cpp:

Go to the source code of this file.

Functions
std::array< double, 3 >	averageVelocity (const std::unique_ptr< ParticleContainer > &particle_container)

Function Documentation

averageVelocity()

std::array<double, 3> averageVelocity

(

const std::unique_ptr< ParticleContainer > &

particle_container

)

Definition at line 10 of file RelativeThermostat.cpp.

                                                                                                {
    std::array<double, 3> average_velocity = {0, 0, 0};
    size_t count = 0;
    for (auto& particle : *particle_container) {
        if (particle.isLocked()) continue;
        average_velocity = average_velocity + particle.getV();
        count++;
    }
    average_velocity = (1.0 / count) * average_velocity;
    return average_velocity;
}