00001
00002 /***************************************************************************
00003  *  rht_circle.h - Header of circle shape model
00004  *                 using Randomized Hough Transform
00005  *
00006  *  Generated: Tue Jun 28 2005
00007  *  Copyright  2005  Hu Yuxiao      <Yuxiao.Hu@rwth-aachen.de>
00008  *
00009  ****************************************************************************/
00010
00011 /*  This program is free software; you can redistribute it and/or modify
00012  *  it under the terms of the GNU General Public License as published by
00013  *  the Free Software Foundation; either version 2 of the License, or
00014  *  (at your option) any later version. A runtime exception applies to
00015  *  this software (see LICENSE.GPL_WRE file mentioned below for details).
00016  *
00017  *  This program is distributed in the hope that it will be useful,
00018  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
00019  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
00020  *  GNU Library General Public License for more details.
00021  *
00022  *  Read the full text in the LICENSE.GPL_WRE file in the doc directory.
00023  */
00024
00025 #ifndef __FIREVISION_RHT_CIRCLE_H_
00026 #define __FIREVISION_RHT_CIRCLE_H_
00027 
00028 #include <vector>
00029 #include <iostream>
00030
00031 #include <utils/math/types.h>
00032 #include <fvutils/base/types.h>
00033 #include <models/shape/circle.h>
00034 #include <models/shape/accumulators/ht_accum.h>
00035
00036 class ROI;
00037
00038 class RhtCircleModel: public ShapeModel
00039 {
00040  private:
00041   std::vector<Circle> m_Circles;
00042   RhtAccumulator accumulator;
00043   static const float RHT_MIN_RADIUS;
00044   static const float RHT_MAX_RADIUS;
00045
00046  public:
00047   RhtCircleModel(void);
00048   virtual ~RhtCircleModel(void);
00049
00050   std::string   getName(void) const {return std::string("RhtCircleModel");}
00051   int           parseImage(unsigned char* buffer, ROI *roi);
00052   int           getShapeCount(void) const;
00053   Circle*       getShape(int id) const;
00054   Circle*       getMostLikelyShape(void) const;
00055
00056  private:
00057   void          calcCircle(     // for calculating circles from 3 points
00058                            const fawkes::point_t& p1,
00059                            const fawkes::point_t& p2,
00060                            const fawkes::point_t& p3,
00061                            center_in_roi_t& center,
00062                            float& radius);
00063 };
00064
00065 #endif // __FIREVISION_RHT_CIRCLE_H_
00066