GridHealer.h

/******************************************************************************************************
 * (C) 2014 markummitchell@github.com. This file is part of Engauge Digitizer, which is released      *
 * under GNU General Public License version 2 (GPLv2) or (at your option) any later version. See file *
 * LICENSE or go to gnu.org/licenses for details. Distribution requires prior written permission.     *
 ******************************************************************************************************/
 
#ifndef GRID_HEALER_H
#define GRID_HEALER_H
 
#include <QMap>
#include <QPointF>
#include <QVector>
 
class DocumentModelGridRemoval;
class QImage;
 
enum PixelState {
  PIXEL_STATE_BACKGROUND, 
  PIXEL_STATE_FOREGROUND, 
  PIXEL_STATE_REMOVED,    
  PIXEL_STATE_ADJACENT,   
  PIXEL_STATE_HEALED,     
  NUM_PIXEL_STATES
};
 
typedef int BoundaryGroup;
 
typedef int PixelStateOrBoundaryGroup;
 
typedef QMap<BoundaryGroup, QPointF> GroupNumberToPoint;
 
class GridHealer
{
 public:
  GridHealer(const QImage &imageBefore,
             const DocumentModelGridRemoval &modelGridRemoval);
 
  void erasePixel (int xCol,
                   int yRow);
 
  void heal (QImage &imageToHeal);
 
 private:
  GridHealer();
 
  void connectCloseGroups(QImage &imageToHeal);
  void groupContiguousAdjacentPixels();
  void recursiveSearchForAdjacentPixels (int boundaryGroup,
                                         int row,
                                         int col,
                                         int &centroidCount,
                                         double &rowCentroidSum,
                                         double &colCentroidSum);
 
  // Mirror of original image
  QVector<QVector<PixelStateOrBoundaryGroup> > m_pixels;
 
  BoundaryGroup m_boundaryGroupNext;
 
  GroupNumberToPoint m_groupNumberToCentroid;
 
  GroupNumberToPoint m_groupNumberToPixel;
 
  DocumentModelGridRemoval m_modelGridRemoval;
};
 
#endif // GRID_HEALER_H