Add convenience function for SPLIT_INDEX constants

Bug: 349828130
Flag: com.android.wm.shell.enable_flexible_split
Test: No effect on behavior, just a small refactor.
Change-Id: I5498519f804f6e8a39374d7cb405eaedcac230d8