Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
#ifndef DataFormats_SoATemplate_test_FakeSoA_h
#define DataFormats_SoATemplate_test_FakeSoA_h

// A SoA-like class (with fake alignment (padding))
#include <cstddef>
#include <iostream>
#include <memory>

#define myassert(A)                                                                            \
  if (not(A)) {                                                                                \
    std::cerr << "Failed assertion: " #A " at  " __FILE__ "(" << __LINE__ << ")" << std::endl; \
    abort();                                                                                   \
  }

class FakeSoA {
public:
  static constexpr size_t padding_ = 128;

  // A fake SoA with 2 columns of uint16_t and uin32_t, plus fake padding.
  static size_t computeBufferSize(size_t nElements) {
    return nElements * (sizeof(uint16_t) + sizeof(uint32_t)) + padding_ + 0x400;
  }

  FakeSoA(std::byte *buffer, size_t nElements) { constFromBufferImpl(buffer, nElements); }

  FakeSoA() : size_(0) { std::cout << "At end of FakeSoA::FakeSoA()" << std::endl; }

  template <typename T>
  void allocateAndIoRead(T &onfile) {
    std::cout << "allocateAndIoRead begin" << std::endl;
    auto buffSize = FakeSoA::computeBufferSize(onfile.size_);
    auto buffer = new std::byte[buffSize];
    std::cout << "Buffer first byte after (alloc) =" << buffer + buffSize << std::endl;
    constFromBufferImpl(buffer, onfile.size_);
    memcpy(a16_, onfile.a16_, sizeof(uint16_t) * onfile.size_);
    memcpy(b32_, onfile.b32_, sizeof(uint32_t) * onfile.size_);
    std::cout << "allocateAndIoRead end" << std::endl;
  }

  void dump() {
    std::cout << "size=" << size_ << " buffer=" << buffer_.get() << " a16=" << a16_ << " b32=" << b32_
              << " (b32 - a16)="
              << reinterpret_cast<std::byte *>(reinterpret_cast<intptr_t>(b32_) - reinterpret_cast<intptr_t>(a16_))
              << " buffer size=" << computeBufferSize(size_) << "(" << std::hex << computeBufferSize(size_) << ")"
              << std::endl;
  }

  void dumpData() { std::cout << "a16_[0]=" << a16_[0] << " b32_[0]=" << b32_[0] << std::endl; }

  void fill() {
    for (int i = 0; i < size_; i++) {
      a16_[i] = 42 + i;
      b32_[i] = 24 + i;
    }
  }

  bool check() {
    bool result = true;
    for (int i = 0; i < size_; i++) {
      if (a16_[i] != 42 + i) {
        std::cout << "a16 mismatch at i=" << i << "(" << a16_[i] << "/" << 42 + i << ")" << std::endl;
        result = false;
      }
      if (b32_[i] != 24 + (uint32_t)i) {
        std::cout << "b32 mismatch at i=" << i << "(" << b32_[i] << "/" << 24 + i << ")" << std::endl;
        result = false;
      }
    }
    return result;
  }

private:
  void constFromBufferImpl(std::byte *buffer, size_t nElements) {
    buffer_.reset(buffer);
    size_ = nElements;
    a16_ = reinterpret_cast<uint16_t *>(buffer);
    buffer += nElements * sizeof(uint16_t) + padding_;
    b32_ = reinterpret_cast<uint32_t *>(buffer);
    buffer += nElements * sizeof(uint32_t);
    std::cout << "Buffer first byte after (const) =" << buffer << std::endl;
    std::cout << "At end of FakeSoA::constFromBufferImpl(std::byte * buffer, size_t nElements): ";
    dump();
  }

  int size_;
  uint16_t *a16_ = nullptr;                      //[size_]
  uint32_t *b32_ = nullptr;                      //[size_]
  std::unique_ptr<std::byte> buffer_ = nullptr;  //!
};

#endif  // DataFormats_SoATemplate_test_FakeSoA_h