loop_shift.h

Go to the documentation of this file.

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

class LoopShift : public Transformation {
    structured_control_flow::StructuredLoop& loop_;
    symbolic::Expression offset_;
 
    std::string shifted_container_name_;
 
public:
    explicit LoopShift(structured_control_flow::StructuredLoop& loop);

    LoopShift(structured_control_flow::StructuredLoop& loop, const symbolic::Expression& offset);
 
    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 LoopShift from_json(builder::StructuredSDFGBuilder& builder, const nlohmann::json& j);

    const std::string& shifted_container_name() const {
        return shifted_container_name_;
    }
};
 
} // namespace transformations
} // namespace sdfg