HTWKVision/hypotheses_generator.h at master · NaoHTWK/HTWKVision · 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
#ifndef HYPOTHESES_GENERATOR_H
#define HYPOTHESES_GENERATOR_H

#include <cstdint>

#include <vector>

#include "field_color_detector.h"
#include "integral_image.h"
#include "object_hypothesis.h"
#include <localization_utils.h>

namespace htwk {

class HypothesesGenerator
{
public:
    HypothesesGenerator() = default;
    HypothesesGenerator(const HypothesesGenerator&) = delete;
    HypothesesGenerator(HypothesesGenerator&&) = delete;
    HypothesesGenerator& operator=(const HypothesesGenerator&) = delete;
    HypothesesGenerator& operator=(HypothesesGenerator&&) = delete;
    virtual ~HypothesesGenerator() = default;

    virtual void proceed(uint8_t *img, const std::vector<int>& fieldborder, CamPose& cam_pose, IntegralImage *integralImg) = 0;
    virtual std::vector<ObjectHypothesis> getHypotheses() const = 0;
    virtual int* getRatingImg() = 0;
    virtual uint8_t* getDebugImg() = 0;
    virtual void setDebugActive(bool active) = 0;

};

} // htwk

#endif // HYPOTHESES_GENERATOR_H