Iterator.H

Go to the documentation of this file.

/* Copyright 2021-2022 The AMReX Community
 *
 * Authors: Axel Huebl
 * License: BSD-3-Clause-LBNL
 */
#pragma once
 
#include "pyImpactX.H"
 
#include <AMReX_BoxArray.H>
#include <AMReX_DistributionMapping.H>
#include <AMReX_FArrayBox.H>
#include <AMReX_FabArray.H>
#include <AMReX_FabArrayBase.H>
#include <AMReX_MultiFab.H>
 
#include <memory>
#include <string>
 
 
namespace pyImpactX
{
    template< typename T_Iterator >
    T_Iterator &
    iterator_next( T_Iterator & it )
    {
        py::object self = py::cast(it);
        if (!py::hasattr(self, "first_or_done"))
            self.attr("first_or_done") = true;
 
        bool first_or_done = self.attr("first_or_done").cast<bool>();
        if (first_or_done) {
            first_or_done = false;
            self.attr("first_or_done") = first_or_done;
        }
        else
            ++it;
        if( !it.isValid() )
        {
            first_or_done = true;
            it.Finalize();
            throw py::stop_iteration();
        }
        return it;
     }
}