extract_simplifier.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_PARDE_LOWERED_EXTRACT_SIMPLIFIER_H_
#define BACKENDS_TOFINO_BF_P4C_PARDE_LOWERED_EXTRACT_SIMPLIFIER_H_
 
#include <boost/range/adaptor/reversed.hpp>
 
#include "backends/tofino/bf-p4c/parde/clot/clot_info.h"
#include "backends/tofino/bf-p4c/parde/lowered/helpers.h"
#include "backends/tofino/bf-p4c/phv/phv_fields.h"
 
namespace Parde::Lowered {
 
struct ExtractSimplifier {
    const PhvInfo &phv;
    const ClotInfo &clot;
 
    ExtractSimplifier(const PhvInfo &phv, const ClotInfo &clot) : phv(phv), clot(clot) {}
 
    using ExtractSequence = std::vector<const IR::BFN::LoweredExtractPhv *>;
 
    void add(const IR::BFN::Extract *extract);
    void add(const IR::BFN::ExtractClot *extract);
    void add(const IR::BFN::ExtractPhv *extract);
 
    template <typename InputBufferRValType>
    static const IR::BFN::LoweredExtractPhv *createMergedExtract(PHV::Container container,
                                                                 const ExtractSequence &extracts,
                                                                 nw_byteinterval bufferRange);
    template <typename InputBufferRValType>
    static const ExtractSequence mergeExtractsFor(PHV::Container container,
                                                  const ExtractSequence &extracts);
 
    void sortExtractPhvs(IR::Vector<IR::BFN::LoweredParserPrimitive> &loweredExtracts);
 
    const IR::BFN::LoweredExtractPhv *mergeExtractsForConstants(PHV::Container container,
                                                                const ExtractSequence &extracts);
 
    IR::Vector<IR::BFN::LoweredParserPrimitive> lowerExtracts(
        std::map<gress_t, std::map<unsigned, unsigned>> clotTagToCsumUnit);
 
    ordered_map<PHV::Container, ExtractSequence> extractFromPacketByContainer;
    ordered_map<PHV::Container, ExtractSequence> extractFromBufferByContainer;
    ordered_map<PHV::Container, ExtractSequence> extractConstantByContainer;
 
    ordered_map<Clot *, std::vector<const IR::BFN::ExtractClot *>> clotExtracts;
};
 
}  // namespace Parde::Lowered
 
#endif /* BACKENDS_TOFINO_BF_P4C_PARDE_LOWERED_EXTRACT_SIMPLIFIER_H_ */