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

ImageToImageViewModel

class ImageToImageViewModel(val dispatchersProvider: DispatchersProvider, val getConfigurationUseCase: GetConfigurationUseCase, val getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase, val getForgeModulesUseCase: GetForgeModulesUseCase, val fetchAndGetArliAiModelsUseCase: FetchAndGetArliAiModelsUseCase, val isADetailerAvailableUseCase: IsADetailerAvailableUseCase, 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 wakeLockInterActor: WakeLockInterActor, val platformServices: GenerationPlatformServices, val buildInfoProvider: BuildInfoProvider, val generationFormUpdateEvent: GenerationFormUpdateEvent, val dimensionValidator: DimensionValidator, val localGenerationBenchmarkGateProvider: () -> LocalGenerationBenchmarkGate, val imageSaver: ImageSaver, val imageSharer: ImageSharer, val router: ImageToImageRouter, val platformActions: ImageToImagePlatformActions, val onError: (Throwable) -> Unit = {}) : BaseMviViewModel<ImageToImageState, ImageToImageIntent, EmptyEffect> (source)

View-model for the img2img screen.

It keeps source-image state, shared generation controls, provider configuration, benchmark gating, and platform image-picking actions coordinated for the image-to-image flow.

Constructors

ImageToImageViewModel

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

Properties

private val _effect: Channel<E>

private val _state: MutableStateFlow<S>

private val actionHandler: ImageToImageActionHandler

aDetailerAvailabilityKey

private var aDetailerAvailabilityKey: StableDiffusionSamplersKey?

aDetailerAvailable

private var aDetailerAvailable: Boolean?

private var arliAiModels: List<String>?

arliAiModelsKey

private var arliAiModelsKey: String?

backgroundTaskManager

private val backgroundTaskManager: BackgroundTaskManager

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

buildInfoProvider

private val buildInfoProvider: BuildInfoProvider

protected val currentState: ImageToImageState

dimensionValidator

private val dimensionValidator: DimensionValidator

dispatchersProvider

private val dispatchersProvider: DispatchersProvider

open override val effect: Flow<EmptyEffect>

effectDispatcher

private val effectDispatcher: CoroutineDispatcher

fetchAndGetArliAiModelsUseCase

private val fetchAndGetArliAiModelsUseCase: FetchAndGetArliAiModelsUseCase

private var forgeModules: List<ForgeModule>?

forgeModulesKey

private var forgeModulesKey: StableDiffusionSamplersKey?

generationFormUpdateEvent

private val generationFormUpdateEvent: GenerationFormUpdateEvent

getConfigurationUseCase

private val getConfigurationUseCase: GetConfigurationUseCase

getForgeModulesUseCase

private val getForgeModulesUseCase: GetForgeModulesUseCase

getRandomImageUseCase

private val getRandomImageUseCase: GetRandomImageUseCase

getStableDiffusionSamplersUseCase

private val getStableDiffusionSamplersUseCase: GetStableDiffusionSamplersUseCase

private val imageSaver: ImageSaver

private val imageSharer: ImageSharer

imageToImageUseCase

private val imageToImageUseCase: ImageToImageUseCase

intentProcessor

private val intentProcessor: ImageToImageIntentProcessor

interruptGenerationUseCase

private val interruptGenerationUseCase: InterruptGenerationUseCase

isADetailerAvailableUseCase

private val isADetailerAvailableUseCase: IsADetailerAvailableUseCase

localGenerationBenchmarkGateProvider

private val localGenerationBenchmarkGateProvider: () -> LocalGenerationBenchmarkGate

observeHordeProcessStatusUseCase

private val observeHordeProcessStatusUseCase: ObserveHordeProcessStatusUseCase

observeLocalDiffusionProcessStatusUseCase

private val observeLocalDiffusionProcessStatusUseCase: ObserveLocalDiffusionProcessStatusUseCase

private val onError: (Throwable) -> Unit

platformActions

private val platformActions: ImageToImagePlatformActions

platformServices

private val platformServices: GenerationPlatformServices

preferenceManager

private val preferenceManager: PreferenceManager

private val router: ImageToImageRouter

saveGenerationResultUseCase

private val saveGenerationResultUseCase: SaveGenerationResultUseCase

saveLastResultToCacheUseCase

private val saveLastResultToCacheUseCase: SaveLastResultToCacheUseCase

stableDiffusionSamplers

private var stableDiffusionSamplers: List<String>?

stableDiffusionSamplersKey

private var stableDiffusionSamplersKey: StableDiffusionSamplersKey?

open override val state: StateFlow<ImageToImageState>

wakeLockInterActor

private val wakeLockInterActor: WakeLockInterActor

Functions

expect open fun addCloseable(closeable: AutoCloseable)

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

applyGenerationResult

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

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

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

internal expect fun clear()

consumePendingGenerationFormUpdate

private fun consumePendingGenerationFormUpdate()

protected fun emitEffect(effect: EmptyEffect)

protected fun emitState(state: ImageToImageState)

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

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

loadADetailerAvailability

private fun loadADetailerAvailability(force: Boolean = false)

loadArliAiModels

private fun loadArliAiModels()

loadConfiguration

private fun loadConfiguration()

loadForgeModules

private fun loadForgeModules()

private fun loadSamplers()

observeGenerationFormUpdate

private fun observeGenerationFormUpdate()

observeGenerationProgress

private fun observeGenerationProgress()

observeSettings

private fun observeSettings()

protected expect open fun onCleared()

open override fun processIntent(intent: ImageToImageIntent)

refreshArliAiModelsIfNeeded

private fun refreshArliAiModelsIfNeeded(settings: Settings)

refreshStableDiffusionMetadataIfNeeded

private fun refreshStableDiffusionMetadataIfNeeded(settings: Settings)

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