revng-revng/tools/model/apply/Main.cpp

/// \file Main.cpp

//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#include <string>

#include "llvm/Support/CommandLine.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/raw_ostream.h"

#include "revng/Model/ToolHelpers.h"
#include "revng/Support/InitRevng.h"
#include "revng/Support/MetaAddress/YAMLTraits.h"
#include "revng/Support/YAMLTraits.h"
#include "revng/TupleTree/TupleTreeDiff.h"

using namespace llvm;

static cl::OptionCategory ThisToolCategory("Tool options", "");

static cl::opt<std::string> PathModel(cl::Positional,
                                      cl::cat(ThisToolCategory),
                                      cl::desc("<input model>"),
                                      cl::value_desc("model"));

static cl::opt<std::string> DiffPath(cl::Positional,
                                     cl::cat(ThisToolCategory),
                                     cl::desc("<model diff>"),
                                     cl::init("-"),
                                     cl::value_desc("model"));

static ModelOutputOptions<false> Options(ThisToolCategory);

int main(int Argc, char *Argv[]) {
  revng::InitRevng X(Argc, Argv, "", { &ThisToolCategory });

  ExitOnError ExitOnError;

  using Type = TupleTree<model::Binary>;
  auto Model = Type::fromFileOrSTDIN(PathModel);
  if (not Model)
    ExitOnError(Model.takeError());

  using TypeDiff = TupleTreeDiff<model::Binary>;
  auto Diff = ExitOnError(fromFileOrSTDIN<TypeDiff>(DiffPath));

  ExitOnError(Diff.apply(*Model));

  auto DesiredOutput = Options.getDesiredOutput();
  ExitOnError(Model->toFile(Options.getPath()));

  return EXIT_SUCCESS;
}