HashGrid.h

Go to the documentation of this file.

// Copyright (c) 2002-2023, The OpenMS Team -- EKU Tuebingen, ETH Zurich, and FU Berlin
// SPDX-License-Identifier: BSD-3-Clause
//
// --------------------------------------------------------------------------
// $Maintainer: Lars Nilse $
// $Authors: Bastian Blank $
// --------------------------------------------------------------------------
 
#include <OpenMS/CONCEPT/Types.h>
#include <OpenMS/DATASTRUCTURES/DPosition.h>
#include <boost/array.hpp>
#include <boost/functional/hash.hpp>
#include <boost/unordered/unordered_map.hpp>
#include <cmath>
#include <iterator>
#include <limits>
 
#ifndef OPENMS_COMPARISON_CLUSTERING_HASHGRID_H
  #define OPENMS_COMPARISON_CLUSTERING_HASHGRID_H
 
namespace OpenMS
{
  template<typename Cluster>
  class HashGrid
  {
  public:
    // XXX: Check is there is another type handy in OpenMS already
    typedef DPosition<2, double> ClusterCenter;
 
    typedef DPosition<2, Int64> CellIndex;
 
    typedef typename boost::unordered_multimap<ClusterCenter, Cluster> CellContent;
 
    typedef boost::unordered_map<CellIndex, CellContent> Grid;
 
    typedef typename CellContent::key_type key_type;
    typedef typename CellContent::mapped_type mapped_type;
    typedef typename CellContent::value_type value_type;
 
  private:
    class Iterator
    {
    private:
      friend class HashGrid;
 
      typedef typename Grid::iterator grid_iterator;
      typedef typename CellContent::iterator cell_iterator;
 
      using iterator_category = std::input_iterator_tag;
      using value_type = HashGrid::value_type;
      using reference = value_type&;
      using pointer = value_type*;
      using difference_type = std::ptrdiff_t;
 
 
      Grid& grid_;
      grid_iterator grid_it_;
      cell_iterator cell_it_;
 
      // Search for next non-empty cell
      void searchNextCell_()
      {
        while (cell_it_ == grid_it_->second.end())
        {
          grid_it_++;
 
          // If we are at the last cell, set cell iterator to something well-known
          if (grid_it_ == grid_.end())
          {
            cell_it_ = cell_iterator();
            return;
          }
 
          cell_it_ = grid_it_->second.begin();
        }
      }
 
    public:
      explicit Iterator(Grid& grid) : grid_(grid), grid_it_(grid.end())
      {
      }
 
      Iterator(Grid& grid, grid_iterator grid_it, cell_iterator cell_it) : grid_(grid), grid_it_(grid_it), cell_it_(cell_it)
      {
        searchNextCell_();
      }
 
      Iterator& operator++()
      {
        ++cell_it_;
        searchNextCell_();
        return *this;
      }
 
      const Iterator operator++(int)
      {
        Iterator ret(*this);
        operator++();
        return ret;
      }
 
      bool operator==(const Iterator& rhs) const
      {
        return grid_it_ == rhs.grid_it_ && cell_it_ == rhs.cell_it_;
      }
 
      bool operator!=(const Iterator& rhs) const
      {
        return !(*this == rhs);
      }
 
      value_type& operator*() const
      {
        return *cell_it_;
      }
 
      value_type* operator->() const
      {
        return &*cell_it_;
      }
 
      const CellIndex index() const
      {
        return grid_it_->first;
      }
    };
 
    class ConstIterator
    {
    private:
      friend class HashGrid;
 
      typedef typename Grid::const_iterator grid_iterator;
      typedef typename CellContent::const_iterator cell_iterator;
 
      using iterator_category = std::input_iterator_tag;
      using value_type = HashGrid::value_type;
      using reference = value_type&;
      using pointer = value_type*;
      using difference_type = std::ptrdiff_t;
 
      const Grid& grid_;
      grid_iterator grid_it_;
      cell_iterator cell_it_;
 
      // Search for next non-empty cell
      void searchNextCell_()
      {
        while (cell_it_ == grid_it_->second.end())
        {
          grid_it_++;
 
          // If we are at the last cell, set cell iterator to something well-known
          if (grid_it_ == grid_.end())
          {
            cell_it_ = cell_iterator();
            return;
          }
 
          cell_it_ = grid_it_->second.begin();
        }
      }
 
    public:
      ConstIterator(const Grid& grid) : grid_(grid), grid_it_(grid.end())
      {
      }
 
      ConstIterator(const Grid& grid, grid_iterator grid_it, cell_iterator cell_it) : grid_(grid), grid_it_(grid_it), cell_it_(cell_it)
      {
        searchNextCell_();
      }
 
      ConstIterator(const Iterator& it) : grid_(it.grid_), grid_it_(it.grid_it_), cell_it_(it.cell_it_)
      {
      }
 
      ConstIterator& operator++()
      {
        ++cell_it_;
        searchNextCell_();
        return *this;
      }
 
      const ConstIterator operator++(int)
      {
        ConstIterator ret(*this);
        operator++();
        return ret;
      }
 
      bool operator==(const ConstIterator& rhs) const
      {
        return grid_it_ == rhs.grid_it_ && cell_it_ == rhs.cell_it_;
      }
 
      bool operator!=(const ConstIterator& rhs) const
      {
        return !(*this == rhs);
      }
 
      const value_type& operator*() const
      {
        return *cell_it_;
      }
 
      const value_type* operator->() const
      {
        return &*cell_it_;
      }
 
      const CellIndex index() const
      {
        return grid_it_->first;
      }
    };
 
  public:
    typedef ConstIterator const_iterator;
    typedef Iterator iterator;
    typedef typename Grid::const_iterator const_grid_iterator;
    typedef typename Grid::iterator grid_iterator;
    typedef typename CellContent::const_iterator const_cell_iterator;
    typedef typename CellContent::iterator cell_iterator;
    typedef typename CellContent::size_type size_type;
 
  private:
    Grid cells_;
    CellIndex grid_dimension_;
 
  public:
    const ClusterCenter cell_dimension;
 
    const CellIndex& grid_dimension;
 
  public:
    explicit HashGrid(const ClusterCenter& c_dimension) : cells_(), grid_dimension_(), cell_dimension(c_dimension), grid_dimension(grid_dimension_)
    {
    }
 
    cell_iterator insert(const value_type& v)
    {
      const CellIndex cellkey = cellindexAtClustercenter_(v.first);
      CellContent& cell = cells_[cellkey];
      updateGridDimension_(cellkey);
      return cell.insert(v);
    }
 
    void erase(iterator pos)
    {
      CellContent& cell = pos.grid_it_->second;
      cell.erase(pos.cell_it_);
    }
 
    size_type erase(const key_type& key)
    {
      const CellIndex cellkey = cellindexAtClustercenter_(key);
      auto cell = cells_.find(cellkey);
      if (cell == cells_.end())
        return 0;
      return cell->second.erase(key);
    }
 
    void clear()
    {
      cells_.clear();
    }
 
    iterator begin()
    {
      grid_iterator grid_it = cells_.begin();
      if (grid_it == cells_.end())
        return end();
 
      cell_iterator cell_it = grid_it->second.begin();
      return iterator(cells_, grid_it, cell_it);
    }
 
    const_iterator begin() const
    {
      const_grid_iterator grid_it = cells_.begin();
      if (grid_it == cells_.end())
        return end();
 
      const_cell_iterator cell_it = grid_it->second.begin();
      return const_iterator(cells_, grid_it, cell_it);
    }
 
    iterator end()
    {
      return iterator(cells_);
    }
 
    const_iterator end() const
    {
      return const_iterator(cells_);
    }
 
    bool empty() const
    {
      return size() == 0;
    }
 
    size_type size() const
    {
      size_type ret = 0;
 
      for (const_grid_iterator it = grid_begin(); it != grid_end(); ++it)
      {
        ret += it->second.size();
      }
 
      return ret;
    }
 
    const_grid_iterator grid_begin() const
    {
      return cells_.begin();
    }
 
    const_grid_iterator grid_end() const
    {
      return cells_.end();
    }
 
    const typename Grid::mapped_type& grid_at(const CellIndex& x) const
    {
      return cells_.at(x);
    }
 
    typename Grid::mapped_type& grid_at(const CellIndex& x)
    {
      return cells_.at(x);
    }
 
    const_grid_iterator grid_find(const CellIndex& x) const
    {
      return cells_.find(x);
    }
 
    grid_iterator grid_find(const CellIndex& x)
    {
      return cells_.find(x);
    }
 
 
    grid_iterator grid_begin()
    {
      return cells_.begin();
    }
    grid_iterator grid_end()
    {
      return cells_.end();
    }
 
  private:
    // XXX: Replace with proper operator
    CellIndex cellindexAtClustercenter_(const ClusterCenter& key)
    {
      CellIndex ret;
      typename CellIndex::iterator it = ret.begin();
      typename ClusterCenter::const_iterator lit = key.begin(), rit = cell_dimension.begin();
      for (; it != ret.end(); ++it, ++lit, ++rit)
      {
        double t = std::floor(*lit / *rit);
        if (t < std::numeric_limits<Int64>::min() || t > std::numeric_limits<Int64>::max())
          throw Exception::OutOfRange(__FILE__, __LINE__, OPENMS_PRETTY_FUNCTION);
        *it = static_cast<Int64>(t);
      }
      return ret;
    }
 
    void updateGridDimension_(const CellIndex& d)
    {
      typename CellIndex::const_iterator it_new = d.begin();
      typename CellIndex::iterator it_cur = grid_dimension_.begin();
      for (; it_new != d.end(); ++it_new, ++it_cur)
      {
        if (*it_cur < *it_new)
          *it_cur = *it_new;
      }
    }
  };
 
  template<UInt N, typename T>
  std::size_t hash_value(const DPosition<N, T>& b)
  {
    boost::hash<T> hasher;
    std::size_t hash = 0;
    for (typename DPosition<N, T>::const_iterator it = b.begin(); it != b.end(); ++it)
      hash ^= hasher(*it);
    return hash;
  }
 
} // namespace OpenMS
 
#endif /* OPENMS_COMPARISON_CLUSTERING_HASHGRID_H */