GPUImage源碼閱讀(七)

概述

GPUImage是一個著名的圖像處理開源庫,它讓你能夠在圖片、視頻、相機上使用GPU加速的濾鏡和其它特效。與CoreImage框架相比,可以根據GPUImage提供的接口,使用自定義的濾鏡。項目地址:https://github.com/BradLarson/GPUImage
這篇文章主要是閱讀GPUImage框架中的 GPUImageFilterPipeline、GPUImageFilterGroup 這幾個類的源碼。這兩個類都和組合濾鏡相關,但是它們又有不同的地方。GPUImageFilterPipeline 繼承自NSObject,也沒有實現其它協議,而GPUImageFilterGroup繼承自GPUImageOutput 實現了GPUImageInput協議。因此,最大的區別就是GPUImageFilterGroup自身可以作為響應鏈的一員,而GPUImageFilterPipeline則不能。以下是源碼內容:
GPUImageFilterPipeline
GPUImageFilterGroup

實現效果

  • 從文件中讀取濾鏡配置。


    filterConfig.png
  • 直接配置濾鏡。


    filter.png
  • 自定義組合濾鏡。


    GPUImageFilterGroup.png

GPUImageFilterPipeline

GPUImageFilterPipeline 繼承自NSObject,它的主要作用是管理濾鏡鏈,自身不能參與響應鏈中。可以用來構建簡單的濾鏡組合。如果濾鏡比較復雜或是涉及到多個紋理的處理,GPUImageFilterGroup則是更好的選擇。

  • 屬性。
// filter數組
@property (strong) NSMutableArray *filters;
// 輸入對象
@property (strong) GPUImageOutput *input;
// 輸出對象
@property (strong) id <GPUImageInput> output;

  • 構造方法。可以通過指定filter數組以及配置字典、文件來構造GPUImageFilterPipeline。
- (id) initWithOrderedFilters:(NSArray*) filters input:(GPUImageOutput*)input output:(id <GPUImageInput>)output;
- (id) initWithConfiguration:(NSDictionary*) configuration input:(GPUImageOutput*)input output:(id <GPUImageInput>)output;
- (id) initWithConfigurationFile:(NSURL*) configuration input:(GPUImageOutput*)input output:(id <GPUImageInput>)output;

// 根據輸入的filter數組、GPUImageOutput、GPUImageInput構建GPUImageFilterPipeline
- (id)initWithOrderedFilters:(NSArray *)filters input:(GPUImageOutput *)input output:(id <GPUImageInput>)output {
    self = [super init];
    if (self) {
        self.input = input;
        self.output = output;
        self.filters = [NSMutableArray arrayWithArray:filters];
        [self _refreshFilters];
    }
    return self;
}

// 根據filter配置字典、GPUImageOutput、GPUImageInput構建GPUImageFilterPipeline
- (id)initWithConfiguration:(NSDictionary *)configuration input:(GPUImageOutput *)input output:(id <GPUImageInput>)output {
    self = [super init];
    if (self) {
        self.input = input;
        self.output = output;
        if (![self _parseConfiguration:configuration]) {
            NSLog(@"Sorry, a parsing error occurred.");
            abort();
        }
        [self _refreshFilters];
    }
    return self;
}

// 根據filter配置文件的URL、GPUImageOutput、GPUImageInput構建GPUImageFilterPipeline
- (id)initWithConfigurationFile:(NSURL *)configuration input:(GPUImageOutput *)input output:(id <GPUImageInput>)output {
    return [self initWithConfiguration:[NSDictionary dictionaryWithContentsOfURL:configuration] input:input output:output];
}

// 解析配置文件
- (BOOL)_parseConfiguration:(NSDictionary *)configuration {
    NSArray *filters = [configuration objectForKey:@"Filters"];
    if (!filters) {
        return NO;
    }
    
    NSError *regexError = nil;
    // 匹配配置文件參數如:float(1.0), CGPoint(1.0, 2.0) 等類型
    NSRegularExpression *parsingRegex = [NSRegularExpression regularExpressionWithPattern:@"(float|CGPoint|NSString)\\((.*?)(?:,\\s*(.*?))*\\)"
                                                                                  options:0
                                                                                    error:&regexError];
    
    // It's faster to put them into an array and then pass it to the filters property than it is to call [self addFilter:] every time
    NSMutableArray *orderedFilters = [NSMutableArray arrayWithCapacity:[filters count]];
    for (NSDictionary *filter in filters) {
        // 由FilterName生成相應的實例對象
        NSString *filterName = [filter objectForKey:@"FilterName"];
        Class theClass = NSClassFromString(filterName);
        GPUImageOutput<GPUImageInput> *genericFilter = [[theClass alloc] init];
        // Set up the properties
        NSDictionary *filterAttributes;
        // 解析Attributes,并傳遞參數
        if ((filterAttributes = [filter objectForKey:@"Attributes"])) {
            for (NSString *propertyKey in filterAttributes) {
                // Set up the selector
                SEL theSelector = NSSelectorFromString(propertyKey);
                NSInvocation *inv = [NSInvocation invocationWithMethodSignature:[theClass instanceMethodSignatureForSelector:theSelector]];
                [inv setSelector:theSelector];
                [inv setTarget:genericFilter];
                
                // check selector given with parameter
                if ([propertyKey hasSuffix:@":"]) {
                    
                    stringValue = nil;
                    
                    // Then parse the arguments
                    NSMutableArray *parsedArray;
                    if ([[filterAttributes objectForKey:propertyKey] isKindOfClass:[NSArray class]]) {
                        NSArray *array = [filterAttributes objectForKey:propertyKey];
                        parsedArray = [NSMutableArray arrayWithCapacity:[array count]];
                        for (NSString *string in array) {
                            NSTextCheckingResult *parse = [parsingRegex firstMatchInString:string
                                                                                   options:0
                                                                                     range:NSMakeRange(0, [string length])];

                            NSString *modifier = [string substringWithRange:[parse rangeAtIndex:1]];
                            if ([modifier isEqualToString:@"float"]) {
                                // Float modifier, one argument
                                CGFloat value = [[string substringWithRange:[parse rangeAtIndex:2]] floatValue];
                                [parsedArray addObject:[NSNumber numberWithFloat:value]];
                                [inv setArgument:&value atIndex:2];
                            } else if ([modifier isEqualToString:@"CGPoint"]) {
                                // CGPoint modifier, two float arguments
                                CGFloat x = [[string substringWithRange:[parse rangeAtIndex:2]] floatValue];
                                CGFloat y = [[string substringWithRange:[parse rangeAtIndex:3]] floatValue];
                                CGPoint value = CGPointMake(x, y);
                                [parsedArray addObject:[NSValue valueWithCGPoint:value]];
                            } else if ([modifier isEqualToString:@"NSString"]) {
                                // NSString modifier, one string argument
                                stringValue = [[string substringWithRange:[parse rangeAtIndex:2]] copy];
                                [inv setArgument:&stringValue atIndex:2];
                                
                            } else {
                                return NO;
                            }
                        }
                        [inv setArgument:&parsedArray atIndex:2];
                    } else {
                        NSString *string = [filterAttributes objectForKey:propertyKey];
                        NSTextCheckingResult *parse = [parsingRegex firstMatchInString:string
                                                                               options:0
                                                                                 range:NSMakeRange(0, [string length])];
                        
                        NSString *modifier = [string substringWithRange:[parse rangeAtIndex:1]];
                        if ([modifier isEqualToString:@"float"]) {
                            // Float modifier, one argument
                            CGFloat value = [[string substringWithRange:[parse rangeAtIndex:2]] floatValue];
                            [inv setArgument:&value atIndex:2];
                        } else if ([modifier isEqualToString:@"CGPoint"]) {
                            // CGPoint modifier, two float arguments
                            CGFloat x = [[string substringWithRange:[parse rangeAtIndex:2]] floatValue];
                            CGFloat y = [[string substringWithRange:[parse rangeAtIndex:3]] floatValue];
                            CGPoint value = CGPointMake(x, y);
                            [inv setArgument:&value atIndex:2];
                        } else if ([modifier isEqualToString:@"NSString"]) {
                            // NSString modifier, one string argument
                            stringValue = [[string substringWithRange:[parse rangeAtIndex:2]] copy];
                            [inv setArgument:&stringValue atIndex:2];
                            
                        } else {
                            return NO;
                        }
                    }
                }
                

                [inv invoke];
            }
        }
        [orderedFilters addObject:genericFilter];
    }
    self.filters = orderedFilters;
    
    return YES;
}
  • 配置文件示例
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
<plist version="1.0">
<dict>
    <key>Filters</key>
    <array>
        <dict>
            <key>FilterName</key>
            <string>GPUImageGammaFilter</string>
            <key>Attributes</key>
            <dict>
                <key>setGamma:</key>
                <string>float(1.0)</string>
            </dict>
        </dict>
        <dict>
            <key>FilterName</key>
            <string>GPUImageBrightnessFilter</string>
            <key>Attributes</key>
            <dict>
                <key>setBrightness:</key>
                <string>float(0.2)</string>
            </dict>
        </dict>
    </array>
</dict>
</plist>
  • 其它方法
// filter的增加、刪除、替換
- (void) addFilter:(GPUImageOutput<GPUImageInput> *)filter;
- (void) addFilter:(GPUImageOutput<GPUImageInput> *)filter atIndex:(NSUInteger)insertIndex;
- (void) replaceFilterAtIndex:(NSUInteger)index withFilter:(GPUImageOutput<GPUImageInput> *)filter;
- (void) replaceAllFilters:(NSArray *) newFilters;
- (void) removeFilter:(GPUImageOutput<GPUImageInput> *)filter;
- (void) removeFilterAtIndex:(NSUInteger)index;
- (void) removeAllFilters;

// 由final filter生成圖片
- (UIImage *) currentFilteredFrame;
- (UIImage *) currentFilteredFrameWithOrientation:(UIImageOrientation)imageOrientation;
- (CGImageRef) newCGImageFromCurrentFilteredFrame;

// 在特定位置增加filter
- (void)addFilter:(GPUImageOutput<GPUImageInput> *)filter atIndex:(NSUInteger)insertIndex {
    [self.filters insertObject:filter atIndex:insertIndex];
    [self _refreshFilters];
}

// 在末尾增加filter
- (void)addFilter:(GPUImageOutput<GPUImageInput> *)filter {
    [self.filters addObject:filter];
    [self _refreshFilters];
}

// 替換filter
- (void)replaceFilterAtIndex:(NSUInteger)index withFilter:(GPUImageOutput<GPUImageInput> *)filter {
    [self.filters replaceObjectAtIndex:index withObject:filter];
    [self _refreshFilters];
}

// 刪除filter
- (void) removeFilter:(GPUImageOutput<GPUImageInput> *)filter;
{
    [self.filters removeObject:filter];
    [self _refreshFilters];
}

// 刪除特定索引的filter
- (void)removeFilterAtIndex:(NSUInteger)index {
    [self.filters removeObjectAtIndex:index];
    [self _refreshFilters];
}

// 刪除所有filter
- (void)removeAllFilters {
    [self.filters removeAllObjects];
    [self _refreshFilters];
}

// 替換所有filter
- (void)replaceAllFilters:(NSArray *)newFilters {
    self.filters = [NSMutableArray arrayWithArray:newFilters];
    [self _refreshFilters];
}

// 更新響應鏈
- (void)_refreshFilters {
    // 將input作為響應鏈源
    id prevFilter = self.input;
    GPUImageOutput<GPUImageInput> *theFilter = nil;
    
    // 循環添加
    for (int i = 0; i < [self.filters count]; i++) {
        theFilter = [self.filters objectAtIndex:i];
        [prevFilter removeAllTargets];
        [prevFilter addTarget:theFilter];
        prevFilter = theFilter;
    }
    
    [prevFilter removeAllTargets];
    
    // 最后將output加入響應鏈  
    if (self.output != nil) {
        [prevFilter addTarget:self.output];
    }
}

// 由final filter生成圖像
- (UIImage *)currentFilteredFrame {
    return [(GPUImageOutput<GPUImageInput> *)[_filters lastObject] imageFromCurrentFramebuffer];
}

// 根據imageOrientation生成圖像
- (UIImage *)currentFilteredFrameWithOrientation:(UIImageOrientation)imageOrientation {
  return [(GPUImageOutput<GPUImageInput> *)[_filters lastObject] imageFromCurrentFramebufferWithOrientation:imageOrientation];
}

// 由final filter生成圖像
- (CGImageRef)newCGImageFromCurrentFilteredFrame {
    return [(GPUImageOutput<GPUImageInput> *)[_filters lastObject] newCGImageFromCurrentlyProcessedOutput];
}

GPUImageFilterGroup

GPUImageFilterGroup繼承自GPUImageOutput ,實現了GPUImageInput協議。因此,可以自身可以作為獨立的濾鏡參與響應鏈中。相比GPUImageFilterPipeline,GPUImageFilterGroup功能更強大。

  • 屬性
@property(readwrite, nonatomic, strong) GPUImageOutput<GPUImageInput> *terminalFilter;
@property(readwrite, nonatomic, strong) NSArray *initialFilters;
@property(readwrite, nonatomic, strong) GPUImageOutput<GPUImageInput> *inputFilterToIgnoreForUpdates; 
  • 方法
// 增加濾鏡
- (void)addFilter:(GPUImageOutput<GPUImageInput> *)newFilter;
- (GPUImageOutput<GPUImageInput> *)filterAtIndex:(NSUInteger)filterIndex;
- (NSUInteger)filterCount;

// 增加濾鏡
- (void)addFilter:(GPUImageOutput<GPUImageInput> *)newFilter;
{
    [filters addObject:newFilter];
}

// 獲取濾鏡
- (GPUImageOutput<GPUImageInput> *)filterAtIndex:(NSUInteger)filterIndex;
{
    return [filters objectAtIndex:filterIndex];
}

// 濾鏡數量
- (NSUInteger)filterCount;
{
    return [filters count];
}

// 
- (void)newFrameReadyAtTime:(CMTime)frameTime atIndex:(NSInteger)textureIndex;
{
    for (GPUImageOutput<GPUImageInput> *currentFilter in _initialFilters)
    {
        if (currentFilter != self.inputFilterToIgnoreForUpdates)
        {
            [currentFilter newFrameReadyAtTime:frameTime atIndex:textureIndex];
        }
    }
}

// 設置幀緩存對象
- (void)setInputFramebuffer:(GPUImageFramebuffer *)newInputFramebuffer atIndex:(NSInteger)textureIndex;
{
    for (GPUImageOutput<GPUImageInput> *currentFilter in _initialFilters)
    {
        [currentFilter setInputFramebuffer:newInputFramebuffer atIndex:textureIndex];
    }
}

實現過程

  • 從文件中讀取濾鏡配置。
// 加載圖片
GPUImagePicture *picture = [[GPUImagePicture alloc] initWithImage:[UIImage imageNamed:@"1.jpg"]];

// 配置文件
NSURL *file = [[NSBundle mainBundle] URLForResource:@"filterConfig" withExtension:@"plist"];

// 濾鏡組合
_pipleLine = [[GPUImageFilterPipeline alloc] initWithConfigurationFile:file input:picture output:_imageView];

[picture processImage];
  • 直接配置濾鏡。
// 加載圖片
GPUImagePicture *picture = [[GPUImagePicture alloc] initWithImage:[UIImage imageNamed:@"2.jpg"]];

// filters
GPUImageRGBFilter *rgbFilter = [[GPUImageRGBFilter alloc] init];
GPUImagePerlinNoiseFilter *noiseFilter = [[GPUImagePerlinNoiseFilter alloc] init];

// 配置文件
NSArray *filters = @[rgbFilter, noiseFilter];

// 濾鏡組合
_pipleLine = [[GPUImageFilterPipeline alloc] initWithOrderedFilters:filters input:picture output:_imageView];

[picture processImage];
  • 自定義組合濾鏡。
#import "QMGrayBilateralBlendFilter.h"

NSString *const kGrayBilateralBlendFragmentShaderString = SHADER_STRING
(
 varying highp vec2 textureCoordinate;
 varying highp vec2 textureCoordinate2;
 
 uniform sampler2D inputImageTexture;
 uniform sampler2D inputImageTexture2;
 
 void main()
 {
     highp vec4 gray = texture2D(inputImageTexture, textureCoordinate);
     highp vec4 bilateral = texture2D(inputImageTexture2, textureCoordinate2);
     
     highp float range = distance(textureCoordinate, vec2(0.5, 0.5));
     
     highp vec4 dstClor = bilateral;
     if (range < 0.4) {
         dstClor = gray;
     }
     
     gl_FragColor = dstClor;
     
     //gl_FragColor = vec4(vec3(mix(gray, bilateral, 0.6)), 1.0);
 }
 );

@implementation QMGrayBilateralBlendFilter

- (instancetype)init
{
    if (self = [super init])
    {
        // GrayscaleFilter
        GPUImageGrayscaleFilter *grayFilter = [[GPUImageGrayscaleFilter alloc] init];
        [self addFilter:grayFilter];
        
        // BilateralFilter
        GPUImageBilateralFilter *bilateralFilter = [[GPUImageBilateralFilter alloc] init];
        bilateralFilter.distanceNormalizationFactor = 16.0;
        [self addFilter:bilateralFilter];
        
        // GPUImageTwoPassFilter
        GPUImageTwoInputFilter *twoPassFilter = [[GPUImageTwoInputFilter alloc] initWithFragmentShaderFromString:kGrayBilateralBlendFragmentShaderString];
        
        // GPUImageHSBFilter
        GPUImageHSBFilter *hsbFilter = [[GPUImageHSBFilter alloc] init];
        
        [grayFilter addTarget:twoPassFilter];
        [bilateralFilter addTarget:twoPassFilter];
        [twoPassFilter addTarget:hsbFilter];
        
        self.initialFilters = @[grayFilter, bilateralFilter];
        self.terminalFilter = hsbFilter;
        
    }
    return self;
}
@end

總結

GPUImageFilterPipeline、GPUImageFilterGroup 都可用于組合濾鏡,GPUImageFilterPipeline相對比較簡單,可定制程度較低,而GPUImageFilterGroup則功能比較強大,可定制程度高。在選用的時候可以根據自己的需求,選用不同的濾鏡組合。

源碼地址:GPUImage源碼閱讀系列 https://github.com/QinminiOS/GPUImage
系列文章地址:GPUImage源碼閱讀 http://www.lxweimin.com/nb/11749791

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容

  • 發現 關注 消息 iOS 第三方庫、插件、知名博客總結 作者大灰狼的小綿羊哥哥關注 2017.06.26 09:4...
    肇東周閱讀 12,180評論 4 61
  • 村子的山坡旁,是羊群的歡樂場。每天都能看到一群無憂無慮的小羊,是不是這就是我們理想的天堂。
    FiftytwoHzwhale閱讀 267評論 0 0
  • 快樂的時光總是飛逝,和我家妞在一起的時光真好 看,美不? 兩個人一起呆著,一起騎車,一起跑步,一起看電影,一起做飯...
    Deanbian閱讀 241評論 2 0
  • 早上好!#幸福實修#~每天進步1%#幸福實修10班@胡玲美07--蘇州# 20170822(29/30) 【幸福三...
    胡玲美hlm閱讀 208評論 3 2
  • 獲客成本不斷高企,急尋擴張的醫藥電商需要品牌塑建和更多流量。 近2年來,最火爆的當屬網絡直播。2016年作為直播元...
    此老張非彼老張閱讀 961評論 1 1