// @(#)root/minuit2:$Id$
// Authors: M. Winkler, F. James, L. Moneta, A. Zsenei   2003-2005

/**********************************************************************
 *                                                                    *
 * Copyright (c) 2005 LCG ROOT Math team,  CERN/PH-SFT                *
 *                                                                    *
 **********************************************************************/

#ifndef ROOT_Minuit2_MnPrint
#define ROOT_Minuit2_MnPrint

#include "Minuit2/MnConfig.h"

#include <cassert>
#include <ios>
#include <iostream>
#include <sstream>
#include <string>
#include <utility>
#include <functional>

namespace ROOT {
namespace Minuit2 {

/**
    define std::ostream operators for output
*/

class FunctionMinimum;
std::ostream &operator<<(std::ostream &, const FunctionMinimum &);

class MinimumState;
std::ostream &operator<<(std::ostream &, const MinimumState &);

class MnUserParameters;
std::ostream &operator<<(std::ostream &, const MnUserParameters &);

class MnUserCovariance;
std::ostream &operator<<(std::ostream &, const MnUserCovariance &);

class MnGlobalCorrelationCoeff;
std::ostream &operator<<(std::ostream &, const MnGlobalCorrelationCoeff &);

class MnUserParameterState;
std::ostream &operator<<(std::ostream &, const MnUserParameterState &);

class MnMachinePrecision;
std::ostream &operator<<(std::ostream &, const MnMachinePrecision &);

class MinosError;
std::ostream &operator<<(std::ostream &, const MinosError &);

class ContoursError;
std::ostream &operator<<(std::ostream &, const ContoursError &);

// std::pair<double, double> is used by MnContour
std::ostream &operator<<(std::ostream &os, const std::pair<double, double> &point);

/* Design notes: 1) We want to delay the costly conversion from object references to
   strings to a point after we have decided whether or not to
   show that string to the user at all. 2) We want to offer a customization point for
   external libraries that want to replace the MnPrint logging. The actual
   implementation is in a separate file, MnPrintImpl.cxx file that external libraries
   can replace with their own implementation.
*/

// logging class for messages of varying severity
class MnPrint {
public:
   // want this to be an enum class for strong typing...
   enum class Verbosity { Error = 0, Warn = 1, Info = 2, Debug = 3, Trace = 4 };

   // ...but also want the values accessible from MnPrint scope for convenience
   static constexpr auto eError = Verbosity::Error;
   static constexpr auto eWarn = Verbosity::Warn;
   static constexpr auto eInfo = Verbosity::Info;
   static constexpr auto eDebug = Verbosity::Debug;
   static constexpr auto eTrace = Verbosity::Trace;

   // used for one-line printing of fcn minimum state
   class Oneline {
   public:
      Oneline(double fcn, double edm, int ncalls, int iter = -1);
      Oneline(const MinimumState &state, int iter = -1);
      Oneline(const FunctionMinimum &fmin, int iter = -1);

   private:
      double fFcn, fEdm;
      int fNcalls, fIter;

      friend std::ostream &operator<<(std::ostream &os, const Oneline &x);
   };

   MnPrint(const char *prefix, int level = MnPrint::GlobalLevel());
   ~MnPrint();

   // set global print level and return the previous one
   static int SetGlobalLevel(int level);

   // return current global print level
   static int GlobalLevel();

   // Whether to show the full prefix stack or only the end
   static void ShowPrefixStack(bool yes);

   static void AddFilter(const char *prefix);
   static void ClearFilter();

   // set print level and return the previous one
   int SetLevel(int level);

   // return current print level
   int Level() const;

   template <class... Ts>
   void Error(const Ts &... args)
   {
      Log(eError, args...);
   }

   template <class... Ts>
   void Warn(const Ts &... args)
   {
      Log(eWarn, args...);
   }

   template <class... Ts>
   void Info(const Ts &... args)
   {
      Log(eInfo, args...);
   }

   template <class... Ts>
   void Debug(const Ts &... args)
   {
      Log(eDebug, args...);
   }

   template <class... Ts>
   void Trace(const Ts &... args)
   {
      Log(eTrace, args...);
   }

private:
   // low level logging
   template <class... Ts>
   void Log(Verbosity level, const Ts &... args)
   {
      if (Level() < static_cast<int>(level))
         return;
      if (Hidden())
         return;

      std::ostringstream os;
      StreamPrefix(os);
      StreamArgs(os, args...);
      Impl(level, os.str());
   }

   static void StreamPrefix(std::ostringstream &os);

   // returns true if filters are installed and message is not selected by any filter
   static bool Hidden();

   // see MnPrintImpl.cxx
   static void Impl(Verbosity level, const std::string &s);

   // TMP to handle lambda argument correctly, exploiting overload resolution rules
   template <class T>
   static auto HandleLambda(std::ostream &os, const T &t, int) -> decltype(t(os), void())
   {
      t(os);
   }

   template <class T>
   static void HandleLambda(std::ostream &os, const T &t, float)
   {
      os << t;
   }

   static void StreamArgs(std::ostringstream &) {}

   // end of recursion
   template <class T>
   static void StreamArgs(std::ostringstream &os, const T &t)
   {
      os << " ";
      HandleLambda(os, t, 0);
   }

   template <class T, class... Ts>
   static void StreamArgs(std::ostringstream &os, const T &t, const Ts &... ts)
   {
      os << " " << t;
      StreamArgs(os, ts...);
   }

   int fLevel;
};

} // namespace Minuit2
} // namespace ROOT

#endif // ROOT_Minuit2_MnPrint