Added implementation of methods in MPPImage test utils
This commit is contained in:
parent
d3c83185ee
commit
218bfc8f69
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue
Block a user