// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkPVLogger.h"

#include "vtkObjectFactory.h"

#include <map>
#include <vtksys/SystemTools.hxx>

namespace
{
vtkLogger::Verbosity DefaultVerbosity = vtkLogger::VERBOSITY_TRACE;

using MapType = std::map<int, vtkLogger::Verbosity>;
MapType& get_map()
{
  static MapType the_map{};
  return the_map;
}

vtkLogger::Verbosity get_verbosity(int key, const char* envvar = nullptr)
{
  auto& map = get_map();
  auto iter = map.find(key);
  if (iter != map.end())
  {
    return iter->second != vtkLogger::VERBOSITY_INVALID ? iter->second : DefaultVerbosity;
  }

  if (envvar != nullptr)
  {
    if (const char* envval = vtksys::SystemTools::GetEnv(envvar))
    {
      auto env_verbosity = vtkLogger::ConvertToVerbosity(envval);
      if (env_verbosity != vtkLogger::VERBOSITY_INVALID)
      {
        vtkLogF(TRACE, "processed environment variable `%s=%s` for log verbosity", envvar, envval);

        // cache for later.
        map.insert(std::make_pair(key, env_verbosity));
        return env_verbosity;
      }
      else
      {
        vtkLogF(TRACE, "ignoring environment variable `%s`, invalid value `%s`", envvar, envval);
      }
    }
  }

  // either envvar not defined or was invalid. we still cache the value since
  // lookup is faster but flag it so that if DefaultVerbosity changes, we report
  // the changed value correctly. We use VERBOSITY_INVALID as an indicator that
  // default value should be used.
  map.insert(std::make_pair(key, vtkLogger::VERBOSITY_INVALID));
  return DefaultVerbosity;
}

void set_verbosity(int key, vtkLogger::Verbosity verbosity)
{
  get_map()[key] = verbosity;
}

const int PipelineVerbosityKey = 1;
const int PluginVerbosityKey = 2;
const int DataMovementVerbosityKey = 3;
const int RenderingVerbosityKey = 4;
const int ApplicationVerbosityKey = 5;
const int ExecutionVerbosityKey = 6;
const int CatalystVerbosityKey = 7;
}

//----------------------------------------------------------------------------
vtkPVLogger::vtkPVLogger() = default;

//----------------------------------------------------------------------------
vtkPVLogger::~vtkPVLogger() = default;

//----------------------------------------------------------------------------
vtkLogger::Verbosity vtkPVLogger::GetDefaultVerbosity()
{
  return DefaultVerbosity;
}

//----------------------------------------------------------------------------
void vtkPVLogger::SetDefaultVerbosity(vtkLogger::Verbosity value)
{
  DefaultVerbosity = value;
}

//----------------------------------------------------------------------------
vtkLogger::Verbosity vtkPVLogger::GetPipelineVerbosity()
{
  return get_verbosity(PipelineVerbosityKey, "PARAVIEW_LOG_PIPELINE_VERBOSITY");
}

//----------------------------------------------------------------------------
void vtkPVLogger::SetPipelineVerbosity(vtkLogger::Verbosity value)
{
  if (value > vtkLogger::VERBOSITY_INVALID && value <= vtkLogger::VERBOSITY_MAX)
  {
    set_verbosity(PipelineVerbosityKey, value);
  }
  else
  {
    vtkLogF(WARNING, "ignoring invalid verbosity %d", value);
  }
}

//----------------------------------------------------------------------------
vtkLogger::Verbosity vtkPVLogger::GetExecutionVerbosity()
{
  return get_verbosity(ExecutionVerbosityKey, "PARAVIEW_LOG_EXECUTION_VERBOSITY");
}

//----------------------------------------------------------------------------
void vtkPVLogger::SetExecutionVerbosity(vtkLogger::Verbosity value)
{
  if (value > vtkLogger::VERBOSITY_INVALID && value <= vtkLogger::VERBOSITY_MAX)
  {
    set_verbosity(ExecutionVerbosityKey, value);
  }
  else
  {
    vtkLogF(WARNING, "ignoring invalid verbosity %d", value);
  }
}

//----------------------------------------------------------------------------
vtkLogger::Verbosity vtkPVLogger::GetPluginVerbosity()
{
  return get_verbosity(PluginVerbosityKey, "PARAVIEW_LOG_PLUGIN_VERBOSITY");
}

//----------------------------------------------------------------------------
void vtkPVLogger::SetPluginVerbosity(vtkLogger::Verbosity value)
{
  if (value > vtkLogger::VERBOSITY_INVALID && value <= vtkLogger::VERBOSITY_MAX)
  {
    set_verbosity(PluginVerbosityKey, value);
  }
  else
  {
    vtkLogF(WARNING, "ignoring invalid verbosity %d", value);
  }
}

//----------------------------------------------------------------------------
vtkLogger::Verbosity vtkPVLogger::GetDataMovementVerbosity()
{
  return get_verbosity(DataMovementVerbosityKey, "PARAVIEW_LOG_DATA_MOVEMENT_VERBOSITY");
}

//----------------------------------------------------------------------------
void vtkPVLogger::SetDataMovementVerbosity(vtkLogger::Verbosity value)
{
  if (value > vtkLogger::VERBOSITY_INVALID && value <= vtkLogger::VERBOSITY_MAX)
  {
    set_verbosity(DataMovementVerbosityKey, value);
  }
  else
  {
    vtkLogF(WARNING, "ignoring invalid verbosity %d", value);
  }
}

//----------------------------------------------------------------------------
vtkLogger::Verbosity vtkPVLogger::GetRenderingVerbosity()
{
  return get_verbosity(RenderingVerbosityKey, "PARAVIEW_LOG_RENDERING_VERBOSITY");
}

//----------------------------------------------------------------------------
void vtkPVLogger::SetRenderingVerbosity(vtkLogger::Verbosity value)
{
  if (value > vtkLogger::VERBOSITY_INVALID && value <= vtkLogger::VERBOSITY_MAX)
  {
    set_verbosity(RenderingVerbosityKey, value);
  }
  else
  {
    vtkLogF(WARNING, "ignoring invalid verbosity %d", value);
  }
}

//----------------------------------------------------------------------------
vtkLogger::Verbosity vtkPVLogger::GetApplicationVerbosity()
{
  return get_verbosity(ApplicationVerbosityKey, "PARAVIEW_LOG_APPLICATION_VERBOSITY");
}

//----------------------------------------------------------------------------
void vtkPVLogger::SetApplicationVerbosity(vtkLogger::Verbosity value)
{
  if (value > vtkLogger::VERBOSITY_INVALID && value <= vtkLogger::VERBOSITY_MAX)
  {
    set_verbosity(ApplicationVerbosityKey, value);
  }
  else
  {
    vtkLogF(WARNING, "ignoring invalid verbosity %d", value);
  }
}

//----------------------------------------------------------------------------
vtkLogger::Verbosity vtkPVLogger::GetCatalystVerbosity()
{
  return get_verbosity(CatalystVerbosityKey, "PARAVIEW_LOG_CATALYST_VERBOSITY");
}

//----------------------------------------------------------------------------
void vtkPVLogger::SetCatalystVerbosity(vtkLogger::Verbosity value)
{
  if (value > vtkLogger::VERBOSITY_INVALID && value <= vtkLogger::VERBOSITY_MAX)
  {
    set_verbosity(CatalystVerbosityKey, value);
  }
  else
  {
    vtkLogF(WARNING, "ignoring invalid verbosity %d", value);
  }
}

//----------------------------------------------------------------------------
void vtkPVLogger::PrintSelf(ostream& os, vtkIndent indent)
{
  this->Superclass::PrintSelf(os, indent);
}
