ArrowTableSlicingCache.h

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.
 
#ifndef ARROWTABLESLICINGCACHE_H
#define ARROWTABLESLICINGCACHE_H
 
#include "Framework/ServiceHandle.h"
#include <arrow/array.h>
#include <gsl/span>
 
namespace o2::framework
{
using ListVector = std::vector<std::vector<int64_t>>;
 
struct SliceInfoPtr {
  gsl::span<int const> values;
  gsl::span<int64_t const> counts;
 
  std::pair<int64_t, int64_t> getSliceFor(int value) const;
};
 
struct SliceInfoUnsortedPtr {
  gsl::span<int const> values;
  ListVector const* groups;
 
  gsl::span<int64_t const> getSliceFor(int value) const;
};
 
struct Entry {
  std::string binding;
  std::string key;
  bool enabled;
 
  Entry(std::string b, std::string k, bool e = true)
    : binding{b},
      key{k},
      enabled{e}
  {
  }
};
 
using Cache = std::vector<Entry>;
 
void updatePairList(Cache& list, std::string const& binding, std::string const& key, bool enabled);
 
struct ArrowTableSlicingCacheDef {
  constexpr static ServiceKind service_kind = ServiceKind::Global;
  Cache bindingsKeys;
  Cache bindingsKeysUnsorted;
 
  void setCaches(Cache&& bsks);
  void setCachesUnsorted(Cache&& bsks);
};
 
struct ArrowTableSlicingCache {
  constexpr static ServiceKind service_kind = ServiceKind::Stream;
 
  Cache bindingsKeys;
  std::vector<std::shared_ptr<arrow::NumericArray<arrow::Int32Type>>> values;
  std::vector<std::shared_ptr<arrow::NumericArray<arrow::Int64Type>>> counts;
 
  Cache bindingsKeysUnsorted;
  std::vector<std::vector<int>> valuesUnsorted;
  std::vector<ListVector> groups;
 
  ArrowTableSlicingCache(Cache&& bsks, Cache&& bsksUnsorted = {});
 
  // set caching information externally
  void setCaches(Cache&& bsks, Cache&& bsksUnsorted = {});
 
  // update slicing info cache entry (assumes it is already present)
  arrow::Status updateCacheEntry(int pos, std::shared_ptr<arrow::Table> const& table);
  arrow::Status updateCacheEntryUnsorted(int pos, std::shared_ptr<arrow::Table> const& table);
 
  // helper to locate cache position
  std::pair<int, bool> getCachePos(Entry const& bindingKey) const;
  int getCachePosSortedFor(Entry const& bindingKey) const;
  int getCachePosUnsortedFor(Entry const& bindingKey) const;
 
  // get slice from cache for a given value
  SliceInfoPtr getCacheFor(Entry const& bindingKey) const;
  SliceInfoUnsortedPtr getCacheUnsortedFor(Entry const& bindingKey) const;
  SliceInfoPtr getCacheForPos(int pos) const;
  SliceInfoUnsortedPtr getCacheUnsortedForPos(int pos) const;
 
  static void validateOrder(Entry const& bindingKey, std::shared_ptr<arrow::Table> const& input);
};
} // namespace o2::framework
 
#endif // ARROWTABLESLICINGCACHE_H