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

Last change on this file was 5801, checked in by cameron, 4 weeks ago

Additional Alphabet analysis and transformation

File size: 5.2 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_group.h>
11#include <re/re_analysis.h>
12#include <re/re_memoizer.hpp>
13#include <re/printer_re.h>
14#include <UCD/ucd_compiler.hpp>
15#include <UCD/resolve_properties.h>
16#include <boost/container/flat_set.hpp>
17#include <cc/multiplex_CCs.h>
18#include <sstream>
19#include <iostream>
20#include <functional>
21#include <llvm/Support/raw_ostream.h>
22
23using namespace boost::container;
24using namespace llvm;
25
26namespace re {
27 
28RE * multiplex(RE * const re,
29               const std::vector<const CC *> & UnicodeSets,
30               const std::vector<std::vector<unsigned>> & exclusiveSetIDs) {
31
32    Memoizer memoizer;
33
34    std::function<RE *(RE *)> multiplex = [&](RE * const re) -> RE * {
35        if (CC * cc = dyn_cast<CC>(re)) {
36            const auto index = find(UnicodeSets.begin(), UnicodeSets.end(), cc) - UnicodeSets.begin();
37            const auto exclusive_IDs = exclusiveSetIDs[index];
38            CC * CC_union = makeCC();
39            for (auto i : exclusive_IDs) {
40                CC_union = makeCC(CC_union, makeCC(i));
41            }
42            return CC_union;
43        } else if (Name * name = dyn_cast<Name>(re)) {
44            auto f = memoizer.find(name);
45            if (f == memoizer.end()) {
46                if (LLVM_LIKELY(name->getDefinition() != nullptr)) {
47                    if (CC * cc = dyn_cast<CC>(name->getDefinition())) {
48                        const auto index = find(UnicodeSets.begin(), UnicodeSets.end(), cc) - UnicodeSets.begin();
49                        const auto exclusive_IDs = exclusiveSetIDs[index];
50                        CC * CC_union = makeCC();
51                        for (auto i : exclusive_IDs) {
52                            CC_union = makeCC(CC_union, makeCC(i));
53                        }
54                        name->setDefinition(CC_union);
55                    } else {
56                        multiplex(name->getDefinition());
57                    }
58                } else {
59                    UndefinedNameError(name);
60                }
61                return memoizer.memoize(name);
62            } else {
63                return *f;
64            }
65        } else if (Seq * seq = dyn_cast<Seq>(re)) {
66            for (auto si = seq->begin(); si != seq->end(); ++si) {
67                *si = multiplex(*si);
68            }
69        } else if (Alt * alt = dyn_cast<Alt>(re)) {
70            for (auto ai = alt->begin(); ai != alt->end(); ++ai) {
71                *ai = multiplex(*ai);
72            }
73        } else if (Rep * rep = dyn_cast<Rep>(re)) {
74            rep->setRE(multiplex(rep->getRE()));
75        } else if (Assertion * a = dyn_cast<Assertion>(re)) {
76            a->setAsserted(multiplex(a->getAsserted()));
77        } else if (Diff * diff = dyn_cast<Diff>(re)) {
78            diff->setLH(multiplex(diff->getLH()));
79            diff->setRH(multiplex(diff->getRH()));
80        } else if (Intersect * ix = dyn_cast<Intersect>(re)) {
81            ix->setLH(multiplex(ix->getLH()));
82            ix->setRH(multiplex(ix->getRH()));
83        } else if (Group * g = dyn_cast<Group>(re)) {
84            g->setRE(multiplex(g->getRE()));
85        }
86        return re;
87    };
88
89    return multiplex(re);
90}   
91
92
93RE * transformCCs(cc::MultiplexedAlphabet * mpx, RE * re) {
94    if (CC * cc = dyn_cast<CC>(re)) {
95        if (cc->getAlphabet() == mpx->getSourceAlphabet()) {
96            re = mpx->transformCC(cc);
97        }
98    } else if (Name * name = dyn_cast<Name>(re)) {
99        if (LLVM_LIKELY(name->getDefinition() != nullptr)) {
100            RE * xfrm = transformCCs(mpx, name->getDefinition());
101            if (name->getType() == Name::Type::ZeroWidth)
102                re = makeZeroWidth(name->getName(), xfrm);
103            else
104                re = makeName(name->getName(), xfrm);
105        } else {
106            UndefinedNameError(name);
107        }
108    } else if (Seq * seq = dyn_cast<Seq>(re)) {
109        std::vector<RE *> list;
110        list.reserve(seq->size());
111        for (RE * item : *seq) {
112            item = transformCCs(mpx, item);
113            list.push_back(item);
114        }
115        re = makeSeq(list.begin(), list.end());
116    } else if (Alt * alt = dyn_cast<Alt>(re)) {
117        std::vector<RE *> list;
118        list.reserve(alt->size());
119        for (RE * item : *alt) {
120            item = transformCCs(mpx, item);
121            list.push_back(item);
122        }
123        re = makeAlt(list.begin(), list.end());
124    } else if (Assertion * a = dyn_cast<Assertion>(re)) {
125        re = makeAssertion(transformCCs(mpx, a->getAsserted()), a->getKind(), a->getSense());
126    } else if (Rep * rep = dyn_cast<Rep>(re)) {
127        RE * expr = transformCCs(mpx, rep->getRE());
128        re = makeRep(expr, rep->getLB(), rep->getUB());
129    } else if (Diff * diff = dyn_cast<Diff>(re)) {
130        re = makeDiff(transformCCs(mpx, diff->getLH()), transformCCs(mpx, diff->getRH()));
131    } else if (Intersect * e = dyn_cast<Intersect>(re)) {
132        re = makeIntersect(transformCCs(mpx, e->getLH()), transformCCs(mpx, e->getRH()));
133    } else if (Group * g = dyn_cast<Group>(re)) {
134        re = makeGroup(g->getMode(), transformCCs(mpx, g->getRE()), g->getSense());
135    }
136    return re;
137};
138
139
140}
Note: See TracBrowser for help on using the repository browser.