FeatureMapArrowIO.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>
 
#ifdef WITH_PARQUET
 
#include <OpenMS/CONCEPT/Types.h>
#include <OpenMS/KERNEL/FeatureMap.h>
#include <OpenMS/FORMAT/MSExperimentArrowExport.h>
 
#include <memory>
 
// Forward declarations
namespace arrow
{
  class Table;
}
 
namespace OpenMS
{
 
class OPENMS_DLLAPI FeatureMapArrowIO
{
public:
  // ==================== Export methods ====================
 
  static std::shared_ptr<arrow::Table> exportFeaturesToArrow(
    const FeatureMap& feature_map);
 
  static std::shared_ptr<arrow::Table> exportPSMsToArrow(
    const FeatureMap& feature_map);
 
  static bool exportToParquet(
    const FeatureMap& feature_map,
    const String& directory,
    const ParquetWriteConfig& config = ParquetWriteConfig{});
 
  // ==================== Import methods ====================
 
  static bool importFeaturesFromArrow(
    const std::shared_ptr<arrow::Table>& table,
    FeatureMap& feature_map);
 
  static bool importPSMsFromArrow(
    const std::shared_ptr<arrow::Table>& table,
    FeatureMap& feature_map);
 
  static bool importFromParquet(
    const String& directory,
    FeatureMap& feature_map);
};
 
} // namespace OpenMS
 
#endif // WITH_PARQUET