ClickHouse/dbms/src/Columns/ColumnConst.cpp

189 lines
5.2 KiB
C++
Raw Normal View History

#include <DB/DataTypes/DataTypeString.h>
2013-03-27 10:07:23 +00:00
#include <DB/DataTypes/DataTypeFixedString.h>
#include <DB/DataTypes/DataTypeArray.h>
2012-08-26 11:14:52 +00:00
2011-11-06 06:22:52 +00:00
#include <DB/Columns/ColumnString.h>
#include <DB/Columns/ColumnConst.h>
#include <DB/Columns/ColumnArray.h>
#include <DB/Columns/ColumnTuple.h>
2013-03-27 10:07:23 +00:00
#include <DB/Columns/ColumnFixedString.h>
#include <DB/Columns/ColumnNullable.h>
#include <DB/DataTypes/DataTypeTuple.h>
#include <ext/enumerate.hpp>
2011-11-06 06:22:52 +00:00
namespace DB
{
2013-03-06 13:56:11 +00:00
2017-03-11 00:52:20 +00:00
namespace ErrorCodes
{
extern const int TOO_LARGE_STRING_SIZE;
extern const int LOGICAL_ERROR;
extern const int NOT_IMPLEMENTED;
}
template <>
ColumnPtr ColumnConst<Null>::convertToFullColumn() const
{
2016-08-16 22:40:05 +00:00
/// We basically create a column whose rows have NULL values.
ColumnPtr full_col = std::make_shared<ColumnUInt8>(size(), 0);
ColumnPtr null_map = std::make_shared<ColumnUInt8>(size(), 1);
return std::make_shared<ColumnNullable>(full_col, null_map);
}
2011-11-06 06:22:52 +00:00
template <> ColumnPtr ColumnConst<String>::convertToFullColumn() const
{
if (!data_type || typeid_cast<const DataTypeString *>(&*data_type))
2013-03-27 10:07:23 +00:00
{
auto res = std::make_shared<ColumnString>();
2013-03-27 10:07:23 +00:00
ColumnString::Offsets_t & offsets = res->getOffsets();
ColumnString::Chars_t & vec = res->getChars();
2011-11-06 06:22:52 +00:00
2013-03-27 10:07:23 +00:00
size_t string_size = data.size() + 1;
size_t offset = 0;
offsets.resize(s);
vec.resize(s * string_size);
2011-11-06 06:22:52 +00:00
2013-03-27 10:07:23 +00:00
for (size_t i = 0; i < s; ++i)
{
memcpy(&vec[offset], data.data(), string_size);
offset += string_size;
offsets[i] = offset;
}
return res;
2011-11-06 06:22:52 +00:00
}
else if (const DataTypeFixedString * type = typeid_cast<const DataTypeFixedString *>(&*data_type))
2013-03-27 10:07:23 +00:00
{
size_t n = type->getN();
2011-11-06 06:22:52 +00:00
2013-03-27 10:07:23 +00:00
if (data.size() > n)
throw Exception("Too long value for " + type->getName(), ErrorCodes::TOO_LARGE_STRING_SIZE);
auto res = std::make_shared<ColumnFixedString>(n);
ColumnFixedString::Chars_t & vec = res->getChars();
2013-03-27 10:07:23 +00:00
vec.resize_fill(n * s);
2013-03-27 10:07:23 +00:00
size_t offset = 0;
for (size_t i = 0; i < s; ++i)
{
memcpy(&vec[offset], data.data(), data.size());
offset += n;
}
return res;
2013-03-27 10:07:23 +00:00
}
else
throw Exception("Invalid data type in ColumnConstString: " + data_type->getName(), ErrorCodes::LOGICAL_ERROR);
2011-11-06 06:22:52 +00:00
}
2012-08-26 06:48:39 +00:00
ColumnPtr ColumnConst<Array>::convertToFullColumn() const
2012-08-26 06:48:39 +00:00
{
2013-03-27 10:07:23 +00:00
if (!data_type)
throw Exception("No data type specified for ColumnConstArray", ErrorCodes::LOGICAL_ERROR);
const DataTypeArray * type = typeid_cast<const DataTypeArray *>(&*data_type);
2013-03-27 10:07:23 +00:00
if (!type)
throw Exception("Non-array data type specified for ColumnConstArray", ErrorCodes::LOGICAL_ERROR);
const Array & array = getDataFromHolderImpl();
size_t array_size = array.size();
const auto & nested_type = type->getNestedType();
ColumnPtr nested_column;
if (nested_type->isNull())
{
/// Special case: an array of Null is actually an array of Nullable(UInt8).
nested_column = std::make_shared<ColumnNullable>(
std::make_shared<ColumnUInt8>(), std::make_shared<ColumnUInt8>());
}
else
nested_column = type->getNestedType()->createColumn();
2012-09-17 04:31:25 +00:00
auto res = std::make_shared<ColumnArray>(nested_column);
2012-08-26 11:14:52 +00:00
ColumnArray::Offsets_t & offsets = res->getOffsets();
offsets.resize(s);
for (size_t i = 0; i < s; ++i)
{
offsets[i] = (i + 1) * array_size;
for (size_t j = 0; j < array_size; ++j)
nested_column->insert(array[j]);
2012-08-26 11:14:52 +00:00
}
return res;
2012-08-26 06:48:39 +00:00
}
2013-03-06 13:56:11 +00:00
StringRef ColumnConst<Array>::getDataAt(size_t n) const
{
throw Exception("Method getDataAt is not supported for " + this->getName(), ErrorCodes::NOT_IMPLEMENTED);
}
UInt64 ColumnConst<Array>::get64(size_t n) const
{
throw Exception("Method get64 is not supported for " + this->getName(), ErrorCodes::NOT_IMPLEMENTED);
}
StringRef ColumnConst<Array>::getDataAtWithTerminatingZero(size_t n) const
{
throw Exception("Method getDataAt is not supported for " + this->getName(), ErrorCodes::NOT_IMPLEMENTED);
}
ColumnPtr ColumnConst<Tuple>::convertToFullColumn() const
2016-07-10 07:24:24 +00:00
{
return convertToTupleOfConstants()->convertToFullColumnIfConst();
}
ColumnPtr ColumnConst<Tuple>::convertToTupleOfConstants() const
{
if (!data_type)
throw Exception("No data type specified for ColumnConstTuple", ErrorCodes::LOGICAL_ERROR);
const DataTypeTuple * type = typeid_cast<const DataTypeTuple *>(&*data_type);
if (!type)
throw Exception("Non-Tuple data type specified for ColumnConstTuple", ErrorCodes::LOGICAL_ERROR);
/// Create columns for each element and convert to full columns.
const DataTypes & element_types = type->getElements();
size_t tuple_size = element_types.size();
Block block;
for (size_t i = 0; i < tuple_size; ++i)
block.insert(ColumnWithTypeAndName{
2016-07-10 07:24:24 +00:00
element_types[i]->createConstColumn(s, static_cast<const TupleBackend &>(*data)[i]),
element_types[i],
""});
return std::make_shared<ColumnTuple>(block);
}
void ColumnConst<Tuple>::getExtremes(Field & min, Field & max) const
{
2016-07-10 07:36:37 +00:00
min = *data;
max = *data;
}
StringRef ColumnConst<Tuple>::getDataAt(size_t n) const
{
throw Exception("Method getDataAt is not supported for " + this->getName(), ErrorCodes::NOT_IMPLEMENTED);
}
UInt64 ColumnConst<Tuple>::get64(size_t n) const
{
throw Exception("Method get64 is not supported for " + this->getName(), ErrorCodes::NOT_IMPLEMENTED);
}
StringRef ColumnConst<Tuple>::getDataAtWithTerminatingZero(size_t n) const
{
throw Exception("Method getDataAt is not supported for " + this->getName(), ErrorCodes::NOT_IMPLEMENTED);
}
2011-11-06 06:22:52 +00:00
}