Added implementation of methods in MPPImage test utils

This commit is contained in:
Prianka Liz Kariat 2023-09-06 18:32:08 +05:30
parent d3c83185ee
commit 218bfc8f69

View File

@ -14,6 +14,7 @@
#import "mediapipe/tasks/ios/test/vision/utils/sources/MPPImage+TestUtils.h" #import "mediapipe/tasks/ios/test/vision/utils/sources/MPPImage+TestUtils.h"
// TODO: Remove this category after all tests are migrated to the new methods.
@interface UIImage (FileUtils) @interface UIImage (FileUtils)
+ (nullable UIImage *)imageFromBundleWithClass:(Class)classObject + (nullable UIImage *)imageFromBundleWithClass:(Class)classObject
@ -37,6 +38,28 @@
@implementation MPPImage (TestUtils) @implementation MPPImage (TestUtils)
+ (MPPImage *)imageWithFileInfo:(MPPFileInfo *)fileInfo {
if (!fileInfo.path) return nil;
UIImage *image = [[UIImage alloc] initWithContentsOfFile:fileInfo.path];
if (!image) return nil;
return [[MPPImage alloc] initWithUIImage:image error:nil];
}
+ (MPPImage *)imageWithFileInfo:(MPPFileInfo *)fileInfo
orientation:(UIImageOrientation)orientation {
if (!fileInfo.path) return nil;
UIImage *image = [[UIImage alloc] initWithContentsOfFile:fileInfo.path];
if (!image) return nil;
return [[MPPImage alloc] initWithUIImage:image orientation:orientation error:nil];
}
// TODO: Remove after all tests are migrated
+ (nullable MPPImage *)imageFromBundleWithClass:(Class)classObject + (nullable MPPImage *)imageFromBundleWithClass:(Class)classObject
fileName:(NSString *)name fileName:(NSString *)name
ofType:(NSString *)type { ofType:(NSString *)type {
@ -45,6 +68,7 @@
return [[MPPImage alloc] initWithUIImage:image error:nil]; return [[MPPImage alloc] initWithUIImage:image error:nil];
} }
// TODO: Remove after all tests are migrated
+ (nullable MPPImage *)imageFromBundleWithClass:(Class)classObject + (nullable MPPImage *)imageFromBundleWithClass:(Class)classObject
fileName:(NSString *)name fileName:(NSString *)name
ofType:(NSString *)type ofType:(NSString *)type