MatrixMarkerToBinary.cc

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2026 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
/*
 * This file is based on the work on AMGCL library (version march 2026)
 * which can be found at https://github.com/ddemidov/amgcl.
 *
 * Copyright (c) 2012-2022 Denis Demidov <dennis.demidov@gmail.com>
 * SPDX-License-Identifier: MIT
 */
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include <iostream>
#include <string>
#include <complex>
 
#include <boost/program_options.hpp>
#include "arccore/alina/AlinaUtils.h"
#include "arccore/alina/ValueTypeComplex.h"
#include "arccore/alina/IO.h"
 
using namespace Arcane;
namespace io = Alina::IO;
namespace po = boost::program_options;
using Alina::precondition;
 
//---------------------------------------------------------------------------
template <class T>
void convert(Alina::IO::mm_reader& ifile, const std::string& ofile)
{
  std::ofstream f(ofile, std::ios::binary);
  precondition(f, "Failed to open output file for writing.");
 
  if (ifile.is_sparse()) {
    size_t rows, cols;
    std::vector<ptrdiff_t> ptr, col;
    std::vector<T> val;
 
    std::tie(rows, cols) = ifile(ptr, col, val);
 
    precondition(io::write(f, rows), "File I/O error.");
    precondition(io::write(f, ptr), "File I/O error.");
    precondition(io::write(f, col), "File I/O error.");
    precondition(io::write(f, val), "File I/O error.");
 
    std::cout
    << "Wrote " << rows << " by " << cols << " sparse matrix, "
    << ptr.back() << " nonzeros" << std::endl;
  }
  else {
    size_t rows, cols;
    std::vector<T> val;
 
    std::tie(rows, cols) = ifile(val);
 
    precondition(io::write(f, rows), "File I/O error.");
    precondition(io::write(f, cols), "File I/O error.");
    precondition(io::write(f, val), "File I/O error.");
 
    std::cout
    << "Wrote " << rows << " by " << cols << " dense matrix"
    << std::endl;
  }
}
 
//---------------------------------------------------------------------------
int main(int argc, char* argv[])
{
  po::options_description desc("Options");
 
  desc.add_options()("help,h", "Show this help.")("input,i", po::value<std::string>()->required(),
                                                  "Input matrix in the MatrixMarket format.")("output,o", po::value<std::string>()->required(),
                                                                                              "Output binary file.");
 
  po::variables_map vm;
  po::store(po::parse_command_line(argc, argv, desc), vm);
 
  if (vm.count("help")) {
    std::cout << desc << std::endl;
    return 0;
  }
 
  po::notify(vm);
 
  io::mm_reader read(vm["input"].as<std::string>());
  precondition(!read.is_integer(), "Integer matrices are not supported!");
 
  if (read.is_complex()) {
    convert<std::complex<double>>(read, vm["output"].as<std::string>());
  }
  else {
    convert<double>(read, vm["output"].as<std::string>());
  }
}