source: icGREP/icgrep-devel/icgrep/re/re_multiplex.cpp @ 5786

Last change on this file since 5786 was 5784, checked in by cameron, 17 months ago

Restructuring step/tidy-up for re_passes

File size: 3.1 KB
Line 
1#include "re_multiplex.h"
2#include <re/re_name.h>
3#include <re/re_alt.h>
4#include <re/re_cc.h>
5#include <re/re_seq.h>
6#include <re/re_rep.h>
7#include <re/re_diff.h>
8#include <re/re_intersect.h>
9#include <re/re_assertion.h>
10#include <re/re_analysis.h>
11#include <re/re_memoizer.hpp>
12#include <UCD/ucd_compiler.hpp>
13#include <UCD/resolve_properties.h>
14#include <boost/container/flat_set.hpp>
15#include <cc/multiplex_CCs.h>
16#include <sstream>
17#include <iostream>
18#include <functional>
19
20using namespace boost::container;
21using namespace llvm;
22
23namespace re {
24 
25RE * multiplex(RE * const re,
26               const std::vector<const CC *> & UnicodeSets,
27               const std::vector<std::vector<unsigned>> & exclusiveSetIDs) {
28
29    Memoizer memoizer;
30
31    std::function<RE *(RE *)> multiplex = [&](RE * const re) -> RE * {
32        if (CC * cc = dyn_cast<CC>(re)) {
33            const auto index = find(UnicodeSets.begin(), UnicodeSets.end(), cc) - UnicodeSets.begin();
34            const auto exclusive_IDs = exclusiveSetIDs[index];
35            CC * CC_union = makeCC();
36            for (auto i : exclusive_IDs) {
37                CC_union = makeCC(CC_union, makeCC(i));
38            }
39            return CC_union;
40        } else if (Name * name = dyn_cast<Name>(re)) {
41            auto f = memoizer.find(name);
42            if (f == memoizer.end()) {
43                if (LLVM_LIKELY(name->getDefinition() != nullptr)) {
44                    if (CC * cc = dyn_cast<CC>(name->getDefinition())) {
45                        const auto index = find(UnicodeSets.begin(), UnicodeSets.end(), cc) - UnicodeSets.begin();
46                        const auto exclusive_IDs = exclusiveSetIDs[index];
47                        CC * CC_union = makeCC();
48                        for (auto i : exclusive_IDs) {
49                            CC_union = makeCC(CC_union, makeCC(i));
50                        }
51                        name->setDefinition(CC_union);
52                    } else {
53                        multiplex(name->getDefinition());
54                    }
55                } else {
56                    UndefinedNameError(name);
57                }
58                return memoizer.memoize(name);
59            } else {
60                return *f;
61            }
62        } else if (Seq * seq = dyn_cast<Seq>(re)) {
63            for (auto si = seq->begin(); si != seq->end(); ++si) {
64                *si = multiplex(*si);
65            }
66        } else if (Alt * alt = dyn_cast<Alt>(re)) {
67            for (auto ai = alt->begin(); ai != alt->end(); ++ai) {
68                *ai = multiplex(*ai);
69            }
70        } else if (Rep * rep = dyn_cast<Rep>(re)) {
71            rep->setRE(multiplex(rep->getRE()));
72        } else if (Assertion * a = dyn_cast<Assertion>(re)) {
73            a->setAsserted(multiplex(a->getAsserted()));
74        } else if (Diff * diff = dyn_cast<Diff>(re)) {
75            diff->setLH(multiplex(diff->getLH()));
76            diff->setRH(multiplex(diff->getRH()));
77        } else if (Intersect * ix = dyn_cast<Intersect>(re)) {
78            ix->setLH(multiplex(ix->getLH()));
79            ix->setRH(multiplex(ix->getRH()));
80        }
81        return re;
82    };
83
84    return multiplex(re);
85}   
86
87}
Note: See TracBrowser for help on using the repository browser.