MaskedLatticeStatsDataProvider.h

Go to the documentation of this file.

//# Copyright (C) 2000,2001
//# Associated Universities, Inc. Washington DC, USA.
//#
//# This library is free software; you can redistribute it and/or modify it
//# under the terms of the GNU Library General Public License as published by
//# the Free Software Foundation; either version 2 of the License, or (at your
//# option) any later version.
//#
//# This library is distributed in the hope that it will be useful, but WITHOUT
//# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
//# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
//# License for more details.
//#
//# You should have received a copy of the GNU Library General Public License
//# along with this library; if not, write to the Free Software Foundation,
//# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
//#
//# Correspondence concerning AIPS++ should be addressed as follows:
//#        Internet email: aips2-request@nrao.edu.
//#        Postal address: AIPS++ Project Office
//#                        National Radio Astronomy Observatory
//#                        520 Edgemont Road
//#                        Charlottesville, VA 22903-2475 USA
//#
//# $Id: Array.h 21545 2015-01-22 19:36:35Z gervandiepen $

#ifndef LATTICES_MASKEDLATTICESTATSDATAPROVIDER_H
#define LATTICES_MASKEDLATTICESTATSDATAPROVIDER_H

#include <casacore/lattices/LatticeMath/LatticeStatsDataProviderBase.h>

#include <casacore/lattices/Lattices/MaskedLattice.h>
#include <casacore/lattices/Lattices/MaskedLatticeIterator.h>

#include <casacore/casa/aips.h>

namespace casacore {

// Data provider which allows stats framework to iterate through a masked lattice.

template <class T> class MaskedLatticeStatsDataProvider
        : public LatticeStatsDataProviderBase<T> {
public:

        // default constructor. Must set lattice after construction but before
        // using the object

        MaskedLatticeStatsDataProvider();

        // <src>iteratorLimitBytes</src> is related to the size of the lattice.
        // If the lattice is greater than this size, then a lattice iterator will
        // be used to step through the lattice. If less, then all the data in the
        // values in the lattice are retrieved in a single chunk. The advantage of
        // the iterator is that less memory is used. The disadvantage is there is
        // a significant performace cost, so if the lattice is small, it is better to
        // get all its values in a single chunk and forgo the iterator. This is particularly
        // true when looping for a large number of iterations and creating a
        // MaskedLatticeStatsDataProvider each loop (in that case, you probably will want
        // to create a single object before the loop and use setLattice() to update
        // its lattice).
        MaskedLatticeStatsDataProvider(
                MaskedLattice<T>& lattice, uInt iteratorLimitBytes=4096*4096
        );

        ~MaskedLatticeStatsDataProvider();

        void operator++();

        uInt estimatedSteps() const;

        // Are there any data sets left to provide?
        Bool atEnd() const;

        // Take any actions necessary to finalize the provider. This will be called when
        // atEnd() returns True.
        void finalize();

        // get the count of elements in the current data set. When implementing this method, be
        // certain to take stride into account; ie for a data set with nominally 100 elements that
        // is to have a stride of two, this method should return 50.
        uInt64 getCount();

        // get the current data set
        const T* getData();

        // Get the associated mask of the current dataset. Only called if hasMask() returns True;
        const Bool* getMask();

        // returns something reasonable based on the lattice size.
        uInt getNMaxThreads() const;

        // Does the current data set have an associated mask?
        Bool hasMask() const;

        // reset the provider to point to the first data set it manages.
        void reset();

        // set the lattice. Automatically resets the lattice iterator.
        // <src>iteratorLimitBytes</src> is related to the size of the lattice.
        // If the lattice is greater than this size, then a lattice iterator will
        // be used to step through the lattice. If less, then all the data in the
        // values in the lattice are retrieved in a single chunk. The advantage of
        // the iterator is that less memory is used. The disadvantage is there is
        // a significant performace cost, so if the lattice is small, it is better to
        // get all its values in a single chunk and forgo the iterator. This is particularly
        // true when looping for a large number of iterations and creating a
        // MaskedLatticeStatsDataProvider each loop (in that case, you probably will want
        // to create a single object before the loop and use setLattice() to update
        // its lattice).
        void setLattice(const MaskedLattice<T>& lattice, uInt iteratorLimitBytes=4096*4096);

        // <group>
        // see base class documentation.
        void updateMaxPos(const std::pair<Int64, Int64>& maxpos);

        void updateMinPos(const std::pair<Int64, Int64>& minpos);
        // </group>

private:

        CountedPtr<RO_MaskedLatticeIterator<T> > _iter;
        Array<T> _currentSlice;
        Array<Bool> _currentMaskSlice;
        const T* _currentPtr;
        const Bool* _currentMaskPtr;
        Bool _delData, _delMask, _atEnd;
        uInt _nMaxThreads;

        void _freeStorage();

};

}

#ifndef CASACORE_NO_AUTO_TEMPLATES
#include <casacore/lattices/LatticeMath/MaskedLatticeStatsDataProvider.tcc>
#endif //# CASACORE_NO_AUTO_TEMPLATES

#endif