Trie.h

Go to the documentation of this file.

#ifndef DataFormats_Common_Trie_h
#define DataFormats_Common_Trie_h
/*
** 
** 
** Copyright (C) 2006 Julien Lemoine
** This program is free software; you can redistribute it and/or modify
** it under the terms of the GNU Lesser General Public License as published by
** the Free Software Foundation; either version 2 of the License, or
** (at your option) any later version.
** 
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU Lesser General Public License for more details.
** 
** You should have received a copy of the GNU Lesser General Public License
** along with this program; if not, write to the Free Software
** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
**
**
**   modified by Vincenzo Innocente on 15/08/2007
**
*/


#include <list>
#include <string>

namespace edm
{
  // fwd declaration
  template <typename T>
  class TrieNode;

  template <typename T>
  class TrieFactory
    {
    public:
      TrieFactory(unsigned paquetSize);
      ~TrieFactory();

    private:
      TrieFactory();
      TrieFactory(const TrieFactory &e);
      TrieFactory& operator=(const TrieFactory &e);

    public:
      TrieNode<T>* newNode(const T &value);
      void clear();

    private:
      unsigned          _paquetSize;
      std::list<TrieNode<T>*>   _allocatedNodes;
      TrieNode<T>       *_lastNodes;
      unsigned          _nbUsedInLastNodes;
    };
}


namespace edm
{
  template<typename T>
  class TrieNodeIter;


  template <typename T>
  class TrieNode
  {
  public:

    typedef TrieNodeIter<T> const_iterator;

    TrieNode();
    ~TrieNode();

  private:
    TrieNode(const TrieNode &e);
    TrieNode& operator=(const TrieNode &e);

  public:
    void setValue(const T &val);
    const T& value() const;

    const TrieNode<T>* brother() const;
    TrieNode<T>* brother();
    unsigned char brotherLabel() const;


    TrieNodeIter<T> begin() const;
    TrieNodeIter<T> end() const;

    // get first sub Node
    const TrieNode<T>* subNode() const;
    TrieNode<T>* subNode();
    unsigned char subNodeLabel() const;

    // Looking for a sub node
    const TrieNode<T>* subNodeByLabel(unsigned char chr) const;
    TrieNode<T>* subNodeByLabel(unsigned char chr);

    // add an edge
    void addSubNode(unsigned char chr, TrieNode<T> *node);

    void display(std::ostream &os, unsigned offset, unsigned char label) const;

    void clear();

  private:
    template <typename Node>
    Node _sequentialSearch(Node first, unsigned char label,
               unsigned char val) const;
    void _setBrother(TrieNode<T> *brother, unsigned char brotherLabel);
    void _addBrother(unsigned char chr, TrieNode<T> *brother);
    const TrieNode<T>* _getBrother(unsigned char chr) const;
    TrieNode<T>* _getBrother(unsigned char chr);

  private:
    TrieNode<T>     *_brother;
    unsigned char   _brotherLabel;
    TrieNode<T>     *_firstSubNode;
    unsigned char   _firstSubNodeLabel;
    T           _value;
  };
}

#include <ostream>

namespace edm
{
  //fwd declaration
  template <typename T>
    class TrieFactory;
  template <typename T>
    class TrieNode;

  template <typename T>
    class Trie
    {
    public:
      Trie(const T &empty);
      ~Trie();

    private:
      Trie();
      Trie(const Trie &e);
      Trie& operator=(const Trie &e);
    
    public:
      void insert(std::string const & str, const T &value);
      void insert(const char *str, unsigned strLen, const T &value);
      void setEntry(std::string const & str, const T &value);
      void setEntry(const char *str, unsigned strLen, const T &value);
      const T& find(std::string const & str) const;
      const T& find(const char *str, unsigned strLen) const;
      const TrieNode<T>* node(std::string const & str) const;
      const TrieNode<T>* node(const char *str, unsigned strLen) const;
      const TrieNode<T>* initialNode() const;
      void display(std::ostream &os);
      void clear();

    private:
      TrieNode<T>* _addEntry(const char *str, unsigned strLen);

    private:
      T         _empty;
      TrieFactory<T>    *_factory;
      TrieNode<T>       *_initialNode;
    };
}



#include <boost/iterator/iterator_facade.hpp>

#include<string>
#include<ostream>


// iterators and visitors

namespace edm{

  template<typename T>
  class TrieNodeIter
    : public boost::iterator_facade<TrieNodeIter<T>,
                    TrieNode<T> const, 
                    boost::forward_traversal_tag >
  {
    
  public:
    typedef TrieNodeIter<T> self;
    typedef TrieNode<T> const node_base;
    TrieNodeIter()
      : m_node(0), m_label(0)
    {}
    
    explicit TrieNodeIter(node_base* p)
      : m_node(p ? p->subNode() : 0), 
    m_label(p ? p->subNodeLabel() : 0)
    {}
    
    unsigned char label() const { return m_label;}
  private:
    friend class boost::iterator_core_access;
    
    void increment() {
      m_label = m_node->brotherLabel();
      m_node = m_node->brother(); 
    }
    
    bool equal(self const& other) const
    {
      return this->m_node == other.m_node;
    }
    
    node_base& dereference() const { return *m_node; }
    
    node_base* m_node;
    unsigned char m_label;
  };
  
  
  template<typename V, typename T>
  void walkTrie(V & v,  TrieNode<T> const  & n, std::string const & label="") {
    typedef TrieNodeIter<T> node_iterator;
    node_iterator e;
    for (node_iterator p(&n); p!=e; ++p) {
      v((*p).value(),label+(char)p.label());
      walkTrie(v,*p,label+(char)p.label());
    }
  }
   


  template<typename V, typename T>
  bool iterateTrieLeaves(V & v,  TrieNode<T> const  & n, std::string const & label="") {
    typedef TrieNodeIter<T> node_iterator;
    node_iterator e;
    node_iterator p(&n);
    if (p==e) return true;
    for (; p!=e; ++p) {
      if (iterateTrieLeaves(v,*p,label+(char)p.label()) )
    v((*p).value(),label+(char)p.label());
    }
    return false;
  }


}

//
//----------------------------------------------------------------
//
// implementations


template <typename T>
edm::TrieFactory<T>::TrieFactory(unsigned paquetSize) :
  _paquetSize(paquetSize), _lastNodes(0x0), _nbUsedInLastNodes(0)
{
  _lastNodes = new TrieNode<T>[paquetSize];
}

template <typename T>
edm::TrieFactory<T>::~TrieFactory()
{
  typename std::list<TrieNode<T>*>::const_iterator it;

  for (it = _allocatedNodes.begin(); it != _allocatedNodes.end(); ++it)
    delete[] *it;
  if (_lastNodes)
    delete[] _lastNodes;
}

template <typename T>
edm::TrieNode<T>* edm::TrieFactory<T>::newNode(const T &value)
{
  if (_nbUsedInLastNodes == _paquetSize)
    {
      _allocatedNodes.push_back(_lastNodes);
      _nbUsedInLastNodes = 0;
      _lastNodes = new TrieNode<T>[_paquetSize];
    }
  TrieNode<T> *res = &_lastNodes[_nbUsedInLastNodes];
  ++_nbUsedInLastNodes;
  res->setValue(value);
  res->clear();
  return res;
}

template <typename T>
void edm::TrieFactory<T>::clear()
{
  typename std::list<TrieNode<T>*>::const_iterator it;
  for (it = _allocatedNodes.begin(); it != _allocatedNodes.end(); ++it)
    delete[] *it;
  _allocatedNodes.clear();
  _nbUsedInLastNodes = 0;
}


template <typename T>
edm::TrieNode<T>::TrieNode() :
  _brother(0), _brotherLabel(0), _firstSubNode(0), _firstSubNodeLabel(0),_value()
{
}

template <typename T>
edm::TrieNode<T>::~TrieNode()
{
  // do not delete _brother and _firstSubNode because they are
  // allocated by factory (TrieFactory) and factory will delete them
}


template <typename T>   
edm::TrieNodeIter<T> 
edm::TrieNode<T>::begin() const {
  return const_iterator(this);
}
template <typename T>   
edm::TrieNodeIter<T> 
edm::TrieNode<T>::end() const {
  return const_iterator(0);
}

template <typename T>
void edm::TrieNode<T>::setValue(const T &val)
{
  _value = val;
}

template <typename T>
const T& edm::TrieNode<T>::value() const
{
  return _value;
}

template <typename T>
const edm::TrieNode<T>* edm::TrieNode<T>::brother() const
{
  return _brother;
}

template <typename T>
edm::TrieNode<T>* edm::TrieNode<T>::brother()
{
  return _brother;
}

template <typename T>
const edm::TrieNode<T>* edm::TrieNode<T>::_getBrother(unsigned char chr) const
{
  const TrieNode<T> *brother = _brother;
  return _sequentialSearch(brother, _brotherLabel, chr);
}

template <typename T>
edm::TrieNode<T>* edm::TrieNode<T>::_getBrother(unsigned char chr)
{
  return _sequentialSearch(_brother, _brotherLabel, chr);
}

template <typename T>
void edm::TrieNode<T>::_addBrother(unsigned char chr, TrieNode<T> *brother)
{
  if (!_brother || _brotherLabel > chr)
    {
      brother->_setBrother(_brother, _brotherLabel);
      _brother = brother;
      _brotherLabel = chr;
    }
  else
    _brother->_addBrother(chr, brother);
}

template <typename T>
unsigned char edm::TrieNode<T>::brotherLabel() const
{
  return _brotherLabel;
}

template <typename T>
const edm::TrieNode<T>* edm::TrieNode<T>::subNode() const
{
  return _firstSubNode;
}

template <typename T>
edm::TrieNode<T>* edm::TrieNode<T>::subNode()
{
  return _firstSubNode;
}

template <typename T>
unsigned char edm::TrieNode<T>::subNodeLabel() const
{
  return _firstSubNodeLabel;
}

template <typename T>
const edm::TrieNode<T>* edm::TrieNode<T>::subNodeByLabel(unsigned char chr) const
{
  const TrieNode<T> *first = _firstSubNode;
  return _sequentialSearch(first, _firstSubNodeLabel, chr);
}

template <typename T>
edm::TrieNode<T>* edm::TrieNode<T>::subNodeByLabel(unsigned char chr)
{
  return _sequentialSearch(_firstSubNode, _firstSubNodeLabel, chr);
}

template <typename T>
void edm::TrieNode<T>::addSubNode(unsigned char chr, TrieNode<T> *node)
{
  if (!_firstSubNode || _firstSubNodeLabel > chr)
    {
      node->_setBrother(_firstSubNode, _firstSubNodeLabel);
      _firstSubNode = node;
      _firstSubNodeLabel = chr;
    }
  else
    _firstSubNode->_addBrother(chr, node);
}

template <typename T>
template <typename Node>
inline Node edm::TrieNode<T>::_sequentialSearch(Node first, unsigned char label, unsigned char val) const
{
  if (first && label <= val)
    {
      if (label == val)
    return first;
      return first->_getBrother(val);
    }
  return 0x0;
}

template <typename T>
void edm::TrieNode<T>::_setBrother(TrieNode<T> *brother, unsigned char brotherLabel)
{
  _brother = brother;
  _brotherLabel = brotherLabel;
}

template <typename T>
void edm::TrieNode<T>::display(std::ostream &os, unsigned offset, unsigned char label) const
{
  unsigned int i;
  for (i = 0; i < offset; ++i)
    os << " ";
  if (label)
    os << "label[" << label << "] ";
  os << "value[" << _value << "]" << std::endl;
  if (_firstSubNode)
    _firstSubNode->display(os, offset + 2, _firstSubNodeLabel);
  if (_brother)
    _brother->display(os, offset, _brotherLabel);
}

template <typename T>
void edm::TrieNode<T>::clear()
{
  _brother = 0x0;
  _brotherLabel = 0;
  _firstSubNode = 0x0;
  _firstSubNodeLabel = 0;
}


#include <vector>
#include <algorithm>
#include <string>
#include <cassert>

#include "FWCore/Utilities/interface/EDMException.h"


namespace edm {
  namespace detailsTrie  {
    inline void errorInsert(std::string const & key) {
      Exception::throwThis(errors::InvalidReference,
    "Trie::insert called with a key already in collection;\n"
    "key value: ",
    key.c_str());
    }
  }
}

template <typename T>
edm::Trie<T>::Trie(const T &empty) :
  _empty(empty), _factory(0x0), _initialNode(0x0)
{
  // initialize nodes by paquets of 10000
  _factory = new TrieFactory<T>(10000);
  _initialNode = _factory->newNode(_empty);
}

template <typename T>
edm::Trie<T>::~Trie()
{
  delete _factory;
}

template <typename T>
void edm::Trie<T>::setEntry(std::string const & str, const T &value)
{
  setEntry(str.c_str(),str.size(),value);
}
template <typename T>
void edm::Trie<T>::setEntry(const char *str, unsigned strLen, const T &value)
{
  TrieNode<T>   *node = _addEntry(str, strLen);
  node->setValue(value);
}

template <typename T>
edm::TrieNode<T>* edm::Trie<T>::_addEntry(const char *str, unsigned strLen)
{
  unsigned  pos = 0;
  bool      found = true;
  TrieNode<T>   *node = _initialNode, *previous = 0x0;

  // Look for the part of the word which is in Trie
  while (found && pos < strLen)
    {
      found = false;
      previous = node;
      node = node->subNodeByLabel(str[pos]);
      if (node)
    {
      found = true;
      ++pos;
    }
    }

  // Add part of the word which is not in Trie
  if (!node || pos != strLen)
    {
      node = previous;
      for (unsigned i = pos; i < strLen; ++i)
    {
      TrieNode<T> *newNode = _factory->newNode(_empty);
      node->addSubNode(str[pos], newNode);
      node = newNode;
      ++pos;
    }
    }
  assert(node != 0x0);
  return node;
}


template <typename T>
void edm::Trie<T>::insert(std::string const & str, const T &value)
{
  insert(str.c_str(),str.size(),value);
}

template <typename T>
void edm::Trie<T>::insert(const char *str, unsigned strLen, const T &value)
{
  TrieNode<T>   *node = _addEntry(str, strLen);
  
  // Set the value on the last node
  if (node && node->value() != _empty)
    detailsTrie::errorInsert(std::string(str,strLen));
  node->setValue(value);
}

template <typename T>
const T& edm::Trie<T>::find(std::string const & str) const {
  return find(str.c_str(),str.size());
}

template <typename T>
const T& edm::Trie<T>::find(const char *str, unsigned strLen) const
{
  unsigned      pos = 0;
  bool          found = true;
  const TrieNode<T> *node = _initialNode;
  
  while (found && pos < strLen)
    {
      found = false;
      node = node->subNodeByLabel(str[pos]);
      if (node)
    {
      found = true;
      ++pos;
    }
    }
  if (node && pos == strLen) // The word is complet in the automaton
    return node->value();
  return _empty;
}

template <typename T>
edm::TrieNode<T> const * 
edm::Trie<T>::node(std::string const & str) const {
    return node(str.c_str(),str.size());
  }

template <typename T>
edm::TrieNode<T> const * 
edm::Trie<T>::node(const char *str, unsigned strLen) const {
  unsigned      pos = 0;
  bool          found = true;
  const TrieNode<T> *node = _initialNode;
    
  while (found && pos < strLen)
    {
      found = false;
      node = node->subNodeByLabel(str[pos]);
      if (node)
    {
      found = true;
      ++pos;
    }
    }
  return node;
}


template <typename T>
const edm::TrieNode<T>* edm::Trie<T>::initialNode() const
{
  return _initialNode;
}

template <typename T>
void edm::Trie<T>::clear()
{
  _factory->clear();
  _initialNode = _factory->newNode(_empty);
}

template <typename T>
void edm::Trie<T>::display(std::ostream &os)
{
  if (_initialNode)
    _initialNode->display(os, 0, 0);
}

#endif   //  DataFormats_Common_Trie_h