summaryrefslogtreecommitdiffhomepage
path: root/tracker-easy/point-extractor.h
blob: f275769d4ad1fa80cfe89eace249be4fe394548d (plain)
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
/*
 * Copyright (c) 2019 Stephane Lenclud
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 */

#pragma once

#include "settings.h"
#include <vector>

#include <opencv2/core.hpp>
#include <opencv2/imgproc.hpp>


namespace EasyTracker
{
    class PointExtractor 
    {
    public:
        PointExtractor();
        // extracts points from frame and draws some processing info into frame, if draw_output is set
        // dt: time since last call in seconds
        void ExtractPoints(const cv::Mat& aFrame, cv::Mat* aPreview, int aNeededPointCount, std::vector<cv::Point>& aPoints);

        void UpdateSettings();

        // Settings
        Settings iSettings;
        // Our frame with a channel size of 8 bits
        cv::Mat iFrameChannelSizeOne;
        // Our frame with a single 8 bits channel
        cv::Mat iFrameGray;
        //
        std::vector<std::vector<cv::Point> > iContours;

        // Take a copy of settings to avoid dead lock
        int iMinPointSize;
        int iMaxPointSize;

    };

}