BlockArray.h

   1 #ifndef _BLOCK_ARRAY_H_
   2 #define _BLOCK_ARRAY_H_
   3 #include "Tools.h"
   4 #include <iostream>
   5 #include <stdexcept>
   6
   7 class BlockArray
   8 {
   9 private:
  10     ulong* data;
  11     ulong n;
  12     ulong index;
  13     ulong blockLength;
  14 public:
  15     BlockArray(ulong len, ulong blockLen) {
  16        n = len;
  17        blockLength = blockLen;
  18        data = new ulong[n*blockLength/W +1];
  19        for (ulong i = 0; i < n*blockLength/W +1; ++i)
  20            data[i] = 0;
  21     }
  22     ~BlockArray() {
  23        delete [] data;
  24     }
  25
  26     BlockArray& operator[](ulong i)  {
  27        index = i;
  28        return *this;
  29     }
  30
  31     void operator=(const ulong x) {
  32        Tools::SetField(data,blockLength,index,x);
  33     }
  34
  35     BlockArray& operator=(const BlockArray& ba) {
  36        if (this == &ba) return *this;
  37        ulong value = Tools::GetField(ba.data, ba.blockLength, ba.index);
  38        Tools::SetField(data,blockLength,index,value);
  39        return *this;
  40     }
  41
  42     operator ulong() {
  43        return Tools::GetField(data,blockLength,index);
  44     }
  45
  46     ulong spaceInBits() {
  47         return n*blockLength+W; // plus 4 ulong's
  48     }
  49
  50     /**
  51      * Saving data fields:
  52      *     ulong n;
  53      *     ulong blockLength;
  54      *     ulong* data;
  55      */
  56     void Save(FILE *file) const
  57     {
  58         if (std::fwrite(&(this->n), sizeof(ulong), 1, file) != 1)
  59             throw std::runtime_error("BlockArray::Save(): file write error (n).");
  60         if (std::fwrite(&(this->blockLength), sizeof(ulong), 1, file) != 1)
  61             throw std::runtime_error("BlockArray::Save(): file write error (blockLength).");
  62
  63         for (ulong offset = 0; offset < n*blockLength/W+1; offset ++)
  64         {
  65             if (std::fwrite(this->data + offset, sizeof(ulong), 1, file) != 1)
  66                 throw std::runtime_error("BlockArray::Save(): file write error (data).");
  67         }
  68     }
  69
  70     /**
  71      * Load from file
  72      */
  73     BlockArray(FILE *file)
  74     {
  75         if (std::fread(&(this->n), sizeof(ulong), 1, file) != 1)
  76             throw std::runtime_error("BlockArray::Load(): file read error (n).");
  77         if (std::fread(&(this->blockLength), sizeof(ulong), 1, file) != 1)
  78             throw std::runtime_error("BlockArray::Load(): file read error (blockLength).");
  79
  80         data = new ulong[n*blockLength/W+1];
  81         for (ulong offset = 0; offset < n*blockLength/W+1; offset ++)
  82         {
  83             if (std::fread(this->data + offset, sizeof(ulong), 1, file) != 1)
  84                 throw std::runtime_error("BlockArray::Load(): file read error (data).");
  85         }
  86     }
  87 };
  88
  89 #endif
  90