Mercurial > hg > nsaunier > traffic-intelligence

comparison trajectorymanagement/src/TrajectoryElement.h @ 1159:e1e7acef8eab

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
moved trajectory management library into Traffic Intelligence
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Mon, 22 Feb 2021 22:09:35 -0500
parents
children
comparison
equal deleted inserted replaced
1158:7eb972942f22 1159:e1e7acef8eab
1 #ifndef TRAJECTORYELEMENT_H_
2 #define TRAJECTORYELEMENT_H_
3
4 #include <istream>
5 #include <ostream>
6
7 #include "PointOperations.h"
8
9 /**
10 * TrajectoryElement class.
11 *
12 * The Trajectory class is composed of components called TrajectoryElement.
13 */
14 template<typename T>
15 class TrajectoryElement {
16 public:
17 /**
18 * Constructor.
19 */
20 TrajectoryElement() {}
21
22 /**
23 * Constructor.
24 *
25 * @param[in] frameNumber number of a frame
26 * @param[in] position position
27 */
28 TrajectoryElement(unsigned int frameNumber, const T& point) {
29 setFrameNumber(frameNumber);
30 setPoint(point);
31 }
32
33 /**
34 * Constructor.
35 *
36 * @param[in] trajectoryElement element of the trajectory
37 */
38 TrajectoryElement(const TrajectoryElement& trajectoryElement) {
39 setFrameNumber(trajectoryElement.getFrameNumber());
40 setPoint(trajectoryElement.getPoint());
41 }
42
43 /**
44 * Get number of a frame.
45 *
46 * @return number of a frame
47 */
48 unsigned int getFrameNumber() const {
49 return frameNumber;
50 }
51
52 /**
53 * Set number of a frame.
54 *
55 * @param[in] frameNumber number of a frame
56 */
57 void setFrameNumber(unsigned int iframeNumber) {
58 frameNumber = iframeNumber;
59 }
60
61 /**
62 * Get position.
63 *
64 * @return position
65 */
66 const T &getPoint() const {
67 return point;
68 }
69
70 /**
71 * Set position.
72 *
73 * @param[in] position position
74 */
75 void setPoint(const T& ipoint) {
76 point = ipoint;
77 }
78
79 /**
80 * Shift position.
81 *
82 * @param[in] t vector by which the position of the element should be shifted.
83 */
84 void shift(const T& t) {
85 point += t;
86 }
87
88 private:
89 /**
90 * Frame number.
91 */
92 unsigned int frameNumber;
93
94 /**
95 * Position of the element.
96 */
97 T point;
98 };
99
100 template<typename T>
101 static inline bool operator==(const TrajectoryElement<T>& a, const TrajectoryElement<T>& b) {
102 return (a.getFrameNumber() == b.getFrameNumber()) && (a.getPoint() == b.getPoint());
103 }
104
105 template<typename Ti, typename Tr>
106 std::basic_istream<Ti>& operator>>(std::basic_istream<Ti>& in, TrajectoryElement<Tr>& trajectoryElement) {
107 unsigned int frameNumber;
108 Tr point;
109 if (in >> frameNumber >> point) {
110 trajectoryElement.setFrameNumber(frameNumber);
111 trajectoryElement.setPoint(point);
112 }
113 return in;
114 }
115
116 template<typename Tr>
117 std::ostream& operator<<(std::ostream& out, const TrajectoryElement<Tr>& trajectoryElement) {
118 out << trajectoryElement.getFrameNumber() << " " << trajectoryElement.getPoint();
119 return out;
120 }
121
122 #endif /* TRAJECTORYELEMENT_H_ */