InlineUIContainer dosen't correctly render with TextProperty=Wrap

[Danich27rus]

Describe the bug

Right now if Inlines have InlineUIContainer inside and we have TextProperty=Wrap on TextBlock, everything is fine until InlineUIContainer renders.

After that, if i back to smaller width, InlineUiContainer will be rendered incorrectly, on top of my text.

I don't know what to do right now. I thought about using WrapPanel, but that means i need to rewrote whole system in my App.

To Reproduce

1. Create inline.
2. Add runs and InlineUIContainer.
3. Set inlines on some TextBlock with changable width.
4. Try to change size on TextBlock
5. InlineUIContainer will show and never hide.

Expected behavior

1. Create inline.
2. Add runs and InlineUIContainer.
3. Set inlines on some TextBlock with changable width.
4. Try to change size on TextBlock
5. InlineUIContainer will proper show and hide as width changes.

Avalonia version

11.1.3

OS

Windows

Additional context

No response

[Danich27rus]

Looks like this code

                            if (drawable is EmbeddedControlRun controlRun
                                && controlRun.Control is Control control)
                            {
                                control.Arrange(
                                    new Rect(new Point(currentX, currentY),
                                    new Size(control.DesiredSize.Width, textLine.Height)));
                            }

in ArrangeOverride dosen't rearrange control when it calls again because it dosen't store it anywhere. I think it can be solved if there will be some kind of pool of all controls. I can do a PR, but i don't think store list of controls is a good thing.

[Danich27rus]

Created super simple fix that was written in 2 minutes:

        List<Control> _aabb = new();

        protected override Size ArrangeOverride(Size finalSize)
        {
            var scale = LayoutHelper.GetLayoutScale(this);
            var padding = LayoutHelper.RoundLayoutThickness(Padding, scale, scale);

            //Fixes: #11019
            if (finalSize.Width < _constraint.Width)
            {
                _textLayout?.Dispose();
                _textLayout = null;
                _constraint = finalSize.Deflate(padding);
            }

            if (HasComplexContent)
            {
                var currentY = padding.Top;

                foreach (var cont in _aabb)
                {
                    cont.Arrange(
                       new Rect(-cont.Bounds.Width, -cont.Bounds.Height, cont.Bounds.Width, cont.Bounds.Height));
                }

                foreach (var textLine in TextLayout.TextLines)
                {
                    var currentX = padding.Left + textLine.Start;

                    foreach (var run in textLine.TextRuns)
                    {
                        if (run is DrawableTextRun drawable)
                        {
                            if (drawable is EmbeddedControlRun controlRun
                                && controlRun.Control is Control control)
                            {
                                control.Arrange(
                                    new Rect(new Point(currentX, currentY),
                                    new Size(control.DesiredSize.Width, textLine.Height)));
                                if (!_aabb.Contains(control))
                                    _aabb.Add(control);
                            }

                            currentX += drawable.Size.Width;
                        }
                    }

                    currentY += textLine.Height;
                }
            }

            return finalSize;
        }

Works as i wanted.