feature/work/com.shifthackz.aisdv1.work.task/ImageToImageTask

ImageToImageTask

internal class ImageToImageTask(context: Context, workerParameters: WorkerParameters, pushNotificationManager: PushNotificationManager, activityIntentProvider: ActivityIntentProvider, preferenceManager: PreferenceManager, observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase, observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase, observeBonsaiProcessStatusUseCase: ObserveBonsaiProcessStatusUseCase, interruptGenerationUseCase: InterruptGenerationUseCase, val backgroundWorkObserver: BackgroundWorkObserver, val imageToImageUseCase: ImageToImageUseCase, val fileProviderDescriptor: FileProviderDescriptor) : CoreGenerationWorker(source)

Background img2img worker.

The task reads the cached serialized payload, subscribes to provider progress, runs the image-to-image use case, and publishes the result or failure through the common generation notification flow.

Constructors

ImageToImageTask

constructor(context: Context, workerParameters: WorkerParameters, pushNotificationManager: PushNotificationManager, activityIntentProvider: ActivityIntentProvider, preferenceManager: PreferenceManager, observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase, observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase, observeBonsaiProcessStatusUseCase: ObserveBonsaiProcessStatusUseCase, interruptGenerationUseCase: InterruptGenerationUseCase, backgroundWorkObserver: BackgroundWorkObserver, imageToImageUseCase: ImageToImageUseCase, fileProviderDescriptor: FileProviderDescriptor)

Properties

activityIntentProvider

private val activityIntentProvider: ActivityIntentProvider

Exposes the activityIntentProvider value used by the SDAI background work feature layer.

applicationContext

val applicationContext: @NonNull Context

backgroundExecutor

@get:RestrictTo(value = [RestrictTo.Scope.LIBRARY_GROUP])

open val backgroundExecutor: @NonNull Executor

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

coroutineContext

open val ~~coroutineContext~~: CoroutineDispatcher

coroutineScope

private val coroutineScope: CoroutineScope

fileProviderDescriptor

private val fileProviderDescriptor: FileProviderDescriptor

foregroundInfoAsync

val foregroundInfoAsync: ListenableFuture<ForegroundInfo>

genericNotificationId

open override val genericNotificationId: Int

Exposes the genericNotificationId value used by the SDAI background work feature layer.

val id: @NonNull UUID

imageToImageUseCase

private val imageToImageUseCase: ImageToImageUseCase

inputData

val inputData: @NonNull Data

interruptGenerationUseCase

private val interruptGenerationUseCase: InterruptGenerationUseCase

isStopped

val isStopped: Boolean

isUsed

@get:RestrictTo(value = [RestrictTo.Scope.LIBRARY_GROUP])

val isUsed: Boolean

mAppContext

private var mAppContext: @NonNull Context

mStopReason

private val mStopReason: AtomicInteger?

mUsed

private var mUsed: Boolean

mWorkerParams

private var mWorkerParams: @NonNull WorkerParameters

network

@get:RequiresApi(value = 28)

val network: @Nullable Network?

notificationId

open override val notificationId: Int

Exposes the notificationId value used by the SDAI background work feature layer.

observeBonsaiProcessStatusUseCase

private val observeBonsaiProcessStatusUseCase: ObserveBonsaiProcessStatusUseCase

observeHordeProcessStatusUseCase

private val observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase

observeLocalDiffusionProcessStatusUseCase

private val observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase

params

private val params: WorkerParameters

preferenceManager

private val preferenceManager: PreferenceManager

pushNotificationManager

private val pushNotificationManager: PushNotificationManager

Exposes the pushNotificationManager value used by the SDAI background work feature layer.

runAttemptCount

@get:IntRange(from = 0)

val runAttemptCount: Int

source

protected val source: ServerSource

stopReason

@get:RequiresApi(value = 31)

val stopReason: Int

Functions

clearSubscriptions

protected fun clearSubscriptions()

doWork

open suspend override fun doWork(): ListenableWorker.Result

getForegroundInfo

open suspend fun getForegroundInfo(): ForegroundInfo

getForegroundInfoAsync

override fun getForegroundInfoAsync(): ListenableFuture<ForegroundInfo>

handleCancellation

protected suspend fun handleCancellation()

handleError

protected fun handleError(t: Throwable)

handleProcess

protected fun handleProcess()

handleStart

protected fun handleStart()

handleSuccess

protected fun handleSuccess(result: List<AiGenerationResult>)

listenBonsaiStatus

protected fun listenBonsaiStatus()

listenHordeStatus

protected fun listenHordeStatus()

listenLocalDiffusionStatus

protected fun listenLocalDiffusionStatus()

listenSourceStatus

protected fun listenSourceStatus()

onStopped

override fun onStopped()

setForeground

suspend fun setForeground(foregroundInfo: ForegroundInfo)

setForegroundAsync

fun setForegroundAsync(foregroundInfo: @NonNull ForegroundInfo): @NonNull ListenableFuture<Void?>

setForegroundNotification

fun setForegroundNotification(title: String, body: String, subText: String? = null, progress: Pair<Int, Int>? = null, silent: Boolean = false, canCancel: Boolean = false)

Executes the setForegroundNotification step in the SDAI background work feature layer.

setProgress

suspend fun setProgress(data: Data)

setProgressAsync

open fun setProgressAsync(data: @NonNull Data): @NonNull ListenableFuture<Void?>

setUsed

@RestrictTo(value = [RestrictTo.Scope.LIBRARY_GROUP])

fun setUsed()

showGenericNotification

fun showGenericNotification(text: String, body: String?)

Executes the showGenericNotification step in the SDAI background work feature layer.

startWork

override fun startWork(): ListenableFuture<ListenableWorker.Result>

stop

@RestrictTo(value = [RestrictTo.Scope.LIBRARY_GROUP])

fun stop(reason: Int)