Changeset 34c32f0


Ignore:
Timestamp:
Feb 1, 2022, 12:06:24 PM (23 months ago)
Author:
caparsons <caparson@…>
Branches:
ADT, ast-experimental, enum, forall-pointer-decay, master, pthread-emulation, qualifiedEnum
Children:
ab1a9ea
Parents:
3e5db5b4 (diff), 7b2c8c3c (diff)
Note: this is a merge changeset, the changes displayed below correspond to the merge itself.
Use the (diff) links above to see all the changes relative to each parent.
Message:

Merge branch 'master' of plg.uwaterloo.ca:software/cfa/cfa-cc

Files:
2 added
21 edited

Legend:

Unmodified
Added
Removed
  • libcfa/src/concurrency/kernel.cfa

    r3e5db5b4 r34c32f0  
    4242
    4343#if !defined(__CFA_NO_STATISTICS__)
    44         #define __STATS( ...) __VA_ARGS__
     44        #define __STATS_DEF( ...) __VA_ARGS__
    4545#else
    46         #define __STATS( ...)
     46        #define __STATS_DEF( ...)
    4747#endif
    4848
     
    122122static thread$ * __next_thread(cluster * this);
    123123static thread$ * __next_thread_slow(cluster * this);
     124static thread$ * __next_thread_search(cluster * this);
    124125static inline bool __must_unpark( thread$ * thrd ) __attribute((nonnull(1)));
    125126static void __run_thread(processor * this, thread$ * dst);
     
    187188                MAIN_LOOP:
    188189                for() {
    189                         #define OLD_MAIN 1
    190                         #if OLD_MAIN
    191190                        // Check if there is pending io
    192191                        __maybe_io_drain( this );
     
    196195
    197196                        if( !readyThread ) {
    198                                 __tls_stats()->io.flush.idle++;
     197                                __IO_STATS__(true, io.flush.idle++; )
    199198                                __cfa_io_flush( this, 0 );
    200199
     200                                readyThread = __next_thread( this->cltr );
     201                        }
     202
     203                        if( !readyThread ) for(5) {
     204                                __IO_STATS__(true, io.flush.idle++; )
     205
    201206                                readyThread = __next_thread_slow( this->cltr );
     207
     208                                if( readyThread ) break;
     209
     210                                __cfa_io_flush( this, 0 );
    202211                        }
    203212
     
    211220
    212221                                // Confirm the ready-queue is empty
    213                                 readyThread = __next_thread_slow( this->cltr );
     222                                readyThread = __next_thread_search( this->cltr );
    214223                                if( readyThread ) {
    215224                                        // A thread was found, cancel the halt
    216225                                        mark_awake(this->cltr->procs, * this);
    217226
    218                                         #if !defined(__CFA_NO_STATISTICS__)
    219                                                 __tls_stats()->ready.sleep.cancels++;
    220                                         #endif
     227                                        __STATS__(true, ready.sleep.cancels++; )
    221228
    222229                                        // continue the mai loop
     
    245252
    246253                        if(this->io.pending && !this->io.dirty) {
    247                                 __tls_stats()->io.flush.dirty++;
     254                                __IO_STATS__(true, io.flush.dirty++; )
    248255                                __cfa_io_flush( this, 0 );
    249256                        }
    250 
    251                         #else
    252                                 #warning new kernel loop
    253                         SEARCH: {
    254                                 /* paranoid */ verify( ! __preemption_enabled() );
    255 
    256                                 // First, lock the scheduler since we are searching for a thread
    257                                 ready_schedule_lock();
    258 
    259                                 // Try to get the next thread
    260                                 readyThread = pop_fast( this->cltr );
    261                                 if(readyThread) { ready_schedule_unlock(); break SEARCH; }
    262 
    263                                 // If we can't find a thread, might as well flush any outstanding I/O
    264                                 if(this->io.pending) { __cfa_io_flush( this, 0 ); }
    265 
    266                                 // Spin a little on I/O, just in case
    267                                 for(5) {
    268                                         __maybe_io_drain( this );
    269                                         readyThread = pop_fast( this->cltr );
    270                                         if(readyThread) { ready_schedule_unlock(); break SEARCH; }
    271                                 }
    272 
    273                                 // no luck, try stealing a few times
    274                                 for(5) {
    275                                         if( __maybe_io_drain( this ) ) {
    276                                                 readyThread = pop_fast( this->cltr );
    277                                         } else {
    278                                                 readyThread = pop_slow( this->cltr );
    279                                         }
    280                                         if(readyThread) { ready_schedule_unlock(); break SEARCH; }
    281                                 }
    282 
    283                                 // still no luck, search for a thread
    284                                 readyThread = pop_search( this->cltr );
    285                                 if(readyThread) { ready_schedule_unlock(); break SEARCH; }
    286 
    287                                 // Don't block if we are done
    288                                 if( __atomic_load_n(&this->do_terminate, __ATOMIC_SEQ_CST) ) {
    289                                         ready_schedule_unlock();
    290                                         break MAIN_LOOP;
    291                                 }
    292 
    293                                 __STATS( __tls_stats()->ready.sleep.halts++; )
    294 
    295                                 // Push self to idle stack
    296                                 ready_schedule_unlock();
    297                                 if(!mark_idle(this->cltr->procs, * this)) goto SEARCH;
    298                                 ready_schedule_lock();
    299 
    300                                 // Confirm the ready-queue is empty
    301                                 __maybe_io_drain( this );
    302                                 readyThread = pop_search( this->cltr );
    303                                 ready_schedule_unlock();
    304 
    305                                 if( readyThread ) {
    306                                         // A thread was found, cancel the halt
    307                                         mark_awake(this->cltr->procs, * this);
    308 
    309                                         __STATS( __tls_stats()->ready.sleep.cancels++; )
    310 
    311                                         // continue the main loop
    312                                         break SEARCH;
    313                                 }
    314 
    315                                 __STATS( if(this->print_halts) __cfaabi_bits_print_safe( STDOUT_FILENO, "PH:%d - %lld 0\n", this->unique_id, rdtscl()); )
    316                                 __cfadbg_print_safe(runtime_core, "Kernel : core %p waiting on eventfd %d\n", this, this->idle_fd);
    317 
    318                                 {
    319                                         eventfd_t val;
    320                                         ssize_t ret = read( this->idle_fd, &val, sizeof(val) );
    321                                         if(ret < 0) {
    322                                                 switch((int)errno) {
    323                                                 case EAGAIN:
    324                                                 #if EAGAIN != EWOULDBLOCK
    325                                                         case EWOULDBLOCK:
    326                                                 #endif
    327                                                 case EINTR:
    328                                                         // No need to do anything special here, just assume it's a legitimate wake-up
    329                                                         break;
    330                                                 default:
    331                                                         abort( "KERNEL : internal error, read failure on idle eventfd, error(%d) %s.", (int)errno, strerror( (int)errno ) );
    332                                                 }
    333                                         }
    334                                 }
    335 
    336                                         __STATS( if(this->print_halts) __cfaabi_bits_print_safe( STDOUT_FILENO, "PH:%d - %lld 1\n", this->unique_id, rdtscl()); )
    337 
    338                                 // We were woken up, remove self from idle
    339                                 mark_awake(this->cltr->procs, * this);
    340 
    341                                 // DON'T just proceed, start looking again
    342                                 continue MAIN_LOOP;
    343                         }
    344 
    345                 RUN_THREAD:
    346                         /* paranoid */ verify( ! __preemption_enabled() );
    347                         /* paranoid */ verify( readyThread );
    348 
    349                         // Reset io dirty bit
    350                         this->io.dirty = false;
    351 
    352                         // We found a thread run it
    353                         __run_thread(this, readyThread);
    354 
    355                         // Are we done?
    356                         if( __atomic_load_n(&this->do_terminate, __ATOMIC_SEQ_CST) ) break MAIN_LOOP;
    357 
    358                         if(this->io.pending && !this->io.dirty) {
    359                                 __cfa_io_flush( this, 0 );
    360                         }
    361 
    362                         ready_schedule_lock();
    363                         __maybe_io_drain( this );
    364                         ready_schedule_unlock();
    365                         #endif
    366257                }
    367258
     
    474365                                break RUNNING;
    475366                        case TICKET_UNBLOCK:
    476                                 #if !defined(__CFA_NO_STATISTICS__)
    477                                         __tls_stats()->ready.threads.threads++;
    478                                 #endif
     367                                __STATS__(true, ready.threads.threads++; )
    479368                                // This is case 2, the racy case, someone tried to run this thread before it finished blocking
    480369                                // In this case, just run it again.
     
    491380        __cfadbg_print_safe(runtime_core, "Kernel : core %p finished running thread %p\n", this, thrd_dst);
    492381
    493         #if !defined(__CFA_NO_STATISTICS__)
    494                 __tls_stats()->ready.threads.threads--;
    495         #endif
     382        __STATS__(true, ready.threads.threads--; )
    496383
    497384        /* paranoid */ verify( ! __preemption_enabled() );
     
    504391        thread$ * thrd_src = kernelTLS().this_thread;
    505392
    506         __STATS( thrd_src->last_proc = kernelTLS().this_processor; )
     393        __STATS_DEF( thrd_src->last_proc = kernelTLS().this_processor; )
    507394
    508395        // Run the thread on this processor
     
    556443        // Dereference the thread now because once we push it, there is not guaranteed it's still valid.
    557444        struct cluster * cl = thrd->curr_cluster;
    558         __STATS(bool outside = hint == UNPARK_LOCAL && thrd->last_proc && thrd->last_proc != kernelTLS().this_processor; )
     445        __STATS_DEF(bool outside = hint == UNPARK_LOCAL && thrd->last_proc && thrd->last_proc != kernelTLS().this_processor; )
    559446
    560447        // push the thread to the cluster ready-queue
     
    607494
    608495        ready_schedule_lock();
    609                 thread$ * thrd;
    610                 for(25) {
    611                         thrd = pop_slow( this );
    612                         if(thrd) goto RET;
    613                 }
    614                 thrd = pop_search( this );
    615 
    616                 RET:
     496                thread$ * thrd = pop_slow( this );
     497        ready_schedule_unlock();
     498
     499        /* paranoid */ verify( ! __preemption_enabled() );
     500        return thrd;
     501}
     502
     503// KERNEL ONLY
     504static inline thread$ * __next_thread_search(cluster * this) with( *this ) {
     505        /* paranoid */ verify( ! __preemption_enabled() );
     506
     507        ready_schedule_lock();
     508                thread$ * thrd = pop_search( this );
    617509        ready_schedule_unlock();
    618510
     
    856748
    857749static bool mark_idle(__cluster_proc_list & this, processor & proc) {
    858         #if !defined(__CFA_NO_STATISTICS__)
    859                 __tls_stats()->ready.sleep.halts++;
    860         #endif
     750        __STATS__(true, ready.sleep.halts++; )
    861751
    862752        proc.idle_wctx.fd = 0;
     
    951841                unsigned tail = *ctx->cq.tail;
    952842                if(head == tail) return false;
    953                 #if OLD_MAIN
    954                         ready_schedule_lock();
    955                         ret = __cfa_io_drain( proc );
    956                         ready_schedule_unlock();
    957                 #else
    958                         ret = __cfa_io_drain( proc );
    959                 #endif
     843                ready_schedule_lock();
     844                ret = __cfa_io_drain( proc );
     845                ready_schedule_unlock();
    960846        #endif
    961847        return ret;
  • libcfa/src/concurrency/kernel/fwd.hfa

    r3e5db5b4 r34c32f0  
    396396                                if( !(in_kernel) ) enable_interrupts(); \
    397397                        }
     398                        #if defined(CFA_HAVE_LINUX_IO_URING_H)
     399                                #define __IO_STATS__(in_kernel, ...) { \
     400                                        if( !(in_kernel) ) disable_interrupts(); \
     401                                        with( *__tls_stats() ) { \
     402                                                __VA_ARGS__ \
     403                                        } \
     404                                        if( !(in_kernel) ) enable_interrupts(); \
     405                                }
     406                        #else
     407                                #define __IO_STATS__(in_kernel, ...)
     408                        #endif
    398409                #else
    399410                        #define __STATS__(in_kernel, ...)
     411                        #define __IO_STATS__(in_kernel, ...)
    400412                #endif
    401413        }
  • src/AST/Stmt.hpp

    r3e5db5b4 r34c32f0  
    99// Author           : Aaron B. Moss
    1010// Created On       : Wed May  8 13:00:00 2019
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Fri May 17 12:45:00 2019
    13 // Update Count     : 5
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:38:53 2022
     13// Update Count     : 12
    1414//
    1515
     
    1717
    1818#include <list>
    19 #include <utility>                // for move
     19#include <utility>                                                                              // for move
    2020#include <vector>
    2121
    2222#include "Label.hpp"
    23 #include "Node.hpp"               // for node, ptr
     23#include "Node.hpp"                                                                             // for node, ptr
    2424#include "ParseNode.hpp"
    2525#include "Visitor.hpp"
     
    2727
    2828// Must be included in *all* AST classes; should be #undef'd at the end of the file
    29 #define MUTATE_FRIEND \
     29#define MUTATE_FRIEND                                                                                                   \
    3030    template<typename node_t> friend node_t * mutate(const node_t * node); \
    3131        template<typename node_t> friend node_t * shallowCopy(const node_t * node);
    3232
    3333namespace ast {
    34 
    3534class Expr;
    3635
    37 /// Base statement node
     36// Base statement node
    3837class Stmt : public ParseNode {
    39 public:
     38  public:
    4039        std::vector<Label> labels;
    4140
    4241        Stmt( const CodeLocation & loc, std::vector<Label> && labels = {} )
    43         : ParseNode(loc), labels(std::move(labels)) {}
     42                : ParseNode(loc), labels(std::move(labels)) {}
    4443
    4544        Stmt(const Stmt& o) : ParseNode(o), labels(o.labels) {}
    4645
    4746        const Stmt * accept( Visitor & v ) const override = 0;
    48 private:
     47  private:
    4948        Stmt * clone() const override = 0;
    5049        MUTATE_FRIEND
    5150};
    5251
    53 /// Compound statement `{ ... }`
     52// Compound statement: { ... }
    5453class CompoundStmt final : public Stmt {
    55 public:
     54  public:
    5655        std::list<ptr<Stmt>> kids;
    5756
    5857        CompoundStmt(const CodeLocation & loc, std::list<ptr<Stmt>> && ks = {},
    59                 std::vector<Label>&& labels = {} )
    60         : Stmt(loc, std::move(labels)), kids(std::move(ks)) {}
     58                                 std::vector<Label>&& labels = {} )
     59                : Stmt(loc, std::move(labels)), kids(std::move(ks)) {}
    6160
    6261        CompoundStmt( const CompoundStmt& o );
     
    6766
    6867        const CompoundStmt * accept( Visitor & v ) const override { return v.visit( this ); }
    69 private:
     68  private:
    7069        CompoundStmt * clone() const override { return new CompoundStmt{ *this }; }
    7170        MUTATE_FRIEND
    7271};
    7372
    74 /// Empty statment `;`
     73// Empty statment: ;
    7574class NullStmt final : public Stmt {
    76 public:
     75  public:
    7776        NullStmt( const CodeLocation & loc, std::vector<Label> && labels = {} )
    78         : Stmt(loc, std::move(labels)) {}
     77                : Stmt(loc, std::move(labels)) {}
    7978
    8079        const NullStmt * accept( Visitor & v ) const override { return v.visit( this ); }
    81 private:
     80  private:
    8281        NullStmt * clone() const override { return new NullStmt{ *this }; }
    8382        MUTATE_FRIEND
    8483};
    8584
    86 /// Expression wrapped by statement
     85// Expression wrapped by statement
    8786class ExprStmt final : public Stmt {
    88 public:
     87  public:
    8988        ptr<Expr> expr;
    9089
    9190        ExprStmt( const CodeLocation& loc, const Expr* e, std::vector<Label>&& labels = {} )
    92         : Stmt(loc, std::move(labels)), expr(e) {}
    93 
    94         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    95 private:
     91                : Stmt(loc, std::move(labels)), expr(e) {}
     92
     93        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     94  private:
    9695        ExprStmt * clone() const override { return new ExprStmt{ *this }; }
    9796        MUTATE_FRIEND
    9897};
    9998
    100 /// Assembly statement `asm ... ( "..." : ... )`
     99// Assembly statement: asm ... ( "..." : ... )
    101100class AsmStmt final : public Stmt {
    102 public:
     101  public:
    103102        bool isVolatile;
    104103        ptr<Expr> instruction;
     
    108107
    109108        AsmStmt( const CodeLocation & loc, bool isVolatile, const Expr * instruction,
    110                 std::vector<ptr<Expr>> && output, std::vector<ptr<Expr>> && input,
    111                 std::vector<ptr<ConstantExpr>> && clobber, std::vector<Label> && gotoLabels,
    112                 std::vector<Label> && labels = {})
    113         : Stmt(loc, std::move(labels)), isVolatile(isVolatile), instruction(instruction),
    114           output(std::move(output)), input(std::move(input)), clobber(std::move(clobber)),
    115           gotoLabels(std::move(gotoLabels)) {}
    116 
    117         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    118 private:
     109                         std::vector<ptr<Expr>> && output, std::vector<ptr<Expr>> && input,
     110                         std::vector<ptr<ConstantExpr>> && clobber, std::vector<Label> && gotoLabels,
     111                         std::vector<Label> && labels = {})
     112                : Stmt(loc, std::move(labels)), isVolatile(isVolatile), instruction(instruction),
     113                  output(std::move(output)), input(std::move(input)), clobber(std::move(clobber)),
     114                  gotoLabels(std::move(gotoLabels)) {}
     115
     116        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     117  private:
    119118        AsmStmt * clone() const override { return new AsmStmt{ *this }; }
    120119        MUTATE_FRIEND
    121120};
    122121
    123 /// C-preprocessor directive `#...`
     122// C-preprocessor directive: #...
    124123class DirectiveStmt final : public Stmt {
    125 public:
     124  public:
    126125        std::string directive;
    127126
    128127        DirectiveStmt( const CodeLocation & loc, const std::string & directive,
    129                 std::vector<Label> && labels = {} )
    130         : Stmt(loc, std::move(labels)), directive(directive) {}
    131 
    132         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    133 private:
     128                                   std::vector<Label> && labels = {} )
     129                : Stmt(loc, std::move(labels)), directive(directive) {}
     130
     131        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     132  private:
    134133        DirectiveStmt * clone() const override { return new DirectiveStmt{ *this }; }
    135134        MUTATE_FRIEND
    136135};
    137136
    138 /// If conditional statement `if (...) ... else ...`
     137// If statement: if (...) ... else ...
    139138class IfStmt final : public Stmt {
    140 public:
     139  public:
    141140        ptr<Expr> cond;
    142141        ptr<Stmt> thenPart;
     
    145144
    146145        IfStmt( const CodeLocation & loc, const Expr * cond, const Stmt * thenPart,
    147                 const Stmt * elsePart = nullptr, std::vector<ptr<Stmt>> && inits = {},
    148                 std::vector<Label> && labels = {} )
    149         : Stmt(loc, std::move(labels)), cond(cond), thenPart(thenPart), elsePart(elsePart),
    150           inits(std::move(inits)) {}
    151 
    152         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    153 private:
     146                        const Stmt * elsePart = nullptr, std::vector<ptr<Stmt>> && inits = {},
     147                        std::vector<Label> && labels = {} )
     148                : Stmt(loc, std::move(labels)), cond(cond), thenPart(thenPart), elsePart(elsePart),
     149                  inits(std::move(inits)) {}
     150
     151        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     152  private:
    154153        IfStmt * clone() const override { return new IfStmt{ *this }; }
    155154        MUTATE_FRIEND
    156155};
    157156
    158 /// Switch or choose conditional statement `switch (...) { ... }`
     157// Switch or choose statement: switch (...) { ... }
    159158class SwitchStmt final : public Stmt {
    160 public:
     159  public:
    161160        ptr<Expr> cond;
    162161        std::vector<ptr<Stmt>> stmts;
    163162
    164163        SwitchStmt( const CodeLocation & loc, const Expr * cond, std::vector<ptr<Stmt>> && stmts,
    165                 std::vector<Label> && labels = {} )
    166         : Stmt(loc, std::move(labels)), cond(cond), stmts(std::move(stmts)) {}
    167 
    168         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    169 private:
     164                                std::vector<Label> && labels = {} )
     165                : Stmt(loc, std::move(labels)), cond(cond), stmts(std::move(stmts)) {}
     166
     167        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     168  private:
    170169        SwitchStmt * clone() const override { return new SwitchStmt{ *this }; }
    171170        MUTATE_FRIEND
    172171};
    173172
    174 /// Case label `case ...:` `default:`
     173// Case label: case ...: or default:
    175174class CaseStmt final : public Stmt {
    176 public:
    177         /// Null for the default label.
     175  public:
     176        // Null for the default label.
    178177        ptr<Expr> cond;
    179178        std::vector<ptr<Stmt>> stmts;
    180179
    181180        CaseStmt( const CodeLocation & loc, const Expr * cond, std::vector<ptr<Stmt>> && stmts,
    182                 std::vector<Label> && labels = {} )
    183         : Stmt(loc, std::move(labels)), cond(cond), stmts(std::move(stmts)) {}
     181                          std::vector<Label> && labels = {} )
     182                : Stmt(loc, std::move(labels)), cond(cond), stmts(std::move(stmts)) {}
    184183
    185184        bool isDefault() const { return !cond; }
    186185
    187186        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    188 private:
     187  private:
    189188        CaseStmt * clone() const override { return new CaseStmt{ *this }; }
    190189        MUTATE_FRIEND
    191190};
    192191
    193 /// While loop `while (...) ...` `do ... while (...);
     192// While loop: while (...) ... else ... or do ... while (...) else ...;
    194193class WhileStmt final : public Stmt {
    195 public:
     194  public:
    196195        ptr<Expr> cond;
    197196        ptr<Stmt> body;
     197        ptr<Stmt> elsePart;
    198198        std::vector<ptr<Stmt>> inits;
    199199        bool isDoWhile;
    200200
    201201        WhileStmt( const CodeLocation & loc, const Expr * cond, const Stmt * body,
    202                 std::vector<ptr<Stmt>> && inits, bool isDoWhile = false, std::vector<Label> && labels = {} )
    203         : Stmt(loc, std::move(labels)), cond(cond), body(body), inits(std::move(inits)),
    204           isDoWhile(isDoWhile) {}
    205 
    206         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    207 private:
     202                           std::vector<ptr<Stmt>> && inits, bool isDoWhile = false, std::vector<Label> && labels = {} )
     203                : Stmt(loc, std::move(labels)), cond(cond), body(body), inits(std::move(inits)), isDoWhile(isDoWhile) {}
     204
     205        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     206  private:
    208207        WhileStmt * clone() const override { return new WhileStmt{ *this }; }
    209208        MUTATE_FRIEND
    210209};
    211210
    212 /// For loop `for (... ; ... ; ...) ...`
     211// For loop: for (... ; ... ; ...) ... else ...
    213212class ForStmt final : public Stmt {
    214 public:
     213  public:
    215214        std::vector<ptr<Stmt>> inits;
    216215        ptr<Expr> cond;
    217216        ptr<Expr> inc;
    218217        ptr<Stmt> body;
     218        ptr<Stmt> elsePart;
    219219
    220220        ForStmt( const CodeLocation & loc, std::vector<ptr<Stmt>> && inits, const Expr * cond,
    221                 const Expr * inc, const Stmt * body, std::vector<Label> && labels = {} )
    222         : Stmt(loc, std::move(labels)), inits(std::move(inits)), cond(cond), inc(inc),
    223           body(body) {}
    224 
    225         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    226 private:
     221                         const Expr * inc, const Stmt * body, std::vector<Label> && labels = {} )
     222                : Stmt(loc, std::move(labels)), inits(std::move(inits)), cond(cond), inc(inc), body(body) {}
     223
     224        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     225  private:
    227226        ForStmt * clone() const override { return new ForStmt{ *this }; }
    228227        MUTATE_FRIEND
    229228};
    230229
    231 /// Branch control flow statement `goto ...` `break` `continue` `fallthru`
     230// Branch control flow statement: goto ... or break or continue or fallthru
    232231class BranchStmt final : public Stmt {
    233 public:
     232  public:
    234233        enum Kind { Goto, Break, Continue, FallThrough, FallThroughDefault };
    235234        static constexpr size_t kindEnd = 1 + (size_t)FallThroughDefault;
     
    241240
    242241        BranchStmt( const CodeLocation & loc, Kind kind, Label target,
    243                 std::vector<Label> && labels = {} );
     242                                std::vector<Label> && labels = {} );
    244243        BranchStmt( const CodeLocation & loc, const Expr * computedTarget,
    245                 std::vector<Label> && labels = {} )
    246         : Stmt(loc, std::move(labels)), originalTarget(loc), target(loc),
    247           computedTarget(computedTarget), kind(Goto) {}
     244                                std::vector<Label> && labels = {} )
     245                : Stmt(loc, std::move(labels)), originalTarget(loc), target(loc),
     246                  computedTarget(computedTarget), kind(Goto) {}
    248247
    249248        const char * kindName() const { return kindNames[kind]; }
    250249
    251250        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    252 private:
     251  private:
    253252        BranchStmt * clone() const override { return new BranchStmt{ *this }; }
    254253        MUTATE_FRIEND
     
    257256};
    258257
    259 /// Return statement `return ...`
     258// Return statement: return ...
    260259class ReturnStmt final : public Stmt {
    261 public:
     260  public:
    262261        ptr<Expr> expr;
    263262
    264263        ReturnStmt( const CodeLocation & loc, const Expr * expr, std::vector<Label> && labels = {} )
    265         : Stmt(loc, std::move(labels)), expr(expr) {}
    266 
    267         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    268 private:
     264                : Stmt(loc, std::move(labels)), expr(expr) {}
     265
     266        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     267  private:
    269268        ReturnStmt * clone() const override { return new ReturnStmt{ *this }; }
    270269        MUTATE_FRIEND
    271270};
    272271
    273 /// Kind of exception
     272// Kind of exception
    274273enum ExceptionKind { Terminate, Resume };
    275274
    276 /// Throw statement `throw ...`
     275// Throw statement: throw ...
    277276class ThrowStmt final : public Stmt {
    278 public:
     277  public:
    279278        ptr<Expr> expr;
    280279        ptr<Expr> target;
     
    284283                const CodeLocation & loc, ExceptionKind kind, const Expr * expr, const Expr * target,
    285284                std::vector<Label> && labels = {} )
    286         : Stmt(loc, std::move(labels)), expr(expr), target(target), kind(kind) {}
    287 
    288         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    289 private:
     285                : Stmt(loc, std::move(labels)), expr(expr), target(target), kind(kind) {}
     286
     287        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     288  private:
    290289        ThrowStmt * clone() const override { return new ThrowStmt{ *this }; }
    291290        MUTATE_FRIEND
    292291};
    293292
    294 /// Try statement `try { ... } ...`
     293// Try statement: try { ... } ...
    295294class TryStmt final : public Stmt {
    296 public:
     295  public:
    297296        ptr<CompoundStmt> body;
    298297        std::vector<ptr<CatchStmt>> handlers;
     
    303302                std::vector<ptr<CatchStmt>> && handlers, const FinallyStmt * finally,
    304303                std::vector<Label> && labels = {} )
    305         : Stmt(loc, std::move(labels)), body(body), handlers(std::move(handlers)), finally(finally) {}
    306 
    307         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    308 private:
     304                : Stmt(loc, std::move(labels)), body(body), handlers(std::move(handlers)), finally(finally) {}
     305
     306        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     307  private:
    309308        TryStmt * clone() const override { return new TryStmt{ *this }; }
    310309        MUTATE_FRIEND
    311310};
    312311
    313 /// Catch clause of try statement
     312// Catch clause of try statement
    314313class CatchStmt final : public Stmt {
    315 public:
     314  public:
    316315        ptr<Decl> decl;
    317316        ptr<Expr> cond;
     
    322321                const CodeLocation & loc, ExceptionKind kind, const Decl * decl, const Expr * cond,
    323322                const Stmt * body, std::vector<Label> && labels = {} )
    324         : Stmt(loc, std::move(labels)), decl(decl), cond(cond), body(body), kind(kind) {}
    325 
    326         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    327 private:
     323                : Stmt(loc, std::move(labels)), decl(decl), cond(cond), body(body), kind(kind) {}
     324
     325        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     326  private:
    328327        CatchStmt * clone() const override { return new CatchStmt{ *this }; }
    329328        MUTATE_FRIEND
    330329};
    331330
    332 /// Finally clause of try statement
     331// Finally clause of try statement
    333332class FinallyStmt final : public Stmt {
    334 public:
     333  public:
    335334        ptr<CompoundStmt> body;
    336335
    337336        FinallyStmt( const CodeLocation & loc, const CompoundStmt * body,
    338                 std::vector<Label> && labels = {} )
    339         : Stmt(loc, std::move(labels)), body(body) {}
    340 
    341         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    342 private:
     337                                 std::vector<Label> && labels = {} )
     338                : Stmt(loc, std::move(labels)), body(body) {}
     339
     340        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     341  private:
    343342        FinallyStmt * clone() const override { return new FinallyStmt{ *this }; }
    344343        MUTATE_FRIEND
    345344};
    346345
    347 /// Suspend statement
     346// Suspend statement
    348347class SuspendStmt final : public Stmt {
    349 public:
     348  public:
    350349        ptr<CompoundStmt> then;
    351350        enum Type { None, Coroutine, Generator } type = None;
    352351
    353352        SuspendStmt( const CodeLocation & loc, const CompoundStmt * then, Type type, std::vector<Label> && labels = {} )
    354         : Stmt(loc, std::move(labels)), then(then), type(type) {}
    355 
    356         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    357 private:
     353                : Stmt(loc, std::move(labels)), then(then), type(type) {}
     354
     355        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     356  private:
    358357        SuspendStmt * clone() const override { return new SuspendStmt{ *this }; }
    359358        MUTATE_FRIEND
    360359};
    361360
    362 /// Wait for concurrency statement `when (...) waitfor (... , ...) ... timeout(...) ... else ...`
     361// Waitfor statement: when (...) waitfor (... , ...) ... timeout(...) ... else ...
    363362class WaitForStmt final : public Stmt {
    364 public:
     363  public:
    365364        struct Target {
    366365                ptr<Expr> func;
     
    390389
    391390        WaitForStmt( const CodeLocation & loc, std::vector<Label> && labels = {} )
    392         : Stmt(loc, std::move(labels)) {}
    393 
    394         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    395 private:
     391                : Stmt(loc, std::move(labels)) {}
     392
     393        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     394  private:
    396395        WaitForStmt * clone() const override { return new WaitForStmt{ *this }; }
    397396        MUTATE_FRIEND
    398397};
    399398
    400 /// Any declaration in a (compound) statement.
     399// Any declaration in a (compound) statement.
    401400class DeclStmt final : public Stmt {
    402 public:
     401  public:
    403402        ptr<Decl> decl;
    404403
    405404        DeclStmt( const CodeLocation & loc, const Decl * decl, std::vector<Label> && labels = {} )
    406         : Stmt(loc, std::move(labels)), decl(decl) {}
    407 
    408         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    409 private:
     405                : Stmt(loc, std::move(labels)), decl(decl) {}
     406
     407        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     408  private:
    410409        DeclStmt * clone() const override { return new DeclStmt{ *this }; }
    411410        MUTATE_FRIEND
    412411};
    413412
    414 /// Represents an implicit application of a constructor or destructor.
     413// Represents an implicit application of a constructor or destructor.
    415414class ImplicitCtorDtorStmt final : public Stmt {
    416 public:
     415  public:
    417416        ptr<Stmt> callStmt;
    418417
    419418        ImplicitCtorDtorStmt( const CodeLocation & loc, const Stmt * callStmt,
    420                 std::vector<Label> && labels = {} )
    421         : Stmt(loc, std::move(labels)), callStmt(callStmt) {}
    422 
    423         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    424 private:
     419                                                  std::vector<Label> && labels = {} )
     420                : Stmt(loc, std::move(labels)), callStmt(callStmt) {}
     421
     422        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     423  private:
    425424        ImplicitCtorDtorStmt * clone() const override { return new ImplicitCtorDtorStmt{ *this }; }
    426425        MUTATE_FRIEND
    427426};
    428427
    429 /// Mutex Statement
     428// Mutex Statement
    430429class MutexStmt final : public Stmt {
    431 public:
     430  public:
    432431        ptr<Stmt> stmt;
    433432        std::vector<ptr<Expr>> mutexObjs;
    434433
    435434        MutexStmt( const CodeLocation & loc, const Stmt * stmt,
    436                 std::vector<ptr<Expr>> && mutexes, std::vector<Label> && labels = {} )
    437         : Stmt(loc, std::move(labels)), stmt(stmt), mutexObjs(std::move(mutexes)) {}
    438 
    439         const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
    440 private:
     435                           std::vector<ptr<Expr>> && mutexes, std::vector<Label> && labels = {} )
     436                : Stmt(loc, std::move(labels)), stmt(stmt), mutexObjs(std::move(mutexes)) {}
     437
     438        const Stmt * accept( Visitor & v ) const override { return v.visit( this ); }
     439  private:
    441440        MutexStmt * clone() const override { return new MutexStmt{ *this }; }
    442441        MUTATE_FRIEND
    443442};
    444 
    445 }
     443} // namespace ast
    446444
    447445#undef MUTATE_FRIEND
    448446
    449447// Local Variables: //
    450 // tab-width: 4 //
    451448// mode: c++ //
    452 // compile-command: "make install" //
    453449// End: //
  • src/ControlStruct/ExceptTranslateNew.cpp

    r3e5db5b4 r34c32f0  
    99// Author           : Andrew Beach
    1010// Created On       : Mon Nov  8 11:53:00 2021
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Mon Nov  8 16:50:00 2021
    13 // Update Count     : 0
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 18:49:58 2022
     13// Update Count     : 1
    1414//
    1515
     
    2222
    2323namespace ControlStruct {
    24 
    25 namespace {
    2624
    2725class TranslateThrowsCore : public ast::WithGuards {
     
    128126}
    129127
    130 } // namespace
    131 
    132128void translateThrows( ast::TranslationUnit & transUnit ) {
    133129        ast::Pass<TranslateThrowsCore>::run( transUnit );
  • src/ControlStruct/FixLabels.cpp

    r3e5db5b4 r34c32f0  
    99// Author           : Andrew Beach
    1010// Created On       : Mon Nov  1 09:39:00 2021
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Mon Nov  8 10:53:00 2021
    13 // Update Count     : 3
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:19:17 2022
     13// Update Count     : 9
    1414//
    1515
     
    2020#include "AST/Stmt.hpp"
    2121#include "ControlStruct/MultiLevelExit.hpp"
     22using namespace ast;
    2223
    2324namespace ControlStruct {
    24 
    25 namespace {
    26 
    27 class FixLabelsCore final : public ast::WithGuards {
     25class FixLabelsCore final : public WithGuards {
    2826        LabelToStmt labelTable;
    29 public:
     27  public:
    3028        FixLabelsCore() : labelTable() {}
    3129
    32         void previsit( const ast::FunctionDecl * );
    33         const ast::FunctionDecl * postvisit( const ast::FunctionDecl * );
    34         void previsit( const ast::Stmt * );
    35         void previsit( const ast::BranchStmt * );
    36         void previsit( const ast::LabelAddressExpr * );
     30        void previsit( const FunctionDecl * );
     31        const FunctionDecl * postvisit( const FunctionDecl * );
     32        void previsit( const Stmt * );
     33        void previsit( const BranchStmt * );
     34        void previsit( const LabelAddressExpr * );
    3735
    38         void setLabelsDef( const std::vector<ast::Label> &, const ast::Stmt * );
    39         void setLabelsUsage( const ast::Label & );
     36        void setLabelsDef( const std::vector<Label> &, const Stmt * );
     37        void setLabelsUsage( const Label & );
    4038};
    4139
    42 void FixLabelsCore::previsit( const ast::FunctionDecl * ) {
     40void FixLabelsCore::previsit( const FunctionDecl * ) {
    4341        GuardValue( labelTable ).clear();
    4442}
    4543
    46 const ast::FunctionDecl * FixLabelsCore::postvisit(
    47                 const ast::FunctionDecl * decl ) {
     44const FunctionDecl * FixLabelsCore::postvisit(
     45        const FunctionDecl * decl ) {
    4846        if ( nullptr == decl->stmts ) return decl;
    4947        for ( auto kvp : labelTable ) {
    5048                if ( nullptr == kvp.second ) {
    5149                        SemanticError( kvp.first.location,
    52                                 "Use of undefined label: " + kvp.first.name );
     50                                                   "Use of undefined label: " + kvp.first.name );
    5351                }
    5452        }
    55         return ast::mutate_field( decl, &ast::FunctionDecl::stmts,
    56                 multiLevelExitUpdate( decl->stmts.get(), labelTable ) );
     53        return mutate_field( decl, &FunctionDecl::stmts,
     54                                                 multiLevelExitUpdate( decl->stmts.get(), labelTable ) );
    5755}
    5856
    59 void FixLabelsCore::previsit( const ast::Stmt * stmt ) {
     57void FixLabelsCore::previsit( const Stmt * stmt ) {
    6058        if ( !stmt->labels.empty() ) {
    6159                setLabelsDef( stmt->labels, stmt );
     
    6361}
    6462
    65 void FixLabelsCore::previsit( const ast::BranchStmt * stmt ) {
     63void FixLabelsCore::previsit( const BranchStmt * stmt ) {
    6664        if ( !stmt->labels.empty() ) {
    6765                setLabelsDef( stmt->labels, stmt );
     
    7270}
    7371
    74 void FixLabelsCore::previsit( const ast::LabelAddressExpr * expr ) {
     72void FixLabelsCore::previsit( const LabelAddressExpr * expr ) {
    7573        assert( !expr->arg.empty() );
    7674        setLabelsUsage( expr->arg );
     
    7876
    7977void FixLabelsCore::setLabelsDef(
    80                 const std::vector<ast::Label> & labels, const ast::Stmt * stmt ) {
     78        const std::vector<Label> & labels, const Stmt * stmt ) {
    8179        assert( !labels.empty() );
    8280        assert( stmt );
     
    8987                        // Duplicate definition, this is an error.
    9088                        SemanticError( label.location,
    91                                 "Duplicate definition of label: " + label.name );
     89                                                   "Duplicate definition of label: " + label.name );
    9290                } else {
    9391                        // Perviously used, but not defined until now.
     
    9896
    9997// Label was used, if it is new add it to the table.
    100 void FixLabelsCore::setLabelsUsage( const ast::Label & label ) {
     98void FixLabelsCore::setLabelsUsage( const Label & label ) {
    10199        if ( labelTable.find( label ) == labelTable.end() ) {
    102100                labelTable[ label ] = nullptr;
     
    104102}
    105103
    106 } // namespace
    107 
    108 void fixLabels( ast::TranslationUnit & translationUnit ) {
    109         ast::Pass<FixLabelsCore>::run( translationUnit );
     104void fixLabels( TranslationUnit & translationUnit ) {
     105        Pass<FixLabelsCore>::run( translationUnit );
    110106}
    111 
    112107} // namespace ControlStruct
    113108
  • src/ControlStruct/FixLabels.hpp

    r3e5db5b4 r34c32f0  
    99// Author           : Andrew Beach
    1010// Created On       : Mon Nov  1 09:36:00 2021
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Mon Nov  1 09:40:00 2021
    13 // Update Count     : 0
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:18:43 2022
     13// Update Count     : 2
    1414//
    1515
     
    1717
    1818namespace ast {
    19         class TranslationUnit;
     19class TranslationUnit;
    2020}
    2121
    2222namespace ControlStruct {
    23 
    24 /// normalizes label definitions and generates multi-level exit labels
     23// normalizes label definitions and generates multi-level exit labels
    2524void fixLabels( ast::TranslationUnit & translationUnit );
    26 
    2725}
    2826
  • src/ControlStruct/ForExprMutator.h

    r3e5db5b4 r34c32f0  
    1010// Created On       : Mon May 18 07:44:20 2015
    1111// Last Modified By : Peter A. Buhr
    12 // Last Modified On : Thu Aug 17 15:32:48 2017
    13 // Update Count     : 5
     12// Last Modified On : Sun Jan 30 09:14:46 2022
     13// Update Count     : 6
    1414//
    1515
     
    2424        class ForExprMutator {
    2525          public:
    26                 Statement *postmutate( IfStmt * );
    27                 Statement *postmutate( ForStmt * );
    28                 Statement *postmutate( WhileStmt * );
     26                Statement * postmutate( IfStmt * );
     27                Statement * postmutate( ForStmt * );
     28                Statement * postmutate( WhileStmt * );
    2929        };
    3030} // namespace ControlStruct
  • src/ControlStruct/HoistControlDecls.cpp

    r3e5db5b4 r34c32f0  
    99// Author           : Andrew Beach
    1010// Created On       : Fri Dec  3 15:34:00 2021
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Fri Dec  3 15:34:00 2021
    13 // Update Count     : 0
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 18:52:35 2022
     13// Update Count     : 23
    1414//
    1515
     
    1919#include "AST/Pass.hpp"
    2020#include "AST/TranslationUnit.hpp"
     21using namespace ast;
    2122
    2223namespace ControlStruct {
    2324
    24 namespace {
     25template<typename StmtT>
     26const Stmt * hoist( const StmtT * stmt ) {
     27        // If no hoisting is needed, then make no changes.
    2528
    26 template<typename StmtT>
    27 const ast::Stmt * hoist( const StmtT * stmt ) {
    28         // If no hoisting is needed, then make no changes.
    29         if ( 0 == stmt->inits.size() ) {
     29        if ( stmt->inits.size() == 0 ) {                                        // no declarations ?
     30                // if ( /* no conditional declarations */ ...  ) ...
    3031                return stmt;
    31         }
     32        } // if
    3233
    33         // Put initializers and the old statement in the compound statement.
    34         ast::CompoundStmt * block = new ast::CompoundStmt( stmt->location );
    35         StmtT * mutStmt = ast::mutate( stmt );
    36         for ( const ast::Stmt * next : mutStmt->inits ) {
     34        // Put hoist declarations and modified statement in a compound statement.
     35
     36        CompoundStmt * block = new CompoundStmt( stmt->location ); // create empty compound statement
     37        //    {
     38        //    }
     39
     40        for ( const Stmt * next : stmt->inits ) {                       // link conditional declarations into compound
    3741                block->kids.push_back( next );
    3842        }
    39         mutStmt->inits.clear();
    40         block->kids.push_back( mutStmt );
     43        //    if ( int x = f(), y = g(); ... ) ...
     44        // link declarations into compound statement
     45        //    {
     46        //         int x = f();
     47        //         int y = g();
     48        //    }
     49
     50        StmtT * mutStmt = mutate( stmt );                                       // create mutate handle to change statement
     51        mutStmt->inits.clear();                                                         // remove declarations
     52        //    if ( ... ) ...
     53
     54        block->kids.push_back( mutStmt );                                       // link modified statement into compound
     55        //    {
     56        //        int x = f();
     57        //        int y = g();
     58        //        if ( ... ) ...
     59        //    }
    4160        return block;
    4261}
    4362
    44 struct HoistControlCore {
    45         const ast::Stmt * postvisit( const ast::IfStmt * stmt ) {
    46                 return hoist<ast::IfStmt>( stmt );
     63struct hoistControlDeclsCore {
     64        // Statements with declarations in conditional.
     65        const Stmt * postvisit( const IfStmt * stmt ) {
     66                return hoist<IfStmt>( stmt );
    4767        }
    48         const ast::Stmt * postvisit( const ast::ForStmt * stmt ) {
    49                 return hoist<ast::ForStmt>( stmt );
     68        const Stmt * postvisit( const ForStmt * stmt ) {
     69                return hoist<ForStmt>( stmt );
    5070        }
    51         const ast::Stmt * postvisit( const ast::WhileStmt * stmt ) {
    52                 return hoist<ast::WhileStmt>( stmt );
     71        const Stmt * postvisit( const WhileStmt * stmt ) {
     72                return hoist<WhileStmt>( stmt );
    5373        }
    5474};
    5575
    56 } // namespace
    57 
    58 /// Hoist initialization out of for statements.
    59 void hoistControlDecls( ast::TranslationUnit & translationUnit ) {
    60         ast::Pass<HoistControlCore>::run( translationUnit );
     76// Hoist initialization out of for statements.
     77void hoistControlDecls( TranslationUnit & translationUnit ) {
     78        Pass<hoistControlDeclsCore>::run( translationUnit );
    6179}
    6280
  • src/ControlStruct/HoistControlDecls.hpp

    r3e5db5b4 r34c32f0  
    99// Author           : Andrew Beach
    1010// Created On       : Fri Dec  3 15:31:00 2021
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Fri Dec  3 15:31:00 2021
    13 // Update Count     : 0
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:25:07 2022
     13// Update Count     : 3
    1414//
    1515
     
    1717
    1818namespace ast {
    19         class TranslationUnit;
     19class TranslationUnit;
    2020}
    2121
    2222namespace ControlStruct {
    23 
    24 /// Hoist initialization out of control flow statements.
     23// Hoist declarations out of control flow statements into compound statement.
    2524void hoistControlDecls( ast::TranslationUnit & translationUnit );
    26 
    2725} // namespace ControlStruct
    2826
  • src/ControlStruct/LabelFixer.cc

    r3e5db5b4 r34c32f0  
    99// Author           : Rodolfo G. Esteves
    1010// Created On       : Mon May 18 07:44:20 2015
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Tue Jan 21 10:32:00 2020
    13 // Update Count     : 160
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:28:31 2022
     13// Update Count     : 161
    1414//
    1515
     
    2727
    2828namespace ControlStruct {
    29         bool LabelFixer::Entry::insideLoop() {
    30                 return ( dynamic_cast< ForStmt * > ( definition ) ||
    31                         dynamic_cast< WhileStmt * > ( definition )  );
     29bool LabelFixer::Entry::insideLoop() {
     30        return ( dynamic_cast< ForStmt * > ( definition ) ||
     31                dynamic_cast< WhileStmt * > ( definition )  );
     32}
     33
     34LabelFixer::LabelFixer( LabelGenerator * gen ) : generator ( gen ) {
     35        if ( generator == 0 )
     36                generator = LabelGenerator::getGenerator();
     37}
     38
     39void LabelFixer::previsit( FunctionDecl * ) {
     40        // need to go into a nested function in a fresh state
     41        GuardValue( labelTable );
     42        labelTable.clear();
     43}
     44
     45void LabelFixer::postvisit( FunctionDecl * functionDecl ) {
     46        PassVisitor<MultiLevelExitMutator> mlem( resolveJumps(), generator );
     47        // We start in the body so we can stop when we hit another FunctionDecl.
     48        maybeMutate( functionDecl->statements, mlem );
     49}
     50
     51// prune to at most one label definition for each statement
     52void LabelFixer::previsit( Statement * stmt ) {
     53        std::list< Label > &labels = stmt->get_labels();
     54
     55        if ( ! labels.empty() ) {
     56                // only remember one label for each statement
     57                Label current = setLabelsDef( labels, stmt );
     58        } // if
     59}
     60
     61void LabelFixer::previsit( BranchStmt * branchStmt ) {
     62        previsit( ( Statement *)branchStmt );
     63
     64        // for labeled branches, add an entry to the label table
     65        Label target = branchStmt->get_target();
     66        if ( target != "" ) {
     67                setLabelsUsg( target, branchStmt );
     68        }
     69}
     70
     71void LabelFixer::previsit( LabelAddressExpr * addrExpr ) {
     72        Label & target = addrExpr->arg;
     73        assert( target != "" );
     74        setLabelsUsg( target, addrExpr );
     75}
     76
     77
     78// Sets the definition of the labelTable entry to be the provided statement for every label in
     79// the list parameter. Happens for every kind of statement.
     80Label LabelFixer::setLabelsDef( std::list< Label > & llabel, Statement * definition ) {
     81        assert( definition != 0 );
     82        assert( llabel.size() > 0 );
     83
     84        for ( std::list< Label >::iterator i = llabel.begin(); i != llabel.end(); i++ ) {
     85                Label & l = *i;
     86                l.set_statement( definition ); // attach statement to the label to be used later
     87                if ( labelTable.find( l ) == labelTable.end() ) {
     88                        // All labels on this statement need to use the same entry,
     89                        // so this should only be created once.
     90                        // undefined and unused until now, add an entry
     91                        labelTable[ l ] = new Entry( definition );
     92                } else if ( labelTable[ l ]->defined() ) {
     93                        // defined twice, error
     94                        SemanticError( l.get_statement()->location,
     95                                "Duplicate definition of label: " + l.get_name() );
     96                } else {
     97                        // used previously, but undefined until now -> link with this entry
     98                        // Question: Is changing objects important?
     99                        delete labelTable[ l ];
     100                        labelTable[ l ] = new Entry( definition );
     101                } // if
     102        } // for
     103
     104        // Produce one of the labels attached to this statement to be temporarily used as the
     105        // canonical label.
     106        return labelTable[ llabel.front() ]->get_label();
     107}
     108
     109// A label was used, add it to the table if it isn't already there
     110template< typename UsageNode >
     111void LabelFixer::setLabelsUsg( Label orgValue, UsageNode *use ) {
     112        assert( use != 0 );
     113
     114        // add label with an unknown origin
     115        if ( labelTable.find( orgValue ) == labelTable.end() ) {
     116                labelTable[ orgValue ] = new Entry( 0 );
     117        }
     118}
     119
     120// Builds a table that maps a label to its defining statement.
     121std::map<Label, Statement * > * LabelFixer::resolveJumps() throw ( SemanticErrorException ) {
     122        std::map< Label, Statement * > *ret = new std::map< Label, Statement * >();
     123        for ( std::map< Label, Entry * >::iterator i = labelTable.begin(); i != labelTable.end(); ++i ) {
     124                if ( ! i->second->defined() ) {
     125                        SemanticError( i->first.get_statement()->location, "Use of undefined label: " + i->first.get_name() );
     126                }
     127                (*ret)[ i->first ] = i->second->get_definition();
    32128        }
    33129
    34         LabelFixer::LabelFixer( LabelGenerator * gen ) : generator ( gen ) {
    35                 if ( generator == 0 )
    36                         generator = LabelGenerator::getGenerator();
    37         }
    38 
    39         void LabelFixer::previsit( FunctionDecl * ) {
    40                 // need to go into a nested function in a fresh state
    41                 GuardValue( labelTable );
    42                 labelTable.clear();
    43         }
    44 
    45         void LabelFixer::postvisit( FunctionDecl * functionDecl ) {
    46                 PassVisitor<MultiLevelExitMutator> mlem( resolveJumps(), generator );
    47                 // We start in the body so we can stop when we hit another FunctionDecl.
    48                 maybeMutate( functionDecl->statements, mlem );
    49         }
    50 
    51         // prune to at most one label definition for each statement
    52         void LabelFixer::previsit( Statement * stmt ) {
    53                 std::list< Label > &labels = stmt->get_labels();
    54 
    55                 if ( ! labels.empty() ) {
    56                         // only remember one label for each statement
    57                         Label current = setLabelsDef( labels, stmt );
    58                 } // if
    59         }
    60 
    61         void LabelFixer::previsit( BranchStmt * branchStmt ) {
    62                 previsit( ( Statement *)branchStmt );
    63 
    64                 // for labeled branches, add an entry to the label table
    65                 Label target = branchStmt->get_target();
    66                 if ( target != "" ) {
    67                         setLabelsUsg( target, branchStmt );
    68                 }
    69         }
    70 
    71         void LabelFixer::previsit( LabelAddressExpr * addrExpr ) {
    72                 Label & target = addrExpr->arg;
    73                 assert( target != "" );
    74                 setLabelsUsg( target, addrExpr );
    75         }
    76 
    77 
    78         // Sets the definition of the labelTable entry to be the provided statement for every label in
    79         // the list parameter. Happens for every kind of statement.
    80         Label LabelFixer::setLabelsDef( std::list< Label > & llabel, Statement * definition ) {
    81                 assert( definition != 0 );
    82                 assert( llabel.size() > 0 );
    83 
    84                 for ( std::list< Label >::iterator i = llabel.begin(); i != llabel.end(); i++ ) {
    85                         Label & l = *i;
    86                         l.set_statement( definition ); // attach statement to the label to be used later
    87                         if ( labelTable.find( l ) == labelTable.end() ) {
    88                                 // All labels on this statement need to use the same entry,
    89                                 // so this should only be created once.
    90                                 // undefined and unused until now, add an entry
    91                                 labelTable[ l ] = new Entry( definition );
    92                         } else if ( labelTable[ l ]->defined() ) {
    93                                 // defined twice, error
    94                                 SemanticError( l.get_statement()->location,
    95                                         "Duplicate definition of label: " + l.get_name() );
    96                         } else {
    97                                 // used previously, but undefined until now -> link with this entry
    98                                 // Question: Is changing objects important?
    99                                 delete labelTable[ l ];
    100                                 labelTable[ l ] = new Entry( definition );
    101                         } // if
    102                 } // for
    103 
    104                 // Produce one of the labels attached to this statement to be temporarily used as the
    105                 // canonical label.
    106                 return labelTable[ llabel.front() ]->get_label();
    107         }
    108 
    109         // A label was used, add it to the table if it isn't already there
    110         template< typename UsageNode >
    111         void LabelFixer::setLabelsUsg( Label orgValue, UsageNode *use ) {
    112                 assert( use != 0 );
    113 
    114                 // add label with an unknown origin
    115                 if ( labelTable.find( orgValue ) == labelTable.end() ) {
    116                         labelTable[ orgValue ] = new Entry( 0 );
    117                 }
    118         }
    119 
    120         // Builds a table that maps a label to its defining statement.
    121         std::map<Label, Statement * > * LabelFixer::resolveJumps() throw ( SemanticErrorException ) {
    122                 std::map< Label, Statement * > *ret = new std::map< Label, Statement * >();
    123                 for ( std::map< Label, Entry * >::iterator i = labelTable.begin(); i != labelTable.end(); ++i ) {
    124                         if ( ! i->second->defined() ) {
    125                                 SemanticError( i->first.get_statement()->location, "Use of undefined label: " + i->first.get_name() );
    126                         }
    127                         (*ret)[ i->first ] = i->second->get_definition();
    128                 }
    129 
    130                 return ret;
    131         }
     130        return ret;
     131}
    132132}  // namespace ControlStruct
    133133
  • src/ControlStruct/LabelFixer.h

    r3e5db5b4 r34c32f0  
    1010// Created On       : Mon May 18 07:44:20 2015
    1111// Last Modified By : Peter A. Buhr
    12 // Last Modified On : Sat Jul 22 09:17:24 2017
    13 // Update Count     : 34
     12// Last Modified On : Mon Jan 31 22:28:04 2022
     13// Update Count     : 35
    1414//
    1515
     
    2626
    2727namespace ControlStruct {
    28         /// normalizes label definitions and generates multi-level exit labels
    29         class LabelGenerator;
     28// normalizes label definitions and generates multi-level exit labels
     29class LabelGenerator;
    3030
    31         class LabelFixer final : public WithGuards {
    32           public:
    33                 LabelFixer( LabelGenerator *gen = 0 );
     31class LabelFixer final : public WithGuards {
     32  public:
     33        LabelFixer( LabelGenerator *gen = 0 );
    3434
    35                 std::map < Label, Statement * > *resolveJumps() throw ( SemanticErrorException );
     35        std::map < Label, Statement * > *resolveJumps() throw ( SemanticErrorException );
    3636
    37                 // Declarations
    38                 void previsit( FunctionDecl *functionDecl );
    39                 void postvisit( FunctionDecl *functionDecl );
     37        // Declarations
     38        void previsit( FunctionDecl *functionDecl );
     39        void postvisit( FunctionDecl *functionDecl );
    4040
    41                 // Statements
    42                 void previsit( Statement *stmt );
    43                 void previsit( BranchStmt *branchStmt );
     41        // Statements
     42        void previsit( Statement *stmt );
     43        void previsit( BranchStmt *branchStmt );
    4444
    45                 // Expressions
    46                 void previsit( LabelAddressExpr *addrExpr );
     45        // Expressions
     46        void previsit( LabelAddressExpr *addrExpr );
    4747
    48                 Label setLabelsDef( std::list< Label > &, Statement *definition );
    49                 template< typename UsageNode >
    50                 void setLabelsUsg( Label, UsageNode *usage = 0 );
     48        Label setLabelsDef( std::list< Label > &, Statement *definition );
     49        template< typename UsageNode >
     50        void setLabelsUsg( Label, UsageNode *usage = 0 );
     51
     52  private:
     53        class Entry {
     54                public:
     55                Entry( Statement *to ) : definition( to ) {}
     56                bool defined() { return ( definition != 0 ); }
     57                bool insideLoop();
     58
     59                Label get_label() const { return label; }
     60                void set_label( Label lab ) { label = lab; }
     61
     62                Statement *get_definition() const { return definition; }
     63                void set_definition( Statement *def ) { definition = def; }
    5164
    5265          private:
    53                 class Entry {
    54                         public:
    55                         Entry( Statement *to ) : definition( to ) {}
    56                         bool defined() { return ( definition != 0 ); }
    57                         bool insideLoop();
     66                Label label;
     67                Statement *definition;
     68        };
    5869
    59                         Label get_label() const { return label; }
    60                         void set_label( Label lab ) { label = lab; }
    61 
    62                         Statement *get_definition() const { return definition; }
    63                         void set_definition( Statement *def ) { definition = def; }
    64 
    65                   private:
    66                         Label label;
    67                         Statement *definition;
    68                 };
    69 
    70                 std::map < Label, Entry *> labelTable;
    71                 LabelGenerator *generator;
    72         };
     70        std::map < Label, Entry *> labelTable;
     71        LabelGenerator *generator;
     72};
    7373} // namespace ControlStruct
    7474
  • src/ControlStruct/LabelGenerator.cc

    r3e5db5b4 r34c32f0  
    99// Author           : Rodolfo G. Esteves
    1010// Created On       : Mon May 18 07:44:20 2015
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Mon Nov  8 10:18:00 2021
    13 // Update Count     : 17
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:30:26 2022
     13// Update Count     : 28
    1414//
    1515
     
    1717#include <sstream>              // for ostringstream
    1818#include <list>                 // for list
     19using namespace std;
    1920
    2021#include "LabelGenerator.h"
    2122
    22 #include "AST/Attribute.hpp"
    23 #include "AST/Label.hpp"
    24 #include "AST/Stmt.hpp"
    2523#include "SynTree/Attribute.h"  // for Attribute
    2624#include "SynTree/Label.h"      // for Label, operator<<
     
    2826
    2927namespace ControlStruct {
    30 
    3128int LabelGenerator::current = 0;
    3229LabelGenerator * LabelGenerator::labelGenerator = nullptr;
    3330
    34         LabelGenerator * LabelGenerator::getGenerator() {
    35                 if ( LabelGenerator::labelGenerator == 0 )
    36                         LabelGenerator::labelGenerator = new LabelGenerator();
    37                 return labelGenerator;
    38         }
    39 
    40         Label LabelGenerator::newLabel( std::string suffix, Statement * stmt ) {
    41                 std::ostringstream os;
    42                 os << "__L" << current++ << "__" << suffix;
    43                 if ( stmt && ! stmt->get_labels().empty() ) {
    44                         os << "_" << stmt->get_labels().front() << "__";
    45                 } // if
    46                 std::string ret = os.str();
    47                 Label l( ret );
    48                 l.get_attributes().push_back( new Attribute("unused") );
    49                 return l;
    50         }
    51 
    52 ast::Label LabelGenerator::newLabel(
    53                 const std::string & suffix, const ast::Stmt * stmt ) {
    54         assert( stmt );
    55 
    56         std::ostringstream os;
    57         os << "__L" << current++ << "__" << suffix;
    58         if ( stmt && !stmt->labels.empty() ) {
    59                 os << "_" << stmt->labels.front() << "__";
    60         }
    61         ast::Label ret_label( stmt->location, os.str() );
    62         ret_label.attributes.push_back( new ast::Attribute( "unused" ) );
    63         return ret_label;
     31LabelGenerator * LabelGenerator::getGenerator() {
     32        if ( LabelGenerator::labelGenerator == 0 )
     33                LabelGenerator::labelGenerator = new LabelGenerator();
     34        return labelGenerator;
    6435}
    6536
     37Label LabelGenerator::newLabel( string suffix, Statement * stmt ) {
     38        ostringstream os;
     39        os << "__L_OLD" << current++ << "__" << suffix;
     40        if ( stmt && ! stmt->get_labels().empty() ) {
     41                os << "_" << stmt->get_labels().front() << "__";
     42        } // if
     43        string ret = os.str();
     44        Label l( ret );
     45        l.get_attributes().push_back( new Attribute( "unused" ) );
     46        return l;
     47}
    6648} // namespace ControlStruct
    6749
    6850// Local Variables: //
    69 // tab-width: 4 //
    7051// mode: c++ //
    71 // compile-command: "make install" //
    7252// End: //
  • src/ControlStruct/LabelGenerator.h

    r3e5db5b4 r34c32f0  
    99// Author           : Rodolfo G. Esteves
    1010// Created On       : Mon May 18 07:44:20 2015
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Mon Nov  8 10:16:00 2021
    13 // Update Count     : 8
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:30:10 2022
     13// Update Count     : 16
    1414//
    1515
     
    2121
    2222class Statement;
     23
    2324namespace ast {
    24         class Stmt;
    25         class Label;
     25class Stmt;
     26class Label;
    2627}
    2728
    2829namespace ControlStruct {
    29 
    3030class LabelGenerator {
    3131        static int current;
    3232        static LabelGenerator *labelGenerator;
    33 protected:
     33  protected:
    3434        LabelGenerator() {}
    35 public:
     35  public:
    3636        static LabelGenerator *getGenerator();
    3737        static Label newLabel(std::string suffix, Statement * stmt = nullptr);
    38         static ast::Label newLabel( const std::string&, const ast::Stmt * );
    39         static void reset() { current = 0; }
    40         static void rewind() { current--; }
    4138};
    42 
    4339} // namespace ControlStruct
    4440
  • src/ControlStruct/MultiLevelExit.cpp

    r3e5db5b4 r34c32f0  
    99// Author           : Andrew Beach
    1010// Created On       : Mon Nov  1 13:48:00 2021
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Mon Nov  8 10:56:00 2021
    13 // Update Count     : 2
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:35:08 2022
     13// Update Count     : 28
    1414//
    1515
     
    1818#include "AST/Pass.hpp"
    1919#include "AST/Stmt.hpp"
    20 #include "ControlStruct/LabelGenerator.h"
     20#include "LabelGeneratorNew.hpp"
    2121
    2222#include <set>
     23using namespace std;
     24using namespace ast;
    2325
    2426namespace ControlStruct {
    25 
    26 namespace {
    27 
    2827class Entry {
    29 public:
    30         const ast::Stmt * stmt;
    31 private:
     28  public:
     29        const Stmt * stmt;
     30  private:
    3231        // Organized like a manual ADT. Avoids creating a bunch of dead data.
    3332        struct Target {
    34                 ast::Label label;
     33                Label label;
    3534                bool used = false;
    36                 Target( const ast::Label & label ) : label( label ) {}
     35                Target( const Label & label ) : label( label ) {}
    3736                Target() : label( CodeLocation() ) {}
    3837        };
     
    4140
    4241        enum Kind {
    43                 ForStmt, WhileStmt, CompoundStmt, IfStmt, CaseStmt, SwitchStmt, TryStmt
     42                ForStmtK, WhileStmtK, CompoundStmtK, IfStmtK, CaseStmtK, SwitchStmtK, TryStmtK
    4443        } kind;
    4544
    4645        bool fallDefaultValid = true;
    4746
    48         static ast::Label & useTarget( Target & target ) {
     47        static Label & useTarget( Target & target ) {
    4948                target.used = true;
    5049                return target.label;
    5150        }
    5251
    53 public:
    54         Entry( const ast::ForStmt * stmt, ast::Label breakExit, ast::Label contExit ) :
    55                 stmt( stmt ), firstTarget( breakExit ), secondTarget( contExit ), kind( ForStmt ) {}
    56         Entry( const ast::WhileStmt * stmt, ast::Label breakExit, ast::Label contExit ) :
    57                 stmt( stmt ), firstTarget( breakExit ), secondTarget( contExit ), kind( WhileStmt ) {}
    58         Entry( const ast::CompoundStmt *stmt, ast::Label breakExit ) :
    59                 stmt( stmt ), firstTarget( breakExit ), secondTarget(), kind( CompoundStmt ) {}
    60         Entry( const ast::IfStmt *stmt, ast::Label breakExit ) :
    61                 stmt( stmt ), firstTarget( breakExit ), secondTarget(), kind( IfStmt ) {}
    62         Entry( const ast::CaseStmt *stmt, ast::Label fallExit ) :
    63                 stmt( stmt ), firstTarget( fallExit ), secondTarget(), kind( CaseStmt ) {}
    64         Entry( const ast::SwitchStmt *stmt, ast::Label breakExit, ast::Label fallDefaultExit ) :
    65                 stmt( stmt ), firstTarget( breakExit ), secondTarget( fallDefaultExit ), kind( SwitchStmt ) {}
    66         Entry( const ast::TryStmt *stmt, ast::Label breakExit ) :
    67                 stmt( stmt ), firstTarget( breakExit ), secondTarget(), kind( TryStmt ) {}
    68 
    69         bool isContTarget() const { return kind <= WhileStmt; }
    70         bool isBreakTarget() const { return CaseStmt != kind; }
    71         bool isFallTarget() const { return CaseStmt == kind; }
    72         bool isFallDefaultTarget() const { return SwitchStmt == kind; }
     52  public:
     53        Entry( const ForStmt * stmt, Label breakExit, Label contExit ) :
     54                stmt( stmt ), firstTarget( breakExit ), secondTarget( contExit ), kind( ForStmtK ) {}
     55        Entry( const WhileStmt * stmt, Label breakExit, Label contExit ) :
     56                stmt( stmt ), firstTarget( breakExit ), secondTarget( contExit ), kind( WhileStmtK ) {}
     57        Entry( const CompoundStmt *stmt, Label breakExit ) :
     58                stmt( stmt ), firstTarget( breakExit ), secondTarget(), kind( CompoundStmtK ) {}
     59        Entry( const IfStmt *stmt, Label breakExit ) :
     60                stmt( stmt ), firstTarget( breakExit ), secondTarget(), kind( IfStmtK ) {}
     61        Entry( const CaseStmt *stmt, Label fallExit ) :
     62                stmt( stmt ), firstTarget( fallExit ), secondTarget(), kind( CaseStmtK ) {}
     63        Entry( const SwitchStmt *stmt, Label breakExit, Label fallDefaultExit ) :
     64                stmt( stmt ), firstTarget( breakExit ), secondTarget( fallDefaultExit ), kind( SwitchStmtK ) {}
     65        Entry( const TryStmt *stmt, Label breakExit ) :
     66                stmt( stmt ), firstTarget( breakExit ), secondTarget(), kind( TryStmtK ) {}
     67
     68        bool isContTarget() const { return kind <= WhileStmtK; }
     69        bool isBreakTarget() const { return kind != CaseStmtK; }
     70        bool isFallTarget() const { return kind == CaseStmtK; }
     71        bool isFallDefaultTarget() const { return kind == SwitchStmtK; }
    7372
    7473        // These routines set a target as being "used" by a BranchStmt
    75         ast::Label useContExit() { assert( kind <= WhileStmt ); return useTarget(secondTarget); }
    76         ast::Label useBreakExit() { assert( CaseStmt != kind ); return useTarget(firstTarget); }
    77         ast::Label useFallExit() { assert( CaseStmt == kind );  return useTarget(firstTarget); }
    78         ast::Label useFallDefaultExit() { assert( SwitchStmt == kind ); return useTarget(secondTarget); }
     74        Label useContExit() { assert( kind <= WhileStmtK ); return useTarget(secondTarget); }
     75        Label useBreakExit() { assert( kind != CaseStmtK ); return useTarget(firstTarget); }
     76        Label useFallExit() { assert( kind == CaseStmtK );  return useTarget(firstTarget); }
     77        Label useFallDefaultExit() { assert( kind == SwitchStmtK ); return useTarget(secondTarget); }
    7978
    8079        // These routines check if a specific label for a statement is used by a BranchStmt
    81         bool isContUsed() const { assert( kind <= WhileStmt ); return secondTarget.used; }
    82         bool isBreakUsed() const { assert( CaseStmt != kind ); return firstTarget.used; }
    83         bool isFallUsed() const { assert( CaseStmt == kind ); return firstTarget.used; }
    84         bool isFallDefaultUsed() const { assert( SwitchStmt == kind ); return secondTarget.used; }
     80        bool isContUsed() const { assert( kind <= WhileStmtK ); return secondTarget.used; }
     81        bool isBreakUsed() const { assert( kind != CaseStmtK ); return firstTarget.used; }
     82        bool isFallUsed() const { assert( kind == CaseStmtK ); return firstTarget.used; }
     83        bool isFallDefaultUsed() const { assert( kind == SwitchStmtK ); return secondTarget.used; }
    8584        void seenDefault() { fallDefaultValid = false; }
    8685        bool isFallDefaultValid() const { return fallDefaultValid; }
    8786};
    8887
    89 // Helper predicates used in std::find_if calls (it doesn't take methods):
     88// Helper predicates used in find_if calls (it doesn't take methods):
    9089bool isBreakTarget( const Entry & entry ) {
    9190        return entry.isBreakTarget();
     
    105104
    106105struct MultiLevelExitCore final :
    107                 public ast::WithVisitorRef<MultiLevelExitCore>,
    108                 public ast::WithShortCircuiting, public ast::WithGuards {
     106        public WithVisitorRef<MultiLevelExitCore>,
     107        public WithShortCircuiting, public WithGuards {
    109108        MultiLevelExitCore( const LabelToStmt & lt );
    110109
    111         void previsit( const ast::FunctionDecl * );
    112 
    113         const ast::CompoundStmt * previsit( const ast::CompoundStmt * );
    114         const ast::BranchStmt * postvisit( const ast::BranchStmt * );
    115         void previsit( const ast::WhileStmt * );
    116         const ast::WhileStmt * postvisit( const ast::WhileStmt * );
    117         void previsit( const ast::ForStmt * );
    118         const ast::ForStmt * postvisit( const ast::ForStmt * );
    119         const ast::CaseStmt * previsit( const ast::CaseStmt * );
    120         void previsit( const ast::IfStmt * );
    121         const ast::IfStmt * postvisit( const ast::IfStmt * );
    122         void previsit( const ast::SwitchStmt * );
    123         const ast::SwitchStmt * postvisit( const ast::SwitchStmt * );
    124         void previsit( const ast::ReturnStmt * );
    125         void previsit( const ast::TryStmt * );
    126         void postvisit( const ast::TryStmt * );
    127         void previsit( const ast::FinallyStmt * );
    128 
    129         const ast::Stmt * mutateLoop( const ast::Stmt * body, Entry& );
     110        void previsit( const FunctionDecl * );
     111
     112        const CompoundStmt * previsit( const CompoundStmt * );
     113        const BranchStmt * postvisit( const BranchStmt * );
     114        void previsit( const WhileStmt * );
     115        const WhileStmt * postvisit( const WhileStmt * );
     116        void previsit( const ForStmt * );
     117        const ForStmt * postvisit( const ForStmt * );
     118        const CaseStmt * previsit( const CaseStmt * );
     119        void previsit( const IfStmt * );
     120        const IfStmt * postvisit( const IfStmt * );
     121        void previsit( const SwitchStmt * );
     122        const SwitchStmt * postvisit( const SwitchStmt * );
     123        void previsit( const ReturnStmt * );
     124        void previsit( const TryStmt * );
     125        void postvisit( const TryStmt * );
     126        void previsit( const FinallyStmt * );
     127
     128        const Stmt * mutateLoop( const Stmt * body, Entry& );
    130129
    131130        const LabelToStmt & target_table;
    132         std::set<ast::Label> fallthrough_labels;
    133         std::vector<Entry> enclosing_control_structures;
    134         ast::Label break_label;
     131        set<Label> fallthrough_labels;
     132        vector<Entry> enclosing_control_structures;
     133        Label break_label;
    135134        bool inFinally;
    136135
     
    140139        const LoopNode * posthandleLoopStmt( const LoopNode * loopStmt );
    141140
    142         std::list<ast::ptr<ast::Stmt>> fixBlock(
    143                 const std::list<ast::ptr<ast::Stmt>> & kids, bool caseClause );
     141        list<ptr<Stmt>> fixBlock(
     142                const list<ptr<Stmt>> & kids, bool caseClause );
    144143
    145144        template<typename UnaryPredicate>
    146145        auto findEnclosingControlStructure( UnaryPredicate pred ) {
    147                 return std::find_if( enclosing_control_structures.rbegin(),
    148                         enclosing_control_structures.rend(), pred );
     146                return find_if( enclosing_control_structures.rbegin(),
     147                                                enclosing_control_structures.rend(), pred );
    149148        }
    150149};
    151150
    152 ast::NullStmt * labelledNullStmt(
    153                 const CodeLocation & cl, const ast::Label & label ) {
    154         return new ast::NullStmt( cl, std::vector<ast::Label>{ label } );
     151NullStmt * labelledNullStmt(
     152        const CodeLocation & cl, const Label & label ) {
     153        return new NullStmt( cl, vector<Label>{ label } );
    155154}
    156155
     
    160159{}
    161160
    162 void MultiLevelExitCore::previsit( const ast::FunctionDecl * ) {
     161void MultiLevelExitCore::previsit( const FunctionDecl * ) {
    163162        visit_children = false;
    164163}
    165164
    166 const ast::CompoundStmt * MultiLevelExitCore::previsit(
    167                 const ast::CompoundStmt * stmt ) {
     165const CompoundStmt * MultiLevelExitCore::previsit(
     166        const CompoundStmt * stmt ) {
    168167        visit_children = false;
    169168
     
    171170        bool isLabeled = !stmt->labels.empty();
    172171        if ( isLabeled ) {
    173                 ast::Label breakLabel = LabelGenerator::newLabel( "blockBreak", stmt );
     172                Label breakLabel = newLabel( "blockBreak", stmt );
    174173                enclosing_control_structures.emplace_back( stmt, breakLabel );
    175174                GuardAction( [this]() { enclosing_control_structures.pop_back(); } );
    176175        }
    177176
    178         auto mutStmt = ast::mutate( stmt );
     177        auto mutStmt = mutate( stmt );
    179178        // A child statement may set the break label.
    180         mutStmt->kids = std::move( fixBlock( stmt->kids, false ) );
     179        mutStmt->kids = move( fixBlock( stmt->kids, false ) );
    181180
    182181        if ( isLabeled ) {
     
    191190
    192191size_t getUnusedIndex(
    193                 const ast::Stmt * stmt, const ast::Label & originalTarget ) {
     192        const Stmt * stmt, const Label & originalTarget ) {
    194193        const size_t size = stmt->labels.size();
    195194
    196         // If the label is empty, we can skip adding the unused attribute:
    197         if ( originalTarget.empty() ) return size;
     195        // If the label is empty, do not add unused attribute.
     196  if ( originalTarget.empty() ) return size;
    198197
    199198        // Search for a label that matches the originalTarget.
    200199        for ( size_t i = 0 ; i < size ; ++i ) {
    201                 const ast::Label & label = stmt->labels[i];
     200                const Label & label = stmt->labels[i];
    202201                if ( label == originalTarget ) {
    203                         for ( const ast::Attribute * attr : label.attributes ) {
     202                        for ( const Attribute * attr : label.attributes ) {
    204203                                if ( attr->name == "unused" ) return size;
    205204                        }
     
    208207        }
    209208        assertf( false, "Could not find label '%s' on statement %s",
    210                 originalTarget.name.c_str(), toString( stmt ).c_str() );
    211 }
    212 
    213 const ast::Stmt * addUnused(
    214                 const ast::Stmt * stmt, const ast::Label & originalTarget ) {
     209                         originalTarget.name.c_str(), toString( stmt ).c_str() );
     210}
     211
     212const Stmt * addUnused(
     213        const Stmt * stmt, const Label & originalTarget ) {
    215214        size_t i = getUnusedIndex( stmt, originalTarget );
    216215        if ( i == stmt->labels.size() ) {
    217216                return stmt;
    218217        }
    219         ast::Stmt * mutStmt = ast::mutate( stmt );
    220         mutStmt->labels[i].attributes.push_back( new ast::Attribute( "unused" ) );
     218        Stmt * mutStmt = mutate( stmt );
     219        mutStmt->labels[i].attributes.push_back( new Attribute( "unused" ) );
    221220        return mutStmt;
    222221}
     
    224223// This routine updates targets on enclosing control structures to indicate which
    225224//     label is used by the BranchStmt that is passed
    226 const ast::BranchStmt * MultiLevelExitCore::postvisit( const ast::BranchStmt * stmt ) {
    227         std::vector<Entry>::reverse_iterator targetEntry =
     225const BranchStmt * MultiLevelExitCore::postvisit( const BranchStmt * stmt ) {
     226        vector<Entry>::reverse_iterator targetEntry =
    228227                enclosing_control_structures.rend();
    229228
    230229        // Labels on different stmts require different approaches to access
    231230        switch ( stmt->kind ) {
    232         case ast::BranchStmt::Goto:
     231          case BranchStmt::Goto:
    233232                return stmt;
    234         case ast::BranchStmt::Continue:
    235         case ast::BranchStmt::Break: {
    236                 bool isContinue = stmt->kind == ast::BranchStmt::Continue;
    237                 // Handle unlabeled break and continue.
    238                 if ( stmt->target.empty() ) {
    239                         if ( isContinue ) {
    240                                 targetEntry = findEnclosingControlStructure( isContinueTarget );
    241                         } else {
    242                                 if ( enclosing_control_structures.empty() ) {
    243                                         SemanticError( stmt->location,
    244                                                 "'break' outside a loop, 'switch', or labelled block" );
    245                                 }
    246                                 targetEntry = findEnclosingControlStructure( isBreakTarget );
    247                         }
    248                 // Handle labeled break and continue.
    249                 } else {
    250                         // Lookup label in table to find attached control structure.
    251                         targetEntry = findEnclosingControlStructure(
    252                                 [ targetStmt = target_table.at(stmt->target) ](auto entry){
    253                                         return entry.stmt == targetStmt;
    254                                 } );
    255                 }
    256                 // Ensure that selected target is valid.
    257                 if ( targetEntry == enclosing_control_structures.rend() || ( isContinue && !isContinueTarget( *targetEntry ) ) ) {
    258                         SemanticError(
    259                                 stmt->location,
    260                                 toString( (isContinue ? "'continue'" : "'break'"),
    261                                         " target must be an enclosing ",
    262                                         (isContinue ? "loop: " : "control structure: "),
    263                                         stmt->originalTarget ) );
    264                 }
    265                 break;
    266         }
    267         // handle fallthrough in case/switch stmts
    268         case ast::BranchStmt::FallThrough: {
    269                 targetEntry = findEnclosingControlStructure( isFallthroughTarget );
    270                 // Check that target is valid.
    271                 if ( targetEntry == enclosing_control_structures.rend() ) {
    272                         SemanticError( stmt->location, "'fallthrough' must be enclosed in a 'switch' or 'choose'" );
    273                 }
    274                 if ( !stmt->target.empty() ) {
    275                         // Labelled fallthrough: target must be a valid fallthough label.
    276                         if ( !fallthrough_labels.count( stmt->target ) ) {
    277                                 SemanticError( stmt->location, toString( "'fallthrough' target must be a later case statement: ", stmt->originalTarget ) );
    278                         }
    279                         return new ast::BranchStmt(
    280                                 stmt->location, ast::BranchStmt::Goto, stmt->originalTarget );
    281                 }
    282                 break;
    283         }
    284         case ast::BranchStmt::FallThroughDefault: {
    285                 targetEntry = findEnclosingControlStructure( isFallthroughDefaultTarget );
    286 
    287                 // Check that this is in a switch or choose statement.
    288                 if ( targetEntry == enclosing_control_structures.rend() ) {
    289                         SemanticError( stmt->location, "'fallthrough' must be enclosed in a 'switch' or 'choose'" );
    290                 }
    291 
    292                 // Check that the switch or choose has a default clause.
    293                 auto switchStmt = strict_dynamic_cast< const ast::SwitchStmt * >(
    294                         targetEntry->stmt );
    295                 bool foundDefault = false;
    296                 for ( auto subStmt : switchStmt->stmts ) {
    297                         const ast::CaseStmt * caseStmt = subStmt.strict_as<ast::CaseStmt>();
    298                         if ( caseStmt->isDefault() ) {
    299                                 foundDefault = true;
    300                                 break;
    301                         }
    302                 }
    303                 if ( !foundDefault ) {
    304                         SemanticError( stmt->location, "'fallthrough default' must be enclosed in a 'switch' or 'choose' control structure with a 'default' clause" );
    305                 }
    306                 break;
    307         }
    308         default:
     233          case BranchStmt::Continue:
     234          case BranchStmt::Break: {
     235                  bool isContinue = stmt->kind == BranchStmt::Continue;
     236                  // Handle unlabeled break and continue.
     237                  if ( stmt->target.empty() ) {
     238                          if ( isContinue ) {
     239                                  targetEntry = findEnclosingControlStructure( isContinueTarget );
     240                          } else {
     241                                  if ( enclosing_control_structures.empty() ) {
     242                                          SemanticError( stmt->location,
     243                                                                         "'break' outside a loop, 'switch', or labelled block" );
     244                                  }
     245                                  targetEntry = findEnclosingControlStructure( isBreakTarget );
     246                          }
     247                          // Handle labeled break and continue.
     248                  } else {
     249                          // Lookup label in table to find attached control structure.
     250                          targetEntry = findEnclosingControlStructure(
     251                                  [ targetStmt = target_table.at(stmt->target) ](auto entry){
     252                                          return entry.stmt == targetStmt;
     253                                  } );
     254                  }
     255                  // Ensure that selected target is valid.
     256                  if ( targetEntry == enclosing_control_structures.rend() || ( isContinue && !isContinueTarget( *targetEntry ) ) ) {
     257                          SemanticError( stmt->location, toString( (isContinue ? "'continue'" : "'break'"),
     258                                                        " target must be an enclosing ", (isContinue ? "loop: " : "control structure: "),
     259                                                        stmt->originalTarget ) );
     260                  }
     261                  break;
     262          }
     263          // handle fallthrough in case/switch stmts
     264          case BranchStmt::FallThrough: {
     265                  targetEntry = findEnclosingControlStructure( isFallthroughTarget );
     266                  // Check that target is valid.
     267                  if ( targetEntry == enclosing_control_structures.rend() ) {
     268                          SemanticError( stmt->location, "'fallthrough' must be enclosed in a 'switch' or 'choose'" );
     269                  }
     270                  if ( !stmt->target.empty() ) {
     271                          // Labelled fallthrough: target must be a valid fallthough label.
     272                          if ( !fallthrough_labels.count( stmt->target ) ) {
     273                                  SemanticError( stmt->location, toString( "'fallthrough' target must be a later case statement: ",
     274                                                                                                                   stmt->originalTarget ) );
     275                          }
     276                          return new BranchStmt(
     277                                  stmt->location, BranchStmt::Goto, stmt->originalTarget );
     278                  }
     279                  break;
     280          }
     281          case BranchStmt::FallThroughDefault: {
     282                  targetEntry = findEnclosingControlStructure( isFallthroughDefaultTarget );
     283
     284                  // Check if in switch or choose statement.
     285                  if ( targetEntry == enclosing_control_structures.rend() ) {
     286                          SemanticError( stmt->location, "'fallthrough' must be enclosed in a 'switch' or 'choose'" );
     287                  }
     288
     289                  // Check if switch or choose has default clause.
     290                  auto switchStmt = strict_dynamic_cast< const SwitchStmt * >( targetEntry->stmt );
     291                  bool foundDefault = false;
     292                  for ( auto subStmt : switchStmt->stmts ) {
     293                          const CaseStmt * caseStmt = subStmt.strict_as<CaseStmt>();
     294                          if ( caseStmt->isDefault() ) {
     295                                  foundDefault = true;
     296                                  break;
     297                          }
     298                  }
     299                  if ( ! foundDefault ) {
     300                          SemanticError( stmt->location, "'fallthrough default' must be enclosed in a 'switch' or 'choose'"
     301                                                         "control structure with a 'default' clause" );
     302                  }
     303                  break;
     304          }
     305          default:
    309306                assert( false );
    310307        }
    311308
    312309        // Branch error checks: get the appropriate label name:
    313         // (This label will always be replaced.)
    314         ast::Label exitLabel( CodeLocation(), "" );
     310        // (This label is always replaced.)
     311        Label exitLabel( CodeLocation(), "" );
    315312        switch ( stmt->kind ) {
    316         case ast::BranchStmt::Break:
     313          case BranchStmt::Break:
    317314                assert( !targetEntry->useBreakExit().empty() );
    318315                exitLabel = targetEntry->useBreakExit();
    319316                break;
    320         case ast::BranchStmt::Continue:
     317          case BranchStmt::Continue:
    321318                assert( !targetEntry->useContExit().empty() );
    322319                exitLabel = targetEntry->useContExit();
    323320                break;
    324         case ast::BranchStmt::FallThrough:
     321          case BranchStmt::FallThrough:
    325322                assert( !targetEntry->useFallExit().empty() );
    326323                exitLabel = targetEntry->useFallExit();
    327324                break;
    328         case ast::BranchStmt::FallThroughDefault:
     325          case BranchStmt::FallThroughDefault:
    329326                assert( !targetEntry->useFallDefaultExit().empty() );
    330327                exitLabel = targetEntry->useFallDefaultExit();
    331328                // Check that fallthrough default comes before the default clause.
    332329                if ( !targetEntry->isFallDefaultValid() ) {
    333                         SemanticError( stmt->location,
    334                                 "'fallthrough default' must precede the 'default' clause" );
     330                        SemanticError( stmt->location, "'fallthrough default' must precede the 'default' clause" );
    335331                }
    336332                break;
    337         default:
     333          default:
    338334                assert(0);
    339335        }
     
    342338        targetEntry->stmt = addUnused( targetEntry->stmt, stmt->originalTarget );
    343339
    344         // Replace this with a goto to make later passes more uniform.
    345         return new ast::BranchStmt( stmt->location, ast::BranchStmt::Goto, exitLabel );
    346 }
    347 
    348 void MultiLevelExitCore::previsit( const ast::WhileStmt * stmt ) {
     340        // Replace with goto to make later passes more uniform.
     341        return new BranchStmt( stmt->location, BranchStmt::Goto, exitLabel );
     342}
     343
     344void MultiLevelExitCore::previsit( const WhileStmt * stmt ) {
    349345        return prehandleLoopStmt( stmt );
    350346}
    351347
    352 const ast::WhileStmt * MultiLevelExitCore::postvisit( const ast::WhileStmt * stmt ) {
     348const WhileStmt * MultiLevelExitCore::postvisit( const WhileStmt * stmt ) {
    353349        return posthandleLoopStmt( stmt );
    354350}
    355351
    356 void MultiLevelExitCore::previsit( const ast::ForStmt * stmt ) {
     352void MultiLevelExitCore::previsit( const ForStmt * stmt ) {
    357353        return prehandleLoopStmt( stmt );
    358354}
    359355
    360 const ast::ForStmt * MultiLevelExitCore::postvisit( const ast::ForStmt * stmt ) {
     356const ForStmt * MultiLevelExitCore::postvisit( const ForStmt * stmt ) {
    361357        return posthandleLoopStmt( stmt );
    362358}
     
    364360// Mimic what the built-in push_front would do anyways. It is O(n).
    365361void push_front(
    366                 std::vector<ast::ptr<ast::Stmt>> & vec, const ast::Stmt * element ) {
     362        vector<ptr<Stmt>> & vec, const Stmt * element ) {
    367363        vec.emplace_back( nullptr );
    368364        for ( size_t i = vec.size() - 1 ; 0 < i ; --i ) {
    369                 vec[ i ] = std::move( vec[ i - 1 ] );
     365                vec[ i ] = move( vec[ i - 1 ] );
    370366        }
    371367        vec[ 0 ] = element;
    372368}
    373369
    374 const ast::CaseStmt * MultiLevelExitCore::previsit( const ast::CaseStmt * stmt ) {
     370const CaseStmt * MultiLevelExitCore::previsit( const CaseStmt * stmt ) {
    375371        visit_children = false;
    376372
    377         // If it is the default, mark the default as seen.
     373        // If default, mark seen.
    378374        if ( stmt->isDefault() ) {
    379375                assert( !enclosing_control_structures.empty() );
     
    382378
    383379        // The cond may not exist, but if it does update it now.
    384         visitor->maybe_accept( stmt, &ast::CaseStmt::cond );
     380        visitor->maybe_accept( stmt, &CaseStmt::cond );
    385381
    386382        // Just save the mutated node for simplicity.
    387         ast::CaseStmt * mutStmt = ast::mutate( stmt );
    388 
    389         ast::Label fallLabel = LabelGenerator::newLabel( "fallThrough", stmt );
    390         if ( !mutStmt->stmts.empty() ) {
     383        CaseStmt * mutStmt = mutate( stmt );
     384
     385        Label fallLabel = newLabel( "fallThrough", stmt );
     386        if ( ! mutStmt->stmts.empty() ) {
    391387                // Ensure that the stack isn't corrupted by exceptions in fixBlock.
    392388                auto guard = makeFuncGuard(
    393389                        [&](){ enclosing_control_structures.emplace_back( mutStmt, fallLabel ); },
    394390                        [this](){ enclosing_control_structures.pop_back(); }
    395                 );
     391                        );
    396392
    397393                // These should already be in a block.
    398                 auto block = ast::mutate( mutStmt->stmts.front().strict_as<ast::CompoundStmt>() );
     394                auto block = mutate( mutStmt->stmts.front().strict_as<CompoundStmt>() );
    399395                block->kids = fixBlock( block->kids, true );
    400396
    401397                // Add fallthrough label if necessary.
    402                 assert( !enclosing_control_structures.empty() );
     398                assert( ! enclosing_control_structures.empty() );
    403399                Entry & entry = enclosing_control_structures.back();
    404400                if ( entry.isFallUsed() ) {
     
    407403                }
    408404        }
    409         assert( !enclosing_control_structures.empty() );
     405        assert( ! enclosing_control_structures.empty() );
    410406        Entry & entry = enclosing_control_structures.back();
    411         assertf( dynamic_cast< const ast::SwitchStmt * >( entry.stmt ),
    412                 "Control structure enclosing a case clause must be a switch, but is: %s",
    413                 toString( entry.stmt ).c_str() );
     407        assertf( dynamic_cast< const SwitchStmt * >( entry.stmt ),
     408                         "Control structure enclosing a case clause must be a switch, but is: %s",
     409                         toString( entry.stmt ).c_str() );
    414410        if ( mutStmt->isDefault() ) {
    415411                if ( entry.isFallDefaultUsed() ) {
    416412                        // Add fallthrough default label if necessary.
    417413                        push_front( mutStmt->stmts, labelledNullStmt(
    418                                 stmt->location, entry.useFallDefaultExit()
    419                         ) );
     414                                                        stmt->location, entry.useFallDefaultExit()
     415                                                        ) );
    420416                }
    421417        }
     
    423419}
    424420
    425 void MultiLevelExitCore::previsit( const ast::IfStmt * stmt ) {
     421void MultiLevelExitCore::previsit( const IfStmt * stmt ) {
    426422        bool labeledBlock = !stmt->labels.empty();
    427423        if ( labeledBlock ) {
    428                 ast::Label breakLabel = LabelGenerator::newLabel( "blockBreak", stmt );
     424                Label breakLabel = newLabel( "blockBreak", stmt );
    429425                enclosing_control_structures.emplace_back( stmt, breakLabel );
    430426                GuardAction( [this](){ enclosing_control_structures.pop_back(); } );
     
    432428}
    433429
    434 const ast::IfStmt * MultiLevelExitCore::postvisit( const ast::IfStmt * stmt ) {
     430const IfStmt * MultiLevelExitCore::postvisit( const IfStmt * stmt ) {
    435431        bool labeledBlock = !stmt->labels.empty();
    436432        if ( labeledBlock ) {
     
    443439}
    444440
    445 bool isDefaultCase( const ast::ptr<ast::Stmt> & stmt ) {
    446         const ast::CaseStmt * caseStmt = stmt.strict_as<ast::CaseStmt>();
     441bool isDefaultCase( const ptr<Stmt> & stmt ) {
     442        const CaseStmt * caseStmt = stmt.strict_as<CaseStmt>();
    447443        return caseStmt->isDefault();
    448444}
    449445
    450 void MultiLevelExitCore::previsit( const ast::SwitchStmt * stmt ) {
    451         ast::Label label = LabelGenerator::newLabel( "switchBreak", stmt );
    452         auto it = std::find_if( stmt->stmts.rbegin(), stmt->stmts.rend(), isDefaultCase );
    453 
    454         const ast::CaseStmt * defaultCase = it != stmt->stmts.rend()
    455                 ? (it)->strict_as<ast::CaseStmt>() : nullptr;
    456         ast::Label defaultLabel = defaultCase
    457                 ? LabelGenerator::newLabel( "fallThroughDefault", defaultCase )
    458                 : ast::Label( stmt->location, "" );
     446void MultiLevelExitCore::previsit( const SwitchStmt * stmt ) {
     447        Label label = newLabel( "switchBreak", stmt );
     448        auto it = find_if( stmt->stmts.rbegin(), stmt->stmts.rend(), isDefaultCase );
     449
     450        const CaseStmt * defaultCase = it != stmt->stmts.rend()
     451                ? (it)->strict_as<CaseStmt>() : nullptr;
     452        Label defaultLabel = defaultCase
     453                ? newLabel( "fallThroughDefault", defaultCase )
     454                : Label( stmt->location, "" );
    459455        enclosing_control_structures.emplace_back( stmt, label, defaultLabel );
    460456        GuardAction( [this]() { enclosing_control_structures.pop_back(); } );
    461457
    462458        // Collect valid labels for fallthrough. It starts with all labels at
    463         // this level, then removed as we see them in traversal.
    464         for ( const ast::Stmt * stmt : stmt->stmts ) {
    465                 auto * caseStmt = strict_dynamic_cast< const ast::CaseStmt * >( stmt );
     459        // this level, then remove as each is seen during traversal.
     460        for ( const Stmt * stmt : stmt->stmts ) {
     461                auto * caseStmt = strict_dynamic_cast< const CaseStmt * >( stmt );
    466462                if ( caseStmt->stmts.empty() ) continue;
    467                 auto block = caseStmt->stmts.front().strict_as<ast::CompoundStmt>();
    468                 for ( const ast::Stmt * stmt : block->kids ) {
    469                         for ( const ast::Label & l : stmt->labels ) {
     463                auto block = caseStmt->stmts.front().strict_as<CompoundStmt>();
     464                for ( const Stmt * stmt : block->kids ) {
     465                        for ( const Label & l : stmt->labels ) {
    470466                                fallthrough_labels.insert( l );
    471467                        }
     
    474470}
    475471
    476 const ast::SwitchStmt * MultiLevelExitCore::postvisit( const ast::SwitchStmt * stmt ) {
     472const SwitchStmt * MultiLevelExitCore::postvisit( const SwitchStmt * stmt ) {
    477473        assert( !enclosing_control_structures.empty() );
    478474        Entry & entry = enclosing_control_structures.back();
    479475        assert( entry.stmt == stmt );
    480476
    481         // Only run if we need to generate the break label.
     477        // Only run to generate the break label.
    482478        if ( entry.isBreakUsed() ) {
    483479                // To keep the switch statements uniform (all direct children of a
    484480                // SwitchStmt should be CastStmts), append the exit label and break
    485481                // to the last case, create a default case is there are no cases.
    486                 ast::SwitchStmt * mutStmt = ast::mutate( stmt );
     482                SwitchStmt * mutStmt = mutate( stmt );
    487483                if ( mutStmt->stmts.empty() ) {
    488                         mutStmt->stmts.push_back( new ast::CaseStmt(
    489                                 mutStmt->location, nullptr, {} ));
    490                 }
    491 
    492                 auto caseStmt = mutStmt->stmts.back().strict_as<ast::CaseStmt>();
    493                 auto mutCase = ast::mutate( caseStmt );
     484                        mutStmt->stmts.push_back( new CaseStmt(
     485                                                                                  mutStmt->location, nullptr, {} ));
     486                }
     487
     488                auto caseStmt = mutStmt->stmts.back().strict_as<CaseStmt>();
     489                auto mutCase = mutate( caseStmt );
    494490                mutStmt->stmts.back() = mutCase;
    495491
    496                 ast::Label label( mutCase->location, "breakLabel" );
    497                 auto branch = new ast::BranchStmt( mutCase->location, ast::BranchStmt::Break, label );
     492                Label label( mutCase->location, "breakLabel" );
     493                auto branch = new BranchStmt( mutCase->location, BranchStmt::Break, label );
    498494                branch->labels.push_back( entry.useBreakExit() );
    499495                mutCase->stmts.push_back( branch );
     
    504500}
    505501
    506 void MultiLevelExitCore::previsit( const ast::ReturnStmt * stmt ) {
     502void MultiLevelExitCore::previsit( const ReturnStmt * stmt ) {
    507503        if ( inFinally ) {
    508504                SemanticError( stmt->location, "'return' may not appear in a finally clause" );
     
    510506}
    511507
    512 void MultiLevelExitCore::previsit( const ast::TryStmt * stmt ) {
     508void MultiLevelExitCore::previsit( const TryStmt * stmt ) {
    513509        bool isLabeled = !stmt->labels.empty();
    514510        if ( isLabeled ) {
    515                 ast::Label breakLabel = LabelGenerator::newLabel( "blockBreak", stmt );
     511                Label breakLabel = newLabel( "blockBreak", stmt );
    516512                enclosing_control_structures.emplace_back( stmt, breakLabel );
    517513                GuardAction([this](){ enclosing_control_structures.pop_back(); } );
     
    519515}
    520516
    521 void MultiLevelExitCore::postvisit( const ast::TryStmt * stmt ) {
     517void MultiLevelExitCore::postvisit( const TryStmt * stmt ) {
    522518        bool isLabeled = !stmt->labels.empty();
    523519        if ( isLabeled ) {
     
    529525}
    530526
    531 void MultiLevelExitCore::previsit( const ast::FinallyStmt * ) {
    532         GuardAction([this, old = std::move(enclosing_control_structures)](){
    533                 enclosing_control_structures = std::move(old);
    534         });
    535         enclosing_control_structures = std::vector<Entry>();
     527void MultiLevelExitCore::previsit( const FinallyStmt * ) {
     528        GuardAction([this, old = move(enclosing_control_structures)](){
     529                                        enclosing_control_structures = move(old);
     530                                });
     531        enclosing_control_structures = vector<Entry>();
    536532        GuardValue( inFinally ) = true;
    537533}
    538534
    539 const ast::Stmt * MultiLevelExitCore::mutateLoop(
    540                 const ast::Stmt * body, Entry & entry ) {
     535const Stmt * MultiLevelExitCore::mutateLoop(
     536        const Stmt * body, Entry & entry ) {
    541537        if ( entry.isBreakUsed() ) {
    542538                break_label = entry.useBreakExit();
     
    545541        // if continue is used insert a continue label into the back of the body of the loop
    546542        if ( entry.isContUsed() ) {
    547                 ast::CompoundStmt * new_body = new ast::CompoundStmt( body->location );
     543                CompoundStmt * new_body = new CompoundStmt( body->location );
    548544                // {}
    549545                new_body->kids.push_back( body );
     
    567563        // Remember is loop before going onto mutate the body.
    568564        // The labels will be folded in if they are used.
    569         ast::Label breakLabel = LabelGenerator::newLabel( "loopBreak", loopStmt );
    570         ast::Label contLabel = LabelGenerator::newLabel( "loopContinue", loopStmt );
     565        Label breakLabel = newLabel( "loopBreak", loopStmt );
     566        Label contLabel = newLabel( "loopContinue", loopStmt );
    571567        enclosing_control_structures.emplace_back( loopStmt, breakLabel, contLabel );
    572568        // labels are added temporarily to see if they are used and then added permanently in postvisit if ther are used
     
    583579        assert( entry.stmt == loopStmt );
    584580
    585         // Now we check if the labels are used and add them if so.
    586         return ast::mutate_field(
     581        // Now check if the labels are used and add them if so.
     582        return mutate_field(
    587583                loopStmt, &LoopNode::body, mutateLoop( loopStmt->body, entry ) );
    588584        // this call to mutate_field compares loopStmt->body and the result of mutateLoop
     
    591587}
    592588
    593 std::list<ast::ptr<ast::Stmt>> MultiLevelExitCore::fixBlock(
    594                 const std::list<ast::ptr<ast::Stmt>> & kids, bool is_case_clause ) {
    595         // Unfortunately we can't use the automatic error collection.
     589list<ptr<Stmt>> MultiLevelExitCore::fixBlock(
     590        const list<ptr<Stmt>> & kids, bool is_case_clause ) {
     591        // Unfortunately cannot use automatic error collection.
    596592        SemanticErrorException errors;
    597593
    598         std::list<ast::ptr<ast::Stmt>> ret;
     594        list<ptr<Stmt>> ret;
    599595
    600596        // Manually visit each child.
    601         for ( const ast::ptr<ast::Stmt> & kid : kids ) {
     597        for ( const ptr<Stmt> & kid : kids ) {
    602598                if ( is_case_clause ) {
    603599                        // Once a label is seen, it's no longer a valid for fallthrough.
    604                         for ( const ast::Label & l : kid->labels ) {
     600                        for ( const Label & l : kid->labels ) {
    605601                                fallthrough_labels.erase( l );
    606602                        }
     
    616612                        ret.push_back(
    617613                                labelledNullStmt( ret.back()->location, break_label ) );
    618                         break_label = ast::Label( CodeLocation(), "" );
     614                        break_label = Label( CodeLocation(), "" );
    619615                }
    620616        }
     
    626622}
    627623
    628 } // namespace
    629 
    630 const ast::CompoundStmt * multiLevelExitUpdate(
    631         const ast::CompoundStmt * stmt,
    632                 const LabelToStmt & labelTable ) {
     624const CompoundStmt * multiLevelExitUpdate(
     625        const CompoundStmt * stmt,
     626        const LabelToStmt & labelTable ) {
    633627        // Must start in the body, so FunctionDecls can be a stopping point.
    634         ast::Pass<MultiLevelExitCore> visitor( labelTable );
    635         const ast::CompoundStmt * ret = stmt->accept( visitor );
     628        Pass<MultiLevelExitCore> visitor( labelTable );
     629        const CompoundStmt * ret = stmt->accept( visitor );
    636630        return ret;
    637631}
    638 
    639632} // namespace ControlStruct
    640633
  • src/ControlStruct/MultiLevelExit.hpp

    r3e5db5b4 r34c32f0  
    99// Author           : Andrew Beach
    1010// Created On       : Mon Nov  1 13:49:00 2021
    11 // Last Modified By : Andrew Beach
    12 // Last Modified On : Mon Nov  8 10:53:00 2021
    13 // Update Count     : 3
     11// Last Modified By : Peter A. Buhr
     12// Last Modified On : Mon Jan 31 22:34:06 2022
     13// Update Count     : 6
    1414//
    1515
     
    1919
    2020namespace ast {
    21         class CompoundStmt;
    22         class Label;
    23         class Stmt;
     21class CompoundStmt;
     22class Label;
     23class Stmt;
    2424}
    2525
    2626namespace ControlStruct {
    27 
    2827using LabelToStmt = std::map<ast::Label, const ast::Stmt *>;
    2928
    30 /// Mutate a function body to handle multi-level exits.
    31 const ast::CompoundStmt * multiLevelExitUpdate(
    32         const ast::CompoundStmt *, const LabelToStmt & );
    33 
     29// Mutate a function body to handle multi-level exits.
     30const ast::CompoundStmt * multiLevelExitUpdate( const ast::CompoundStmt *, const LabelToStmt & );
    3431}
    3532
  • src/ControlStruct/module.mk

    r3e5db5b4 r34c32f0  
    1010## Author           : Richard C. Bilson
    1111## Created On       : Mon Jun  1 17:49:17 2015
    12 ## Last Modified By : Henry Xue
    13 ## Last Modified On : Tue Jul 20 04:10:50 2021
    14 ## Update Count     : 5
     12## Last Modified By : Peter A. Buhr
     13## Last Modified On : Sat Jan 29 12:04:19 2022
     14## Update Count     : 7
    1515###############################################################################
    1616
     
    2828        ControlStruct/LabelGenerator.cc \
    2929        ControlStruct/LabelGenerator.h \
     30        ControlStruct/LabelGeneratorNew.cpp \
     31        ControlStruct/LabelGeneratorNew.hpp \
    3032        ControlStruct/MLEMutator.cc \
    3133        ControlStruct/MLEMutator.h \
  • src/Parser/ParseNode.h

    r3e5db5b4 r34c32f0  
    1010// Created On       : Sat May 16 13:28:16 2015
    1111// Last Modified By : Peter A. Buhr
    12 // Last Modified On : Wed Jul 14 17:28:53 2021
    13 // Update Count     : 900
     12// Last Modified On : Sat Jan 29 09:45:56 2022
     13// Update Count     : 901
    1414//
    1515
     
    390390Statement * build_expr( ExpressionNode * ctl );
    391391
    392 struct IfCtrl {
    393         IfCtrl( DeclarationNode * decl, ExpressionNode * condition ) :
     392struct CondCtl {
     393        CondCtl( DeclarationNode * decl, ExpressionNode * condition ) :
    394394                init( decl ? new StatementNode( decl ) : nullptr ), condition( condition ) {}
    395395
     
    409409};
    410410
    411 Expression * build_if_control( IfCtrl * ctl, std::list< Statement * > & init );
    412 Statement * build_if( IfCtrl * ctl, StatementNode * then_stmt, StatementNode * else_stmt );
     411Expression * build_if_control( CondCtl * ctl, std::list< Statement * > & init );
     412Statement * build_if( CondCtl * ctl, StatementNode * then_stmt, StatementNode * else_stmt );
    413413Statement * build_switch( bool isSwitch, ExpressionNode * ctl, StatementNode * stmt );
    414414Statement * build_case( ExpressionNode * ctl );
    415415Statement * build_default();
    416 Statement * build_while( IfCtrl * ctl, StatementNode * stmt );
     416Statement * build_while( CondCtl * ctl, StatementNode * stmt );
    417417Statement * build_do_while( ExpressionNode * ctl, StatementNode * stmt );
    418418Statement * build_for( ForCtrl * forctl, StatementNode * stmt );
  • src/Parser/StatementNode.cc

    r3e5db5b4 r34c32f0  
    1010// Created On       : Sat May 16 14:59:41 2015
    1111// Last Modified By : Peter A. Buhr
    12 // Last Modified On : Sat Oct 24 04:20:55 2020
    13 // Update Count     : 383
     12// Last Modified On : Sat Jan 29 09:45:51 2022
     13// Update Count     : 384
    1414//
    1515
     
    7878} // build_expr
    7979
    80 Expression * build_if_control( IfCtrl * ctl, std::list< Statement * > & init ) {
     80Expression * build_if_control( CondCtl * ctl, std::list< Statement * > & init ) {
    8181        if ( ctl->init != 0 ) {
    8282                buildMoveList( ctl->init, init );
     
    100100} // build_if_control
    101101
    102 Statement * build_if( IfCtrl * ctl, StatementNode * then_stmt, StatementNode * else_stmt ) {
     102Statement * build_if( CondCtl * ctl, StatementNode * then_stmt, StatementNode * else_stmt ) {
    103103        Statement * thenb, * elseb = nullptr;
    104104        std::list< Statement * > branches;
     
    145145} // build_default
    146146
    147 Statement * build_while( IfCtrl * ctl, StatementNode * stmt ) {
     147Statement * build_while( CondCtl * ctl, StatementNode * stmt ) {
    148148        std::list< Statement * > branches;
    149149        buildMoveList< Statement, StatementNode >( stmt, branches );
  • src/Parser/parser.yy

    r3e5db5b4 r34c32f0  
    1010// Created On       : Sat Sep  1 20:22:55 2001
    1111// Last Modified By : Peter A. Buhr
    12 // Last Modified On : Fri Oct 15 09:20:17 2021
    13 // Update Count     : 5163
     12// Last Modified On : Sun Jan 30 09:41:13 2022
     13// Update Count     : 5165
    1414//
    1515
     
    238238        WaitForStmt * wfs;
    239239        Expression * constant;
    240         IfCtrl * ifctl;
     240        CondCtl * ifctl;
    241241        ForCtrl * fctl;
    242242        enum OperKinds compop;
     
    327327%type<en> comma_expression                              comma_expression_opt
    328328%type<en> argument_expression_list_opt  argument_expression_list        argument_expression                     default_initializer_opt
    329 %type<ifctl> if_control_expression
     329%type<ifctl> conditional_declaration
    330330%type<fctl> for_control_expression              for_control_expression_list
    331331%type<compop> inclexcl
     
    11231123
    11241124if_statement:
    1125         IF '(' if_control_expression ')' statement                      %prec THEN
     1125        IF '(' conditional_declaration ')' statement            %prec THEN
    11261126                // explicitly deal with the shift/reduce conflict on if/else
    11271127                { $$ = new StatementNode( build_if( $3, maybe_build_compound( $5 ), nullptr ) ); }
    1128         | IF '(' if_control_expression ')' statement ELSE statement
     1128        | IF '(' conditional_declaration ')' statement ELSE statement
    11291129                { $$ = new StatementNode( build_if( $3, maybe_build_compound( $5 ), maybe_build_compound( $7 ) ) ); }
    11301130        ;
    11311131
    1132 if_control_expression:
     1132conditional_declaration:
    11331133        comma_expression
    1134                 { $$ = new IfCtrl( nullptr, $1 ); }
     1134                { $$ = new CondCtl( nullptr, $1 ); }
    11351135        | c_declaration                                                                         // no semi-colon
    1136                 { $$ = new IfCtrl( $1, nullptr ); }
     1136                { $$ = new CondCtl( $1, nullptr ); }
    11371137        | cfa_declaration                                                                       // no semi-colon
    1138                 { $$ = new IfCtrl( $1, nullptr ); }
     1138                { $$ = new CondCtl( $1, nullptr ); }
    11391139        | declaration comma_expression                                          // semi-colon separated
    1140                 { $$ = new IfCtrl( $1, $2 ); }
     1140                { $$ = new CondCtl( $1, $2 ); }
    11411141        ;
    11421142
     
    11931193iteration_statement:
    11941194        WHILE '(' ')' statement                                                         // CFA => while ( 1 )
    1195                 { $$ = new StatementNode( build_while( new IfCtrl( nullptr, new ExpressionNode( build_constantInteger( *new string( "1" ) ) ) ), maybe_build_compound( $4 ) ) ); }
    1196         | WHILE '(' if_control_expression ')' statement         %prec THEN
     1195                { $$ = new StatementNode( build_while( new CondCtl( nullptr, new ExpressionNode( build_constantInteger( *new string( "1" ) ) ) ), maybe_build_compound( $4 ) ) ); }
     1196        | WHILE '(' conditional_declaration ')' statement       %prec THEN
    11971197                { $$ = new StatementNode( build_while( $3, maybe_build_compound( $5 ) ) ); }
    1198         | WHILE '(' if_control_expression ')' statement ELSE statement // CFA
     1198        | WHILE '(' conditional_declaration ')' statement ELSE statement // CFA
    11991199                { SemanticError( yylloc, "Loop default block is currently unimplemented." ); $$ = nullptr; }
    12001200        | DO statement WHILE '(' ')' ';'                                        // CFA => do while( 1 )
  • tests/concurrent/preempt.cfa

    r3e5db5b4 r34c32f0  
     1#include <clock.hfa>
    12#include <fstream.hfa>
    23#include <kernel.hfa>
     
    2223extern void __cfaabi_check_preemption();
    2324
    24 static volatile int counter = 0;
     25static struct {
     26        volatile int counter;
     27        volatile Time prev;
     28        Duration durations[6];
     29} globals;
    2530
    2631thread worker_t {
    2732        int value;
    28         unsigned spin;
     33        unsigned long long spin;
    2934};
    3035
     
    3540
    3641void main(worker_t & this) {
    37         while(TEST(counter < N)) {
    38                 if(this.spin > 100_000_000) abort | "Worker" | this.value | "has been spinning too long! (" | this.spin | ")";
     42        while(TEST(globals.counter < N)) {
     43                if(this.spin > 50_000_000_000) abort | "Worker" | this.value | "has been spinning too long! (" | this.spin | ")";
    3944                __cfaabi_check_preemption();
    40                 if( (counter % 7) == this.value ) {
     45                if( (globals.counter % 7) == this.value ) {
    4146                        __cfaabi_check_preemption();
    42                         int next = __atomic_add_fetch( &counter, 1, __ATOMIC_SEQ_CST );
     47                        #if !defined(TEST_LONG)
     48                                Time now = timeHiRes();
     49                                Duration diff = now - globals.prev;
     50                                globals.prev = now;
     51                        #endif
     52                        int next = __atomic_add_fetch( &globals.counter, 1, __ATOMIC_SEQ_CST );
    4353                        __cfaabi_check_preemption();
    44                         if( (next % 100) == 0 ) printf("%d\n", (int)next);
     54                        if( (next % 100) == 0 ) {
     55                                #if !defined(TEST_LONG)
     56                                        unsigned idx = next / 100;
     57                                        if (idx >= 6) abort | "Idx from next is invalid: " | idx | "vs" | next;
     58                                        globals.durations[idx] = diff;
     59                                        if(diff > 12`s) serr | "Duration suspiciously large:" | diff;
     60                                #endif
     61                                printf("%d\n", (int)next);
     62
     63                        }
    4564                        __cfaabi_check_preemption();
    4665                        this.spin = 0;
     
    5473int main(int argc, char* argv[]) {
    5574        processor p;
     75        globals.counter = 0;
     76        globals.durations[0] = 0;
     77        globals.durations[1] = 0;
     78        globals.durations[2] = 0;
     79        globals.durations[3] = 0;
     80        globals.durations[4] = 0;
     81        globals.durations[5] = 0;
    5682        {
     83                globals.prev = timeHiRes();
    5784                worker_t w0 = 0;
    5885                worker_t w1 = 1;
  • tools/auto-complete.md

    r3e5db5b4 r34c32f0  
    3232
    3333### Zsh
     34
     351 - Add the following somewhere:
     36    #compdef test.py
     37
     38    _test_py() {
     39        local -a options
     40        options=$($words[1] --list-comp)
     41        _alternative "files:filenames:($options)"
     42    }
     43
     44    _test_py "$@"
     45
     462 - Add the path to that file to the "fpath" environment variable.
     47
     483 - In ~/.zshrc add
     49    autoload -U compinit
     50    compinit
     51
     52*How it works:* I don't know ;P
     53
     54
Note: See TracChangeset for help on using the changeset viewer.