OpenSearchModificationAnalysis.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: Timo Sachsenberg $
// $Authors: Timo Sachsenberg $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/METADATA/PeptideIdentificationList.h>
#include <OpenMS/DATASTRUCTURES/String.h>
 
#include <map>
#include <vector>
#include <unordered_map>
#include <unordered_set>
 
namespace OpenMS
{
  class OPENMS_DLLAPI OpenSearchModificationAnalysis
  {
  public:
    
    struct ModificationPattern
    {
      double count = 0.0;           
      std::vector<double> masses;   
      int num_charge_states = 0;    
    };
 
    struct ModificationSummary
    {
      int count;                    
      String name;                  
      int num_charge_states;        
      std::vector<double> masses;   
    };
 
    struct FuzzyDoubleComparator
    {
      double epsilon;
      FuzzyDoubleComparator(double eps = 1e-9) : epsilon(eps) {}
      bool operator()(const double& a, const double& b) const 
      {
        return std::fabs(a - b) >= epsilon && a < b;
      }
    };
 
    using DeltaMassHistogram = std::map<double, double, FuzzyDoubleComparator>;
    using DeltaMassToChargeCount = std::map<double, int, FuzzyDoubleComparator>;
 
    OpenSearchModificationAnalysis() = default;
 
    ~OpenSearchModificationAnalysis() = default;
 
    std::pair<DeltaMassHistogram, DeltaMassToChargeCount> 
    analyzeDeltaMassPatterns(const PeptideIdentificationList& peptide_ids, 
                            bool use_smoothing = false, 
                            bool debug = false) const;
 
    std::vector<ModificationSummary>
    mapDeltaMassesToModifications(const DeltaMassHistogram& delta_mass_histogram,
                                 const DeltaMassToChargeCount& charge_histogram,
                                 PeptideIdentificationList& peptide_ids,
                                 double precursor_mass_tolerance = 5.0,
                                 bool precursor_mass_tolerance_unit_ppm = true,
                                 const String& output_file = "") const;
 
    std::vector<ModificationSummary>
    analyzeModifications(PeptideIdentificationList& peptide_ids,
                        double precursor_mass_tolerance = 5.0,
                        bool precursor_mass_tolerance_unit_ppm = true,
                        bool use_smoothing = false,
                        const String& output_file = "") const;
 
  private:
    
    static double gaussian_(double x, double sigma);
 
    static DeltaMassHistogram smoothDeltaMassHistogram_(const DeltaMassHistogram& histogram, 
                                                       double sigma = 0.001);
 
    static DeltaMassHistogram findPeaksInHistogram_(const DeltaMassHistogram& histogram, 
                                                   double count_threshold = 0.0, 
                                                   double snr = 2.0);
 
    void writeModificationSummary_(const std::vector<ModificationSummary>& modifications,
                                  const String& output_file) const;
  };
 
} // namespace OpenMS