Changeset 59c8dff for src/AST

Timestamp:

Jan 19, 2024, 2:42:58 AM (2 years ago)

Author:

JiadaL <j82liang@…>

Branches:

master

Children:

f988834

Parents:

8b4faf6

Message:

Draft Implementation for enum position pesudo function (posE). EnumPosExpr is mostly irrelevant for now. It is used in development/code probing and will be removed later

Location:

src/AST

Files:

• Expr.cpp (modified) (1 diff)
• Expr.hpp (modified) (1 diff)
• Fwd.hpp (modified) (1 diff)
• Node.cpp (modified) (1 diff)
• Pass.hpp (modified) (1 diff)
• Pass.impl.hpp (modified) (1 diff)
• Print.cpp (modified) (1 diff)
• Visitor.hpp (modified) (1 diff)

src/AST/Expr.cpp

@@ -307 +307 @@
 }
 
+// 
+
+// --- EnumPosExpr
+EnumPosExpr::EnumPosExpr( const CodeLocation & loc, const EnumInstType * ty)
+: Expr( loc, new BasicType{ BasicType::UnsignedInt }), type( ty ) {
+        assert( ty );
+}
+
+EnumPosExpr::EnumPosExpr( const CodeLocation & loc, const Expr * expr )
+: Expr( loc, new BasicType{ BasicType::UnsignedInt }), expr(expr) {
+        assert( expr );
+}
+
+
 // --- LogicalExpr
 

src/AST/Expr.hpp

@@ -548 +548 @@
 };
 
+class EnumPosExpr final : public Expr {
+public:
+        ptr<EnumInstType> type;
+        ptr<Expr> expr;
+        
+        EnumPosExpr( const CodeLocation & loc, const EnumInstType * ty );
+        EnumPosExpr( const CodeLocation & loc, const Expr * expr );
+        const Expr * accept( Visitor & v ) const override { return v.visit( this ); }
+private:
+        EnumPosExpr * clone() const override { return new EnumPosExpr{ *this }; }
+        MUTATE_FRIEND
+};
+
 /// Variants of short-circuiting logical expression
 enum LogicalFlag { OrExpr, AndExpr };

src/AST/Fwd.hpp

@@ -89 +89 @@
 class OffsetofExpr;
 class OffsetPackExpr;
+class EnumPosExpr;
 class LogicalExpr;
 class ConditionalExpr;

src/AST/Node.cpp

@@ -232 +232 @@
 template class ast::ptr_base< ast::OffsetPackExpr, ast::Node::ref_type::weak >;
 template class ast::ptr_base< ast::OffsetPackExpr, ast::Node::ref_type::strong >;
+template class ast::ptr_base< ast::EnumPosExpr, ast::Node::ref_type::weak >;
+template class ast::ptr_base< ast::EnumPosExpr, ast::Node::ref_type::strong >;
 template class ast::ptr_base< ast::LogicalExpr, ast::Node::ref_type::weak >;
 template class ast::ptr_base< ast::LogicalExpr, ast::Node::ref_type::strong >;

src/AST/Pass.hpp

@@ -191 +191 @@
         const ast::Expr *             visit( const ast::OffsetofExpr         * ) override final;
         const ast::Expr *             visit( const ast::OffsetPackExpr       * ) override final;
+        const ast::Expr *             visit( const ast::EnumPosExpr          * ) override final;
         const ast::Expr *             visit( const ast::LogicalExpr          * ) override final;
         const ast::Expr *             visit( const ast::ConditionalExpr      * ) override final;

src/AST/Pass.impl.hpp

@@ -1452 +1452 @@
 
 //--------------------------------------------------------------------------
+// EnumPosExpr
+template< typename core_t>
+const ast::Expr * ast::Pass< core_t >::visit( const ast::EnumPosExpr * node ) {
+        VISIT_START( node );
+
+        if ( __visit_children() ) {
+                guard_symtab guard { *this };
+                maybe_accept( node, &EnumPosExpr::type );
+                maybe_accept( node, &EnumPosExpr::expr );
+        }
+
+        VISIT_END( Expr, node );
+}
+
+//--------------------------------------------------------------------------
 // LogicalExpr
 template< typename core_t >

src/AST/Print.cpp

@@ -1183 +1183 @@
         }
 
+        virtual const ast::Expr * visit( const ast::EnumPosExpr * node ) override final {
+                os << "Enum Position Expression on: ";
+                ++indent;
+                safe_print( node->type );
+                --indent;
+                postprint( node );
+                return node;
+        }
+
         virtual const ast::Expr * visit( const ast::LogicalExpr * node ) override final {
                 os << "Short-circuited operation (" << (node->isAnd ? "and" : "or") << ") on: ";

src/AST/Visitor.hpp

@@ -79 +79 @@
     virtual const ast::Expr *             visit( const ast::OffsetofExpr         * ) = 0;
     virtual const ast::Expr *             visit( const ast::OffsetPackExpr       * ) = 0;
+    virtual const ast::Expr *             visit( const ast::EnumPosExpr          * ) = 0;
     virtual const ast::Expr *             visit( const ast::LogicalExpr          * ) = 0;
     virtual const ast::Expr *             visit( const ast::ConditionalExpr      * ) = 0;