TCPSocketServer/utils.h at master · modelec/TCPSocketServer · GitHub

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
69
70
#pragma once

#include <array>
#include <utility>
#include <vector>
#include <string>
#include <ostream>
#include <cmath>

#define PI 3.14159265358979323846

enum PinceState {
    WHITE_FLOWER,
    PURPLE_FLOWER,
    FLOWER,
    NONE
};

namespace TCPUtils {
    bool startWith(const std::string& str, const std::string& start);

    bool endWith(const std::string& str, const std::string& end);

    bool contains(const std::string& str, const std::string& sub);

    std::vector<std::string> split(const std::string& str, const std::string& delimiter);
}

class ArucoTag {

public:
    ArucoTag(int id, std::string name, std::array<float, 2> pos, std::array<float, 3> rot);

    ArucoTag() = default;

    ArucoTag(const ArucoTag& other) = default;

    [[nodiscard]] int id() const;

    [[nodiscard]] std::string name() const;

    [[nodiscard]] std::array<float, 2> pos() const;

    [[nodiscard]] std::array<float, 3> rot() const;

    void setId(int id);

    void setName(const std::string& name);

    void setPos(float x, float y);

    void setRot(float x, float y, float z);

    ArucoTag& operator=(const ArucoTag& tag);

    void find();

    [[nodiscard]] int getNbFind() const;

    friend std::ostream& operator<<(std::ostream& os, const ArucoTag& tag);

private:
    int _id = -1;
    std::string _name;
    std::array<float, 2> _pos = {0, 0};
    std::array<float, 3> _rot = {0, 0, 0};
    int nbFind = 1;
};

double distanceToTag(const ArucoTag& tag);