testAbstractRefAccessor.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#define BOOST_TEST_MODULE Test AbstractRefAccessor class
#define BOOST_TEST_MAIN
#define BOOST_TEST_DYN_LINK
#include <numeric>
#include <boost/test/unit_test.hpp>
#include "CommonDataFormat/AbstractRefAccessor.h"
#include "CommonDataFormat/AbstractRef.h"
#include "Framework/Logger.h"
 
using namespace o2::dataformats;
 
struct Base {
  int b = 0;
};
 
struct Foo1 : public Base {
  int f1 = 0;
};
 
struct Foo2 : public Foo1 {
  int f2 = 0;
};
 
struct Bar1 : public Base {
  double b1 = 0.;
};
 
struct GloIdx : public AbstractRef<25, 5, 2> {
  enum Source : uint8_t { // provenance of the
    ID0,
    ID1,
    ID2,
    ID3,
    NSources
  };
  using AbstractRef<25, 5, 2>::AbstractRef;
};
 
// basic AbstractRefAccessor
BOOST_AUTO_TEST_CASE(AbstractRefAccess)
{
 
  std::vector<Base> vb(10);
  std::vector<Foo1> vf(10);
  std::array<Foo2, 10> af;
  std::vector<Bar1> bar(10);
 
  std::vector<GloIdx> vid;
 
  for (int i = 0; i < 10; i++) {
    vb[i].b = GloIdx::ID0 * 100 + i;
    vid.emplace_back(i, GloIdx::ID1);
 
    vf[i].b = GloIdx::ID1 * 100 + i;
    vf[i].f1 = 0.5 + 100 + i;
    vid.emplace_back(i, GloIdx::ID2);
 
    af[i].b = GloIdx::ID2 * 100 + i;
    af[i].f2 = 0.8 + 100 + i;
    vid.emplace_back(i, GloIdx::ID3);
 
    bar[i].b = GloIdx::ID3 * 100 + i;
    bar[i].b1 = 0.2 + 300 + i;
    vid.emplace_back(i, GloIdx::ID3);
  }
 
  AbstractRefAccessor<Base, GloIdx::NSources> acc;
 
  acc.registerContainer(vb, GloIdx::ID0);
  acc.registerContainer(vf, GloIdx::ID1);
  acc.registerContainer(af, GloIdx::ID2);
  acc.registerContainer(bar, GloIdx::ID3);
 
  size_t nid = vid.size();
  for (size_t i = 0; i < nid; i++) {
    auto gid = vid[i];
    const auto& obj = acc.get(gid);
    int expect = gid.getSource() * 100 + i / GloIdx::NSources;
    LOG(info) << i << " ? " << obj.b << " == " << expect << " for " << gid.getRaw();
    BOOST_CHECK(obj.b == expect);
  }
 
  const auto& barEl = acc.get_as<Bar1>(vid.back());
  LOG(info) << " ? " << barEl.b1 << " == " << bar.back().b1;
  BOOST_CHECK(barEl.b1 == bar.back().b1);
}