SegmentFactory.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 SEGMENT_FACTORY_H
#define SEGMENT_FACTORY_H
 
#include <QList>
#include <QPointF>
#include <vector>
 
class ColorFilter;
class DocumentModelSegments;
class QGraphicsScene;
class QImage;
class Segment;
 
typedef std::vector<Segment*> SegmentVector;
 
class SegmentFactory
{
public:
  SegmentFactory(QGraphicsScene &scene,
                 bool isGnuplot);
 
  void clearSegments(QList<Segment*> &segments);
 
  QList<QPoint> fillPoints(const DocumentModelSegments &modelSegments,
                           QList<Segment*> segments);
 
  void makeSegments (const QImage &imageFiltered,
                     const DocumentModelSegments &modelSegments,
                     QList<Segment*> &segments,
                     bool useDlg = true);
 
private:
  SegmentFactory();
 
  // Return the number of runs adjacent to the pixels from yStart to yStop (inclusive)
  int adjacentRuns(bool *columnBool,
                   int yStart,
                   int yStop,
                   int height);
 
  // Find the single segment pointer among the adjacent pixels from yStart-1 to yStop+1
  Segment *adjacentSegment(SegmentVector &lastSegment,
                           int yStart,
                           int yStop,
                           int height);
 
  // Return the number of segments adjacent to the pixels from yStart to yStop (inclusive)
  int adjacentSegments(SegmentVector &lastSegment,
                       int yStart,
                       int yStop,
                       int height);
 
  // Process a run of pixels. If there are fewer than two adjacent pixel runs on
  // either side, this run will be added to an existing segment, or the start of
  // a new segment
  void finishRun(bool *lastBool,
                 bool *nextBool,
                 SegmentVector &lastSegment,
                 SegmentVector &currSegment,
                 int x,
                 int yStart,
                 int yStop,
                 int height,
                 const DocumentModelSegments &modelSegments,
                 int* madeLines);
 
  // Initialize one column of boolean flags using the pixels of the specified column
  void loadBool (const ColorFilter &filter,
                 bool *columnBool,
                 const QImage &image,
                 int x);
 
  // Initialize one column of segment pointers
  void loadSegment (SegmentVector &columnSegment,
                    int height);
 
  // Identify the runs in a column, and connect them to segments
  void matchRunsToSegments (int x,
                            int height,
                            bool *lastBool,
                            SegmentVector &lastSegment,
                            bool *currBool,
                            SegmentVector &currSegment,
                            bool *nextBool,
                            const DocumentModelSegments &modelSegments,
                            int *madeLines,
                            int *foldedLines,
                            int *shortLine,
                            QList<Segment*> &segments);
 
  void removeEmptySegments (QList<Segment*> &segments) const;
 
  // Remove unneeded lines belonging to segments that just finished in the previous column.
  // The results of this function are displayed in the debug spew of makeSegments
  void removeUnneededLines(SegmentVector &lastSegment,
                           SegmentVector &currSegment,
                           int height,
                           int *foldedLines,
                           int *shortLines,
                           const DocumentModelSegments &modelSegments,
                           QList<Segment*> &segments);
 
  // Scroll the boolean flags of the right column into the left column
  void scrollBool(bool *left,
                  bool *right,
                  int height);
 
  // Scroll the segment pointers of the right column into the left column
  void scrollSegment(SegmentVector &left,
                     SegmentVector &right,
                     int height);
 
  QGraphicsScene &m_scene;
 
  bool m_isGnuplot;
};
 
#endif // SEGMENT_FACTORY_H