FeatureFinderAlgorithmPicked.h

Go to the documentation of this file.

// --------------------------------------------------------------------------
//                   OpenMS -- Open-Source Mass Spectrometry
// --------------------------------------------------------------------------
// Copyright The OpenMS Team -- Eberhard Karls University Tuebingen,
// ETH Zurich, and Freie Universitaet Berlin 2002-2021.
//
// This software is released under a three-clause BSD license:
//  * Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
//  * Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
//  * Neither the name of any author or any participating institution
//    may be used to endorse or promote products derived from this software
//    without specific prior written permission.
// For a full list of authors, refer to the file AUTHORS.
// --------------------------------------------------------------------------
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL ANY OF THE AUTHORS OR THE CONTRIBUTING
// INSTITUTIONS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
// OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
// WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
// OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
// ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// --------------------------------------------------------------------------
// $Maintainer: Timo Sachsenberg $
// $Authors: Marc Sturm $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/TRANSFORMATIONS/FEATUREFINDER/FeatureFinderAlgorithm.h>
#include <OpenMS/TRANSFORMATIONS/FEATUREFINDER/FeatureFinderAlgorithmPickedHelperStructs.h>
#include <OpenMS/TRANSFORMATIONS/FEATUREFINDER/TraceFitter.h>
 
#include <fstream>
 
namespace OpenMS
{
  class OPENMS_DLLAPI FeatureFinderAlgorithmPicked :
    public FeatureFinderAlgorithm,
    public FeatureFinderDefs
  {
public:
 
    typedef FeatureFinderAlgorithm::MapType MapType;
    typedef MapType::SpectrumType SpectrumType;
    typedef SpectrumType::FloatDataArrays FloatDataArrays;
 
    using FeatureFinderAlgorithm::param_;
    using FeatureFinderAlgorithm::features_;
    using FeatureFinderAlgorithm::ff_;
    using FeatureFinderAlgorithm::defaults_;
 
protected:
    typedef Peak1D PeakType;
    typedef FeatureFinderAlgorithmPickedHelperStructs::Seed Seed;
    typedef FeatureFinderAlgorithmPickedHelperStructs::MassTrace MassTrace;
    typedef FeatureFinderAlgorithmPickedHelperStructs::MassTraces MassTraces;
    typedef FeatureFinderAlgorithmPickedHelperStructs::TheoreticalIsotopePattern TheoreticalIsotopePattern;
    typedef FeatureFinderAlgorithmPickedHelperStructs::IsotopePattern IsotopePattern;
 
public:
    FeatureFinderAlgorithmPicked();
 
    // docu in base class
    void setSeeds(const FeatureMap& seeds) override;
 
    void run() override;
 
    static FeatureFinderAlgorithm* create();
 
    static const String getProductName();
 
protected:
    MapType map_;
    mutable std::ofstream log_;
    bool debug_;
    std::map<String, UInt> aborts_;
    std::map<Seed, String> abort_reasons_;
    FeatureMap seeds_;
 
 
    double pattern_tolerance_; 
    double trace_tolerance_; 
    UInt min_spectra_; 
    UInt max_missing_trace_peaks_; 
    double slope_bound_; 
    double intensity_percentage_; 
    double intensity_percentage_optional_; 
    double optional_fit_improvement_; 
    double mass_window_width_; 
    UInt intensity_bins_; 
    double min_isotope_fit_; 
    double min_trace_score_; 
    double min_rt_span_; 
    double max_rt_span_; 
    double max_feature_intersection_; 
    String reported_mz_; 
 
 
    double intensity_rt_step_;
    double intensity_mz_step_;
    std::vector<std::vector<std::vector<double> > > intensity_thresholds_;
 
    std::vector<TheoreticalIsotopePattern> isotope_distributions_;
 
    // Docu in base class
    void updateMembers_() override;
 
    void abort_(const Seed& seed, const String& reason);
 
    double intersection_(const Feature& f1, const Feature& f2) const;
 
    const TheoreticalIsotopePattern& getIsotopeDistribution_(double mass) const;
 
    double findBestIsotopeFit_(const Seed& center, UInt charge, IsotopePattern& best_pattern) const;
 
    void extendMassTraces_(const IsotopePattern& pattern, MassTraces& traces, Size meta_index_overall) const;
 
    void extendMassTrace_(MassTrace& trace, SignedSize spectrum_index, double mz, bool increase_rt, Size meta_index_overall, double min_rt = 0.0, double max_rt = 0.0) const;
 
    Size nearest_(double pos, const MSSpectrum& spec, Size start) const;
 
    void findIsotope_(double pos, Size spectrum_index, IsotopePattern& pattern, Size pattern_index, Size& peak_index) const;
 
    double positionScore_(double pos1, double pos2, double allowed_deviation) const;
 
    double isotopeScore_(const TheoreticalIsotopePattern& isotopes, IsotopePattern& pattern, bool consider_mz_distances) const;
 
    double intensityScore_(Size spectrum, Size peak) const;
 
    std::unique_ptr<TraceFitter> chooseTraceFitter_(double& tau);
 
    double intensityScore_(Size rt_bin, Size mz_bin, double intensity) const;
 
 
    void cropFeature_(std::shared_ptr<TraceFitter> fitter,
                      const MassTraces& traces,
                      MassTraces& new_traces);
 
    bool checkFeatureQuality_(std::shared_ptr<TraceFitter> fitter,
                              MassTraces& feature_traces,
                              const double& seed_mz, const double& min_feature_score,
                              String& error_msg, double& fit_score, double& correlation, double& final_score);
 
    void writeFeatureDebugInfo_(std::shared_ptr<TraceFitter> fitter,
                                const MassTraces& traces,
                                const MassTraces& new_traces,
                                bool feature_ok, const String error_msg, const double final_score, const Int plot_nr, const PeakType& peak,
                                const String path  = "debug/features/");
 
private:
 
    FeatureFinderAlgorithmPicked& operator=(const FeatureFinderAlgorithmPicked&);
    FeatureFinderAlgorithmPicked(const FeatureFinderAlgorithmPicked&);
  };
 
} // namespace OpenMS