allocator

var (
	// ByAllocatableMemoryAsc Compare the assignable memory of two devices in ascending order
	ByAllocatableMemoryAsc = func(p1, p2 *device.Device) bool {
		return p1.AllocatableMemory() < p2.AllocatableMemory()
	}
	// ByAllocatableMemoryDes Compare the assignable memory of two devices in descending order
	ByAllocatableMemoryDes = func(p1, p2 *device.Device) bool {
		return p1.AllocatableMemory() > p2.AllocatableMemory()
	}
	// ByAllocatableCoresAsc Compare the assignable cores of two devices in ascending order
	ByAllocatableCoresAsc = func(p1, p2 *device.Device) bool {
		return p1.AllocatableCores() < p2.AllocatableCores()
	}
	// ByAllocatableCoresDes Compare the assignable cores of two devices in descending order
	ByAllocatableCoresDes = func(p1, p2 *device.Device) bool {
		return p1.AllocatableCores() > p2.AllocatableCores()
	}
	// ByDeviceIdAsc Compare the device id of two devices in ascending order
	ByDeviceIdAsc = func(p1, p2 *device.Device) bool {
		return p1.GetID() < p2.GetID()
	}
	ByAllocatableNumberDes = func(p1, p2 *device.Device) bool {
		return p1.AllocatableNumber() > p2.AllocatableNumber()
	}
	ByNumaAsc = func(p1, p2 *device.Device) bool {
		return p1.GetNUMA() < p2.GetNUMA()
	}
	ByNodeNameAsc = func(p1, p2 *device.NodeInfo) bool {
		return p1.GetName() < p2.GetName()
	}
	// ByNodeScoreAsc Sort in ascending order based on node scores,
	// to avoid double counting scores, a score cache was used.
	ByNodeScoreAsc = func() func(p1, p2 *device.NodeInfo) bool {
		nodeScoreMap := map[string]float64{}
		return func(p1, p2 *device.NodeInfo) bool {
			p1Score, ok := nodeScoreMap[p1.GetName()]
			if !ok {
				p1Score = GetNodeScore(p1)
				nodeScoreMap[p1.GetName()] = p1Score
			}
			p2Score, ok := nodeScoreMap[p2.GetName()]
			if !ok {
				p2Score = GetNodeScore(p2)
				nodeScoreMap[p2.GetName()] = p2Score
			}
			return p1Score < p2Score
		}
	}
	// ByNodeScoreDes Sort in descending order based on node scores,
	// to avoid double counting scores, a score cache was used.
	ByNodeScoreDes = func() func(p1, p2 *device.NodeInfo) bool {
		nodeScoreMap := map[string]float64{}
		return func(p1, p2 *device.NodeInfo) bool {
			p1Score, ok := nodeScoreMap[p1.GetName()]
			if !ok {
				p1Score = GetNodeScore(p1)
				nodeScoreMap[p1.GetName()] = p1Score
			}
			p2Score, ok := nodeScoreMap[p2.GetName()]
			if !ok {
				p2Score = GetNodeScore(p2)
				nodeScoreMap[p2.GetName()] = p2Score
			}
			return p1Score > p2Score
		}
	}
)