@@ -256,6 +256,75 @@ impl Node { |
| 256 | 256 | } |
| 257 | 257 | } |
| 258 | 258 | |
| 259 | + /// Get the split ratio affecting a window in the given direction. |
| 260 | + /// Returns None if no such split exists. |
| 261 | + pub fn get_split_ratio(&self, window: XWindow, direction: Direction) -> Option<f32> { |
| 262 | + match self { |
| 263 | + Node::Leaf { .. } => None, |
| 264 | + Node::Internal { |
| 265 | + split, |
| 266 | + ratio, |
| 267 | + left, |
| 268 | + right, |
| 269 | + } => { |
| 270 | + // Check if this split is in the right orientation for the direction |
| 271 | + let dominated = match (split, direction) { |
| 272 | + (SplitDirection::Vertical, Direction::Left | Direction::Right) => true, |
| 273 | + (SplitDirection::Horizontal, Direction::Up | Direction::Down) => true, |
| 274 | + _ => false, |
| 275 | + }; |
| 276 | + |
| 277 | + if dominated && (left.contains(window) || right.contains(window)) { |
| 278 | + return Some(*ratio); |
| 279 | + } |
| 280 | + |
| 281 | + // Recurse into children |
| 282 | + if left.contains(window) { |
| 283 | + left.get_split_ratio(window, direction) |
| 284 | + } else if right.contains(window) { |
| 285 | + right.get_split_ratio(window, direction) |
| 286 | + } else { |
| 287 | + None |
| 288 | + } |
| 289 | + } |
| 290 | + } |
| 291 | + } |
| 292 | + |
| 293 | + /// Set the split ratio affecting a window in the given direction. |
| 294 | + /// Returns true if the ratio was set. |
| 295 | + pub fn set_split_ratio(&mut self, window: XWindow, direction: Direction, new_ratio: f32) -> bool { |
| 296 | + match self { |
| 297 | + Node::Leaf { .. } => false, |
| 298 | + Node::Internal { |
| 299 | + split, |
| 300 | + ratio, |
| 301 | + left, |
| 302 | + right, |
| 303 | + } => { |
| 304 | + // Check if this split is in the right orientation for the direction |
| 305 | + let dominated = match (split, direction) { |
| 306 | + (SplitDirection::Vertical, Direction::Left | Direction::Right) => true, |
| 307 | + (SplitDirection::Horizontal, Direction::Up | Direction::Down) => true, |
| 308 | + _ => false, |
| 309 | + }; |
| 310 | + |
| 311 | + if dominated && (left.contains(window) || right.contains(window)) { |
| 312 | + *ratio = new_ratio.clamp(0.1, 0.9); |
| 313 | + return true; |
| 314 | + } |
| 315 | + |
| 316 | + // Recurse into children |
| 317 | + if left.contains(window) { |
| 318 | + left.set_split_ratio(window, direction, new_ratio) |
| 319 | + } else if right.contains(window) { |
| 320 | + right.set_split_ratio(window, direction, new_ratio) |
| 321 | + } else { |
| 322 | + false |
| 323 | + } |
| 324 | + } |
| 325 | + } |
| 326 | + } |
| 327 | + |
| 259 | 328 | /// Swap two windows in the tree. |
| 260 | 329 | pub fn swap(&mut self, a: XWindow, b: XWindow) -> bool { |
| 261 | 330 | // Find and swap the windows |