structInitializers.h

/*
 * Copyright 2018 VMware, Inc.
 * SPDX-FileCopyrightText: 2018 VMware, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef FRONTENDS_P4_STRUCTINITIALIZERS_H_
#define FRONTENDS_P4_STRUCTINITIALIZERS_H_
 
#include "frontends/common/resolveReferences/resolveReferences.h"
#include "frontends/p4/typeChecking/typeChecker.h"
#include "ir/ir.h"
 
namespace P4 {

class CreateStructInitializers : public Transform, public ResolutionContext {
    TypeMap *typeMap;
 
 public:
    explicit CreateStructInitializers(TypeMap *typeMap) : typeMap(typeMap) {
        setName("CreateStructInitializers");
        CHECK_NULL(typeMap);
    }
 
    const IR::Node *postorder(IR::AssignmentStatement *statement) override;
    const IR::Node *postorder(IR::MethodCallExpression *expression) override;
    const IR::Node *postorder(IR::Operation_Relation *expression) override;
    const IR::Node *postorder(IR::Declaration_Variable *statement) override;
    const IR::Node *postorder(IR::ReturnStatement *statement) override;
};
 
class StructInitializers : public PassManager {
 public:
    explicit StructInitializers(TypeMap *typeMap) {
        setName("StructInitializers");
        passes.push_back(new TypeChecking(nullptr, typeMap));
        passes.push_back(new CreateStructInitializers(typeMap));
        passes.push_back(new ClearTypeMap(typeMap));
    }
};
 
}  // namespace P4
 
#endif /* FRONTENDS_P4_STRUCTINITIALIZERS_H_ */