zmort.h

Go to the documentation of this file.

 
#ifndef BRICK_ZMORT_H
#define BRICK_ZMORT_H
 
struct ZMORT {
  unsigned long id; 
  unsigned long dim; 
 
  ZMORT() : id(0ul), dim(0ul) {}
 
  ZMORT(unsigned long id, unsigned long dim) : id(id), dim(dim) {};
 
  inline ZMORT operator[](unsigned long p) {
    unsigned long oid = id;
 
    ZMORT zmort;
    zmort.id = 0;
    zmort.dim = dim + 1ul;
    unsigned long omask = (1ul << dim) - 1ul;
 
    unsigned long i = 0ul;
    while (oid || p) {
      zmort.id |= (((oid & omask) << 1ul) + (p & 1ul)) << (i * zmort.dim);
      oid >>= dim;
      p >>= 1ul;
      ++i;
    }
 
    return zmort;
  }
 
  inline unsigned long operator()(unsigned long d) {
    unsigned long oid = id >> d;
    unsigned long pos = 0;
 
    unsigned long i = 0;
    while (oid) {
      pos |= (oid & 1ul) << i;
      oid >>= dim;
      ++i;
    }
 
    return pos;
  }
 
  inline ZMORT set(unsigned long d, unsigned long p) {
    unsigned long oid = id >> d;
    unsigned long omask = (1ul << dim) - 2ul;
    unsigned long i = d;
 
    ZMORT zmort(id & ((1ul << d) - 1), dim);
    while (oid || p) {
      zmort.id |= ((p & 1ul) + (oid & omask)) << i;
      oid >>= dim;
      p >>= 1ul;
      i += dim;
    }
    return zmort;
  }
 
  inline operator unsigned long() const {
    return id;
  }
};
 
#endif //BRICK_ZMORT_H