source: icGREP/icgrep-devel/icgrep/kernels/charclasses.cpp @ 5757

Last change on this file since 5757 was 5750, checked in by nmedfort, 19 months ago

Minor code clean up

File size: 3.3 KB
Line 
1/*
2 *  Copyright (c) 2017 International Characters.
3 *  This software is licensed to the public under the Open Software License 3.0.
4 */
5
6#include "charclasses.h"
7#include <re/re_toolchain.h>
8#include <kernels/kernel_builder.h>
9#include <UCD/ucd_compiler.hpp>
10#include <cc/cc_compiler.h>
11#include <re/re_name.h>
12#include <boost/uuid/sha1.hpp>
13#include <llvm/Support/raw_ostream.h>
14
15using NameMap = UCD::UCDCompiler::NameMap;
16
17using namespace cc;
18using namespace kernel;
19using namespace pablo;
20using namespace re;
21using namespace llvm;
22using namespace UCD;
23
24inline static std::string sha1sum(const std::string & str) {
25    char buffer[41];    // 40 hex-digits and the terminating null
26    uint32_t digest[5]; // 160 bits in total
27    boost::uuids::detail::sha1 sha1;
28    sha1.process_bytes(str.c_str(), str.size());
29    sha1.get_digest(digest);
30    snprintf(buffer, sizeof(buffer), "%.8x%.8x%.8x%.8x%.8x",
31             digest[0], digest[1], digest[2], digest[3], digest[4]);
32    return std::string(buffer);
33}
34
35inline std::string signature(const std::vector<re::CC *> & ccs) {
36    if (LLVM_UNLIKELY(ccs.empty())) {
37        return "[]";
38    } else {
39        std::string tmp;
40        raw_string_ostream out(tmp);
41        char joiner = '[';
42        for (const auto & set : ccs) {
43            out << joiner;
44            set->print(out);
45            joiner = ',';
46        }
47        out << ']';
48        return out.str();
49    }
50}
51
52CharClassesSignature::CharClassesSignature(const std::vector<CC *> &ccs)
53: mSignature(signature(ccs)) {
54
55}
56
57
58CharClassesKernel::CharClassesKernel(const std::unique_ptr<kernel::KernelBuilder> & iBuilder, std::vector<CC *> && ccs)
59: CharClassesSignature(ccs)
60, PabloKernel(iBuilder,
61              "cc" + sha1sum(mSignature),
62              {Binding{iBuilder->getStreamSetTy(8), "basis"}},
63              {Binding{iBuilder->getStreamSetTy(ccs.size(), 1), "charclasses"}})
64, mCCs(std::move(ccs)) {
65
66}
67
68std::string CharClassesKernel::makeSignature(const std::unique_ptr<kernel::KernelBuilder> &) {
69    return mSignature;
70}
71
72void CharClassesKernel::generatePabloMethod() {
73    CC_Compiler ccc(this, getInput(0));
74    auto & pb = ccc.getBuilder();
75    unsigned n = mCCs.size();
76
77    NameMap nameMap;
78    std::vector<Name *> names;
79    for (unsigned i = 0; i < n; i++) {
80        Name * name = re::makeName("cc" + std::to_string(i), mCCs[i]);
81        nameMap.emplace(name, nullptr);
82        names.push_back(name);
83    }
84
85    UCD::UCDCompiler ucdCompiler(ccc);
86    if (LLVM_UNLIKELY(AlgorithmOptionIsSet(DisableIfHierarchy))) {
87        ucdCompiler.generateWithoutIfHierarchy(nameMap, pb);
88    } else {
89        ucdCompiler.generateWithDefaultIfHierarchy(nameMap, pb);
90    }
91
92    // The first UnicodeSet in the vector ccs represents the last bit of the character class basis bit streams.
93    std::reverse(names.begin(), names.end());
94    for (unsigned i = 0; i < names.size(); i++) {
95        auto t = nameMap.find(names[i]); 
96        if (t != nameMap.end()) {
97            PabloAST * const r = pb.createExtract(getOutput(0), pb.getInteger(i));
98            if (t->first->getType() == Name::Type::Byte) {
99                pb.createAssign(r, ccc.compileCC(dyn_cast<CC>(t->first->getDefinition())));
100            } else {
101                pb.createAssign(r, t->second);
102            }
103        } else {
104            throw std::runtime_error("Can't compile character classes.");
105        }
106    }
107}
108
109
Note: See TracBrowser for help on using the repository browser.