DigestionEnzymeDB.h

Go to the documentation of this file.

// Copyright (c) 2002-present, OpenMS Inc. -- EKU Tuebingen, ETH Zurich, and FU Berlin
// SPDX-License-Identifier: BSD-3-Clause
//
// --------------------------------------------------------------------------
// $Maintainer: Xiao Liang $
// $Authors: Xiao Liang, Chris Bielow $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/CHEMISTRY/DigestionEnzyme.h>
#include <OpenMS/CHEMISTRY/DigestionEnzymeDataProvider.h>
#include <OpenMS/CONCEPT/Exception.h>
#include <OpenMS/DATASTRUCTURES/StringUtils.h>
 
#include <set>
#include <map>
#include <memory>
 
namespace OpenMS
{
  template<typename DigestionEnzymeType, typename InstanceType> class DigestionEnzymeDB
  {
  public:
 
    typedef typename std::set<const DigestionEnzymeType*>::const_iterator ConstEnzymeIterator;
    typedef typename std::set<const DigestionEnzymeType*>::iterator EnzymeIterator;
 
    static const InstanceType* getInstance()
    {
      static InstanceType* db_ = nullptr;
      if (db_ == nullptr)
      {
        db_ = new InstanceType;
      }
      return db_;
    }
 
    virtual ~DigestionEnzymeDB()
    {
      for (ConstEnzymeIterator it = const_enzymes_.begin(); it != const_enzymes_.end(); ++it)
      {
        delete *it;
      }
    }
 
    const DigestionEnzymeType* getEnzyme(const std::string& name) const
    {
      auto pos = enzyme_names_.find(name);
      if (pos == enzyme_names_.end())
      {
        throw Exception::ElementNotFound(__FILE__, __LINE__, OPENMS_PRETTY_FUNCTION, name);
      }
      return pos->second;
    }
 
    const DigestionEnzymeType* getEnzymeByRegEx(const std::string& cleavage_regex) const
    {
      if (!hasRegEx(cleavage_regex))
      {
        // @TODO: why does this use a different exception than "getEnzyme"?
        throw Exception::IllegalArgument(__FILE__, __LINE__, OPENMS_PRETTY_FUNCTION,
                                         std::string("Enzyme with regex " + cleavage_regex + " was not registered in Enzyme DB, register first!").c_str());
      }
      return enzyme_regex_.at(cleavage_regex);
    }
 
    void getAllNames(std::vector<std::string>& all_names) const
    {
      all_names.clear();
      for (ConstEnzymeIterator it = const_enzymes_.begin(); it != const_enzymes_.end(); ++it)
      {
        all_names.push_back((*it)->getName());
      }
    }
 
    bool hasEnzyme(const std::string& name) const
    {
      return (enzyme_names_.contains(name));
    }
 
    bool hasRegEx(const std::string& cleavage_regex) const
    {
      return (enzyme_regex_.contains(cleavage_regex));
    }
 
    bool hasEnzyme(const DigestionEnzymeType* enzyme) const
    {
      return (const_enzymes_.contains(enzyme) );
    }
 
    inline ConstEnzymeIterator beginEnzyme() const { return const_enzymes_.begin(); }  // we only allow constant iterators -- this DB is not meant to be modifiable
    inline ConstEnzymeIterator endEnzyme() const { return const_enzymes_.end(); }
 
  protected:
    DigestionEnzymeDB() = default;
 
    DigestionEnzymeDB(const DigestionEnzymeDB& enzymes_db) = delete;
 
    DigestionEnzymeDB& operator=(const DigestionEnzymeDB& enzymes_db) = delete;
 
    void addEnzyme_(const DigestionEnzymeType* enzyme)
    {
      std::string name = enzyme->getName();
 
      // if an enzyme with the same name exists, remove the old one first
      auto existing = enzyme_names_.find(name);
      if (existing != enzyme_names_.end())
      {
        const DigestionEnzymeType* old = existing->second;
        const_enzymes_.erase(old);
        // remove old name/synonym entries
        std::string old_name = old->getName();
        enzyme_names_.erase(old_name);
        enzyme_names_.erase(StringUtils::toLower(old_name));
        for (const auto& syn : old->getSynonyms())
        {
          enzyme_names_.erase(syn);
        }
        // remove old regex entry
        if (!old->getRegEx().empty())
        {
          enzyme_regex_.erase(old->getRegEx());
        }
        delete old;
      }
 
      // add to internal storage
      const_enzymes_.insert(enzyme);
      // add to internal indices (by name and its synonyms)
      enzyme_names_[name] = enzyme;
      enzyme_names_[StringUtils::toLower(name)] = enzyme;
      for (std::set<std::string>::const_iterator it = enzyme->getSynonyms().begin(); it != enzyme->getSynonyms().end(); ++it)
      {
        enzyme_names_[*it] = enzyme;
      }
      // ... and by regex
      if (enzyme->getRegEx() != "")
      {
        enzyme_regex_[enzyme->getRegEx()] = enzyme;
      }
      return;
    }
 
    void loadFromProviders_(std::vector<std::unique_ptr<DigestionEnzymeDataProvider<DigestionEnzymeType>>>& providers)
    {
      for (auto& provider : providers)
      {
        auto enzymes = provider->loadEnzymes();
        for (auto& enzyme : enzymes)
        {
          addEnzyme_(enzyme.release());
        }
      }
    }
 
    std::map<std::string, const DigestionEnzymeType*> enzyme_names_; 
 
    std::map<std::string, const DigestionEnzymeType*> enzyme_regex_; 
 
    std::set<const DigestionEnzymeType*> const_enzymes_; 
 
  };
}