tesseract::StrideMap::Index Class Reference

#include <stridemap.h>

Public Member Functions
	Index (const StrideMap &stride_map)
	Index (const StrideMap &stride_map, int batch, int y, int x)
int	t () const
int	index (FlexDimensions dimension) const
void	InitToFirst ()
void	InitToLast ()
bool	IsValid () const
bool	IsLast (FlexDimensions dimension) const
int	MaxIndexOfDim (FlexDimensions dim) const
bool	AddOffset (int offset, FlexDimensions dimension)
bool	Increment ()
bool	Decrement ()

Detailed Description

Definition at line 44 of file stridemap.h.

Constructor & Destructor Documentation

Index() [1/2]

tesseract::StrideMap::Index::Index ( const StrideMap &  stride_map )

inlineexplicit

Definition at line 46 of file stridemap.h.

                                                : stride_map_(&stride_map) {
      InitToFirst();
    }

Index() [2/2]

tesseract::StrideMap::Index::Index ( const StrideMap &  stride_map, int  batch, int  y, int  x  )

inline

Definition at line 49 of file stridemap.h.

        : stride_map_(&stride_map) {
      indices_[FD_BATCH] = batch;
      indices_[FD_HEIGHT] = y;
      indices_[FD_WIDTH] = x;
      SetTFromIndices();
    }

Member Function Documentation

AddOffset()

bool tesseract::StrideMap::Index::AddOffset	(	int	offset,
		FlexDimensions	dimension
	)

Definition at line 62 of file stridemap.cpp.

                                                                   {
  indices_[dimension] += offset;
  SetTFromIndices();
  return IsValid();
}

Decrement()

bool tesseract::StrideMap::Index::Decrement

(

)

Definition at line 87 of file stridemap.cpp.

                               {
  for (int d = FD_DIMSIZE - 1; d >= 0; --d) {
    if (indices_[d] > 0) {
      --indices_[d];
      if (d == FD_BATCH) {
        // The upper limits of the other dimensions may have changed as a result
        // of a different batch index, so they have to be reset.
        InitToLastOfBatch(indices_[FD_BATCH]);
      } else {
        t_ -= stride_map_->t_increments_[d];
      }
      return true;
    }
    indices_[d] = MaxIndexOfDim(static_cast<FlexDimensions>(d));
    t_ += stride_map_->t_increments_[d] * indices_[d];
    // Now borrow from the next dimension.
  }
  return false;
}

Increment()

bool tesseract::StrideMap::Index::Increment

(

)

Definition at line 70 of file stridemap.cpp.

                               {
  for (int d = FD_DIMSIZE - 1; d >= 0; --d) {
    if (!IsLast(static_cast<FlexDimensions>(d))) {
      t_ += stride_map_->t_increments_[d];
      ++indices_[d];
      return true;
    }
    t_ -= stride_map_->t_increments_[d] * indices_[d];
    indices_[d] = 0;
    // Now carry to the next dimension.
  }
  return false;
}

index()

int tesseract::StrideMap::Index::index ( FlexDimensions  dimension ) const

inline

Definition at line 58 of file stridemap.h.

{ return indices_[dimension]; }

InitToFirst()

void tesseract::StrideMap::Index::InitToFirst ( )

inline

Definition at line 60 of file stridemap.h.

                       {
      memset(indices_, 0, sizeof(indices_));
      t_ = 0;
    }

InitToLast()

void tesseract::StrideMap::Index::InitToLast ( )

inline

Definition at line 65 of file stridemap.h.

{ InitToLastOfBatch(MaxIndexOfDim(FD_BATCH)); }

IsLast()

bool tesseract::StrideMap::Index::IsLast

(

FlexDimensions

dimension

)

const

Definition at line 37 of file stridemap.cpp.

                                                          {
  return MaxIndexOfDim(dimension) == indices_[dimension];
}

IsValid()

bool tesseract::StrideMap::Index::IsValid

(

)

const

Definition at line 24 of file stridemap.cpp.

                                   {
  // Cheap check first.
  for (int index : indices_) {
    if (index < 0) return false;
  }
  for (int d = 0; d < FD_DIMSIZE; ++d) {
    if (indices_[d] > MaxIndexOfDim(static_cast<FlexDimensions>(d)))
      return false;
  }
  return true;
}

MaxIndexOfDim()

int tesseract::StrideMap::Index::MaxIndexOfDim

(

FlexDimensions

dim

)

const

Definition at line 43 of file stridemap.cpp.

                                                          {
  int max_index = stride_map_->shape_[dim] - 1;
  if (dim == FD_BATCH) return max_index;
  assert(0 <= indices_[FD_BATCH]);
  const size_t batch = indices_[FD_BATCH];
  if (dim == FD_HEIGHT) {
    if (batch >= stride_map_->heights_.size() ||
        stride_map_->heights_[batch] > max_index)
      return max_index;
    return stride_map_->heights_[batch] - 1;
  }
  if (batch >= stride_map_->widths_.size() ||
      stride_map_->widths_[batch] > max_index)
    return max_index;
  return stride_map_->widths_[batch] - 1;
}

t()

int tesseract::StrideMap::Index::t ( ) const

inline

Definition at line 57 of file stridemap.h.

{ return t_; }

---

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

• /tesseract/src/lstm/stridemap.h
• /tesseract/src/lstm/stridemap.cpp