zombiefuck/src/Optimizer.h

#ifndef ZP_OPTIMIZER_H
#define ZP_OPTIMIZER_H

#include "Parser.h"

#include <unordered_map>
#include <memory>

namespace zp
{
    // forward declarations
    enum class Instruction : char;

    struct Block;
    struct InstructionBlock;
    struct UnionBlock;
    struct Loop;

    class AstVisitor;

    struct BlockInstruction;
    struct UnionBlockInstruction;
    struct IOInstruction;
    struct SimpleBlockInstruction;
    struct LinearLoopInstruction;
    struct LoopInstruction;

    class Optimizer;

    using Index = int;
    using Cell = unsigned char;
}


class zp::AstVisitor
{
public:
    std::unique_ptr<BlockInstruction> visit(Block& block);

    virtual std::unique_ptr<BlockInstruction>
        visitLoop(Loop& loop) = 0;
    virtual std::unique_ptr<BlockInstruction>
        visitInstructionBlock(InstructionBlock& ib) = 0;
    virtual std::unique_ptr<BlockInstruction>
        visitIOBlock(IOBlock& io) = 0;
    virtual std::unique_ptr<BlockInstruction>
        visitUnionBlock(UnionBlock& ub) = 0;
};


struct zp::BlockInstruction
{
    virtual ~BlockInstruction(void) = default;
};


struct zp::UnionBlockInstruction :
    public BlockInstruction
{
    std::vector<std::unique_ptr<BlockInstruction>> content;
};


struct zp::IOInstruction :
    public BlockInstruction
{
    /// true on , false on .
    bool read;
    inline BlockInstruction(bool read) : read{read} {}
};


struct zp::SimpleBlockInstruction :
    public BlockInstruction
{
    Index pointerMoved;
    std::unordered_map<Index, Cell> added;
};


struct zp::LinearLoopInstruction :
    public BlockInstruction
{
};


struct zp::LoopInstruction :
    public UnionBlockInstruction
{
};


class zp::Optimizer :
    public AstVisitor
{
public:
    std::unique_ptr<BlockInstruction> visitLoop(Loop& loop) override;
    std::unique_ptr<BlockInstruction> visitInstructionBlock(InstructionBlock& ib) override;
    std::unique_ptr<BlockInstruction> visitIOBlock(IOBlock& io) override;
    std::unique_ptr<BlockInstruction> visitUnionBlock(UnionBlock& ub) override;

};


#endif /* ZP_OPTIMIZER_H */