MSSpectrum.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-2012.
//
// 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: Stephan Aiche$
// $Authors: Marc Sturm $
// --------------------------------------------------------------------------

#ifndef OPENMS_KERNEL_MSSPECTRUM_H
#define OPENMS_KERNEL_MSSPECTRUM_H

#include <OpenMS/METADATA/SpectrumSettings.h>
#include <OpenMS/METADATA/MetaInfoDescription.h>
#include <OpenMS/FORMAT/DB/PersistentObject.h>
#include <OpenMS/KERNEL/RangeManager.h>
#include <OpenMS/KERNEL/ComparatorUtils.h>

namespace OpenMS
{
  class Peak1D;

  template <typename PeakT = Peak1D>
  class MSSpectrum :
    public std::vector<PeakT>,
    public RangeManager<1>,
    public SpectrumSettings,
    public PersistentObject
  {
public:

    class OPENMS_DLLAPI FloatDataArray :
      public MetaInfoDescription,
      public std::vector<Real>
    {};

    class OPENMS_DLLAPI IntegerDataArray :
      public MetaInfoDescription,
      public std::vector<Int>
    {};

    class OPENMS_DLLAPI StringDataArray :
      public MetaInfoDescription,
      public std::vector<String>
    {};

    struct RTLess :
      public std::binary_function<MSSpectrum, MSSpectrum, bool>
    {
      inline bool operator()(const MSSpectrum & a, const MSSpectrum & b) const
      {
        return a.getRT() < b.getRT();
      }

    };



    typedef PeakT PeakType;
    typedef typename PeakType::CoordinateType CoordinateType;
    typedef std::vector<PeakType> ContainerType;
    typedef std::vector<FloatDataArray> FloatDataArrays;
    typedef std::vector<StringDataArray> StringDataArrays;
    typedef std::vector<IntegerDataArray> IntegerDataArrays;



    typedef typename ContainerType::iterator Iterator;
    typedef typename ContainerType::const_iterator ConstIterator;
    typedef typename ContainerType::reverse_iterator ReverseIterator;
    typedef typename ContainerType::const_reverse_iterator ConstReverseIterator;


    MSSpectrum() :
      ContainerType(),
      RangeManager<1>(),
      SpectrumSettings(),
      PersistentObject(),
      retention_time_(-1),
      ms_level_(1),
      name_(),
      float_data_arrays_(),
      string_data_arrays_(),
      integer_data_arrays_()
    {}

    MSSpectrum(const MSSpectrum & source) :
      ContainerType(source),
      RangeManager<1>(source),
      SpectrumSettings(source),
      PersistentObject(source),
      retention_time_(source.retention_time_),
      ms_level_(source.ms_level_),
      name_(source.name_),
      float_data_arrays_(source.float_data_arrays_),
      string_data_arrays_(source.string_data_arrays_),
      integer_data_arrays_(source.integer_data_arrays_)
    {}

    ~MSSpectrum()
    {}

    MSSpectrum & operator=(const MSSpectrum & source)
    {
      if (&source == this) return *this;

      ContainerType::operator=(source);
      RangeManager<1>::operator=(source);
      SpectrumSettings::operator=(source);
      PersistentObject::operator=(source);

      retention_time_ = source.retention_time_;
      ms_level_ = source.ms_level_;
      name_ = source.name_;
      float_data_arrays_ = source.float_data_arrays_;
      string_data_arrays_ = source.string_data_arrays_;
      integer_data_arrays_ = source.integer_data_arrays_;

      return *this;
    }

    bool operator==(const MSSpectrum & rhs) const
    {
      //name_ can differ => it is not checked
      return std::operator==(*this, rhs) &&
             RangeManager<1>::operator==(rhs) &&
             SpectrumSettings::operator==(rhs)  &&
             retention_time_ == rhs.retention_time_ &&
             ms_level_ == rhs.ms_level_ &&
             float_data_arrays_ == rhs.float_data_arrays_ &&
             string_data_arrays_ == rhs.string_data_arrays_ &&
             integer_data_arrays_ == rhs.integer_data_arrays_;
    }

    bool operator!=(const MSSpectrum & rhs) const
    {
      return !(operator==(rhs));
    }

    // Docu in base class (RangeManager)
    virtual void updateRanges()
    {
      this->clearRanges();
      updateRanges_(ContainerType::begin(), ContainerType::end());
    }

    inline DoubleReal getRT() const
    {
      return retention_time_;
    }

    inline void setRT(DoubleReal rt)
    {
      retention_time_ = rt;
    }

    inline UInt getMSLevel() const
    {
      return ms_level_;
    }

    inline void setMSLevel(UInt ms_level)
    {
      ms_level_ = ms_level;
    }

    inline const String & getName() const
    {
      return name_;
    }

    inline void setName(const String & name)
    {
      name_ = name;
    }



    inline const FloatDataArrays & getFloatDataArrays() const
    {
      return float_data_arrays_;
    }

    inline FloatDataArrays & getFloatDataArrays()
    {
      return float_data_arrays_;
    }

    inline const StringDataArrays & getStringDataArrays() const
    {
      return string_data_arrays_;
    }

    inline StringDataArrays & getStringDataArrays()
    {
      return string_data_arrays_;
    }

    inline const IntegerDataArrays & getIntegerDataArrays() const
    {
      return integer_data_arrays_;
    }

    inline IntegerDataArrays & getIntegerDataArrays()
    {
      return integer_data_arrays_;
    }




    void sortByIntensity(bool reverse = false)
    {
      if (float_data_arrays_.empty() && string_data_arrays_.empty() && integer_data_arrays_.empty())
      {
        if (reverse)
        {
          std::sort(ContainerType::begin(), ContainerType::end(), reverseComparator(typename PeakType::IntensityLess()));
        }
        else
        {
          std::sort(ContainerType::begin(), ContainerType::end(), typename PeakType::IntensityLess());
        }
      }
      else
      {
        //sort index list
        std::vector<std::pair<typename PeakType::IntensityType, Size> > sorted_indices;
        sorted_indices.reserve(ContainerType::size());
        for (Size i = 0; i < ContainerType::size(); ++i)
        {
          sorted_indices.push_back(std::make_pair(ContainerType::operator[](i).getIntensity(), i));
        }

        if (reverse)
        {
          std::sort(sorted_indices.begin(), sorted_indices.end(), reverseComparator(PairComparatorFirstElement<std::pair<typename PeakType::IntensityType, Size> >()));
        }
        else
        {
          std::sort(sorted_indices.begin(), sorted_indices.end(), PairComparatorFirstElement<std::pair<typename PeakType::IntensityType, Size> >());
        }

        //apply sorting to ContainerType and to meta data arrays
        ContainerType tmp;
        for (Size i = 0; i < sorted_indices.size(); ++i)
        {
          tmp.push_back(*(ContainerType::begin() + (sorted_indices[i].second)));
        }
        ContainerType::swap(tmp);

        for (Size i = 0; i < float_data_arrays_.size(); ++i)
        {
          std::vector<Real> mda_tmp;
          for (Size j = 0; j < float_data_arrays_[i].size(); ++j)
          {
            mda_tmp.push_back(*(float_data_arrays_[i].begin() + (sorted_indices[j].second)));
          }
          float_data_arrays_[i].swap(mda_tmp);
        }

        for (Size i = 0; i < string_data_arrays_.size(); ++i)
        {
          std::vector<String> mda_tmp;
          for (Size j = 0; j < string_data_arrays_[i].size(); ++j)
          {
            mda_tmp.push_back(*(string_data_arrays_[i].begin() + (sorted_indices[j].second)));
          }
          string_data_arrays_[i].swap(mda_tmp);
        }

        for (Size i = 0; i < integer_data_arrays_.size(); ++i)
        {
          std::vector<Int> mda_tmp;
          for (Size j = 0; j < integer_data_arrays_[i].size(); ++j)
          {
            mda_tmp.push_back(*(integer_data_arrays_[i].begin() + (sorted_indices[j].second)));
          }
          integer_data_arrays_[i].swap(mda_tmp);
        }
      }
    }

    void sortByPosition()
    {
      if (float_data_arrays_.empty())
      {
        std::sort(ContainerType::begin(), ContainerType::end(), typename PeakType::PositionLess());
      }
      else
      {
        //sort index list
        std::vector<std::pair<typename PeakType::PositionType, Size> > sorted_indices;
        sorted_indices.reserve(ContainerType::size());
        for (Size i = 0; i < ContainerType::size(); ++i)
        {
          sorted_indices.push_back(std::make_pair(ContainerType::operator[](i).getPosition(), i));
        }
        std::sort(sorted_indices.begin(), sorted_indices.end(), PairComparatorFirstElement<std::pair<typename PeakType::PositionType, Size> >());

        //apply sorting to ContainerType and to metadataarrays
        ContainerType tmp;
        tmp.reserve(sorted_indices.size());
        for (Size i = 0; i < sorted_indices.size(); ++i)
        {
          tmp.push_back(*(ContainerType::begin() + (sorted_indices[i].second)));
        }
        ContainerType::swap(tmp);

        for (Size i = 0; i < float_data_arrays_.size(); ++i)
        {
          std::vector<Real> mda_tmp;
          mda_tmp.reserve(float_data_arrays_[i].size());
          for (Size j = 0; j < float_data_arrays_[i].size(); ++j)
          {
            mda_tmp.push_back(*(float_data_arrays_[i].begin() + (sorted_indices[j].second)));
          }
          std::swap(float_data_arrays_[i], mda_tmp);
        }

        for (Size i = 0; i < string_data_arrays_.size(); ++i)
        {
          std::vector<String> mda_tmp;
          mda_tmp.reserve(string_data_arrays_[i].size());
          for (Size j = 0; j < string_data_arrays_[i].size(); ++j)
          {
            mda_tmp.push_back(*(string_data_arrays_[i].begin() + (sorted_indices[j].second)));
          }
          std::swap(string_data_arrays_[i], mda_tmp);
        }

        for (Size i = 0; i < integer_data_arrays_.size(); ++i)
        {
          std::vector<Int> mda_tmp;
          mda_tmp.reserve(integer_data_arrays_[i].size());
          for (Size j = 0; j < integer_data_arrays_[i].size(); ++j)
          {
            mda_tmp.push_back(*(integer_data_arrays_[i].begin() + (sorted_indices[j].second)));
          }
          std::swap(integer_data_arrays_[i], mda_tmp);
        }
      }
    }

    bool isSorted() const
    {
      for (Size i = 1; i < this->size(); ++i)
      {
        if (this->operator[](i - 1).getMZ() > this->operator[](i).getMZ()) return false;
      }
      return true;
    }



    Size findNearest(CoordinateType mz) const
    {
      // no peak => no search
      if (ContainerType::size() == 0) throw Exception::Precondition(__FILE__, __LINE__, __PRETTY_FUNCTION__, "There must be at least one peak to determine the nearest peak!");

      // search for position for inserting
      ConstIterator it = MZBegin(mz);
      // border cases
      if (it == ContainerType::begin()) return 0;

      if (it == ContainerType::end()) return ContainerType::size() - 1;

      // the peak before or the current peak are closest
      ConstIterator it2 = it;
      --it2;
      if (std::fabs(it->getMZ() - mz) < std::fabs(it2->getMZ() - mz))
      {
        return Size(it - ContainerType::begin());
      }
      else
      {
        return Size(it2 - ContainerType::begin());
      }
    }

    Iterator MZBegin(CoordinateType mz)
    {
      PeakType p;
      p.setPosition(mz);
      return lower_bound(ContainerType::begin(), ContainerType::end(), p, typename PeakType::PositionLess());
    }

    Iterator MZBegin(Iterator begin, CoordinateType mz, Iterator end)
    {
      PeakType p;
      p.setPosition(mz);
      return lower_bound(begin, end, p, typename PeakType::PositionLess());
    }

    Iterator MZEnd(CoordinateType mz)
    {
      PeakType p;
      p.setPosition(mz);
      return upper_bound(ContainerType::begin(), ContainerType::end(), p, typename PeakType::PositionLess());
    }

    Iterator MZEnd(Iterator begin, CoordinateType mz, Iterator end)
    {
      PeakType p;
      p.setPosition(mz);
      return upper_bound(begin, end, p, typename PeakType::PositionLess());
    }

    ConstIterator MZBegin(CoordinateType mz) const
    {
      PeakType p;
      p.setPosition(mz);
      return lower_bound(ContainerType::begin(), ContainerType::end(), p, typename PeakType::PositionLess());
    }

    ConstIterator MZBegin(ConstIterator begin, CoordinateType mz, ConstIterator end) const
    {
      PeakType p;
      p.setPosition(mz);
      return lower_bound(begin, end, p, typename PeakType::PositionLess());
    }

    ConstIterator MZEnd(CoordinateType mz) const
    {
      PeakType p;
      p.setPosition(mz);
      return upper_bound(ContainerType::begin(), ContainerType::end(), p, typename PeakType::PositionLess());
    }

    ConstIterator MZEnd(ConstIterator begin, CoordinateType mz, ConstIterator end) const
    {
      PeakType p;
      p.setPosition(mz);
      return upper_bound(begin, end, p, typename PeakType::PositionLess());
    }



    void clear(bool clear_meta_data)
    {
      ContainerType::clear();

      if (clear_meta_data)
      {
        clearRanges();
        clearId();
        this->SpectrumSettings::operator=(SpectrumSettings()); // no "clear" method
        retention_time_ = -1.0;
        ms_level_ = 1;
        name_.clear();
        float_data_arrays_.clear();
        string_data_arrays_.clear();
        integer_data_arrays_.clear();
      }
    }

protected:
    // Docu in base class
    virtual void clearChildIds_()
    {}

    DoubleReal retention_time_;

    UInt ms_level_;

    String name_;

    FloatDataArrays float_data_arrays_;

    StringDataArrays string_data_arrays_;

    IntegerDataArrays integer_data_arrays_;
  };

  template <typename PeakT>
  std::ostream & operator<<(std::ostream & os, const MSSpectrum<PeakT> & spec)
  {
    os << "-- MSSPECTRUM BEGIN --" << std::endl;

    //spectrum settings
    os << static_cast<const SpectrumSettings &>(spec);

    //peaklist
    for (typename MSSpectrum<PeakT>::ConstIterator it = spec.begin(); it != spec.end(); ++it)
    {
      os << *it << std::endl;
    }

    os << "-- MSSPECTRUM END --" << std::endl;
    return os;
  }

} // namespace OpenMS

#endif // OPENMS_KERNEL_MSSPECTRUM_H