Tree.h

Go to the documentation of this file.

/*
 * Copyright 2025 Yeong-won Seo
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
 
#include <deque>
 
namespace mllif::mlir {

    class Node final {
        std::string _tag;
        std::string _name;
        std::vector<std::pair<std::string, std::string>> _attributes;
        std::vector<Node> _children;
 
        static std::string Escape(std::string s) {
            std::stringstream ss;
            for (const auto c : s) {
                if (!std::isprint(c)) {
                    ss << "&#" << static_cast<int>(c) << ';';
                }
 
                switch (c) {
                case '"':
                    ss << "&quot;";
                    break;
                case '\'':
                    ss << "&apos;";
                    break;
                case '<':
                    ss << "&lt;";
                    break;
                case '>':
                    ss << "&gt;";
                    break;
                case '&':
                    ss << "&amp;";
                    break;
                default:
                    ss << c;
                    break;
                }
            }
            return ss.str();
        }
 
      public:
        explicit Node(std::string tag, std::string name) : _tag(std::move(tag)), _name(std::move(name)) {}

        std::string &tag() { return _tag; }

        std::string tag() const { return _tag; }

        std::string &name() { return _name; }

        std::string name() const { return _name; }

        std::vector<std::pair<std::string, std::string>> &attributes() { return _attributes; }

        const std::vector<std::pair<std::string, std::string>> &attributes() const { return _attributes; }

        std::vector<Node> &children() { return _children; }

        const std::vector<Node> &children() const { return _children; }

        void print(llvm::raw_ostream &os) const;

        Node *insert_inplace(std::deque<std::string> &path, const std::string &tag);

        Node *insert(std::deque<std::string> &path, const std::string &tag);
    };

    class Tree {
        Node _root;
 
      public:
        Tree() : _root("assembly", "") {}

        Node &root() { return _root; }
    };
 
} // namespace mllif::mlir