OCCT

// Copyright (c) 2020 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.

#include <Message_AttributeMeter.hxx>

#include <Message_Report.hxx>
#include <OSD_Timer.hxx>

#include <Precision.hxx>
#include <Standard_Dump.hxx>

IMPLEMENT_STANDARD_RTTIEXT(Message_AttributeMeter, Message_Attribute)

//=================================================================================================

Message_AttributeMeter::Message_AttributeMeter(const TCollection_AsciiString& theName)
    : Message_Attribute(theName)
{
}

//=================================================================================================

bool Message_AttributeMeter::HasMetric(const Message_MetricType& theMetric) const
{
  return myMetrics.Contains(theMetric);
}

//=================================================================================================

bool Message_AttributeMeter::IsMetricValid(const Message_MetricType& theMetric) const
{
  return std::abs(StartValue(theMetric) - UndefinedMetricValue()) > Precision::Confusion()
         && std::abs(StopValue(theMetric) - UndefinedMetricValue()) > Precision::Confusion();
}

//=================================================================================================

double Message_AttributeMeter::StartValue(const Message_MetricType& theMetric) const
{
  if (!HasMetric(theMetric))
  {
    return UndefinedMetricValue();
  }

  return myMetrics.Seek(theMetric)->first;
}

//=================================================================================================

void Message_AttributeMeter::SetStartValue(const Message_MetricType& theMetric,
                                           const double              theValue)
{
  if (StartToStopValue* aValPtr = myMetrics.ChangeSeek(theMetric))
  {
    aValPtr->first = theValue;
  }
  else
  {
    myMetrics.Add(theMetric, std::make_pair(theValue, UndefinedMetricValue()));
  }
}

//=================================================================================================

double Message_AttributeMeter::StopValue(const Message_MetricType& theMetric) const
{
  if (!HasMetric(theMetric))
  {
    return UndefinedMetricValue();
  }
  return myMetrics.Seek(theMetric)->second;
}

//=================================================================================================

void Message_AttributeMeter::SetStopValue(const Message_MetricType& theMetric,
                                          const double              theValue)
{
  if (StartToStopValue* aValPtr = myMetrics.ChangeSeek(theMetric))
  {
    aValPtr->second = theValue;
  }
}

//=================================================================================================

void Message_AttributeMeter::SetAlertMetrics(const occ::handle<Message_AlertExtended>& theAlert,
                                             const bool theStartValue)
{
  if (theAlert.IsNull())
  {
    return;
  }

  occ::handle<Message_AttributeMeter> aMeterAttribute =
    occ::down_cast<Message_AttributeMeter>(theAlert->Attribute());
  if (aMeterAttribute.IsNull())
  {
    return;
  }

  occ::handle<Message_Report>                       aReport         = Message::DefaultReport(true);
  const NCollection_IndexedMap<Message_MetricType>& anActiveMetrics = aReport->ActiveMetrics();

  // time metrics
  if (anActiveMetrics.Contains(Message_MetricType_WallClock))
  {
    OSD_Timer aTimer;
    aTimer.Start();
    double aTime = OSD_Timer::GetWallClockTime();
    if (theStartValue)
    {
      aMeterAttribute->SetStartValue(Message_MetricType_WallClock, aTime);
    }
    else
    {
      aMeterAttribute->SetStopValue(Message_MetricType_WallClock, aTime);
    }
  }
  if (anActiveMetrics.Contains(Message_MetricType_ProcessCPUUserTime)
      || anActiveMetrics.Contains(Message_MetricType_ProcessCPUSystemTime)
      || anActiveMetrics.Contains(Message_MetricType_ThreadCPUUserTime)
      || anActiveMetrics.Contains(Message_MetricType_ThreadCPUSystemTime))
  {
    if (anActiveMetrics.Contains(Message_MetricType_ProcessCPUUserTime)
        || anActiveMetrics.Contains(Message_MetricType_ProcessCPUSystemTime))
    {
      double aProcessUserTime, aProcessSystemTime;
      OSD_Chronometer::GetProcessCPU(aProcessUserTime, aProcessSystemTime);
      if (anActiveMetrics.Contains(Message_MetricType_ProcessCPUUserTime))
      {
        if (theStartValue)
        {
          aMeterAttribute->SetStartValue(Message_MetricType_ProcessCPUUserTime, aProcessUserTime);
        }
        else
        {
          aMeterAttribute->SetStopValue(Message_MetricType_ProcessCPUUserTime, aProcessUserTime);
        }
      }
      if (anActiveMetrics.Contains(Message_MetricType_ProcessCPUSystemTime))
      {
        if (theStartValue)
        {
          aMeterAttribute->SetStartValue(Message_MetricType_ProcessCPUSystemTime,
                                         aProcessSystemTime);
        }
        else
        {
          aMeterAttribute->SetStopValue(Message_MetricType_ProcessCPUSystemTime,
                                        aProcessSystemTime);
        }
      }
    }
    if (anActiveMetrics.Contains(Message_MetricType_ThreadCPUUserTime)
        || anActiveMetrics.Contains(Message_MetricType_ThreadCPUSystemTime))
    {
      double aThreadUserTime, aThreadSystemTime;
      OSD_Chronometer::GetThreadCPU(aThreadUserTime, aThreadSystemTime);
      if (anActiveMetrics.Contains(Message_MetricType_ThreadCPUUserTime))
      {
        if (theStartValue)
        {
          aMeterAttribute->SetStartValue(Message_MetricType_ThreadCPUUserTime, aThreadUserTime);
        }
        else
        {
          aMeterAttribute->SetStopValue(Message_MetricType_ThreadCPUUserTime, aThreadUserTime);
        }
      }
      if (anActiveMetrics.Contains(Message_MetricType_ThreadCPUSystemTime))
      {
        if (theStartValue)
        {
          aMeterAttribute->SetStartValue(Message_MetricType_ThreadCPUSystemTime, aThreadSystemTime);
        }
        else
        {
          aMeterAttribute->SetStopValue(Message_MetricType_ThreadCPUSystemTime, aThreadSystemTime);
        }
      }
    }
  }

  // memory metrics
  OSD_MemInfo aMemInfo(false);
  aMemInfo.SetActive(false);
  NCollection_IndexedMap<OSD_MemInfo::Counter> aCounters;
  for (NCollection_IndexedMap<Message_MetricType>::Iterator anIterator(anActiveMetrics);
       anIterator.More();
       anIterator.Next())
  {
    OSD_MemInfo::Counter anInfoCounter;
    if (!Message::ToOSDMetric(anIterator.Value(), anInfoCounter))
    {
      continue;
    }

    aCounters.Add(anInfoCounter);
    aMemInfo.SetActive(anInfoCounter, true);
  }
  if (aCounters.IsEmpty())
  {
    return;
  }

  aMemInfo.Update();
  Message_MetricType aMetricType;
  for (NCollection_IndexedMap<OSD_MemInfo::Counter>::Iterator anIterator(aCounters);
       anIterator.More();
       anIterator.Next())
  {
    if (!Message::ToMessageMetric(anIterator.Value(), aMetricType))
    {
      continue;
    }

    if (theStartValue)
    {
      aMeterAttribute->SetStartValue(aMetricType,
                                     (double)aMemInfo.ValuePreciseMiB(anIterator.Value()));
    }
    else
    {
      aMeterAttribute->SetStopValue(aMetricType,
                                    (double)aMemInfo.ValuePreciseMiB(anIterator.Value()));
    }
  }
}

//=================================================================================================

void Message_AttributeMeter::DumpJson(Standard_OStream& theOStream, int theDepth) const
{
  OCCT_DUMP_TRANSIENT_CLASS_BEGIN(theOStream)
  OCCT_DUMP_BASE_CLASS(theOStream, theDepth, Message_Attribute)

  for (NCollection_IndexedDataMap<Message_MetricType, StartToStopValue>::Iterator anIterator(
         myMetrics);
       anIterator.More();
       anIterator.Next())
  {
    OCCT_DUMP_VECTOR_CLASS(theOStream,
                           Message::MetricToString(anIterator.Key()),
                           2,
                           anIterator.Value().first,
                           anIterator.Value().second)
  }
}