MSChromatogram.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: Andreas Bertsch $
// $Authors: Andreas Bertsch $
// --------------------------------------------------------------------------

#ifndef OPENMS_KERNEL_MSCHROMATOGRAM_H
#define OPENMS_KERNEL_MSCHROMATOGRAM_H

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


namespace OpenMS
{
  template <typename PeakT = ChromatogramPeak>
  class MSChromatogram :
    public std::vector<PeakT>,
    public RangeManager<1>,
    public ChromatogramSettings,
    public PersistentObject
  {

public:

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

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

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

    struct MZLess :
      public std::binary_function<MSChromatogram, MSChromatogram, bool>
    {
      inline bool operator()(const MSChromatogram & a, const MSChromatogram & b) const
      {
        return a.getMZ() < b.getMZ();
      }

    };

    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;


    MSChromatogram() :
      ContainerType(),
      RangeManager<1>(),
      ChromatogramSettings(),
      PersistentObject(),
      name_(),
      float_data_arrays_(),
      string_data_arrays_(),
      integer_data_arrays_()
    {}

    MSChromatogram(const MSChromatogram & source) :
      ContainerType(source),
      RangeManager<1>(source),
      ChromatogramSettings(source),
      PersistentObject(source),
      name_(source.name_),
      float_data_arrays_(source.float_data_arrays_),
      string_data_arrays_(source.string_data_arrays_),
      integer_data_arrays_(source.integer_data_arrays_)
    {}

    virtual ~MSChromatogram()
    {}

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

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

      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 MSChromatogram & rhs) const
    {
      //name_ can differ => it is not checked
      return std::operator==(*this, rhs) &&
             RangeManager<1>::operator==(rhs) &&
             ChromatogramSettings::operator==(rhs)  &&
             float_data_arrays_ == rhs.float_data_arrays_ &&
             string_data_arrays_ == rhs.string_data_arrays_ &&
             integer_data_arrays_ == rhs.integer_data_arrays_;
    }

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

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

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

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


    inline DoubleReal getMZ() const
    {
      return getProduct().getMZ();
    }



    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_.size() && integer_data_arrays_.size())
      {
        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;
        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)));
          }
          std::swap(float_data_arrays_[i], 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)));
          }
          std::swap(string_data_arrays_[i], 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)));
          }
          std::swap(integer_data_arrays_[i], mda_tmp);
        }
      }
    }

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



    Size findNearest(CoordinateType rt) 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 = RTBegin(rt);
      //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->getRT() - rt) < std::fabs(it2->getRT() - rt))
      {
        return Size(it - ContainerType::begin());
      }
      else
      {
        return Size(it2 - ContainerType::begin());
      }
    }

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

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

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

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

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

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

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

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

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

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


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

    String name_;

    FloatDataArrays float_data_arrays_;

    StringDataArrays string_data_arrays_;

    IntegerDataArrays integer_data_arrays_;
  };

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

    //chromatogram settings
    os << static_cast<const ChromatogramSettings &>(spec);

    //peaklist
    os << static_cast<const typename MSChromatogram<PeakT>::ContainerType &>(spec);

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

    return os;
  }

} // namespace OpenMS

#endif // OPENMS_KERNEL_MSCHROMATOGRAM_H