UnionBuilder.h

Go to the documentation of this file.

// BSD 3-Clause License; see https://github.com/scikit-hep/awkward/blob/main/LICENSE
 
#ifndef AWKWARD_UNIONBUILDER_H_
#define AWKWARD_UNIONBUILDER_H_
 
#include <vector>
 
#include "awkward/common.h"
#include "awkward/BuilderOptions.h"
#include "awkward/GrowableBuffer.h"
#include "awkward/builder/Builder.h"
 
namespace awkward {
  class TupleBuilder;
  class RecordBuilder;

  class EXPORT_SYMBOL UnionBuilder: public Builder {
  public:
    static const BuilderPtr
      fromsingle(const BuilderOptions& options,
                 const BuilderPtr& firstcontent);

    UnionBuilder(const BuilderOptions& options,
                 GrowableBuffer<int8_t> tags,
                 GrowableBuffer<int64_t> index,
                 std::vector<BuilderPtr>& contents);

    const std::string
      classname() const override;
 
    const std::string
      to_buffers(BuffersContainer& container, int64_t& form_key_id) const override;
 
    int64_t
      length() const override;
 
    void
      clear() override;

    bool
      active() const override;
 
    const BuilderPtr
      null() override;
 
    const BuilderPtr
      boolean(bool x) override;
 
    const BuilderPtr
      integer(int64_t x) override;
 
    const BuilderPtr
      real(double x) override;
 
    const BuilderPtr
      complex(std::complex<double> x) override;
 
    const BuilderPtr
      datetime(int64_t x, const std::string& unit) override;
 
    const BuilderPtr
      timedelta(int64_t x, const std::string& unit) override;
 
    const BuilderPtr
      string(const char* x, int64_t length, const char* encoding) override;
 
    const BuilderPtr
      beginlist() override;
 
    const BuilderPtr
      endlist() override;
 
    const BuilderPtr
      begintuple(int64_t numfields) override;
 
    const BuilderPtr
      index(int64_t index) override;
 
    const BuilderPtr
      endtuple() override;
 
    const BuilderPtr
      beginrecord(const char* name, bool check) override;
 
    void
      field(const char* key, bool check) override;
 
    const BuilderPtr
      endrecord() override;
 
    const BuilderOptions&
      options() const { return options_; }
 
    const GrowableBuffer<int8_t>& tags() const {  return tags_; }
    GrowableBuffer<int8_t>& tags_buffer() {  return tags_; }
 
    const GrowableBuffer<int64_t>& index() const { return index_; }
    GrowableBuffer<int64_t>& index_buffer() { return index_; }
 
    const std::vector<BuilderPtr>& contents() const { return contents_; }
    std::vector<BuilderPtr>& builders() { return contents_; }
 
    int8_t current() { return current_;}
 
  private:
    const BuilderOptions options_;
    GrowableBuffer<int8_t> tags_;
    GrowableBuffer<int64_t> index_;
    std::vector<BuilderPtr> contents_;
    int8_t current_;
  };
}
 
#endif // AWKWARD_UNIONBUILDER_H_