Files
UnrealEngine/Engine/Source/Developer/AutomationDriver/Public/WaitUntil.h
Brandyn / Techy fcc1b09210 init
2026-04-04 15:40:51 -05:00

223 lines
10 KiB
C++

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
#include "CoreMinimal.h"
#include "Delegates/Delegate.h"
#include "HAL/Platform.h"
#include "Misc/Timespan.h"
#include "Templates/Function.h"
#include "Templates/SharedPointer.h"
#define UE_API AUTOMATIONDRIVER_API
class IElementLocator;
/**
* Represents the state of an active wait action for the driver
*/
struct FDriverWaitResponse
{
public:
enum class EState : uint8
{
PASSED,
WAIT,
FAILED,
};
/**
* @return a FDriverWaitResponse with a state of PASSED and a wait of zero
*/
static UE_API FDriverWaitResponse Passed();
/**
* @return a FDriverWaitResponse with a state of WAIT and a wait of 0.5 seconds
*/
static UE_API FDriverWaitResponse Wait();
/**
* @return a FDriverWaitResponse with a state of WAIT and a wait of the specified timespan
*/
static UE_API FDriverWaitResponse Wait(FTimespan Timespan);
/**
* @return a FDriverWaitResponse with a state of FAILED and a wait of zero
*/
static UE_API FDriverWaitResponse Failed();
UE_API FDriverWaitResponse(EState InState, FTimespan InNextWait);
// How long the driver should wait before re-evaluating the wait condition again
const FTimespan NextWait;
// Whether the wait condition is completely finished or should be rescheduled again for execution
const EState State;
};
DECLARE_DELEGATE_RetVal_OneParam(FDriverWaitResponse, FDriverWaitDelegate, const FTimespan& /*TotalWaitTime*/);
DECLARE_DELEGATE_RetVal(bool, FDriverWaitConditionDelegate);
/**
* A fluent wrapper around timespan to enforce obvious differences between specified Timeout and Interval values for waits
*/
class FWaitTimeout
{
public:
static UE_API FWaitTimeout InMilliseconds(double Value);
static UE_API FWaitTimeout InSeconds(double Value);
static UE_API FWaitTimeout InMinutes(double Value);
static UE_API FWaitTimeout InHours(double Value);
UE_API explicit FWaitTimeout(FTimespan InTimespan);
const FTimespan Timespan;
};
/**
* A fluent wrapper around timespan to enforce obvious differences between specified Timeout and Interval values for waits
*/
class FWaitInterval
{
public:
static UE_API FWaitInterval InMilliseconds(double Value);
static UE_API FWaitInterval InSeconds(double Value);
static UE_API FWaitInterval InMinutes(double Value);
static UE_API FWaitInterval InHours(double Value);
UE_API explicit FWaitInterval(FTimespan InTimespan);
const FTimespan Timespan;
};
/**
* Represents a collection of fluent helper functions designed to make accessing and creating driver wait delegates easier
*/
class Until
{
public:
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers elements or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate ElementExists(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers elements or if the specified timeout timespan elapses;
* The element locator is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate ElementExists(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers visible elements or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate ElementIsVisible(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers visible elements or if the specified timeout timespan elapses;
* The element locator is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate ElementIsVisible(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers hidden elements or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate ElementIsHidden(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers hidden elements or if the specified timeout timespan elapses;
* The element locator is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate ElementIsHidden(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers interactable elements or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate ElementIsInteractable(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers interactable elements or if the specified timeout timespan elapses;
* The element locator is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate ElementIsInteractable(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers a scrollable element whose scroll position is at the beginning
* or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate ElementIsScrolledToBeginning(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers a scrollable element whose scroll position is at the beginning
* or if the specified timeout timespan elapses
* The element locator is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate ElementIsScrolledToBeginning(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers a scrollable element whose scroll position is at the end
* or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate ElementIsScrolledToEnd(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers a scrollable element whose scroll position is at the end
* or if the specified timeout timespan elapses
* The element locator is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate ElementIsScrolledToEnd(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes its wait only if the specified element locator discovers an element focused by the default user or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate ElementIsFocusedByKeyboard(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers an element focused by the default user or if the specified timeout timespan elapses;
* The element locator is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate ElementIsFocusedByKeyboard(const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes its wait only if the specified element locator discovers an element focused by the specified user or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate ElementIsFocusedByUser(uint32 UserIndex, const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified element locator discovers an element focused by the specified user or if the specified timeout timespan elapses;
* The element locator is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate ElementIsFocusedByUser(uint32 UserIndex, const TSharedRef<IElementLocator, ESPMode::ThreadSafe>& ElementLocator, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified condition returns true or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate Condition(const TFunction<bool()>& Function, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified condition returns true or if the specified timeout timespan elapses
* The lambda is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate Condition(const TFunction<bool()>& Function, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified condition returns true or if the specified timeout timespan elapses
*/
static UE_API FDriverWaitDelegate Condition(const FDriverWaitConditionDelegate& Delegate, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which completes it's wait only if the specified condition returns true or if the specified timeout timespan elapses
* The delegate is only re-evaluated at the specified wait interval
*/
static UE_API FDriverWaitDelegate Condition(const FDriverWaitConditionDelegate& Delegate, FWaitInterval Interval, FWaitTimeout Timeout);
/**
* Creates a new wait delegate which drives its state off the result of the specified lambda
*/
static UE_API FDriverWaitDelegate Lambda(const TFunction<FDriverWaitResponse(const FTimespan&)>& Value);
};
#undef UE_API