ConnFracStatistics.hpp

/*
  Copyright 2024 Equinor ASA.

  This file is part of the Open Porous Media project (OPM).

  OPM is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  OPM 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 General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef OPM_CONNFRACSTATISTICS_HPP
#define OPM_CONNFRACSTATISTICS_HPP

#include <opm/simulators/wells/RunningStatistics.hpp>

#include <array>
#include <cstddef>
#include <type_traits>

namespace Opm {

template <typename Scalar>
class ConnFracStatistics
{
public:
    enum class Quantity : std::size_t
    {
        Pressure,

        FlowRate,

        Width,

        // -------------------------------------------------------------
        // Helper.  Must be last enumerator.
        NumQuantities,
    };

    using SamplePoint = std::array
        <Scalar, static_cast<std::underlying_type_t<Quantity>>(Quantity::NumQuantities)>;

    template <class Serializer>
    void serializeOp(Serializer& serializer)
    {
        serializer(this->quantity_);
    }

    void reset()
    {
        for (auto& q : this->quantity_) { q.reset(); }
    }

    void addSamplePoint(const SamplePoint& samplePoint)
    {
        for (auto qIdx = 0*samplePoint.size(); qIdx < samplePoint.size(); ++qIdx) {
            this->quantity_[qIdx].addSamplePoint(samplePoint[qIdx]);
        }
    }

    const RunningStatistics<Scalar>& statistics(const Quantity q) const
    {
        return this->quantity_[ static_cast<std::underlying_type_t<Quantity>>(q) ];
    }

    static ConnFracStatistics serializationTestObject()
    {
        auto stat = ConnFracStatistics{};

        stat.quantity_
            .fill(RunningStatistics<Scalar>::serializationTestObject());

        return stat;
    }

    bool operator==(const ConnFracStatistics& that) const
    {
        return this->quantity_ == that.quantity_;
    }

private:
    using StatArray = std::array<
    RunningStatistics<Scalar>,
    static_cast<std::underlying_type_t<Quantity>>(Quantity::NumQuantities)
    >;

    StatArray quantity_{};
};

} // namespace Opm

#endif // OPM_CONNFRACSTATISTICS_HPP