readerWriterHelpers.hpp

/*
 * CoDiPack, a Code Differentiation Package
 *
 * Copyright (C) 2015-2024 Chair for Scientific Computing (SciComp), University of Kaiserslautern-Landau
 * Homepage: http://scicomp.rptu.de
 * Contact:  Prof. Nicolas R. Gauger (codi@scicomp.uni-kl.de)
 *
 * Lead developers: Max Sagebaum, Johannes Blühdorn (SciComp, University of Kaiserslautern-Landau)
 *
 * This file is part of CoDiPack (http://scicomp.rptu.de/software/codi).
 *
 * CoDiPack is free software: you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * CoDiPack is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty
 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 *
 * See the GNU General Public License for more details.
 * You should have received a copy of the GNU
 * General Public License along with CoDiPack.
 * If not, see <http://www.gnu.org/licenses/>.
 *
 * For other licensing options please contact us.
 *
 * Authors:
 *  - SciComp, University of Kaiserslautern-Landau:
 *    - Max Sagebaum
 *    - Johannes Blühdorn
 *    - Former members:
 *      - Tim Albring
 */
#pragma once
 
#include <map>
 
#include "../../config.h"
#include "graphWriters.hpp"
#include "jacobianBinaryReaderWriter.hpp"
#include "jacobianTextReaderWriter.hpp"
#include "mathRepWriter.hpp"
#include "primalBinaryReaderWriter.hpp"
#include "primalTextReaderWriter.hpp"
#include "tapeReaderWriterInterface.hpp"
 
namespace codi {
 
  template<typename T, typename... Args>
  std::unique_ptr<T> make_unique_helper(Args&&... args) {
    return std::unique_ptr<T>(new T(std::forward<Args>(args)...));
  }
 
  template<typename T_Type>
  std::unique_ptr<TapeWriterInterface<T_Type>> createWriter(const std::string& fileName,
                                                            std::vector<typename T_Type::Identifier>& inputVariables,
                                                            std::vector<typename T_Type::Identifier>& outputVariables,
                                                            FileType selectedType) {
    using Type = CODI_DD(T_Type, CODI_DEFAULT_LHS_EXPRESSION);  
    using Tape = typename Type::Tape;                           
 
    std::unique_ptr<TapeWriterInterface<T_Type>> writer = nullptr;
    static constexpr bool IsPrimalValueTape = TapeTraits::IsPrimalValueTape<Tape>::value;
 
    switch (selectedType) {
      case FileType::Text:
        if (IsPrimalValueTape) {
          writer = make_unique_helper<codi::PrimalTextTapeWriter<Type>>(fileName, inputVariables, outputVariables, true,
                                                                        true);
        } else {
          writer = make_unique_helper<codi::JacobianTextTapeWriter<Type>>(fileName, inputVariables, outputVariables,
                                                                          true, true);
        }
        break;
 
      case FileType::Binary:
        if (IsPrimalValueTape) {
          writer = make_unique_helper<codi::PrimalBinaryTapeWriter<Type>>(fileName, inputVariables, outputVariables);
        } else {
          writer = make_unique_helper<codi::JacobianBinaryTapeWriter<Type>>(fileName, inputVariables, outputVariables);
        }
        break;
 
      case FileType::Graph:
        if (IsPrimalValueTape) {
          writer = make_unique_helper<codi::PrimalGraphTapeWriter<Type>>(fileName, inputVariables, outputVariables);
        } else {
          writer =
              make_unique_helper<codi::JacobianGraphTapeWriter<Type>>(fileName, inputVariables, outputVariables, true);
        }
        break;
 
      case FileType::Math:
        if (IsPrimalValueTape) {
          writer = make_unique_helper<codi::MathRepWriter<Type>>(fileName, inputVariables, outputVariables);
        } else {
          CODI_EXCEPTION("The MathRepWriter is not supported for Jacobian tapes.");
        }
        break;
 
      default:
        CODI_EXCEPTION("A valid file format was not selected.");
        break;
    }
 
    return writer;
  };
 
  template<typename T_Type>
  std::unique_ptr<TapeReaderInterface<T_Type>> readTapeFile(std::string const& fileName) {
    // This function uses the file extension provided in the fileName to determine the type of TapeReader.
 
    std::unique_ptr<TapeReaderInterface<T_Type>> reader = nullptr;
    std::string fileExtension;  
 
    size_t dotPosition = fileName.find_last_of(".");
    if (dotPosition != std::string::npos) {
      fileExtension = fileName.substr(dotPosition + 1);
    }
 
    if (fileExtension == "txt") {
      reader = make_unique_helper<JacobianTextTapeReader<T_Type>>();
    } else if (fileExtension == "dat") {
      reader = make_unique_helper<JacobianBinaryTapeReader<T_Type>>();
    } else {
      CODI_EXCEPTION("The given file type is not supported.");
    }
    reader->readFile(fileName);
 
    return reader;
  };
 
  template<typename T_Type>
  std::unique_ptr<TapeReaderInterface<T_Type>> readTapeFile(
      std::string const& fileName, std::vector<typename T_Type::Tape::EvalHandle> const& evalHandles) {
    // This function uses the file extension provided in the fileName to determine the type of TapeReader.
 
    std::unique_ptr<TapeReaderInterface<T_Type>> reader = nullptr;
    std::string fileExtension;  
 
    size_t dotPosition = fileName.find_last_of(".");
    if (dotPosition != std::string::npos) {
      fileExtension = fileName.substr(dotPosition + 1);
    }
 
    if (fileExtension == "txt") {
      reader = make_unique_helper<PrimalTextTapeReader<T_Type>>(evalHandles);
    } else if (fileExtension == "dat") {
      reader = make_unique_helper<PrimalBinaryTapeReader<T_Type>>(evalHandles);
    } else {
      CODI_EXCEPTION("The given file type is not supported.");
    }
 
    reader->readFile(fileName);
 
    return reader;
  };
}