Subversion Repositories OpenCV2-Cookbook

Rev

View as "text/plain" | Blame | Last modification | View Log | RSS feed

1
/*------------------------------------------------------------------------------------------*\
   This file contains material supporting chapter 3 of the cookbook:  
   Computer Vision Programming using the OpenCV Library.
   by Robert Laganiere, Packt Publishing, 2011.

   This program is free software; permission is hereby granted to use, copy, modify,
   and distribute this source code, or portions thereof, for any purpose, without fee,
   subject to the restriction that the copyright notice may not be removed
   or altered from any source or altered source distribution.
   The software is released on an as-is basis and without any warranties of any kind.
   In particular, the software is not guaranteed to be fault-tolerant or free from failure.
   The author disclaims all warranties with regard to this software, any use,
   and any consequent failure, is purely the responsibility of the user.
 
   Copyright (C) 2010-2011 Robert Laganiere, www.laganiere.name
\*------------------------------------------------------------------------------------------*/


#if !defined CD_CNTRLLR
#define CD_CNTRLLR

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include "colordetector.h"

class ColorDetectController {

  private:

        static ColorDetectController *singleton; // pointer to the singleton

        ColorDetector *cdetect;

        // The image to be processed
        cv::Mat image;
        cv::Mat result;
       
  public:
        ColorDetectController() { // private constructor

                  //setting up the application
                  cdetect= new ColorDetector();
        }

          // Sets the color distance threshold
          void setColorDistanceThreshold(int distance) {

                  cdetect->setColorDistanceThreshold(distance);
          }

          // Gets the color distance threshold
          int getColorDistanceThreshold() const {

                  return cdetect->getColorDistanceThreshold();
          }

          // Sets the color to be detected
          void setTargetColor(unsigned char red, unsigned char green, unsigned char blue) {

                  cdetect->setTargetColor(red,green,blue);
          }

          // Gets the color to be detected
          void getTargetColor(unsigned char &red, unsigned char &green, unsigned char &blue) const {

                  cv::Vec3b color= cdetect->getTargetColor();

                  red= color[2];
                  green= color[1];
                  blue= color[0];
          }

          // Sets the input image. Reads it from file.
          bool setInputImage(std::string filename) {

                  image= cv::imread(filename);

                  if (!image.data)
                          return false;
                  else
                          return true;
          }

          // Returns the current input image.
          const cv::Mat getInputImage() const {

                  return image;
          }

          // Performs image processing.
          void process() {

                  result= cdetect->process(image);
          }
         

          // Returns the image result from the latest processing.
          const cv::Mat getLastResult() const {

                  return result;
          }

          // Deletes all processor objects created by the controller.
          ~ColorDetectController() {

                  delete cdetect;
          }

          // Singleton static members
          static ColorDetectController *getInstance() {

                  if (singleton == 0)
                        singleton= new ColorDetectController;

                  return singleton;
          }

          // Releases the singleton instance of this controller.
          static void destroy() {

                  if (singleton != 0) {
                          delete singleton;
                          singleton= 0;
                  }
          }
};

#endif