/
Fixpoint.h
84 lines (70 loc) · 2.45 KB
/
Fixpoint.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/*
* Souffle - A Datalog Compiler
* Copyright (c) 2018, The Souffle Developers. All rights reserved
* Licensed under the Universal Permissive License v 1.0 as shown at:
* - https://opensource.org/licenses/UPL
* - <souffle root>/licenses/SOUFFLE-UPL.txt
*/
/************************************************************************
*
* @file Fixpoint.h
*
* Transformer that repeatedly executes a sub-transformer until no changes are made
*
***********************************************************************/
#pragma once
#include "DebugReporter.h"
#include "ast/transform/Transformer.h"
#include <memory>
#include <set>
#include <string>
#include <utility>
#include <vector>
namespace souffle {
class AstTranslationUnit;
/**
* Transformer that repeatedly executes a sub-transformer until no changes are made
*/
class FixpointTransformer : public MetaTransformer {
public:
FixpointTransformer(std::unique_ptr<AstTransformer> transformer) : transformer(std::move(transformer)) {}
void setDebugReport() override {
if (auto* mt = dynamic_cast<MetaTransformer*>(transformer.get())) {
mt->setDebugReport();
} else {
transformer = std::make_unique<DebugReporter>(std::move(transformer));
}
}
std::vector<AstTransformer*> getSubtransformers() const override {
return {transformer.get()};
}
void setVerbosity(bool verbose) override {
this->verbose = verbose;
if (auto* mt = dynamic_cast<MetaTransformer*>(transformer.get())) {
mt->setVerbosity(verbose);
}
}
void disableTransformers(const std::set<std::string>& transforms) override {
if (auto* mt = dynamic_cast<MetaTransformer*>(transformer.get())) {
mt->disableTransformers(transforms);
} else if (transforms.find(transformer->getName()) != transforms.end()) {
transformer = std::make_unique<NullTransformer>();
}
}
std::string getName() const override {
return "FixpointTransformer";
}
FixpointTransformer* clone() const override {
return new FixpointTransformer(souffle::clone(transformer));
}
private:
std::unique_ptr<AstTransformer> transformer;
bool transform(AstTranslationUnit& translationUnit) override {
bool changed = false;
while (applySubtransformer(translationUnit, transformer.get())) {
changed = true;
}
return changed;
}
};
} // end of namespace souffle