detection.h

//
// Projectname: amos-ss16-proj5
//
// Copyright (c) 2016 de.fau.cs.osr.amos2016.gruppe5
//
// This file is part of the AMOS Project 2016 @ FAU
// (Friedrich-Alexander University Erlangen-Nürnberg)
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public
// License along with this program. If not, see
// <http://www.gnu.org/licenses/>.
//

#ifndef DETECTION_H
#define DETECTION_H

// opencv
#include <opencv2/opencv.hpp>

//local
#include "../StreamDecoder/image.h"
#include "../StreamDecoder/image_view.h"
#include "detector.h"
#include "frame_detection_data.h"

const double default_contrast_value = 3.0;
const int default_brightness_value = 50;

class Detection {

public:

    Detection(Detector * peopleDetector, Detector * vehicleDetector);

    ~Detection();

    FrameDetectionData* ProcessFrame(Image * image);

private:

    Detector * people_detector_; 
    Detector * vehicle_detector_; 
    ImageView * image_view_;    

    cv::Mat ResizeFrame(cv::Mat *frame);
    
    cv::Mat AdjustContrastAndBrightness(cv::Mat *frame, double contrastValue, int brightnessValue);
    

    cv::Mat AdjustContrastAndBrightness(cv::Mat *frame);

};


#endif // DETECTION_H