103 lines
2.3 KiB
C++
103 lines
2.3 KiB
C++
// -*- C++ -*-
|
|
|
|
#ifndef __index_info_h__
|
|
#define __index_info_h__
|
|
|
|
#include <cstdlib>
|
|
#include <cstdint>
|
|
#include <vector>
|
|
#include <memory>
|
|
#include "vindex_base.h"
|
|
#include "index_type.h"
|
|
|
|
namespace MultiArrayTools
|
|
{
|
|
class IndexInfo;
|
|
|
|
class IndexInfo
|
|
{
|
|
public:
|
|
|
|
IndexInfo(IndexInfo&& in) = default;
|
|
IndexInfo& operator=(IndexInfo&& in) = default;
|
|
IndexInfo(const IndexInfo& in) = default;
|
|
IndexInfo& operator=(const IndexInfo& in) = default;
|
|
|
|
template <class IndexClass>
|
|
IndexInfo(const IndexClass& ind, size_t stepSize = 1);
|
|
|
|
template <class IndexClass>
|
|
IndexInfo& reassign(const IndexClass& ind, size_t stepSize = 1);
|
|
|
|
bool operator==(const IndexInfo& in) const;
|
|
bool operator!=(const IndexInfo& in) const;
|
|
|
|
bool operator<=(const IndexInfo& in) const;
|
|
bool operator<(const IndexInfo& in) const;
|
|
bool operator>(const IndexInfo& in) const;
|
|
bool operator>=(const IndexInfo& in) const;
|
|
|
|
|
|
const IndexInfo* getPtr(size_t inum) const;
|
|
std::intptr_t getPtrNum() const;
|
|
size_t dim() const;
|
|
size_t max() const;
|
|
size_t getStepSize(size_t inum) const;
|
|
size_t getStepSize() const;
|
|
IndexType type() const;
|
|
|
|
private:
|
|
|
|
IndexInfo() = default;
|
|
|
|
std::vector<IndexInfo> mNext;
|
|
std::intptr_t mPtrNum;
|
|
size_t mDim;
|
|
size_t mMax;
|
|
size_t mStepSize;
|
|
IndexType mType;
|
|
};
|
|
|
|
template <class IndexClass>
|
|
IndexInfo::IndexInfo(const IndexClass& ind, size_t stepSize) :
|
|
mNext(ind.infoVec()),
|
|
mPtrNum( reinterpret_cast<std::intptr_t>( &ind ) ),
|
|
mDim(ind.vrange()->dim()),
|
|
mMax(ind.max()),
|
|
mStepSize(stepSize),
|
|
mType(ind.type())
|
|
{}
|
|
|
|
template <class IndexClass>
|
|
IndexInfo& IndexInfo::reassign(const IndexClass& ind, size_t stepSize)
|
|
{
|
|
IndexInfo ii(ind, stepSize);
|
|
(*this) = std::move(ii);
|
|
return *this;
|
|
}
|
|
|
|
std::vector<IndexInfo> getRootIndices(const IndexInfo& info);
|
|
|
|
inline size_t getStepSize(const IndexInfo& ii, std::intptr_t j)
|
|
{
|
|
if(ii.type() == IndexType::SINGLE){
|
|
return ii.getPtrNum() == j ? 1 : 0;
|
|
}
|
|
else {
|
|
size_t ss = 0;
|
|
size_t sx = 1;
|
|
for(size_t i = 0; i != ii.dim(); ++i){
|
|
const IndexInfo& itmp = *ii.getPtr(ii.dim()-i-1);
|
|
const size_t max = itmp.max();
|
|
const size_t tmp = getStepSize(itmp, j);
|
|
ss += tmp * sx;
|
|
sx *= max;
|
|
}
|
|
return ss;
|
|
}
|
|
}
|
|
|
|
size_t getStepSize(const std::vector<IndexInfo>& iv, std::intptr_t j);
|
|
} // end namespace MultiArrayTools
|
|
|
|
#endif
|