presentation/com.shifthackz.aisdv1.presentation.screen.txt2img/TextToImageViewModel

TextToImageViewModel

class TextToImageViewModel(val dispatchersProvider: DispatchersProvider, val getConfigurationUseCase: GetConfigurationUseCase, val getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase, val textToImageUseCase: TextToImageUseCase, 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: TextToImageRouter, val onError: (Throwable) -> Unit = {}) : BaseMviViewModel<TextToImageState, TextToImageIntent, EmptyEffect> (source)

Coordinates TextToImageViewModel behavior in the SDAI presentation layer.

Author

Dmitriy Moroz

Constructors

TextToImageViewModel

constructor(dispatchersProvider: DispatchersProvider, getConfigurationUseCase: GetConfigurationUseCase, getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase, textToImageUseCase: TextToImageUseCase, 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: TextToImageRouter, onError: (Throwable) -> Unit = {})

Properties

_effect

private val _effect: Channel<E>

_state

private val _state: MutableStateFlow<S>

actionHandler

private val actionHandler: TextToImageActionHandler

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: TextToImageState

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.

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.

intentProcessor

private val intentProcessor: TextToImageIntentProcessor

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.

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: TextToImageRouter

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<TextToImageState>

textToImageUseCase

private val textToImageUseCase: TextToImageUseCase

Exposes the textToImageUseCase value used by the SDAI presentation layer.

Functions

addCloseable

expect open fun addCloseable(closeable: AutoCloseable)

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

applyGenerationResult

private fun applyGenerationResult(ai: AiGenerationResult)

clear

@MainThread

internal expect fun clear()

consumePendingGenerationFormUpdate

private fun consumePendingGenerationFormUpdate()

emitEffect

protected fun emitEffect(effect: EmptyEffect)

emitState

protected fun emitState(state: TextToImageState)

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: TextToImageIntent)

refreshStableDiffusionSamplersIfNeeded

private fun refreshStableDiffusionSamplersIfNeeded(settings: Settings)

updateState

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