112 lines
2 KiB
C
112 lines
2 KiB
C
|
// -*- C++ -*-
|
||
|
|
|
||
|
|
#ifndef __block_h__
|
||
|
|
#define __block_h__
|
||
|
|
|
||
|
|
#include <cstdlib>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include "base_def.h"
|
||
|
|
|
||
|
|
namespace MultiArrayHelper
|
||
|
|
{
|
||
|
|
|
||
|
|
enum class BlockType {
|
||
|
|
INDEF = 0,
|
||
|
|
BLOCK = 1,
|
||
|
|
VALUE = 2,
|
||
|
|
SPLIT = 3,
|
||
|
|
RESULT = 4
|
||
|
|
};
|
||
|
|
|
||
|
|
// manage vectorization in the future !!
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
class BlockBase
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
BlockBase() = default;
|
||
|
|
BlockBase(size_t size);
|
||
|
|
|
||
|
|
virtual BlockType type() const = 0;
|
||
|
|
|
||
|
|
virtual size_t size() const;
|
||
|
|
virtual const T& operator[](size_t pos) const = 0;
|
||
|
|
|
||
|
|
template <class OpFunction>
|
||
|
|
BlockResult<T> operate(const BlockBase& in);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
size_t mSize;
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
class Block : public BlockBase<T>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
Block() = default;
|
||
|
|
Block(const std::vector<T>& data, size_t begPos, size_t size);
|
||
|
|
|
||
|
|
virtual BlockType type() const override;
|
||
|
|
virtual const T& operator[](size_t pos) const override;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
T* mBegPtr;
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
class BlockValue : public BlockBase<T>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
BlockValue() = default;
|
||
|
|
BlockValue(const T& val, size_t size);
|
||
|
|
|
||
|
|
virtual BlockType type() const override;
|
||
|
|
virtual const T& operator[](size_t pos) const override;
|
||
|
|
|
||
|
|
virtual BlockBase& set(size_t begPos) override;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
T mVal;
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
class SplitBlock : public BlockBase<T>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
SplitBlock() = default;
|
||
|
|
SplitBlock(std::vector<T*>&& begPtrVec);
|
||
|
|
|
||
|
|
virtual BlockType type() const override;
|
||
|
|
virtual const T& operator[](size_t pos) const override;
|
||
|
|
|
||
|
|
virtual BlockBase& set(size_t begPos) override;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
std::vector<T*> mBegPtrVec;
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
class BlockResult : public BlockBase<T>
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
BlockResult() = default;
|
||
|
|
BlockResult(size_t size);
|
||
|
|
|
||
|
|
virtual BlockType type() const override;
|
||
|
|
virtual const T& operator[](size_t pos) const override;
|
||
|
|
virtual T& operator[](size_t i);
|
||
|
|
|
||
|
|
virtual BlockBase& set(size_t begPos) override;
|
||
|
|
|
||
|
|
protected:
|
||
|
|
std::vector<T> mRes;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // end namespace MultiArrayHelper
|
||
|
|
|
||
|
|
#include "block.cc"
|
||
|
|
|
||
|
|
#endif
|