This commit is contained in:
Manav Rathi 2024-05-16 11:13:34 +05:30
parent cfbd7806c8
commit dfd91beaff
No known key found for this signature in database

View file

@ -8,7 +8,6 @@ import { DetectedFace, MLSyncFileContext } from "services/ml/types";
import { EnteFile } from "types/file"; import { EnteFile } from "types/file";
import { getRenderableImage } from "utils/file"; import { getRenderableImage } from "utils/file";
import { clamp } from "utils/image"; import { clamp } from "utils/image";
import { DEFAULT_ML_SYNC_CONFIG } from "./machineLearningService";
class ReaderService { class ReaderService {
async getImageBitmap(fileContext: MLSyncFileContext) { async getImageBitmap(fileContext: MLSyncFileContext) {
@ -29,7 +28,6 @@ class ReaderService {
fileContext.localFile, fileContext.localFile,
); );
} else if ( } else if (
DEFAULT_ML_SYNC_CONFIG.imageSource === "Original" &&
[FILE_TYPE.IMAGE, FILE_TYPE.LIVE_PHOTO].includes( [FILE_TYPE.IMAGE, FILE_TYPE.LIVE_PHOTO].includes(
fileContext.enteFile.metadata.fileType, fileContext.enteFile.metadata.fileType,
) )
@ -38,13 +36,14 @@ class ReaderService {
fileContext.enteFile, fileContext.enteFile,
); );
} else { } else {
// TODO-ML(MR): We don't do it on videos, when will we ever come
// here?
fileContext.imageBitmap = await getThumbnailImageBitmap( fileContext.imageBitmap = await getThumbnailImageBitmap(
fileContext.enteFile, fileContext.enteFile,
); );
} }
fileContext.newMlFile.imageSource = fileContext.newMlFile.imageSource = "Original";
DEFAULT_ML_SYNC_CONFIG.imageSource;
const { width, height } = fileContext.imageBitmap; const { width, height } = fileContext.imageBitmap;
fileContext.newMlFile.imageDimensions = { width, height }; fileContext.newMlFile.imageDimensions = { width, height };