PeptideProteinResolution.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: 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 <OpenMS/METADATA/PeptideIdentificationList.h>
#include <vector>
#include <set>
#include <unordered_map>
 
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::unordered_map<String, Size> prot_acc_to_indist_prot_grp_;
    
    bool statistics_;
    
  public:
    PeptideProteinResolution(bool statistics = false);
 
 
    static void resolve(ProteinIdentification& protein,
                        PeptideIdentificationList& peptides,
                        bool resolve_ties,
                        bool targets_first);
 
    static void run(std::vector<ProteinIdentification>& inferred_protein_id,
                    PeptideIdentificationList& inferred_peptide_ids);
 
    void buildGraph(ProteinIdentification& protein,
                    const PeptideIdentificationList& peptides,
                    bool skip_sort = false);
 
    void resolveGraph(ProteinIdentification& protein,
                      PeptideIdentificationList& peptides);
 
    ConnectedComponent findConnectedComponent(Size& root_prot_grp);
    
 
    void resolveConnectedComponent(ConnectedComponent& conn_comp,
                                    ProteinIdentification& protein,
                                    PeptideIdentificationList& peptides);
};
  
} //namespace OpenMS