presentation/com.shifthackz.aisdv1.presentation.screen.img2img/ImageToImageViewModel

ImageToImageViewModel

class ImageToImageViewModel(val dispatchersProvider: DispatchersProvider, val getConfigurationUseCase: GetConfigurationUseCase, val getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase, val getRandomImageUseCase: GetRandomImageUseCase, val imageToImageUseCase: ImageToImageUseCase, val saveGenerationResultUseCase: SaveGenerationResultUseCase, val saveLastResultToCacheUseCase: SaveLastResultToCacheUseCase, val interruptGenerationUseCase: InterruptGenerationUseCase, val observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase, val observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase, val preferenceManager: PreferenceManager, val backgroundTaskManager: BackgroundTaskManager, val backgroundWorkObserver: BackgroundWorkObserver, val platformServices: GenerationPlatformServices, val buildInfoProvider: BuildInfoProvider, val generationFormUpdateEvent: GenerationFormUpdateEvent, val dimensionValidator: DimensionValidator, val imageSaver: ImageSaver, val imageSharer: ImageSharer, val router: ImageToImageRouter, val platformActions: ImageToImagePlatformActions, val onError: (Throwable) -> Unit = {}) : BaseMviViewModel<ImageToImageState, ImageToImageIntent, EmptyEffect> (source)

Coordinates ImageToImageViewModel behavior in the SDAI presentation layer.

Author

Dmitriy Moroz

Constructors

ImageToImageViewModel

constructor(dispatchersProvider: DispatchersProvider, getConfigurationUseCase: GetConfigurationUseCase, getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase, getRandomImageUseCase: GetRandomImageUseCase, imageToImageUseCase: ImageToImageUseCase, saveGenerationResultUseCase: SaveGenerationResultUseCase, saveLastResultToCacheUseCase: SaveLastResultToCacheUseCase, interruptGenerationUseCase: InterruptGenerationUseCase, observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase, observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase, preferenceManager: PreferenceManager, backgroundTaskManager: BackgroundTaskManager, backgroundWorkObserver: BackgroundWorkObserver, platformServices: GenerationPlatformServices, buildInfoProvider: BuildInfoProvider, generationFormUpdateEvent: GenerationFormUpdateEvent, dimensionValidator: DimensionValidator, imageSaver: ImageSaver, imageSharer: ImageSharer, router: ImageToImageRouter, platformActions: ImageToImagePlatformActions, onError: (Throwable) -> Unit = {})

Properties

_effect

private val _effect: Channel<E>

_state

private val _state: MutableStateFlow<S>

actionHandler

private val actionHandler: ImageToImageActionHandler

backgroundTaskManager

private val backgroundTaskManager: BackgroundTaskManager

Exposes the backgroundTaskManager value used by the SDAI presentation layer.

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

Exposes the backgroundWorkObserver value used by the SDAI presentation layer.

buildInfoProvider

private val buildInfoProvider: BuildInfoProvider

Exposes the buildInfoProvider value used by the SDAI presentation layer.

currentState

protected val currentState: ImageToImageState

dimensionValidator

private val dimensionValidator: DimensionValidator

Exposes the dimensionValidator value used by the SDAI presentation layer.

dispatchersProvider

private val dispatchersProvider: DispatchersProvider

Exposes the dispatchersProvider value used by the SDAI presentation layer.

effect

open override val effect: Flow<EmptyEffect>

effectDispatcher

private val effectDispatcher: CoroutineDispatcher

generationFormUpdateEvent

private val generationFormUpdateEvent: GenerationFormUpdateEvent

Exposes the generationFormUpdateEvent value used by the SDAI presentation layer.

getConfigurationUseCase

private val getConfigurationUseCase: GetConfigurationUseCase

Exposes the getConfigurationUseCase value used by the SDAI presentation layer.

getRandomImageUseCase

private val getRandomImageUseCase: GetRandomImageUseCase

Exposes the getRandomImageUseCase value used by the SDAI presentation layer.

getStableDiffusionSamplersUseCase

private val getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase

Exposes the getStableDiffusionSamplersUseCase value used by the SDAI presentation layer.

imageSaver

private val imageSaver: ImageSaver

Exposes the imageSaver value used by the SDAI presentation layer.

imageSharer

private val imageSharer: ImageSharer

Exposes the imageSharer value used by the SDAI presentation layer.

imageToImageUseCase

private val imageToImageUseCase: ImageToImageUseCase

Exposes the imageToImageUseCase value used by the SDAI presentation layer.

intentProcessor

private val intentProcessor: ImageToImageIntentProcessor

interruptGenerationUseCase

private val interruptGenerationUseCase: InterruptGenerationUseCase

Exposes the interruptGenerationUseCase value used by the SDAI presentation layer.

observeHordeProcessStatusUseCase

private val observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase

Exposes the observeHordeProcessStatusUseCase value used by the SDAI presentation layer.

observeLocalDiffusionProcessStatusUseCase

private val observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase

Exposes the observeLocalDiffusionProcessStatusUseCase value used by the SDAI presentation layer.

onError

private val onError: (Throwable) -> Unit

Exposes the onError value used by the SDAI presentation layer.

platformActions

private val platformActions: ImageToImagePlatformActions

Exposes the platformActions value used by the SDAI presentation layer.

platformServices

private val platformServices: GenerationPlatformServices

Exposes the platformServices value used by the SDAI presentation layer.

preferenceManager

private val preferenceManager: PreferenceManager

Exposes the preferenceManager value used by the SDAI presentation layer.

router

private val router: ImageToImageRouter

Exposes the router value used by the SDAI presentation layer.

saveGenerationResultUseCase

private val saveGenerationResultUseCase: SaveGenerationResultUseCase

Exposes the saveGenerationResultUseCase value used by the SDAI presentation layer.

saveLastResultToCacheUseCase

private val saveLastResultToCacheUseCase: SaveLastResultToCacheUseCase

Exposes the saveLastResultToCacheUseCase value used by the SDAI presentation layer.

stableDiffusionSamplers

private var stableDiffusionSamplers: List<String>?

stableDiffusionSamplersKey

private var stableDiffusionSamplersKey: StableDiffusionSamplersKey?

state

open override val state: StateFlow<ImageToImageState>

Functions

addCloseable

expect open fun addCloseable(closeable: AutoCloseable)

expect fun addCloseable(key: String, closeable: AutoCloseable)

applyGenerationResult

private fun applyGenerationResult(ai: AiGenerationResult)

private fun applyGenerationResult(payload: GenerationFormUpdateEvent.Payload.I2IForm)

private fun applyGenerationResult(ai: AiGenerationResult, inputImage: Boolean)

private fun applyGenerationResult(ai: AiGenerationResult, imageBase64: String?)

clear

@MainThread

internal expect fun clear()

consumePendingGenerationFormUpdate

private fun consumePendingGenerationFormUpdate()

emitEffect

protected fun emitEffect(effect: EmptyEffect)

emitState

protected fun emitState(state: ImageToImageState)

getCloseable

expect fun <T : AutoCloseable> getCloseable(key: String): T?

launch

protected fun launch(dispatcher: CoroutineDispatcher, start: CoroutineStart, block: suspend CoroutineScope.() -> Unit): Job

loadConfiguration

private fun loadConfiguration()

loadSamplers

private fun loadSamplers()

observeGenerationFormUpdate

private fun observeGenerationFormUpdate()

observeGenerationProgress

private fun observeGenerationProgress()

observeSettings

private fun observeSettings()

onCleared

protected expect open fun onCleared()

processIntent

open override fun processIntent(intent: ImageToImageIntent)

refreshStableDiffusionSamplersIfNeeded

private fun refreshStableDiffusionSamplersIfNeeded(settings: Settings)

updateState

protected fun updateState(mutation: (ImageToImageState) -> ImageToImageState)