primalBaseReaderWriter.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 "../../misc/demangleName.hpp"
#include "commonReaderWriterBase.hpp"
 
namespace codi {
  template<typename T_Type>
  struct PrimalBaseTapeWriter : public CommonBaseTapeWriter<T_Type> {
      using Type = CODI_DD(T_Type, CODI_DEFAULT_LHS_EXPRESSION);  
      using Tape = typename Type::Tape;                           
 
      using Identifier = typename Type::Identifier;  
      using Real = typename Type::Real;              
      using EvalHandle = typename Tape::EvalHandle;  
 
      std::map<EvalHandle, size_t> existingEvalHandles;  
      size_t evalHandleCount;                            
      std::vector<std::string> evalHandleStatements;     
 
      PrimalBaseTapeWriter(std::string const& name, std::vector<Identifier> const& in,
                           std::vector<Identifier> const& out)
          : CommonBaseTapeWriter<T_Type>(name, in, out),
            existingEvalHandles(),
            evalHandleCount(0),
            evalHandleStatements() {};
 
    protected:
 
      void generateHandleCreatorFile() {
        FILE* fileHandleCreator = nullptr;
        std::string handleCreatorFileName = this->modifyFileName(".hpp");
        this->openFile(fileHandleCreator, handleCreatorFileName, "w");
 
        std::string functionName = this->modifyFileName("CreateEvalHandles");
        std::string fileHeader =
            "#include <codi.hpp>\n\n"
            "template <typename Tape>\n"
            "std::vector<typename Tape::EvalHandle> " +
            functionName +
            "(){\n\n"
            "  std::vector<typename Tape::EvalHandle> evalHandles;\n"
            "  using Impl = ";
 
        // Add the Impl type to the fileHeader
        fileHeader += demangleName<Tape>() + ";\n\n";
        // Resize the evalHandles
        fileHeader += "  evalHandles.resize(" + std::to_string(evalHandleCount) + ");\n";
 
        // Stmt strings
        std::string frontOfStmt = "  evalHandles[";
        std::string backOfStmt = "] = Tape::StatementEvaluator::template createHandle<";
 
        // Create fileFooter
        std::string fileFooter = "return evalHandles;\n}";
 
        // Header print
        fprintf(fileHandleCreator, "%s", fileHeader.c_str());
 
        // Use CreateHandle for all the statements stored in the lookup table.
        for (size_t handleCounter = 0; handleCounter < evalHandleStatements.size(); handleCounter++) {
          fprintf(fileHandleCreator, "%s%zu%s%s>();\n", frontOfStmt.c_str(), handleCounter, backOfStmt.c_str(),
                  evalHandleStatements[handleCounter].c_str());
        }
        // Footer print
        fprintf(fileHandleCreator, "\n  %s", fileFooter.c_str());
      }
 
      size_t getEvalHandleIndex(EvalHandle const evalHandle, std::string const& evalStatement) {
        auto it = existingEvalHandles.find(evalHandle);
        if (it != existingEvalHandles.end()) {
          return it->second;
        } else {
          existingEvalHandles[evalHandle] = evalHandleCount;
          evalHandleStatements.push_back(evalStatement);
          evalHandleCount += 1;
          return evalHandleCount - 1;
        }
      }
  };
 
}