MascotInfile.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: Nico Pfeifer $
// $Authors: $
// --------------------------------------------------------------------------

#ifndef OPENMS_FORMAT_MASCOTINFILE_H
#define OPENMS_FORMAT_MASCOTINFILE_H

#include <OpenMS/KERNEL/MSExperiment.h>
#include <OpenMS/DATASTRUCTURES/String.h>
#include <OpenMS/SYSTEM/File.h>
#include <OpenMS/CONCEPT/ProgressLogger.h>
#include <OpenMS/KERNEL/StandardTypes.h>

#include <vector>
#include <fstream>

namespace OpenMS
{
  class OPENMS_DLLAPI MascotInfile :
    public ProgressLogger
  {
public:

    MascotInfile();

    virtual ~MascotInfile();

    void store(const String & filename, const PeakSpectrum & spec, DoubleReal mz, DoubleReal retention_time, String search_title);

    void store(const String & filename, const MSExperiment<> & experiment, String search_title);

    template <typename MapType>
    void load(const String & filename, MapType & exp)
    {
      exp.reset();
      if (!File::exists(filename))
      {
        throw Exception::FileNotFound(__FILE__, __LINE__, __PRETTY_FUNCTION__, filename);
      }

      std::ifstream is(filename.c_str());
      std::vector<std::pair<double, double> > spec;
      UInt charge(0);
      double pre_mz(0), pre_int(0), rt(-1);
      String title;
      while (getNextSpectrum_(is, spec, charge, pre_mz, pre_int, rt, title))
      {
        typename MapType::SpectrumType spectrum;
        for (std::vector<std::pair<double, double> >::const_iterator it = spec.begin(); it != spec.end(); ++it)
        {
          typename MapType::PeakType p;
          p.setPosition(it->first);
          p.setIntensity(it->second);
          spectrum.push_back(p);
        }
        spectrum.setMSLevel(2);
        spectrum.getPrecursors().resize(1);
        spectrum.getPrecursors()[0].setMZ(pre_mz);
        spectrum.getPrecursors()[0].setIntensity(pre_int);
        spectrum.getPrecursors()[0].setCharge(charge);
        spectrum.setRT(rt);
        if (title != "")
        {
          spectrum.setMetaValue("TITLE", title);
          title = "";
        }

        exp.push_back(spectrum);

        // clean up
        spec.clear();
        charge = 0;
        pre_mz = 0;
        pre_int = 0;
      }
    }

    const String & getBoundary();
    void setBoundary(const String & boundary);

    const String & getDB();
    void setDB(const String & db);

    const String & getSearchType();
    void setSearchType(const String & search_type);

    const String & getHits();
    void setHits(const String & hits);

    const String & getCleavage();
    void setCleavage(const String & cleavage);

    const String & getMassType();
    void setMassType(const String & mass_type);

    const std::vector<String> & getModifications();
    void setModifications(const std::vector<String> & mods);

    const std::vector<String> & getVariableModifications();
    void setVariableModifications(const std::vector<String> & mods);

    const String & getInstrument();
    void setInstrument(const String & instrument);

    UInt getMissedCleavages();
    void setMissedCleavages(UInt missed_cleavages);

    Real getPrecursorMassTolerance();
    void setPrecursorMassTolerance(Real precursor_mass_tolerance);

    Real getPeakMassTolerance();
    void setPeakMassTolerance(Real ion_mass_tolerance);

    const String & getTaxonomy();
    void setTaxonomy(const String & taxonomy);

    const String & getFormVersion();
    void setFormVersion(const String & form_version);

    const String & getCharges();
    void setCharges(std::vector<Int> & charges);

protected:
    DoubleReal mz_;

    String charges_;

    String search_title_;

    String db_;

    String search_type_;

    String hits_;

    String cleavage_;

    String mass_type_;

    std::vector<String> mods_;

    std::vector<String> variable_mods_;

    String instrument_;

    UInt missed_cleavages_;

    Real precursor_mass_tolerance_;

    Real ion_mass_tolerance_;

    String taxonomy_;

    String form_version_;

    String boundary_;

    DoubleReal retention_time_;

    void writeParameterHeader_(const String & name, FILE * fp, bool line_break = true);

    void writeHeader_(FILE * fp);

    void writeSpectrum_(FILE * fp,
                        const String & filename,
                        const PeakSpectrum & peaks);

    void writeMSExperiment_(FILE * fp,
                            const String & filename,
                            const MSExperiment<> & experiment);

    bool getNextSpectrum_(std::istream & is, std::vector<std::pair<double, double> > & spectrum, UInt & charge, double & precursor_mz, double & precursor_int, double & rt, String & title);
  };

} // namespace OpenMS

#endif // OPENMS_FORMAT_MASCOTINFILE_H