Finishing the renaming.
[flatland.git] / universe.h
1
2 /*
3  *  dyncnn is a deep-learning algorithm for the prediction of
4  *  interacting object dynamics
5  *
6  *  Copyright (c) 2016 Idiap Research Institute, http://www.idiap.ch/
7  *  Written by Francois Fleuret <francois.fleuret@idiap.ch>
8  *
9  *  This file is part of dyncnn.
10  *
11  *  dyncnn is free software: you can redistribute it and/or modify it
12  *  under the terms of the GNU General Public License version 3 as
13  *  published by the Free Software Foundation.
14  *
15  *  dyncnn is distributed in the hope that it will be useful, but
16  *  WITHOUT ANY WARRANTY; without even the implied warranty of
17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  *  General Public License for more details.
19  *
20  *  You should have received a copy of the GNU General Public License
21  *  along with dyncnn.  If not, see <http://www.gnu.org/licenses/>.
22  *
23  */
24
25 #ifndef UNIVERSE_H
26 #define UNIVERSE_H
27
28 #include <iostream>
29 #include <cmath>
30
31 #include "misc.h"
32 #include "canvas.h"
33 #include "polygon.h"
34
35 using namespace std;
36
37 class Universe {
38   scalar_t _width, _height;
39 public:
40   int _nb_max_polygons, _nb_polygons;
41   Polygon **_polygons;
42
43   inline scalar_t width() { return _width; }
44   inline scalar_t height() { return _height; }
45
46   Universe(int nb_max_polygons, scalar_t width, scalar_t height);
47   // The destructor deletes all the added polygons
48   ~Universe();
49
50   void initialize_polygon(Polygon *p);
51   void clear();
52   void add_polygon(Polygon *p);
53   bool collide(Polygon *p);
54
55   // Compute collisions between projections of the polygons on a few
56   // axis to speed up the computation
57   void compute_pseudo_collisions(int nb_axis, int *nb_colliding_axis);
58   void apply_collision_forces(scalar_t dt);
59   bool update(scalar_t dt, scalar_t padding);
60
61   Polygon *pick_polygon(scalar_t x, scalar_t y);
62
63   void draw(Canvas *canvas);
64 };
65
66 #endif