TAS: allow to dump the usage of domains for TAS ResourceFlavors

pkg/cache/snapshot.go

@@ -79,6 +79,19 @@ func (s *Snapshot) Log(log logr.Logger) {
 			"usage", cohort.ResourceNode.Usage,
 		)
 	}
+
+	// Dump TAS snapshots if the feature is enabled
+	if features.Enabled(features.TopologyAwareScheduling) {
+		for cqName, cq := range s.ClusterQueues {
+			for tasFlavor, tasSnapshot := range cq.TASFlavors {
+				log.Info("TAS Snapshot Free Capacity",
+					"clusterQueue", cqName,
+					"resourceFlavor", tasFlavor,
+					"freeCapacityPerDomain", tasSnapshot.FreeCapacityPerDomain(),
+				)
+			}
+		}
+	}
 }
 
 func (c *Cache) Snapshot(ctx context.Context) (*Snapshot, error) {

pkg/cache/tas_flavor_snapshot.go

@@ -241,6 +241,22 @@ func (s *TASFlavorSnapshot) removeTASUsage(domainID utiltas.TopologyDomainID, us
 	s.leaves[domainID].tasUsage.Sub(usage)
 }
 
+func (s *TASFlavorSnapshot) FreeCapacityPerDomain() map[string]int {
+	freeCapacityPerDomain := make(map[string]int)
+
+	for domainID, leaf := range s.leaves {
+		// Calculate the total free capacity for the domain
+		totalFreeCapacity := 0
+		for _, quantity := range leaf.freeCapacity {
+			totalFreeCapacity += int(quantity)
+		}
+
+		freeCapacityPerDomain[string(domainID)] = totalFreeCapacity
+	}
+
+	return freeCapacityPerDomain
+}
+
 type TASPodSetRequests struct {
 	PodSet            *kueue.PodSet
 	SinglePodRequests resources.Requests