MSChromatogramParquetConsumer.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: Justin Sing $
// $Authors: Justin Sing $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/FORMAT/DATAACCESS/MSDataWritingConsumer.h>
#include <OpenMS/OPENSWATHALGO/DATAACCESS/TransitionExperiment.h>
 
#include <memory>
 
namespace OpenMS
{
  class MSChromatogramParquetConsumerImpl;
 
  class OPENMS_DLLAPI MSChromatogramParquetConsumer : public Interfaces::IMSDataConsumer
  {
  public:
    MSChromatogramParquetConsumer(const std::string& filename,
                                  UInt64 run_id,
                                  const std::string& source_file,
                                  const OpenSwath::LightTargetedExperiment& transition_exp);
 
    ~MSChromatogramParquetConsumer() override;
 
    void consumeSpectrum(SpectrumType& s) override;
    void consumeChromatogram(ChromatogramType& c) override;
 
    class OPENMS_DLLAPI EncodedChromatograms
    {
    public:
      EncodedChromatograms();
      ~EncodedChromatograms();
      EncodedChromatograms(EncodedChromatograms&&) noexcept;
      EncodedChromatograms& operator=(EncodedChromatograms&&) noexcept;
 
      Size size() const;
 
    private:
      friend class MSChromatogramParquetConsumer;
      struct Rows;
      std::unique_ptr<Rows> rows_;
    };
 
    void encodeChromatogram(ChromatogramType& c, EncodedChromatograms& rows) const;
 
    void consumeEncodedChromatograms(EncodedChromatograms& rows);
    void finalize();
    void setExpectedSize(Size expectedSpectra, Size expectedChromatograms) override;
    void setExperimentalSettings(const ExperimentalSettings& exp) override;
 
  private:
    std::unique_ptr<MSChromatogramParquetConsumerImpl> impl_;
  };
} // namespace OpenMS