MassTraceDetection.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: Erhan Kenar, Holger Franken, Mohammed Alhigaylan $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/CONCEPT/ProgressLogger.h>
#include <OpenMS/DATASTRUCTURES/DefaultParamHandler.h>
#include <OpenMS/KERNEL/MassTrace.h>
#include <OpenMS/KERNEL/MSExperiment.h>
#include <OpenMS/KERNEL/StandardTypes.h>
 
#include <boost/dynamic_bitset.hpp>
 
namespace OpenMS
{
 
    class OPENMS_DLLAPI MassTraceDetection :
            public DefaultParamHandler,
            public ProgressLogger
    {
    public:
        enum TraceTerminationCriterion
        {
            OUTLIER,        
            SAMPLE_RATE     
        };
 
        MassTraceDetection();
 
        ~MassTraceDetection() override;
 
        void run(const PeakMap &, std::vector<MassTrace> &, const Size max_traces = 0);
 
        void run(PeakMap::ConstAreaIterator & begin, PeakMap::ConstAreaIterator & end, std::vector<MassTrace> & found_masstraces);
 
        bool hasFwhmMz() const { return has_fwhm_mz_; }
        bool hasFwhmIm() const { return has_fwhm_im_; }
        bool hasCentroidIm() const { return has_centroid_im_; }
 
    protected:
        void updateMembers_() override;
        static void updateIterativeWeightedMean_(const double& added_value,
                                                const double& added_intensity,
                                                double& centroid_value,
                                                double& prev_counter,
                                                double& prev_denom);
 
    private:
 
        struct Apex
        {
          Apex(double intensity, Size scan_idx, Size peak_idx);
          double intensity;
          Size scan_idx;
          Size peak_idx;
        };
 
        struct PeakCandidate
        {
          Size idx;
          double mz;
          double intensity;
          double im;
          bool found;
          
          PeakCandidate() : idx(0), mz(-1.0), intensity(-1.0), im(-1.0), found(false) {}
        };
 
        struct TraceExtensionState
        {
          Size scan_counter;
          Size hitting_peak_count;
          Size consecutive_missed;
          bool active;
          
          TraceExtensionState() : scan_counter(0), hitting_peak_count(0), consecutive_missed(0), active(true) {}
        };
 
        void run_(const std::vector<Apex>& chrom_apices,
                  const Size peak_count,
                  const PeakMap & work_exp,
                  const std::vector<Size>& spec_offsets,
                  std::vector<MassTrace> & found_masstraces,
                  const Size max_traces = 0);
 
        void getIMIndices_(const PeakMap& spectra,
                           int& fwhm_meta_idx, bool& has_fwhm_mz,
                           int& im_idx, bool& has_centroid_im,
                           int& im_fwhm_idx, bool& has_fwhm_im) const;
 
        PeakCandidate findBestPeak_(const MSSpectrum& spectrum,
                                    double centroid_mz,
                                    double ftl_sd,
                                    double centroid_im = -1.0) const;
 
        bool isPeakAcceptable_(const PeakCandidate& candidate,
                              double centroid_mz,
                              double ftl_sd,
                              double centroid_im,
                              Size spectrum_idx,
                              const std::vector<Size>& spec_offsets,
                              const boost::dynamic_bitset<>& peak_visited) const;
 
        void processPeak_(const PeakCandidate& candidate,
                         const MSSpectrum& spectrum,
                         std::list<PeakType>& current_trace,
                         std::vector<std::pair<Size, Size>>& gathered_idx,
                         std::vector<double>& fwhms_mz,
                         std::vector<double>& fwhms_im,
                         double& centroid_mz,
                         double& centroid_im,
                         double& prev_counter,
                         double& prev_denom,
                         double& prev_counter_im,
                         double& prev_denom_im,
                         double& ftl_sd,
                         double& intensity_so_far,
                         Size spectrum_idx,
                         bool is_upward_extension);
 
 
        bool isTraceValid_(const std::list<PeakType>& trace,
                          Size total_scans_visited,
                          Size consecutive_missed_down,
                          Size consecutive_missed_up) const;
 
        // Metadata availability flags – set by getIMIndices_ and valid after run_()
        bool has_fwhm_mz_ = false;
        bool has_fwhm_im_ = false;
        bool has_centroid_im_ = false;
 
        // parameter stuff
        double mass_error_ppm_;
        double mass_error_da_;
        double noise_threshold_int_;
        double chrom_peak_snr_;
        double ion_mobility_tolerance_;
        MassTrace::MT_QUANTMETHOD quant_method_;
 
        TraceTerminationCriterion trace_termination_criterion_;
        Size trace_termination_outliers_;
        double min_sample_rate_;
        double min_trace_length_;
        double max_trace_length_;
 
        bool reestimate_mt_sd_;
 
        // Metadata array indices - set during run_()
        mutable int fwhm_meta_idx_ = -1;
        mutable int ion_mobility_idx_ = -1;
        mutable int im_fwhm_idx_ = -1;
    };
}