ConsensusMapArrowExport.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: Timo Sachsenberg $
// $Authors: Timo Sachsenberg $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/config.h>
 
#include <OpenMS/CONCEPT/Types.h>
#include <OpenMS/KERNEL/ConsensusMap.h>
#include <OpenMS/FORMAT/MSExperimentArrowExport.h>
#include <OpenMS/FORMAT/QPXIdentity.h>
 
#include <memory>
#include <string>
 
// Forward declarations
namespace arrow
{
  class Table;
}
 
namespace OpenMS
{
 
class OPENMS_DLLAPI ConsensusMapArrowExport
{
public:
  static std::shared_ptr<arrow::Table> exportToArrow(const ConsensusMap& cmap,
                                                     QPXIdentity::FeatureLinks* out_links = nullptr);
 
  static void requireUnambiguousIdentities(const ConsensusMap& cmap);
 
  static void requireResolvableIdRuns(const ConsensusMap& cmap);
 
  static bool exportToParquet(
    const ConsensusMap& cmap,
    const std::string& filename,
    const ParquetWriteConfig& config = ParquetWriteConfig{},
    QPXIdentity::FeatureLinks* out_links = nullptr);
 
  static bool exportToParquetStreaming(
    const ConsensusMap& cmap,
    const std::string& filename,
    size_t batch_size = 1000000,
    const ParquetWriteConfig& config = ParquetWriteConfig{},
    int n_threads = 1,
    QPXIdentity::FeatureLinks* out_links = nullptr);
};
 
} // namespace OpenMS