Components.h

Go to the documentation of this file.

/* Copyright (C) 2002 Dominic Letourneau (dominic.letourneau@usherbrooke.ca)

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 2.1 of the License, or (at your option) any later version.
   
   This library 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
   Lesser General Public License for more details.
   
   You should have received a copy of the GNU Lesser General Public
   License along with this library; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/
// Components.h: interface for the CComponents class.
//
//////////////////////////////////////////////////////////////////////

#ifndef _COMPONENTS_H_
#define _COMPONENTS_H_

#include <vector>
#include <list>
#include "CRect.h"
#include "BufferedNode.h"
#include "Image.h"

namespace RobotFlow {

#define MAX_COLOR 32     //THIS ALGORITHM CANNOT ALLOW MORE THAN 32 COLORS (CAUSE 32bits for unsigned int)
#define MAX_CELLS 76800  //IMAGE_WIDTH * IMAGE_HEIGHT
#define MAX_RECT 4800    //MAX_CELLS / MIN_RECT_AREA + 1
#define MAX_X_GAP 3      //USER DEFINED X GAP BETWEEN TWO PIXELS OF THE SAME COLOR
#define MAX_Y_GAP 3      //USER DEFINED Y GAP BETWEEN TWO PIXELS OF THE SAME COLOR

//local prototype (for sort)
int CRect_compare(const void *a, const void *b);

class Components : public FD::BufferedNode {


public:
  
  Components();

  Components(std::string nodeName, FD::ParameterSet params);

  ~Components();



  void get_components (Image &image);

  void initialize (int width, int height, int num_colors);

  int get_component_count(){return m_rect_count;}

  std::list<CRect*> & get_color_components(int color);

  void draw_rectangles(int color, unsigned short *image_16bits, unsigned short value);

  void calculate(int output_id, int count, FD::Buffer &out);

  void reset();
  
 private:

  void get_components (unsigned short *image_16bits);
  void get_components (unsigned char *image_8bits);
  void find_runs (unsigned short *image);
  void find_runs (unsigned char *image);

  void recursive_compute(int line_number);

  void sort(int color);
  
  std::vector<std::list<ImageCell *> > m_cell_list[MAX_COLOR];

  //list<CRect*> m_rect_list[MAX_COLOR];
  
  int m_width;
  int m_height;
  int m_color_number;
  int m_cell_count;
  int m_rect_count;
  int m_x_gap;
  int m_y_gap;
  int m_minArea;
  int m_imageID;
  int m_lookupID;
  int m_blobsID;

  ImageCell  *m_cell_array; //reusable cells
  CRect *m_rect_array; //reusable rectangles
  
  CRect *m_current_rect; 

  //the color table
  unsigned int *m_color_table;

  int m_num_colors;

  FD::ObjectRef m_components_data;
  
};

}//namespace RobotFlow

#endif 

---