TagFamily.h

Go to the documentation of this file.

#ifndef TAGFAMILY_H
#define TAGFAMILY_H

#include <climits>
#include <cmath>
#include <stdio.h>
#include <vector>
#include <map>

#include "AprilTags/TagDetection.h"
using namespace std;

namespace AprilTags {

class TagCodes {
public:
  int bits;
  int minHammingDistance;
  std::vector<unsigned long long> codes;
public:
 TagCodes(int bits, int minHammingDistance,
          const unsigned long long* codesA, int num)
   : bits(bits), minHammingDistance(minHammingDistance),
    codes(codesA, codesA+num) // created vector for all entries of codesA
      {}
};

class TagFamily {
public:
  TagFamily(const TagCodes& tagCodes);

  void setErrorRecoveryBits(int b);

  void setErrorRecoveryFraction(float v);

  /* if the bits in w were arranged in a d*d grid and that grid was
   * rotated, what would the new bits in w be?
   * The bits are organized like this (for d = 3):
   *
   *  8 7 6       2 5 8      0 1 2
   *  5 4 3  ==>  1 4 7 ==>  3 4 5    (rotate90 applied twice)
   *  2 1 0       0 3 6      6 7 8
   */
  static unsigned long long rotate90(unsigned long long w, int d);

  static int hammingDistance(unsigned long long a, unsigned long long b);

  static unsigned char popCountReal(unsigned long long w);

  static int popCount(unsigned long long w);

  /*  The corresponding fields of TagDetection will be filled in. */
  void decode(TagDetection& det, unsigned long long rCode) const;

  void printHammingDistances() const;

  int blackBorder;

  int bits;

  int dimension;

  /*  Accounting for rotational ambiguity? The code can recover
   *  (minHammingDistance-1)/2 bit errors.
   */
  int minimumHammingDistance;

  /* The error recovery value determines our position on the ROC
   * curve. We will report codes that are within errorRecoveryBits
   * of a valid code. Small values mean greater rejection of bogus
   * tags (but false negatives). Large values mean aggressive
   * reporting of bad tags (but with a corresponding increase in
   * false positives).
   */
  int errorRecoveryBits;

  std::vector<unsigned long long> codes;

  static const int  popCountTableShift = 12;
  static const unsigned int popCountTableSize = 1 << popCountTableShift;
  static unsigned char popCountTable[popCountTableSize];

  static class TableInitializer {
  public:
    TableInitializer() {
      for (unsigned int i = 0; i < TagFamily::popCountTableSize; i++)
        TagFamily::popCountTable[i] = TagFamily::popCountReal(i);
    }
  } initializer;
};

} // namespace

#endif