TimefoldAI / timefold-solver-python

Timefold Solver is an AI constraint solver for Python to optimize the Vehicle Routing Problem, Employee Rostering, Maintenance Scheduling, Task Assignment, School Timetabling, Cloud Optimization, Conference Scheduling, Job Shop Scheduling, Bin Packing and many more planning problems.
https://timefold.ai
Apache License 2.0
36 stars 3 forks source link

jpyinterpreter - make FOR_ITER use Java iterator loop format when possible #8

Open Christopher-Chianelli opened 4 months ago

Christopher-Chianelli commented 4 months ago

Currently, in order to fully support all the forms a Python iterator can take, jpyinterpreter generates the following code for FOR_ITER:

try {
    do {
        TOS' = next(TOS)
         // code in for block
    } while(true);
} catch (StopIteration e) {
    // code after for block
}

This is highly atypical in Java, and the JVM probably would have a harder time optimizing its standard for iterator loop:

while (TOS.hasNext()) {
    TOS' = TOS.next();
    // code in for block
}
 // code after for block

We can look at TOS to see if it a known iterator type (i.e. PythonIterator), and if so, generate the more typical Java loop.