data/com.shifthackz.aisdv1.data.repository/StableDiffusionCppGenerationRepositoryImpl

StableDiffusionCppGenerationRepositoryImpl

internal class StableDiffusionCppGenerationRepositoryImpl(mediaStoreGateway: MediaStoreGateway, localDataSource: GenerationResultDataSource.Local, backgroundWorkObserver: BackgroundWorkObserver, preferenceManager: PreferenceManager, val stableDiffusionCpp: StableDiffusionCpp, val downloadableLocalDataSource: DownloadableModelDataSource.Local, val modelFileStore: DownloadableModelFileStore) : CoreGenerationRepository, StableDiffusionCppGenerationRepository(source)

Constructors

constructor(mediaStoreGateway: MediaStoreGateway, localDataSource: GenerationResultDataSource.Local, backgroundWorkObserver: BackgroundWorkObserver, preferenceManager: PreferenceManager, stableDiffusionCpp: StableDiffusionCpp, downloadableLocalDataSource: DownloadableModelDataSource.Local, modelFileStore: DownloadableModelFileStore)

Properties

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

Exposes the backgroundWorkObserver value used by the SDAI data layer.

downloadableLocalDataSource

private val downloadableLocalDataSource: DownloadableModelDataSource.Local

localDataSource

private val localDataSource: GenerationResultDataSource.Local

Exposes the localDataSource value used by the SDAI data layer.

mediaStoreGateway

private val mediaStoreGateway: MediaStoreGateway

Exposes the mediaStoreGateway value used by the SDAI data layer.

modelFileStore

private val modelFileStore: DownloadableModelFileStore

preferenceManager

private val preferenceManager: PreferenceManager

Exposes the preferenceManager value used by the SDAI data layer.

sdxlPreferenceManager

private val sdxlPreferenceManager: PreferenceManager

stableDiffusionCpp

private val stableDiffusionCpp: StableDiffusionCpp

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, modelPath: String, modelName: String): 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>

resolveSdxlPath

private fun LocalAiModel.resolveSdxlPath(): String