if_else.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <optional>
 
#include "sdfg/structured_control_flow/sequence.h"
#include "sdfg/symbolic/symbolic.h"
 
namespace sdfg {
 
namespace builder {
class StructuredSDFGBuilder;
}
 
namespace structured_control_flow {

class IfElse : public ControlFlowNode {
    friend class sdfg::builder::StructuredSDFGBuilder;
 
private:
    std::vector<std::unique_ptr<Sequence>> cases_;
    std::vector<symbolic::Condition> conditions_;
    // Lazily computed; reset whenever `conditions_` changes.
    mutable std::optional<bool> is_complete_;
 
    IfElse(size_t element_id, const DebugInfo& debug_info, ControlFlowNode* parent);
 
    bool compute_is_complete() const;
 
    static constexpr size_t REQUIRED_ELEMENT_IDS = 1;
 
public:
    IfElse(const IfElse& node) = delete;
    IfElse& operator=(const IfElse&) = delete;
 
    ElementType type_id() const override {
        return ElementType::IfElse;
    }
 
    static bool classof(const Element& element) {
        return element.type_id() == ElementType::IfElse;
    }
 
    bool accept(visitor::ActualStructuredSDFGVisitor& visitor) override;
 
    void validate(const Function& function) const override;

    size_t size() const;

    std::pair<const Sequence&, const symbolic::Condition> at(size_t i) const;

    std::pair<Sequence&, const symbolic::Condition> at(size_t i);

    int index(const Sequence& child) const;

    bool is_complete() const;

    void replace(const symbolic::Expression old_expression, const symbolic::Expression new_expression) override;
    void replace(const symbolic::ExpressionMapping& replacements) override;
};
 
} // namespace structured_control_flow
} // namespace sdfg