#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "friction_writer.hh"

template <class ScalarVector, class Vector>
FrictionWriter<ScalarVector, Vector>::FrictionWriter(
    Vector const &vertexCoordinates,
    Dune::BitSetVector<1> const &_boundaryNodes)
    : BW(vertexCoordinates, _boundaryNodes),
      coefficientWriter("coefficients", std::fstream::out),
      stateWriter("states", std::fstream::out) {}

template <class ScalarVector, class Vector>
FrictionWriter<ScalarVector, Vector>::~FrictionWriter() {
  stateWriter.close();
  coefficientWriter.close();
}

template <class ScalarVector, class Vector>
void FrictionWriter<ScalarVector, Vector>::writeOther(
    ScalarVector const &coefficient, ScalarVector const &alpha) {
  for (size_t i = 0; i < boundaryNodes.size(); ++i) {
    if (!boundaryNodes[i][0])
      continue;

    coefficientWriter << coefficient[i] << " ";
    stateWriter << alpha[i] << " ";
  }

  stateWriter << std::endl;
  coefficientWriter << std::endl;
}

#include "friction_writer_tmpl.cc"