collapse_to_depth.h

Go to the documentation of this file.

#pragma once
 
#include "sdfg/structured_control_flow/map.h"
#include "sdfg/transformations/transformation.h"
 
namespace sdfg {
namespace transformations {

class CollapseToDepth : public Transformation {
    structured_control_flow::Map& loop_;
    size_t target_loops_;
    bool applied_ = false;
 
    structured_control_flow::Map* outer_loop_ = nullptr;
    structured_control_flow::Map* inner_loop_ = nullptr;
 
public:
    CollapseToDepth(structured_control_flow::Map& loop, size_t target_loops);
 
    virtual std::string name() const override;
 
    virtual bool
    can_be_applied(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;
 
    virtual void apply(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;
 
    virtual void to_json(nlohmann::json& j) const override;
 
    static CollapseToDepth from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);

    structured_control_flow::Map* outer_loop();

    structured_control_flow::Map* inner_loop();
};
 
} // namespace transformations
} // namespace sdfg