ArrowTable.h

Go to the documentation of this file.

// Copyright 2019-2026 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 O2_FRAMEWORK_AODNATIVE_ARROWTABLE_H_
#define O2_FRAMEWORK_AODNATIVE_ARROWTABLE_H_
 
#include "AODNative/Decoder.h"
#include "AODNative/Objects.h"
#include <arrow/memory_pool.h>
#include <arrow/record_batch.h>
#include <arrow/result.h>
#include <arrow/type.h>
#include <memory>
#include <vector>
 
namespace o2::aod::native
{
 
std::shared_ptr<arrow::Schema> physicalSchema(TreeInfo const& tree);
 
std::shared_ptr<arrow::Schema> datasetSchema(std::shared_ptr<arrow::Schema> const& physical);
 
struct ColumnOp {
  enum struct Kind { Values,
                     Booleans,
                     VLA };
  Kind kind;
  BranchInfo const* branch;
  int datasetField;
  size_t byteSize;
  std::shared_ptr<arrow::Buffer> offsets;
};
 
std::vector<ColumnOp> planColumns(TreeInfo const& tree, std::shared_ptr<arrow::Schema> const& schema,
                                  BranchDecoder& decoder, arrow::MemoryPool* pool, int64_t& rows);
 
void executeColumnOp(BranchDecoder& decoder, ColumnOp const& op, uint8_t* dst);
 
std::shared_ptr<arrow::RecordBatch> makeBatch(std::shared_ptr<arrow::Schema> const& schema, std::vector<ColumnOp> const& ops,
                                              std::vector<std::shared_ptr<arrow::Buffer>> const& buffers, int64_t rows);
 
arrow::Result<std::shared_ptr<arrow::RecordBatch>> readTree(ByteSource& source, TreeInfo const& tree, Options const& options,
                                                            arrow::MemoryPool* pool = arrow::default_memory_pool());
 
} // namespace o2::aod::native
 
#endif // O2_FRAMEWORK_AODNATIVE_ARROWTABLE_H_