-
Notifications
You must be signed in to change notification settings - Fork 1
/
center-of-mass.h
68 lines (58 loc) · 1.96 KB
/
center-of-mass.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
/**
* Copyright (C) 2018 Greenweaves Software Limited
*
* This is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this software. If not, see <http://www.gnu.org/licenses/>
*/
#ifndef _CENTRE_OF_MASS_H
#define _CENTRE_OF_MASS_H
#include "treecode.h"
/**
* Used to calculate centre of mass for Internal and External Nodes
*/
class CentreOfMassCalculator : public Node::Visitor {
public:
/**
* Create CentreOfMassCalculator.
* Parameters:
* particles These are the particles whose centre of mass is to be calculated.
*/
CentreOfMassCalculator(std::vector<Particle*> particles);
/**
* When we visit an External Node, record the position and mass of the particle
*/
Node::Visitor::Status visit(Node * node);
/**
* For an internal note we need to accumulate the mass and positions for each child
*/
virtual void propagate(Node * node,Node * child);
/**
* This is called when we finish processing a Node, which means that all children
* have been processed. Store centre of mass.
*/
virtual bool depart(Node * node);
/**
* Make sure every node was processed.
*/
void check_all_particles_processed();
private:
/**
* These are the particles whose centre of mass is to be calculated.
*/
std::vector<Particle*> _particles;
/**
* Indicates whether each particle has been processed
*/
std::vector<bool> _processed_particle;
};
#endif