tesseract::IntFeatureSpace Class Reference

#include <intfeaturespace.h>

Public Member Functions
	IntFeatureSpace ()
void	Init (uint8_t xbuckets, uint8_t ybuckets, uint8_t thetabuckets)
bool	Serialize (FILE *fp) const
int	Size () const
INT_FEATURE_STRUCT	PositionFromIndex (int index) const
int	Index (const INT_FEATURE_STRUCT &f) const
void	IndexFeatures (const INT_FEATURE_STRUCT *features, int num_features, GenericVector< int > *mapped_features) const
void	IndexAndSortFeatures (const INT_FEATURE_STRUCT *features, int num_features, GenericVector< int > *sorted_features) const
int	XYToFeatureIndex (int x, int y) const

Protected Member Functions
int	XBucket (int x) const
int	YBucket (int y) const
int	ThetaBucket (int theta) const
INT_FEATURE_STRUCT	PositionFromBuckets (int x, int y, int theta) const

Protected Attributes
uint8_t	x_buckets_
uint8_t	y_buckets_
uint8_t	theta_buckets_

Detailed Description

Definition at line 38 of file intfeaturespace.h.

Constructor & Destructor Documentation

IntFeatureSpace()

tesseract::IntFeatureSpace::IntFeatureSpace

(

)

Definition at line 26 of file intfeaturespace.cpp.

  : x_buckets_(0), y_buckets_(0), theta_buckets_(0) {
}

Member Function Documentation

Index()

int tesseract::IntFeatureSpace::Index ( const INT_FEATURE_STRUCT &  f ) const

inline

Definition at line 60 of file intfeaturespace.h.

                                               {
    return (XBucket(f.X) * y_buckets_ + YBucket(f.Y)) * theta_buckets_ +
        ThetaBucket(f.Theta);
  }

IndexAndSortFeatures()

void tesseract::IntFeatureSpace::IndexAndSortFeatures	(	const INT_FEATURE_STRUCT *	features,
		int	num_features,
		GenericVector< int > *	sorted_features
	)		const

Definition at line 68 of file intfeaturespace.cpp.

                                               {
  sorted_features->truncate(0);
  for (int f = 0; f < num_features; ++f)
    sorted_features->push_back(Index(features[f]));
  sorted_features->sort();
}

IndexFeatures()

void tesseract::IntFeatureSpace::IndexFeatures	(	const INT_FEATURE_STRUCT *	features,
		int	num_features,
		GenericVector< int > *	mapped_features
	)		const

Definition at line 58 of file intfeaturespace.cpp.

                                                                               {
  mapped_features->truncate(0);
  for (int f = 0; f < num_features; ++f)
    mapped_features->push_back(Index(features[f]));
}

Init()

void tesseract::IntFeatureSpace::Init	(	uint8_t	xbuckets,
		uint8_t	ybuckets,
		uint8_t	thetabuckets
	)

Definition at line 30 of file intfeaturespace.cpp.

                                                                                   {
  x_buckets_ = xbuckets;
  y_buckets_ = ybuckets;
  theta_buckets_ = thetabuckets;
}

PositionFromBuckets()

INT_FEATURE_STRUCT tesseract::IntFeatureSpace::PositionFromBuckets ( int  x, int  y, int  theta  ) const

protected

Definition at line 114 of file intfeaturespace.cpp.

                                                                         {
  INT_FEATURE_STRUCT pos(
      (x * kIntFeatureExtent + kIntFeatureExtent / 2) / x_buckets_,
      (y * kIntFeatureExtent + kIntFeatureExtent / 2) / y_buckets_,
      DivRounded(theta * kIntFeatureExtent, theta_buckets_));
  return pos;
}

PositionFromIndex()

INT_FEATURE_STRUCT tesseract::IntFeatureSpace::PositionFromIndex

(

int

index

)

const

Definition at line 50 of file intfeaturespace.cpp.

                                                                     {
  return PositionFromBuckets(index / (y_buckets_ * theta_buckets_),
                             index / theta_buckets_ % y_buckets_,
                             index % theta_buckets_);
}

Serialize()

bool tesseract::IntFeatureSpace::Serialize

(

FILE *

fp

)

const

Definition at line 38 of file intfeaturespace.cpp.

                                              {
  if (fwrite(&x_buckets_, sizeof(x_buckets_), 1, fp) != 1)
    return false;
  if (fwrite(&y_buckets_, sizeof(y_buckets_), 1, fp) != 1)
    return false;
  if (fwrite(&theta_buckets_, sizeof(theta_buckets_), 1, fp) != 1)
    return false;
  return true;
}

Size()

int tesseract::IntFeatureSpace::Size ( ) const

inline

Definition at line 51 of file intfeaturespace.h.

                   {
    return static_cast<int>(x_buckets_) * y_buckets_ * theta_buckets_;
  }

ThetaBucket()

int tesseract::IntFeatureSpace::ThetaBucket ( int  theta ) const

inlineprotected

Definition at line 89 of file intfeaturespace.h.

                                   {
    int bucket = DivRounded(theta * theta_buckets_, kIntFeatureExtent);
    return Modulo(bucket, theta_buckets_);
  }

XBucket()

int tesseract::IntFeatureSpace::XBucket ( int  x ) const

inlineprotected

Definition at line 79 of file intfeaturespace.h.

                           {
    int bucket = x * x_buckets_ / kIntFeatureExtent;
    return ClipToRange(bucket, 0, static_cast<int>(x_buckets_) - 1);
  }

XYToFeatureIndex()

int tesseract::IntFeatureSpace::XYToFeatureIndex	(	int	x,
		int	y
	)		const

Definition at line 79 of file intfeaturespace.cpp.

                                                        {
  // Round the x,y position to a feature. Search for a valid theta.
  INT_FEATURE_STRUCT feature(x, y, 0);
  int index = -1;
  for (int theta = 0; theta <= UINT8_MAX && index < 0; ++theta) {
    feature.Theta = theta;
    index = Index(feature);
  }
  if (index < 0) {
    tprintf("(%d,%d) does not exist in feature space!\n", x, y);
    return -1;
  }
  feature = PositionFromIndex(index);
  tprintf("Click at (%d, %d) ->(%d, %d), ->(%d, %d)\n",
          x, y, feature.X, feature.Y, x - feature.X, y - feature.Y);
  // Get the relative position of x,y from the rounded feature.
  x -= feature.X;
  y -= feature.Y;
  if (x != 0 || y != 0) {
    double angle = atan2(static_cast<double>(y), static_cast<double>(x)) + M_PI;
    angle *= kIntFeatureExtent / (2.0 * M_PI);
    feature.Theta = static_cast<uint8_t>(angle + 0.5);
    index = Index(feature);
    if (index < 0) {
      tprintf("Feature failed to map to a valid index:");
      feature.print();
      return -1;
    }
    feature = PositionFromIndex(index);
  }
  feature.print();
  return index;
}

YBucket()

int tesseract::IntFeatureSpace::YBucket ( int  y ) const

inlineprotected

Definition at line 83 of file intfeaturespace.h.

                           {
    int bucket = y * y_buckets_ / kIntFeatureExtent;
    return ClipToRange(bucket, 0, static_cast<int>(y_buckets_) - 1);
  }

Member Data Documentation

theta_buckets_

uint8_t tesseract::IntFeatureSpace::theta_buckets_

protected

Definition at line 99 of file intfeaturespace.h.

x_buckets_

uint8_t tesseract::IntFeatureSpace::x_buckets_

protected

Definition at line 97 of file intfeaturespace.h.

y_buckets_

uint8_t tesseract::IntFeatureSpace::y_buckets_

protected

Definition at line 98 of file intfeaturespace.h.

---

The documentation for this class was generated from the following files:

• /tesseract/src/classify/intfeaturespace.h
• /tesseract/src/classify/intfeaturespace.cpp