Files
revng-revng/include/revng/Pipes/ModelGlobal.h
Massimo Fioravanti a8a8da3bae Introduce revng pipes
This commit imports all the revng pipes (and other support utilities) to
be used with `revng-pipeline`. In particular, the pipes necessary for
binary translations have been introduced.
2022-02-08 00:05:03 +01:00

53 lines
1.6 KiB
C++

#pragma once
//
// This file is distributed under the MIT License. See LICENSE.md for details.
//
#include "revng/Model/Binary.h"
#include "revng/Model/LoadModelPass.h"
#include "revng/Pipeline/ContainerSet.h"
#include "revng/Pipeline/Context.h"
#include "revng/Pipeline/Contract.h"
#include "revng/Pipeline/GenericLLVMPipe.h"
#include "revng/Pipeline/LLVMContainer.h"
#include "revng/Pipeline/Loader.h"
#include "revng/Pipeline/SavableObject.h"
#include "revng/Pipeline/Target.h"
#include "revng/TupleTree/TupleTree.h"
namespace revng::pipes {
class ModelGlobal : public pipeline::SavableObject<ModelGlobal> {
private:
ModelWrapper Model;
public:
constexpr static const char *Name = "model.yml";
static const char ID;
llvm::Error storeToDisk(llvm::StringRef Path) const final;
llvm::Error loadFromDisk(llvm::StringRef Path) final;
explicit ModelGlobal(ModelWrapper Model) : Model(std::move(Model)) {}
ModelGlobal() = default;
const ModelWrapper &getModelWrapper() const { return Model; }
ModelWrapper &getModelWrapper() { return Model; }
};
inline const model::Binary &getModelFromContext(const pipeline::Context &Ctx) {
using Wrapper = ModelGlobal;
const auto &Model = llvm::cantFail(Ctx.getGlobal<Wrapper>(Wrapper::Name));
return Model->getModelWrapper().getReadOnlyModel();
}
inline TupleTree<model::Binary> &
getWritableModelFromContext(pipeline::Context &Ctx) {
using Wrapper = ModelGlobal;
const auto &Model = llvm::cantFail(Ctx.getGlobal<Wrapper>(Wrapper::Name));
return Model->getModelWrapper().getWriteableModel();
}
} // namespace revng::pipes