test_ransReverseSymbolLookupTable.cxx

Go to the documentation of this file.

// Copyright 2019-2023 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 Utility test
#define BOOST_TEST_MAIN
#define BOOST_TEST_DYN_LINK
 
#include <numeric>
#include <vector>
#include <iterator>
 
#include <boost/test/unit_test.hpp>
#include <boost/mpl/vector.hpp>
 
#include "rANS/internal/containers/ReverseSymbolLookupTable.h"
#include "rANS/internal/containers/DenseHistogram.h"
#include "rANS/internal/containers/RenormedHistogram.h"
#include "rANS/internal/transform/renorm.h"
#include "rANS/factory.h"
 
using namespace o2::rans;
using namespace o2::rans::internal;
 
template <typename T>
size_t getNUniqueSymbols(const T& container)
{
  return std::count_if(container.begin(), container.end(), [](uint32_t value) { return value != 0; });
};
 
BOOST_AUTO_TEST_CASE(test_empty)
{
  const auto renormedHistogram = renorm(DenseHistogram<uint32_t>{}, RenormingPolicy::ForceIncompressible);
  const ReverseSymbolLookupTable<uint32_t> rLut{renormedHistogram};
 
  const auto size = 0;
  BOOST_CHECK_EQUAL(rLut.size(), size);
 
  const std::vector<int32_t> res;
  BOOST_CHECK_EQUAL_COLLECTIONS(rLut.begin(), rLut.end(), res.begin(), res.end());
}
 
#pragma GCC diagnostic ignored "-Wfree-nonheap-object" // TODO: Remove once this is fixed in GCC
 
BOOST_AUTO_TEST_CASE(test_buildRLUT)
{
  const std::vector<int32_t> A{5, 5, 6, 6, 8, 8, 8, 8, 8, -1, -5, 2, 7, 3};
  const std::vector<uint32_t> histA{1, 0, 0, 0, 1, 0, 0, 1, 1, 0, 2, 2, 1, 5, 1};
  const size_t scaleBits = 17;
  const auto size = (1ull << scaleBits) - 1;
 
  const auto renormedHistogram = renorm(makeDenseHistogram::fromSamples(A.begin(), A.end()), scaleBits, RenormingPolicy::ForceIncompressible);
  const ReverseSymbolLookupTable<int32_t> rLut{renormedHistogram};
 
  BOOST_CHECK_EQUAL(rLut.size(), size);
 
  const auto min = *std::min_element(A.begin(), A.end());
  const std::vector<uint32_t> frequencies{renormedHistogram.begin(), renormedHistogram.end()};
  std::vector<uint32_t> cumulative;
  std::exclusive_scan(frequencies.begin(), frequencies.end(), std::back_inserter(cumulative), 0);
 
  for (size_t i = 0; i < frequencies.size(); ++i) {
    const int symbol = min + i;
    for (size_t cumul = cumulative[i]; cumul < cumulative[i] + frequencies[i]; ++cumul) {
      BOOST_CHECK(cumul < rLut.size());
      BOOST_CHECK_EQUAL(rLut[cumul], symbol);
    }
  }
}