config: add just-mapped window criteria

src/async_engine.rs

@@ -105,7 +105,6 @@ impl AsyncEngine {
                 break;
             }
             self.now.take();
-            self.iteration.fetch_add(1);
             let mut phase = 0;
             while phase < NUM_PHASES {
                 self.queues[phase].swap(&mut *stash);
@@ -121,6 +120,7 @@ impl AsyncEngine {
                     }
                 }
             }
+            self.iteration.fetch_add(1);
             self.yields.swap(&mut *yield_stash);
             while let Some(waker) = yield_stash.pop_front() {
                 waker.wake();
@@ -153,7 +153,7 @@ impl AsyncEngine {
         self.yields.push(waker);
     }
 
-    fn iteration(&self) -> u64 {
+    pub fn iteration(&self) -> u64 {
         self.iteration.get()
     }
 

src/compositor.rs

@@ -18,7 +18,9 @@ use {
         criteria::{
             CritMatcherIds,
             clm::{ClMatcherManager, handle_cl_changes, handle_cl_leaf_events},
-            tlm::{TlMatcherManager, handle_tl_changes, handle_tl_leaf_events},
+            tlm::{
+                TlMatcherManager, handle_tl_changes, handle_tl_just_mapped, handle_tl_leaf_events,
+            },
         },
         damage::{DamageVisualizer, visualize_damage},
         dbus::Dbus,
@@ -484,6 +486,11 @@ fn start_global_event_handlers(
             "tl matcher leaf events",
             handle_tl_leaf_events(state.clone()),
         ),
+        eng.spawn2(
+            "tl matcher just mapped",
+            Phase::Layout,
+            handle_tl_just_mapped(state.clone()),
+        ),
     ]
 }
 

src/config/handler.rs

@@ -2002,6 +2002,7 @@ impl ConfigProxyHandler {
             WindowCriterionIpc::Urgent => mgr.urgent(),
             WindowCriterionIpc::SeatFocus(seat) => mgr.seat_focus(&*self.get_seat(*seat)?),
             WindowCriterionIpc::Fullscreen => mgr.fullscreen(),
+            WindowCriterionIpc::JustMapped => mgr.just_mapped(),
         };
         let cached = Rc::new(CachedCriterion {
             crit: criterion.clone(),

src/criteria/tlm.rs

@@ -15,6 +15,7 @@ use {
                 tlmm_client::TlmMatchClient,
                 tlmm_floating::TlmMatchFloating,
                 tlmm_fullscreen::TlmMatchFullscreen,
+                tlmm_just_mapped::TlmMatchJustMapped,
                 tlmm_kind::TlmMatchKind,
                 tlmm_seat_focus::TlmMatchSeatFocus,
                 tlmm_string::{TlmMatchAppId, TlmMatchTitle},
@@ -49,6 +50,7 @@ bitflags! {
     TL_CHANGED_URGENT      = 1 << 6,
     TL_CHANGED_SEAT_FOCI   = 1 << 7,
     TL_CHANGED_FULLSCREEN  = 1 << 8,
+    TL_CHANGED_JUST_MAPPED = 1 << 9,
 }
 
 type TlmFixedRootMatcher<T> = FixedRootMatcher<ToplevelData, T>;
@@ -57,11 +59,13 @@ pub struct TlMatcherManager {
     ids: Rc<CritMatcherIds>,
     changes: AsyncQueue<Rc<dyn ToplevelNode>>,
     leaf_events: Rc<AsyncQueue<CritLeafEvent<ToplevelData>>>,
+    handle_just_mapped: AsyncQueue<Rc<dyn ToplevelNode>>,
     constant: TlmFixedRootMatcher<CritMatchConstant<ToplevelData>>,
     floating: TlmFixedRootMatcher<TlmMatchFloating>,
     visible: TlmFixedRootMatcher<TlmMatchVisible>,
     urgent: TlmFixedRootMatcher<TlmMatchUrgent>,
     fullscreen: TlmFixedRootMatcher<TlmMatchFullscreen>,
+    just_mapped: TlmFixedRootMatcher<TlmMatchJustMapped>,
     matchers: Rc<RootMatchers>,
 }
 
@@ -94,6 +98,16 @@ pub async fn handle_tl_leaf_events(state: Rc<State>) {
     }
 }
 
+pub async fn handle_tl_just_mapped(state: Rc<State>) {
+    let mgr = &state.tl_matcher_manager;
+    loop {
+        let tl = mgr.handle_just_mapped.pop().await;
+        let data = tl.tl_data();
+        data.just_mapped_scheduled.set(false);
+        data.property_changed(TL_CHANGED_JUST_MAPPED);
+    }
+}
+
 pub type TlmUpstreamNode = dyn CritUpstreamNode<ToplevelData>;
 pub type TlmLeafMatcher = CritLeafMatcher<ToplevelData>;
 
@@ -117,16 +131,19 @@ impl TlMatcherManager {
             visible: bool!(TlmMatchVisible),
             urgent: bool!(TlmMatchUrgent),
             fullscreen: bool!(TlmMatchFullscreen),
+            just_mapped: bool!(TlmMatchJustMapped),
             changes: Default::default(),
             leaf_events: Default::default(),
             ids: ids.clone(),
             matchers,
+            handle_just_mapped: Default::default(),
         }
     }
 
     pub fn clear(&self) {
         self.changes.clear();
         self.leaf_events.clear();
+        self.handle_just_mapped.clear();
     }
 
     pub fn rematch_all(&self, state: &Rc<State>) {
@@ -192,6 +209,7 @@ impl TlMatcherManager {
         fixed_conditional!(TL_CHANGED_VISIBLE, visible);
         fixed_conditional!(TL_CHANGED_URGENT, urgent);
         fixed_conditional!(TL_CHANGED_FULLSCREEN, fullscreen);
+        fixed_conditional!(TL_CHANGED_JUST_MAPPED, just_mapped);
         false
     }
 
@@ -263,6 +281,14 @@ impl TlMatcherManager {
         fixed_conditional!(TL_CHANGED_VISIBLE, visible);
         fixed_conditional!(TL_CHANGED_URGENT, urgent);
         fixed_conditional!(TL_CHANGED_FULLSCREEN, fullscreen);
+        fixed_conditional!(TL_CHANGED_JUST_MAPPED, just_mapped);
+        if changed.contains(TL_CHANGED_JUST_MAPPED)
+            && data.just_mapped()
+            && (self.just_mapped[false].has_downstream() || self.just_mapped[true].has_downstream())
+            && !data.just_mapped_scheduled.replace(true)
+        {
+            self.handle_just_mapped.push(node);
+        }
     }
 
     pub fn title(&self, string: CritLiteralOrRegex) -> Rc<TlmUpstreamNode> {
@@ -297,6 +323,10 @@ impl TlMatcherManager {
         self.urgent[true].clone()
     }
 
+    pub fn just_mapped(&self) -> Rc<TlmUpstreamNode> {
+        self.just_mapped[true].clone()
+    }
+
     pub fn seat_focus(&self, seat: &WlSeatGlobal) -> Rc<TlmUpstreamNode> {
         self.root(TlmMatchSeatFocus::new(seat.id()))
     }

src/criteria/tlm/tlm_matchers.rs

@@ -20,6 +20,7 @@ macro_rules! fixed_root_criterion {
 pub mod tlmm_client;
 pub mod tlmm_floating;
 pub mod tlmm_fullscreen;
+pub mod tlmm_just_mapped;
 pub mod tlmm_kind;
 pub mod tlmm_seat_focus;
 pub mod tlmm_string;

src/criteria/tlm/tlm_matchers/tlmm_just_mapped.rs

@@ -0,0 +1,11 @@
+use crate::{criteria::crit_graph::CritFixedRootCriterion, tree::ToplevelData};
+
+pub struct TlmMatchJustMapped(pub bool);
+
+fixed_root_criterion!(TlmMatchJustMapped, just_mapped);
+
+impl CritFixedRootCriterion<ToplevelData> for TlmMatchJustMapped {
+    fn matches(&self, data: &ToplevelData) -> bool {
+        data.just_mapped()
+    }
+}

src/tree/toplevel.rs

@@ -103,6 +103,7 @@ impl<T: ToplevelNodeBase> ToplevelNode for T {
         let data = self.tl_data();
         let parent_was_none = data.parent.set(Some(parent.clone())).is_none();
         if parent_was_none {
+            data.mapped_during_iteration.set(data.state.eng.iteration());
             data.property_changed(TL_CHANGED_NEW);
         }
         let was_floating = data.is_floating.get();
@@ -308,6 +309,7 @@ pub struct ToplevelData {
     pub workspace: CloneCell<Option<Rc<WorkspaceNode>>>,
     pub title: RefCell<String>,
     pub parent: CloneCell<Option<Rc<dyn ContainingNode>>>,
+    pub mapped_during_iteration: Cell<u64>,
     pub pos: Cell<Rect>,
     pub desired_extents: Cell<Rect>,
     pub seat_state: NodeSeatState,
@@ -325,6 +327,7 @@ pub struct ToplevelData {
     pub slf: Weak<dyn ToplevelNode>,
     pub destroyed: CopyHashMap<CritMatcherId, Weak<dyn CritDestroyListener<ToplevelData>>>,
     pub changed_properties: Cell<TlMatcherChange>,
+    pub just_mapped_scheduled: Cell<bool>,
     pub seat_foci: CopyHashMap<SeatId, ()>,
 }
 
@@ -357,6 +360,7 @@ impl ToplevelData {
             workspace: Default::default(),
             title: RefCell::new(title),
             parent: Default::default(),
+            mapped_during_iteration: Cell::new(0),
             pos: Default::default(),
             desired_extents: Default::default(),
             seat_state: Default::default(),
@@ -372,6 +376,7 @@ impl ToplevelData {
             slf: slf.clone(),
             destroyed: Default::default(),
             changed_properties: Default::default(),
+            just_mapped_scheduled: Cell::new(false),
             seat_foci: Default::default(),
         }
     }
@@ -696,6 +701,10 @@ impl ToplevelData {
         };
         (0, 0)
     }
+
+    pub fn just_mapped(&self) -> bool {
+        self.mapped_during_iteration.get() == self.state.eng.iteration()
+    }
 }
 
 impl Drop for ToplevelData {