ClickHouse/dbms/include/DB/Columns/ColumnFixedArray.h

204 lines
4.6 KiB
C
Raw Normal View History

2010-05-13 16:17:10 +00:00
#ifndef DBMS_CORE_COLUMN_FIXED_ARRAY_H
#define DBMS_CORE_COLUMN_FIXED_ARRAY_H
#include <Poco/SharedPtr.h>
#include <DB/Core/Exception.h>
#include <DB/Core/ErrorCodes.h>
#include <DB/Columns/IColumn.h>
namespace DB
{
using Poco::SharedPtr;
/** Cтолбeц значений типа "массив фиксированного размера".
*/
class ColumnFixedArray : public IColumn
{
public:
/** Создать пустой столбец массивов фиксированного размера n, со типом значений, как в столбце nested_column */
2011-08-09 19:19:00 +00:00
ColumnFixedArray(ColumnPtr nested_column, size_t n_)
2010-05-13 16:17:10 +00:00
: data(nested_column), n(n_)
{
2011-08-07 02:08:22 +00:00
clear();
2010-05-13 16:17:10 +00:00
}
2010-05-21 19:52:50 +00:00
2011-08-28 00:31:30 +00:00
std::string getName() const { return "ColumnFixedArray(" + data->getName() + ")"; }
2011-08-09 19:19:00 +00:00
ColumnPtr cloneEmpty() const
2010-05-21 19:52:50 +00:00
{
return new ColumnFixedArray(data->cloneEmpty(), n);
}
2010-05-13 16:17:10 +00:00
size_t size() const
{
return data->size() / n;
}
Field operator[](size_t index) const
{
Array res;
for (size_t i = n * index; i < n * (index + 1); ++i)
res[i] = (*data)[n * index + i];
return res;
}
StringRef getDataAt(size_t n) const
{
throw Exception("Method getDataAt is not supported for " + getName(), ErrorCodes::NOT_IMPLEMENTED);
}
2010-05-13 16:17:10 +00:00
void cut(size_t start, size_t length)
{
data->cut(n * start, n * length);
}
2010-05-20 19:29:04 +00:00
void insert(const Field & x)
{
const Array & array = get<const Array &>(x);
2010-05-20 19:29:04 +00:00
if (n != array.size())
throw Exception("Size of array doesn't match size of FixedArray column",
ErrorCodes::SIZE_OF_ARRAY_DOESNT_MATCH_SIZE_OF_FIXEDARRAY_COLUMN);
for (size_t i = 0; i < n; ++i)
data->insert(array[i]);
}
void insertFrom(const IColumn & src_, size_t index)
{
const ColumnFixedArray & src = static_cast<const ColumnFixedArray &>(src_);
if (n != src.getN())
throw Exception("Size of array doesn't match size of FixedArray column",
ErrorCodes::SIZE_OF_ARRAY_DOESNT_MATCH_SIZE_OF_FIXEDARRAY_COLUMN);
for (size_t i = 0; i < n; ++i)
data->insertFrom(src.getData(), n * index + i);
}
2010-05-20 19:29:04 +00:00
void insertDefault()
{
for (size_t i = 0; i < n; ++i)
data->insertDefault();
}
2010-05-13 16:17:10 +00:00
void clear()
{
2011-08-07 02:08:22 +00:00
data->clear();
2010-05-13 16:17:10 +00:00
}
/** Более эффективные методы манипуляции */
IColumn & getData()
{
return *data;
}
2011-08-22 20:24:45 +00:00
void filter(const Filter & filt)
{
size_t size = this->size();
if (size != filt.size())
throw Exception("Size of filter doesn't match size of column.", ErrorCodes::SIZES_OF_COLUMNS_DOESNT_MATCH);
if (size == 0)
return;
/// Не слишком оптимально. Можно сделать специализацию для массивов известных типов.
Filter nested_filt(size * n);
for (size_t i = 0; i < size; ++i)
if (filt[i])
memset(&nested_filt[i * n], 1, n);
data->filter(nested_filt);
}
2012-08-27 05:13:14 +00:00
void replicate(const Offsets_t & offsets)
{
throw Exception("Replication of column FixedArray is not implemented.", ErrorCodes::NOT_IMPLEMENTED);
}
2011-09-04 00:22:19 +00:00
void permute(const Permutation & perm)
{
size_t size = this->size();
if (size != perm.size())
throw Exception("Size of permutation doesn't match size of column.", ErrorCodes::SIZES_OF_COLUMNS_DOESNT_MATCH);
if (size == 0)
return;
Permutation nested_perm(size * n);
for (size_t i = 0; i < size; ++i)
for (size_t j = 0; j < n; ++j)
2012-08-03 20:34:19 +00:00
nested_perm[i * n + j] = perm[i] * n + j;
2011-09-04 00:22:19 +00:00
data->permute(nested_perm);
}
int compareAt(size_t p1, size_t p2, const IColumn & rhs_) const
{
const ColumnFixedArray & rhs = static_cast<const ColumnFixedArray &>(rhs_);
/// Не оптимально
for (size_t i = 0; i < n; ++i)
if (int res = data->compareAt(p1 * n + i, p2 * n + i, *rhs.data))
return res;
return 0;
}
2011-09-26 11:05:38 +00:00
struct less
{
const ColumnFixedArray & parent;
const Permutation & nested_perm;
less(const ColumnFixedArray & parent_, const Permutation & nested_perm_) : parent(parent_), nested_perm(nested_perm_) {}
bool operator()(size_t lhs, size_t rhs) const
{
for (size_t i = 0; i < parent.n; ++i)
{
if (nested_perm[lhs * parent.n + i] < nested_perm[rhs * parent.n + i])
return true;
else if (nested_perm[lhs * parent.n + i] > nested_perm[rhs * parent.n + i])
return false;
}
return false;
}
};
Permutation getPermutation() const
{
Permutation nested_perm = data->getPermutation();
size_t s = data->size() / n;
Permutation res(s);
for (size_t i = 0; i < s; ++i)
res[i] = i;
std::sort(res.begin(), res.end(), less(*this, nested_perm));
return res;
}
2012-05-17 19:15:53 +00:00
size_t byteSize() const
2011-08-27 22:43:31 +00:00
{
return data->byteSize() + sizeof(n);
}
2010-05-13 16:17:10 +00:00
const IColumn & getData() const
{
return *data;
}
2011-08-21 03:41:37 +00:00
size_t getN() const
{
return n;
}
2011-08-07 02:08:22 +00:00
protected:
2011-08-09 19:19:00 +00:00
ColumnPtr data;
2010-05-13 16:17:10 +00:00
const size_t n;
};
}
#endif