Summary

Refactored the take_annotations method in ProgramAnnotations to simplify the logic for handling statement annotations. Instead of creating a temporary replacement and then swapping values, the method now takes ownership of the annotations entry directly, clones it if needed for backward jumps, and returns the modified entry.

Type of change

Please check one:

• Performance improvement
• Bug fix (fixes incorrect behavior)
• New feature
• Documentation change with concrete technical impact
• Style, wording, formatting, or typo-only change

Why is this change needed?

The previous implementation of take_annotations was unnecessarily complex, using multiple memory operations (replace, take) and creating temporary objects. This change simplifies the code flow and reduces memory operations by taking ownership of the annotation entry directly.

What was the behavior or documentation before?

The method would retrieve a mutable reference to the annotations, create a temporary replacement, and use std::mem::replace and std::mem::take to swap values around.

What is the behavior or documentation after?

The method now uses take() to get ownership of the annotations directly, clones them only when necessary (for backward jumps), and returns the modified entry without unnecessary memory operations.

Additional context

This change maintains the same functionality while making the code more straightforward and potentially more efficient by reducing memory operations and temporary object creation.