operator.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2010 - 2018 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE.md at
// the top level directory of deal.II.
//
// ---------------------------------------------------------------------


#ifndef dealii_operator_h
#define dealii_operator_h

#include <deal.II/base/config.h>

#include <deal.II/algorithms/any_data.h>

#include <deal.II/base/event.h>

#include <fstream>

DEAL_II_NAMESPACE_OPEN

namespace Algorithms
{
  class OperatorBase : public Subscriptor
  {
  public:
    virtual ~OperatorBase() override = default;

    virtual void
    operator()(AnyData &out, const AnyData &in) = 0;

    virtual void
    notify(const Event &);
    void
    clear_events();

  protected:
    Event notifications;
  };

  template <typename VectorType>
  class OutputOperator : public Subscriptor
  {
  public:
    OutputOperator();

    OutputOperator(const OutputOperator<VectorType> &) = delete;

    virtual ~OutputOperator() override = default;

    void
    initialize_stream(std::ostream &stream);
    void
    set_step(const unsigned int step);
    virtual OutputOperator<VectorType> &
    operator<<(const AnyData &vectors);

  protected:
    unsigned int step;

  private:
    std::ostream *os;
  };

  template <typename VectorType>
  inline void
  OutputOperator<VectorType>::set_step(const unsigned int s)
  {
    step = s;
  }


  template <typename VectorType>
  inline OutputOperator<VectorType> &
  operator<<(OutputOperator<VectorType> &out, unsigned int step)
  {
    out.set_step(step);
    return out;
  }
} // namespace Algorithms

DEAL_II_NAMESPACE_CLOSE

#endif