PeptideProteinResolution.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: Julianus Pfeuffer $
// $Authors: Julianus Pfeuffer $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/CONCEPT/Types.h>
#include <OpenMS/KERNEL/StandardTypes.h>
#include <OpenMS/METADATA/ProteinIdentification.h>
#include <OpenMS/METADATA/PeptideIdentification.h>
#include <vector>
#include <set>
 
namespace OpenMS
{
  struct OPENMS_DLLAPI ConnectedComponent
  {
    std::set<Size> prot_grp_indices;
    std::set<Size> pep_indices;
    friend std::ostream& operator<<(std::ostream& os, const ConnectedComponent& conn_comp);
  };
 
  class OPENMS_DLLAPI PeptideProteinResolution
  {
  
  private:
    typedef std::map<Size, std::set<Size> > IndexMap_;
 
    // TODO WIP for better tie resolution
    // std::vector<std::pair<bool,bool>> indist_prot_grp_td_;
    IndexMap_ indist_prot_grp_to_pep_;
    IndexMap_ pep_to_indist_prot_grp_;
    
    std::map<String, Size> prot_acc_to_indist_prot_grp_;
    
    bool statistics_;
    
  public:
    PeptideProteinResolution(bool statistics = false);
 
 
    static void resolve(ProteinIdentification& protein,
                        std::vector<PeptideIdentification>& peptides,
                        bool resolve_ties,
                        bool targets_first);
 
    static void run(std::vector<ProteinIdentification>& inferred_protein_id, 
                    std::vector<PeptideIdentification>& inferred_peptide_ids);
 
    void buildGraph(ProteinIdentification& protein,
                    const std::vector<PeptideIdentification>& peptides,
                    bool skip_sort = false);
      
    void resolveGraph(ProteinIdentification& protein,
                      std::vector<PeptideIdentification>& peptides);
    
    ConnectedComponent findConnectedComponent(Size& root_prot_grp);
    
 
    void resolveConnectedComponent(ConnectedComponent& conn_comp,
                                    ProteinIdentification& protein,
                                    std::vector<PeptideIdentification>& peptides);
};
  
} //namespace OpenMS