#pragma once

#include <type_traits>

#include <dune/amdis/assembler/FirstOrderTestPartialTrial.hpp>

namespace AMDiS
{
  /**
   * \addtogroup operators
   * @{
   **/

  namespace tag
  {
    struct partialtest_trial
    {
      std::size_t comp;
    };
  }


  /// first-order operator \f$ \langle\partial_i\psi, c\,\phi\rangle \f$
  template <class GridFct, class QuadCreator>
  class GridFunctionOperator<tag::partialtest_trial, GridFct, QuadCreator>
      : public GridFunctionOperator<tag::test_partialtrial, GridFct, QuadCreator>
  {
    using Transposed = GridFunctionOperator<tag::test_partialtrial, GridFct, QuadCreator>;

  public:
    GridFunctionOperator(tag::partialtest_trial tag, GridFct const& expr, QuadCreator const& quadCreator)
      : Transposed(tag::test_partialtrial{tag.comp}, expr, quadCreator)
    {}

    template <class Context, class QuadratureRule,
              class ElementMatrix, class RowNode, class ColNode, bool sameFE, bool sameNode>
    void calculateElementMatrix(Context const& context,
                                QuadratureRule const& quad,
                                ElementMatrix& elementMatrix,
                                RowNode const& rowNode,
                                ColNode const& colNode,
                                std::integral_constant<bool, sameFE> flagSameFE,
                                std::integral_constant<bool, sameNode> flagSameNode)
    {
      auto elementMatrixTransposed = trans(elementMatrix);
      Transposed::calculateElementMatrix(
        context, quad, elementMatrixTransposed, colNode, rowNode, flagSameFE, flagSameNode);
    }
  };

  /** @} **/

} // end namespace AMDiS