/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/
#ifndef itkResourceProbesCollectorBase_h
#define itkResourceProbesCollectorBase_h


#include "itkResourceProbe.h"
#include "itkMemoryUsageObserver.h"

namespace itk
{
/** \class ResourceProbesCollectorBase
 *  \brief Aggregates a set of probes.
 *
 *  This class defines a set of ResourceProbes and assign names to them.
 *  The user can start and stop each one of the probes by addressing them by name.
 *
 *  \sa ResourceProbe
 *
 * \ingroup ITKCommon
 */
template< typename TProbe >
class ITK_TEMPLATE_EXPORT ResourceProbesCollectorBase
{
public:
  using IdType = std::string;
  using MapType = std::map< IdType, TProbe >;

  /** destructor */
  virtual ~ResourceProbesCollectorBase() = default;

  /** Start a probe with a particular name. If the time probe does not
   * exist, it will be created */
  virtual void Start(const char *name);

  /** Stop a time probe identified with a name */
  virtual void Stop(const char *name);

  /** Report the summary of results from all probes */
  virtual void Report(std::ostream & os = std::cout,bool printSystemInfo = true,
                      bool printReportHead = true, bool useTabs = false);

  /** Report the summary of results from a specific probe */
  virtual void Report(const char *name, std::ostream & os = std::cout,
                      bool printSystemInfo = true, bool printReportHead = true,
                      bool useTabs = false);

  /** Expanded report of the summary of results from all probes */
  virtual void ExpandedReport(std::ostream & os = std::cout,bool printSystemInfo = true,
                              bool printReportHead = true, bool useTabs = false);

  /** Expanded report of the summary of results from a specific probe */
  virtual void ExpandedReport(const char *name, std::ostream & os = std::cout,
                              bool printSystemInfo = true, bool printReportHead = true, bool useTabs = false);

  /** JavaScript Object Notation (JSON) expanded report the summary of results from all probes */
  virtual void JSONReport(std::ostream & os = std::cout, bool printSystemInfo = true);

  /** JavaScript Object Notation (JSON) expanded report the summary of results from a specific probe */
  virtual void JSONReport(const char *name, std::ostream & os = std::cout);

  /** Destroy the set of probes. New probes can be created after invoking this
    method. */
  virtual void Clear();


  /** Returns a named Probe. If the name does not exists an exception
   * is thrown. */
  const TProbe & GetProbe(const char*name) const;


protected:
  MapType m_Probes;
};
} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkResourceProbesCollectorBase.hxx"
#endif

#endif //itkResourceProbesCollectorBase_h
