statuniquecount.h

// Copyright 2009-2019 NTESS. Under the terms
// of Contract DE-NA0003525 with NTESS, the U.S.
// Government retains certain rights in this software.
//
// Copyright (c) 2009-2019, NTESS
// All rights reserved.
//
// This file is part of the SST software package. For license
// information, see the LICENSE file in the top level directory of the
// distribution.


#ifndef _H_SST_CORE_UNIQUE_COUNT_STATISTIC_
#define _H_SST_CORE_UNIQUE_COUNT_STATISTIC_

#include <sst/core/sst_types.h>
#include <sst/core/warnmacros.h>

#include <sst/core/statapi/statbase.h>

namespace SST {
class BaseComponent;
namespace Statistics {

/**
        \class UniqueCountStatistic

        Creates a Statistic which counts unique values provided to it.

        @tparam T A template for holding the main data type of this statistic
*/

template <typename T>
class UniqueCountStatistic : public Statistic<T>
{
public:
  SST_ELI_DECLARE_STATISTIC_TEMPLATE(
      UniqueCountStatistic,
      "sst",
      "UniqueCountStatistic",
      SST_ELI_ELEMENT_VERSION(1,0,0),
      "Track unique occurrences of statistic",
      "SST::Statistic<T>")

    UniqueCountStatistic(BaseComponent* comp, const std::string& statName, const std::string& statSubId, Params& statParams)
                : Statistic<T>(comp, statName, statSubId, statParams)
    {
        // Set the Name of this Statistic
        this->setStatisticTypeName("UniqueCount");
    }

    ~UniqueCountStatistic(){};

protected:
    /**
        Present a new value to the Statistic to be included in the unique set
        @param data New data item to be included in the unique set
    */
    void addData_impl(T data) override {
        uniqueSet.insert(data);
    }

private:
    void clearStatisticData() override
    {
      uniqueSet.clear();
    }

    void registerOutputFields(StatisticOutput* statOutput) override
    {
      uniqueCountField = statOutput->registerField<uint64_t>("UniqueItems");
    }

    void outputStatisticData(StatisticOutput* statOutput, bool UNUSED(EndOfSimFlag)) override
    {
      statOutput->outputField(uniqueCountField, (uint64_t) uniqueSet.size());
    }

private:
    std::set<T> uniqueSet;
    StatisticOutput::fieldHandle_t uniqueCountField;

};


} //namespace Statistics
} //namespace SST

#endif