mirror of
https://github.com/NGSolve/netgen.git
synced 2024-12-25 05:20:34 +05:00
Enumerate + Zip iterators
This commit is contained in:
parent
bdc738f87e
commit
a183380cc6
@ -1618,6 +1618,64 @@ namespace ngcore
|
||||
{
|
||||
return ar.Ptr()+i;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
template <typename TIA, typename TIB>
|
||||
class IteratorPair
|
||||
{
|
||||
TIA a;
|
||||
TIB b;
|
||||
public:
|
||||
IteratorPair (TIA _a, TIB _b) : a(_a), b(_b) { ; }
|
||||
|
||||
IteratorPair & operator++() { ++a; ++b; return *this; }
|
||||
bool operator!= (const IteratorPair & it2) { return a != it2.a; }
|
||||
|
||||
auto operator*()
|
||||
{
|
||||
// return pair(*a,*b);
|
||||
return std::pair<decltype(*a), decltype(*b)> (*a, *b); // keep reference
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
template <typename TA, typename TB>
|
||||
class Zip
|
||||
{
|
||||
const TA & a;
|
||||
const TB & b;
|
||||
public:
|
||||
Zip(const TA & _a, const TB & _b) : a(_a), b(_b) { ; }
|
||||
auto begin() const { return IteratorPair(a.begin(), b.begin()); }
|
||||
auto end() const { return IteratorPair(a.end(), b.end()); }
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
inline size_t size (const BaseArrayObject<T> & ao) { return ao.Size(); }
|
||||
|
||||
template <typename TA>
|
||||
class Enumerate
|
||||
{
|
||||
IntRange r;
|
||||
const TA & a;
|
||||
public:
|
||||
Enumerate(const TA & _a) : r(size(_a)), a(_a) { ; }
|
||||
auto begin() const { return IteratorPair(r.begin(), a.begin()); }
|
||||
auto end() const { return IteratorPair(r.end(), a.end()); }
|
||||
};
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
@ -57,7 +57,7 @@ namespace ngcore
|
||||
}
|
||||
|
||||
/// init i[0], i[1]
|
||||
NETGEN_INLINE INT (T ai1, T ai2)
|
||||
constexpr NETGEN_INLINE INT (T ai1, T ai2)
|
||||
{ i[0] = ai1; i[1] = ai2; }
|
||||
|
||||
/// init i[0], i[1], i[2]
|
||||
|
Loading…
Reference in New Issue
Block a user