data/com.shifthackz.aisdv1.data.repository/LocalDiffusionGenerationRepositoryImpl

LocalDiffusionGenerationRepositoryImpl

internal class LocalDiffusionGenerationRepositoryImpl(mediaStoreGateway: MediaStoreGateway, localDataSource: GenerationResultDataSource.Local, backgroundWorkObserver: BackgroundWorkObserver, val preferenceManager: PreferenceManager, val localDiffusion: LocalDiffusion, val downloadableLocalDataSource: DownloadableModelDataSource.Local, val bitmapToBase64Converter: BitmapToBase64Converter) : CoreGenerationRepository, LocalDiffusionGenerationRepository(source)

Implements LocalDiffusionGenerationRepository behavior in the SDAI data layer.

Author

Dmitriy Moroz

Constructors

LocalDiffusionGenerationRepositoryImpl

constructor(mediaStoreGateway: MediaStoreGateway, localDataSource: GenerationResultDataSource.Local, backgroundWorkObserver: BackgroundWorkObserver, preferenceManager: PreferenceManager, localDiffusion: LocalDiffusion, downloadableLocalDataSource: DownloadableModelDataSource.Local, bitmapToBase64Converter: BitmapToBase64Converter)

Properties

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

Exposes the backgroundWorkObserver value used by the SDAI data layer.

bitmapToBase64Converter

private val bitmapToBase64Converter: BitmapToBase64Converter

Exposes the bitmapToBase64Converter value used by the SDAI data layer.

downloadableLocalDataSource

private val downloadableLocalDataSource: DownloadableModelDataSource.Local

Exposes the downloadableLocalDataSource value used by the SDAI data layer.

localDataSource

private val localDataSource: GenerationResultDataSource.Local

Exposes the localDataSource value used by the SDAI data layer.

localDiffusion

private val localDiffusion: LocalDiffusion

Exposes the localDiffusion value used by the SDAI data layer.

mediaStoreGateway

private val mediaStoreGateway: MediaStoreGateway

Exposes the mediaStoreGateway value used by the SDAI data layer.

preferenceManager

private val preferenceManager: PreferenceManager

Exposes the preferenceManager value used by the SDAI data layer.

Functions

exportAsync

private fun exportAsync(result: AiGenerationResult)

Executes the exportAsync step in the SDAI data layer.

exportToMediaStoreAsync

protected suspend fun exportToMediaStoreAsync(result: AiGenerationResult)

Executes the exportToMediaStoreAsync step in the SDAI data layer.

generate

private suspend fun generate(payload: TextToImagePayload): AiGenerationResult

generateFromText

open suspend override fun generateFromText(payload: TextToImagePayload): AiGenerationResult

getInfoAsync

protected suspend fun getInfoAsync(): MediaStoreInfo

Loads SDAI data through getInfoAsync.

insertGenerationResult

protected suspend fun insertGenerationResult(ai: AiGenerationResult): AiGenerationResult

Performs the SDAI side effect handled by insertGenerationResult.

interruptGeneration

open suspend override fun interruptGeneration()

observeStatus

open override fun observeStatus(): Flow<LocalDiffusionStatus>