214 lines
5.6 KiB
C++
214 lines
5.6 KiB
C++
// -*- C++ -*-
|
|
|
|
#ifndef __single_range_h__
|
|
#define __single_range_h__
|
|
|
|
#include <cstdlib>
|
|
#include <vector>
|
|
#include <memory>
|
|
|
|
#include "base_def.h"
|
|
#include "index_base.h"
|
|
#include "range_base.h"
|
|
|
|
namespace MultiArrayTools
|
|
{
|
|
|
|
template <typename U, RangeType TYPE>
|
|
class SingleIndex : public IndexBase<SingleIndex<U,TYPE> >
|
|
{
|
|
public:
|
|
|
|
typedef IndexBase<SingleIndex<U,TYPE> > IB;
|
|
typedef IndefinitIndexBase IIB;
|
|
|
|
DEFAULT_MEMBERS(SingleIndex);
|
|
|
|
SingleIndex(const U& upos, std::shared_ptr<const RangeBase<SingleIndex<U,TYPE> > >& rangePtr);
|
|
|
|
SingleIndex(std::shared_ptr<const RangeBase<SingleIndex<U,TYPE> > >& rangePtr,
|
|
size_t pos = 0);
|
|
|
|
//virtual SingleIndex& operator=(const U& upos);
|
|
|
|
virtual SingleIndex& operator=(size_t pos) override;
|
|
virtual SingleIndex& operator++() override;
|
|
virtual SingleIndex& operator--() override;
|
|
virtual SingleIndex& operator+=(int n) override;
|
|
virtual SingleIndex& operator-=(int n) override;
|
|
|
|
virtual bool operator==(const SingleIndex& i);
|
|
virtual bool operator!=(const SingleIndex& i);
|
|
|
|
virtual MultiRangeType rangeType() const override;
|
|
|
|
virtual U getMetaPos() const;
|
|
virtual SingleIndex& atMeta(const U& metaPos);
|
|
|
|
virtual size_t dim() const override; // = 1
|
|
virtual void copyPos(const SingleIndex& in) override;
|
|
|
|
virtual size_t giveSubStepSize(IndefinitIndexBase* subIndex) override;
|
|
|
|
//virtual void eval() override;
|
|
//virtual bool virt() const override { return false; }
|
|
|
|
protected:
|
|
virtual size_t evaluate(const SingleIndex& in) const override;
|
|
};
|
|
|
|
template <typename U, RangeType TYPE>
|
|
class SingleRange : public RangeBase<SingleIndex<U,TYPE> >
|
|
{
|
|
public:
|
|
typedef typename RangeBase<SingleIndex<U,TYPE> >::IndexType IndexType;
|
|
|
|
static SingleRange<U,TYPE> oType() { return SingleRange<U,TYPE>(); }
|
|
|
|
virtual size_t size() const override;
|
|
|
|
//U get(size_t pos) const;
|
|
const U& get(size_t pos) const;
|
|
size_t getMeta(const U& metaPos) const;
|
|
|
|
virtual MultiRangeType type() const override;
|
|
|
|
SingleIndex<U,TYPE> begin() const override;
|
|
SingleIndex<U,TYPE> end() const override;
|
|
|
|
protected:
|
|
|
|
SingleRange(const std::vector<U>& space);
|
|
|
|
std::vector<U> mSpace;
|
|
};
|
|
|
|
// specializaions
|
|
|
|
template <>
|
|
class SingleRange<int,RangeType::SPACE> : public RangeBase<SingleIndex<int,RangeType::SPACE> >
|
|
{
|
|
public:
|
|
typedef typename RangeBase<SingleIndex<int,RangeType::SPACE> >::IndexType IndexType;
|
|
|
|
static SingleRange<int,RangeType::SPACE> oType() { return SingleRange<int,RangeType::SPACE>(); }
|
|
|
|
virtual size_t size() const override;
|
|
|
|
int get(size_t pos) const;
|
|
size_t getMeta(int metaPos) const;
|
|
|
|
virtual MultiRangeType type() const override;
|
|
|
|
SingleIndex<int,RangeType::SPACE> begin() const override;
|
|
SingleIndex<int,RangeType::SPACE> end() const override;
|
|
|
|
protected:
|
|
|
|
SingleRange(size_t ext);
|
|
|
|
size_t mExt;
|
|
};
|
|
|
|
template <>
|
|
class SingleRange<size_t,RangeType::DISTANCE> : public RangeBase<SingleIndex<size_t,RangeType::DISTANCE> >
|
|
{
|
|
public:
|
|
typedef typename RangeBase<SingleIndex<size_t,RangeType::DISTANCE> >::IndexType IndexType;
|
|
|
|
static SingleRange<size_t,RangeType::DISTANCE> oType() { return SingleRange<size_t,RangeType::DISTANCE>(); }
|
|
|
|
virtual size_t size() const override;
|
|
|
|
size_t get(size_t pos) const;
|
|
size_t getMeta(size_t metaPos) const;
|
|
|
|
virtual MultiRangeType type() const override;
|
|
|
|
SingleIndex<size_t,RangeType::DISTANCE> begin() const override;
|
|
SingleIndex<size_t,RangeType::DISTANCE> end() const override;
|
|
|
|
protected:
|
|
|
|
SingleRange(size_t ext);
|
|
|
|
size_t mExt;
|
|
};
|
|
|
|
template <>
|
|
class SingleRange<size_t,RangeType::ENSEMBLE> : public RangeBase<SingleIndex<size_t,RangeType::ENSEMBLE> >
|
|
{
|
|
public:
|
|
typedef typename RangeBase<SingleIndex<size_t,RangeType::ENSEMBLE> >::IndexType IndexType;
|
|
|
|
static SingleRange<size_t,RangeType::ENSEMBLE> oType() { return SingleRange<size_t,RangeType::ENSEMBLE>(); }
|
|
|
|
virtual size_t size() const override;
|
|
|
|
size_t get(size_t pos) const;
|
|
size_t getMeta(size_t metaPos) const;
|
|
|
|
virtual MultiRangeType type() const override;
|
|
|
|
SingleIndex<size_t,RangeType::ENSEMBLE> begin() const override;
|
|
SingleIndex<size_t,RangeType::ENSEMBLE> end() const override;
|
|
|
|
protected:
|
|
|
|
SingleRange(size_t num);
|
|
|
|
size_t mNum;
|
|
};
|
|
|
|
enum class VET
|
|
{
|
|
VALUE = 0,
|
|
ERROR = 1
|
|
};
|
|
|
|
std::ostream& operator<<(std::ostream& os, VET vet);
|
|
|
|
template <>
|
|
class SingleRange<VET,RangeType::VALUE_ERROR> : public RangeBase<SingleIndex<VET,RangeType::VALUE_ERROR> >
|
|
{
|
|
public:
|
|
typedef typename RangeBase<SingleIndex<VET,RangeType::VALUE_ERROR> >::IndexType IndexType;
|
|
|
|
static SingleRange<VET,RangeType::VALUE_ERROR> oType()
|
|
{ return SingleRange<VET,RangeType::VALUE_ERROR>(); }
|
|
|
|
virtual size_t size() const override;
|
|
|
|
VET get(size_t pos) const;
|
|
size_t getMeta(VET metaPos) const;
|
|
|
|
virtual MultiRangeType type() const override;
|
|
|
|
SingleIndex<VET,RangeType::VALUE_ERROR> begin() const override;
|
|
SingleIndex<VET,RangeType::VALUE_ERROR> end() const override;
|
|
};
|
|
|
|
template <>
|
|
class SingleRange<size_t,RangeType::LORENTZ> : public RangeBase<SingleIndex<size_t,RangeType::LORENTZ> >
|
|
{
|
|
public:
|
|
typedef typename RangeBase<SingleIndex<size_t,RangeType::LORENTZ> >::IndexType IndexType;
|
|
|
|
static SingleRange<size_t,RangeType::LORENTZ> oType() { return SingleRange<size_t,RangeType::LORENTZ>(); }
|
|
|
|
virtual size_t size() const override;
|
|
|
|
size_t get(size_t pos) const;
|
|
size_t getMeta(size_t metaPos) const;
|
|
|
|
virtual MultiRangeType type() const override;
|
|
|
|
SingleIndex<size_t,RangeType::LORENTZ> begin() const override;
|
|
SingleIndex<size_t,RangeType::LORENTZ> end() const override;
|
|
};
|
|
|
|
}
|
|
|
|
#include "single_range.cc"
|
|
|
|
#endif
|