mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-17 05:03:20 +00:00
116 lines
4.5 KiB
C++
116 lines
4.5 KiB
C++
#pragma once
|
|
|
|
#include <DB/Columns/IColumn.h>
|
|
#include <DB/Columns/ColumnVector.h>
|
|
|
|
|
|
namespace DB
|
|
{
|
|
|
|
namespace ErrorCodes
|
|
{
|
|
extern const int ILLEGAL_COLUMN;
|
|
extern const int NOT_IMPLEMENTED;
|
|
extern const int BAD_ARGUMENTS;
|
|
}
|
|
|
|
/** A column of array values.
|
|
* In memory, it is represented as one column of a nested type, whose size is equal to the sum of the sizes of all arrays,
|
|
* and as an array of offsets in it, which allows you to get each element.
|
|
*/
|
|
class ColumnArray final : public IColumn
|
|
{
|
|
public:
|
|
/** On the index i there is an offset to the beginning of the i + 1 -th element. */
|
|
using ColumnOffsets_t = ColumnVector<Offset_t>;
|
|
|
|
/** Create an empty column of arrays with the type of values as in the column `nested_column` */
|
|
explicit ColumnArray(ColumnPtr nested_column, ColumnPtr offsets_column = nullptr);
|
|
|
|
std::string getName() const override;
|
|
ColumnPtr cloneResized(size_t size) const override;
|
|
size_t size() const override;
|
|
Field operator[](size_t n) const override;
|
|
void get(size_t n, Field & res) const override;
|
|
StringRef getDataAt(size_t n) const override;
|
|
void insertData(const char * pos, size_t length) override;
|
|
StringRef serializeValueIntoArena(size_t n, Arena & arena, char const *& begin) const override;
|
|
const char * deserializeAndInsertFromArena(const char * pos) override;
|
|
void updateHashWithValue(size_t n, SipHash & hash) const override;
|
|
void insertRangeFrom(const IColumn & src, size_t start, size_t length) override;
|
|
void insert(const Field & x) override;
|
|
void insertFrom(const IColumn & src_, size_t n) override;
|
|
void insertDefault() override;
|
|
void popBack(size_t n) override;
|
|
ColumnPtr filter(const Filter & filt, ssize_t result_size_hint) const override;
|
|
ColumnPtr permute(const Permutation & perm, size_t limit) const override;
|
|
int compareAt(size_t n, size_t m, const IColumn & rhs_, int nan_direction_hint) const override;
|
|
void getPermutation(bool reverse, size_t limit, int nan_direction_hint, Permutation & res) const override;
|
|
void reserve(size_t n) override;
|
|
size_t byteSize() const override;
|
|
size_t allocatedSize() const override;
|
|
ColumnPtr replicate(const Offsets_t & replicate_offsets) const override;
|
|
ColumnPtr convertToFullColumnIfConst() const override;
|
|
void getExtremes(Field & min, Field & max) const override;
|
|
|
|
bool hasEqualOffsets(const ColumnArray & other) const;
|
|
|
|
/** More efficient methods of manipulation */
|
|
IColumn & getData() { return *data.get(); }
|
|
const IColumn & getData() const { return *data.get(); }
|
|
|
|
ColumnPtr & getDataPtr() { return data; }
|
|
const ColumnPtr & getDataPtr() const { return data; }
|
|
|
|
Offsets_t & ALWAYS_INLINE getOffsets()
|
|
{
|
|
return static_cast<ColumnOffsets_t &>(*offsets.get()).getData();
|
|
}
|
|
|
|
const Offsets_t & ALWAYS_INLINE getOffsets() const
|
|
{
|
|
return static_cast<const ColumnOffsets_t &>(*offsets.get()).getData();
|
|
}
|
|
|
|
ColumnPtr & getOffsetsColumn() { return offsets; }
|
|
const ColumnPtr & getOffsetsColumn() const { return offsets; }
|
|
|
|
Columns scatter(ColumnIndex num_columns, const Selector & selector) const override
|
|
{
|
|
return scatterImpl<ColumnArray>(num_columns, selector);
|
|
}
|
|
|
|
private:
|
|
ColumnPtr data;
|
|
ColumnPtr offsets; /// Displacements can be shared across multiple columns - to implement nested data structures.
|
|
|
|
size_t ALWAYS_INLINE offsetAt(size_t i) const { return i == 0 ? 0 : getOffsets()[i - 1]; }
|
|
size_t ALWAYS_INLINE sizeAt(size_t i) const { return i == 0 ? getOffsets()[0] : (getOffsets()[i] - getOffsets()[i - 1]); }
|
|
|
|
|
|
/// Multiply values if the nested column is ColumnVector<T>.
|
|
template <typename T>
|
|
ColumnPtr replicateNumber(const Offsets_t & replicate_offsets) const;
|
|
|
|
/// Multiply the values if the nested column is ColumnString. The code is too complicated.
|
|
ColumnPtr replicateString(const Offsets_t & replicate_offsets) const;
|
|
|
|
/** Non-constant arrays of constant values are quite rare.
|
|
* Most functions can not work with them, and does not create such columns as a result.
|
|
* An exception is the function `replicate`(see FunctionsMiscellaneous.h), which has service meaning for the implementation of lambda functions.
|
|
* Only for its sake is the implementation of the `replicate` method for ColumnArray(ColumnConst).
|
|
*/
|
|
ColumnPtr replicateConst(const Offsets_t & replicate_offsets) const;
|
|
|
|
|
|
/// Specializations for the filter function.
|
|
template <typename T>
|
|
ColumnPtr filterNumber(const Filter & filt, ssize_t result_size_hint) const;
|
|
|
|
ColumnPtr filterString(const Filter & filt, ssize_t result_size_hint) const;
|
|
ColumnPtr filterGeneric(const Filter & filt, ssize_t result_size_hint) const;
|
|
};
|
|
|
|
|
|
}
|