histogram.h

Go to the documentation of this file.

 
/******************************************************************************
The MIT License(MIT)
 
Embedded Template Library.
https://github.com/ETLCPP/etl
https://www.etlcpp.com
 
Copyright(c) 2021 John Wellbelove
 
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files(the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and / or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions :
 
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
 
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
******************************************************************************/
 
#ifndef ETL_HISTOGRAM_INCLUDED
#define ETL_HISTOGRAM_INCLUDED
 
#include "platform.h"
#include "algorithm.h"
#include "array.h"
#include "flat_map.h"
#include "functional.h"
#include "integral_limits.h"
#include "static_assert.h"
#include "type_traits.h"
 
namespace etl
{
  namespace private_histogram
  {
    //***************************************************************************
    //***************************************************************************
    template <typename TCount, size_t Max_Size_>
    class histogram_common
    {
    public:
 
      ETL_STATIC_ASSERT(etl::is_integral<TCount>::value, "Only integral count allowed");
 
      static ETL_CONSTANT size_t Max_Size = Max_Size_;
 
      typedef typename etl::array<TCount, Max_Size>::const_iterator const_iterator;
 
      //*********************************
      //*********************************
      const_iterator begin() const
      {
        return accumulator.begin();
      }
 
      //*********************************
      //*********************************
      const_iterator cbegin() const
      {
        return accumulator.cbegin();
      }
 
      //*********************************
      //*********************************
      const_iterator end() const
      {
        return accumulator.end();
      }
 
      //*********************************
      //*********************************
      const_iterator cend() const
      {
        return accumulator.cend();
      }
 
      //*********************************
      //*********************************
      void clear()
      {
        accumulator.fill(TCount(0));
      }
 
      //*********************************
      //*********************************
      ETL_CONSTEXPR size_t size() const
      {
        return Max_Size;
      }
 
      //*********************************
      //*********************************
      ETL_CONSTEXPR size_t max_size() const
      {
        return Max_Size;
      }
 
      //*********************************
      //*********************************
      size_t count() const
      {
        return etl::accumulate(accumulator.begin(), accumulator.end(), size_t(0));
      }
 
    protected:
 
      etl::array<TCount, Max_Size> accumulator;
    };
 
    template <typename TCount, size_t Max_Size_>
    ETL_CONSTANT size_t histogram_common<TCount, Max_Size_>::Max_Size;
  } // namespace private_histogram
 
  //***************************************************************************
  //***************************************************************************
  template <typename TKey, typename TCount, size_t Max_Size, int32_t Start_Index = etl::integral_limits<int32_t>::max>
  class histogram
    : public etl::private_histogram::histogram_common<TCount, Max_Size>
    , public etl::unary_function<TKey, void>
  {
  public:
 
    ETL_STATIC_ASSERT(etl::is_integral<TKey>::value, "Only integral keys allowed");
    ETL_STATIC_ASSERT(etl::is_integral<TCount>::value, "Only integral count allowed");
 
    typedef TKey   key_type;
    typedef TCount count_type;
    typedef TCount value_type;
 
    //*********************************
    //*********************************
    histogram()
    {
      this->accumulator.fill(count_type(0));
    }
 
    //*********************************
    //*********************************
    template <typename TIterator>
    histogram(TIterator first, TIterator last)
    {
      this->accumulator.fill(count_type(0));
      add(first, last);
    }
 
    //*********************************
    //*********************************
    histogram(const histogram& other)
    {
      this->accumulator = other.accumulator;
    }
 
#if ETL_USING_CPP11
    //*********************************
    //*********************************
    histogram(histogram&& other)
    {
      this->accumulator = etl::move(other.accumulator);
    }
#endif
 
    //*********************************
    //*********************************
    histogram& operator=(const histogram& rhs)
    {
      this->accumulator = rhs.accumulator;
 
      return *this;
    }
 
#if ETL_USING_CPP11
    //*********************************
    //*********************************
    histogram& operator=(histogram&& rhs)
    {
      this->accumulator = etl::move(rhs.accumulator);
 
      return *this;
    }
#endif
 
    //*********************************
    //*********************************
    void add(key_type key)
    {
      ++this->accumulator[static_cast<size_t>(key - Start_Index)];
    }
 
    //*********************************
    //*********************************
    template <typename TIterator>
    void add(TIterator first, TIterator last)
    {
      while (first != last)
      {
        add(*first);
        ++first;
      }
    }
 
    //*********************************
    //*********************************
    void operator()(key_type key)
    {
      add(key);
    }
 
    //*********************************
    //*********************************
    template <typename TIterator>
    void operator()(TIterator first, TIterator last)
    {
      add(first, last);
    }
 
    //*********************************
    //*********************************
    value_type operator[](key_type key) const
    {
      return this->accumulator[static_cast<size_t>(key - Start_Index)];
    }
  };
 
  //***************************************************************************
  //***************************************************************************
  template <typename TKey, typename TCount, size_t Max_Size>
  class histogram<TKey, TCount, Max_Size, etl::integral_limits<int32_t>::max>
    : public etl::private_histogram::histogram_common<TCount, Max_Size>
    , public etl::unary_function<TKey, void>
  {
  public:
 
    ETL_STATIC_ASSERT(etl::is_integral<TKey>::value, "Only integral keys allowed");
    ETL_STATIC_ASSERT(etl::is_integral<TCount>::value, "Only integral count allowed");
 
    typedef TKey   key_type;
    typedef TCount count_type;
    typedef TCount value_type;
 
    //*********************************
    //*********************************
    explicit histogram(key_type start_index_)
      : start_index(start_index_)
    {
      this->accumulator.fill(count_type(0));
    }
 
    //*********************************
    //*********************************
    template <typename TIterator>
    histogram(key_type start_index_, TIterator first, TIterator last)
      : start_index(start_index_)
    {
      this->accumulator.fill(count_type(0));
      add(first, last);
    }
 
    //*********************************
    //*********************************
    histogram(const histogram& other)
    {
      this->accumulator = other.accumulator;
    }
 
#if ETL_USING_CPP11
    //*********************************
    //*********************************
    histogram(histogram&& other)
    {
      this->accumulator = etl::move(other.accumulator);
    }
#endif
 
    //*********************************
    //*********************************
    histogram& operator=(const histogram& rhs)
    {
      this->accumulator = rhs.accumulator;
 
      return *this;
    }
 
#if ETL_USING_CPP11
    //*********************************
    //*********************************
    histogram& operator=(histogram&& rhs)
    {
      this->accumulator = etl::move(rhs.accumulator);
 
      return *this;
    }
#endif
 
    //*********************************
    //*********************************
    void add(key_type key)
    {
      ++this->accumulator[static_cast<size_t>(key - start_index)];
    }
 
    //*********************************
    //*********************************
    template <typename TIterator>
    void add(TIterator first, TIterator last)
    {
      while (first != last)
      {
        add(*first);
        ++first;
      }
    }
 
    //*********************************
    //*********************************
    void operator()(key_type key)
    {
      add(key);
    }
 
    //*********************************
    //*********************************
    template <typename TIterator>
    void operator()(TIterator first, TIterator last)
    {
      add(first, last);
    }
 
    //*********************************
    //*********************************
    value_type operator[](key_type key) const
    {
      return this->accumulator[static_cast<size_t>(key - start_index)];
    }
 
  private:
 
    key_type start_index;
  };
 
  //***************************************************************************
  //***************************************************************************
  template <typename TKey, typename TCount, size_t Max_Size_>
  class sparse_histogram : public etl::unary_function<TKey, void>
  {
  private:
 
    typedef etl::flat_map<TKey, TCount, Max_Size_> accumulator_type;
 
  public:
 
    ETL_STATIC_ASSERT(etl::is_integral<TCount>::value, "Only integral count allowed");
 
    static ETL_CONSTANT size_t Max_Size = Max_Size_;
 
    typedef TKey                                      key_type;
    typedef TCount                                    count_type;
    typedef typename accumulator_type::value_type     value_type;
    typedef typename accumulator_type::const_iterator const_iterator;
 
  public:
 
    //*********************************
    //*********************************
    sparse_histogram() {}
 
    //*********************************
    //*********************************
    template <typename TIterator>
    sparse_histogram(TIterator first, TIterator last)
    {
      add(first, last);
    }
 
    //*********************************
    //*********************************
    sparse_histogram(const sparse_histogram& other)
    {
      this->accumulator = other.accumulator;
    }
 
#if ETL_USING_CPP11
    //*********************************
    //*********************************
    sparse_histogram(sparse_histogram&& other)
    {
      accumulator = etl::move(other.accumulator);
    }
#endif
 
    //*********************************
    //*********************************
    sparse_histogram& operator=(const sparse_histogram& rhs)
    {
      accumulator = rhs.accumulator;
 
      return *this;
    }
 
#if ETL_USING_CPP11
    //*********************************
    //*********************************
    sparse_histogram& operator=(sparse_histogram&& rhs)
    {
      accumulator = etl::move(rhs.accumulator);
 
      return *this;
    }
#endif
 
    //*********************************
    //*********************************
    const_iterator begin() const
    {
      return accumulator.begin();
    }
 
    //*********************************
    //*********************************
    const_iterator cbegin() const
    {
      return accumulator.cbegin();
    }
 
    //*********************************
    //*********************************
    const_iterator end() const
    {
      return accumulator.begin();
    }
 
    //*********************************
    //*********************************
    const_iterator cend() const
    {
      return accumulator.cbegin();
    }
 
    //*********************************
    //*********************************
    void add(const key_type& key)
    {
      ++accumulator[key];
    }
 
    //*********************************
    //*********************************
    template <typename TIterator>
    void add(TIterator first, TIterator last)
    {
      while (first != last)
      {
        add(*first);
        ++first;
      }
    }
 
    //*********************************
    //*********************************
    void operator()(const key_type& key)
    {
      add(key);
    }
 
    //*********************************
    //*********************************
    template <typename TIterator>
    void operator()(TIterator first, TIterator last)
    {
      add(first, last);
    }
 
    //*********************************
    //*********************************
    const value_type& operator[](const key_type& key) const
    {
      static const value_type unused(key_type(), count_type(0));
 
      typename accumulator_type::const_iterator itr = accumulator.find(key);
 
      if (itr != accumulator.end())
      {
        return *itr;
      }
      else
      {
        return unused;
      }
    }
 
    //*********************************
    //*********************************
    void clear()
    {
      accumulator.clear();
    }
 
    //*********************************
    //*********************************
    size_t size() const
    {
      return accumulator.size();
    }
 
    //*********************************
    //*********************************
    ETL_CONSTEXPR size_t max_size() const
    {
      return Max_Size;
    }
 
    //*********************************
    //*********************************
    size_t count() const
    {
      count_type sum = count_type(0);
 
      const_iterator itr = accumulator.begin();
 
      while (itr != accumulator.end())
      {
        sum += (*itr).second;
        ++itr;
      }
 
      return static_cast<size_t>(sum);
    }
 
  private:
 
    etl::flat_map<key_type, count_type, Max_Size> accumulator;
  };
 
  template <typename TKey, typename TCount, size_t Max_Size_>
  ETL_CONSTANT size_t sparse_histogram<TKey, TCount, Max_Size_>::Max_Size;
} // namespace etl
 
#endif