#include "TransportPDESolver.hpp"


void TransportPDESolver::run(SampleSolution &solution) {
  assemble->SetTimeSeries(solution.U);
  solution.U.GetMesh().PrintInfo();
  assemble->GetTimeSeries().PrintInfo();
  timeInt->Method(assemble, solution.U);
}

void TransportPDESolver::computeQ(SampleSolution &solution) {
  if (quantity == "Energy") solution.Q = assemble->Energy(solution.U);
  else if (quantity == "Mass") solution.Q = assemble->Mass(solution.U);
  else if (quantity == "Error") solution.Q = assemble->Error(0, solution.U);
  else if (quantity == "Inflow") solution.Q = assemble->InflowOutflow(solution.U).first;
  else if (quantity == "Outflow") solution.Q = assemble->InflowOutflow(solution.U).second;
//    else if (quantity == "L2Error") solution.Q = assemble->L2Error(solution.U);
//    else if (quantity == "EnergyError") solution.Q = assemble->EnergyError(solution.U);
  else Exit("Quantity of interest not implemented")
}

void TransportPDESolver::computeCost(SampleSolution &solution) {
}

void TransportPDESolver::plotSolution(SampleSolution &solution) {
}