iOS 创建PDF文件并在文件中绘制波形图和文字

// 1、.m文件

//
//  CreatePDFImage.m
//  Helo
//
//  Created by 杨峰 on 2017/12/25.
//  Copyright © 2017年 seedmorn. All rights reserved.
//

#import "CreatePDFImage.h"
#import <CoreText/CoreText.h>

@implementation CreatePDFImage

- (instancetype)init {
    self = [super init];
    if (self) {
        // defaults File Name
        self.fileName = @"EcgImageView";
        
        // defaults String
        self.Title = @"Health Report";
        self.Message_1 = @"The report is for reference only";
        self.Message_2 = @"Name:fabio Sex:Male Age:45 Height:170cm Weight:83kg";
        self.Message_3 = @"Average HR: 77bpm";
        self.Message_4 = @"Gain: 10mm/mv Walking speed: 25mm/s Date: 2017-11-29 11:58:28";
        
        // defaults values
        self.FileWidth = 900;
        self.CellHeight = 45;
        self.CellPointCount = 2800;
        self.CellGap = 30;
        self.Top = 100;
        self.Left = 10;
        self.Right = 10;
        self.GridWidth = 15;
        self.GridLineWidth = 0.5;
        self.LineWidth = 1.0;
        
    }
    return self;
}

- (void)CreatePdfFile:(NSArray *)array {
    
    // 1.设置页面大小
    CGFloat myPageWidth = _FileWidth;
    CGFloat myPageHeight = [UIScreen mainScreen].bounds.size.height;
    CGRect mediaBox = CGRectMake (0, 0, myPageWidth, myPageHeight);
    
    // 2.设置pdf文档存储的路径
    NSArray *paths = NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES);
    NSString *documentsDirectory = paths[0];
    NSString *filePath = [documentsDirectory stringByAppendingString:[NSString stringWithFormat:@"/%@.pdf",_fileName]];
    const char *cfilePath = [filePath UTF8String];
    CFStringRef pathRef = CFStringCreateWithCString(NULL, cfilePath, kCFStringEncodingUTF8);
    
    // 3.设置当前pdf页面的属性
    CFStringRef myKeys[1];
    CFTypeRef myValues[1];
    myKeys[0] = kCGPDFContextMediaBox;
    myValues[0] = (CFTypeRef) CFDataCreate(NULL,(const UInt8 *)&mediaBox, sizeof (CGRect));
    CFDictionaryRef pageDictionary = CFDictionaryCreate(NULL, (const void **) myKeys, (const void **) myValues, 1,
                                                        &kCFTypeDictionaryKeyCallBacks, & kCFTypeDictionaryValueCallBacks);
    
    // 4.获取pdf绘图上下文
    CGContextRef myPDFContext = MyPDFContextCreate (&mediaBox, pathRef);
    
    // 5.开始描绘第一页页面
    CGPDFContextBeginPage(myPDFContext, pageDictionary);
    
    //开始画pdf
    [self drawViewss:myPDFContext width:myPageWidth height:myPageHeight array:array];
    
    // 绘制title
    CGFloat width_0 = [self getStringWidthFontSize:17 string:_Title];
    CGRect rect_0 = CGRectMake((myPageWidth - width_0) / 2.0, myPageHeight - _Top / 2.0, width_0, 20);
    [self DrawTextWith:myPDFContext string:_Title color:[UIColor blackColor] fontSize:17 rect:rect_0 fontName:CFSTR("Helvetica-Bold")];
    
    // 绘制message_1
    CGFloat width_1 = [self getStringWidthFontSize:11 string:_Message_1];
    CGRect rect_1 = CGRectMake(myPageWidth - width_1, myPageHeight - 30, width_1, 20);
    [self DrawTextWith:myPDFContext string:_Message_1 color:[UIColor grayColor] fontSize:11 rect:rect_1 fontName:CFSTR("Helvetica")];
    
    // 绘制message_2
    CGFloat width_2 = [self getStringWidthFontSize:13 string:_Message_2];
    CGRect rect_2 = CGRectMake((myPageWidth - width_2) / 2.0, myPageHeight - (_Top - 10), width_2, 20);
    [self DrawTextWith:myPDFContext string:_Message_2 color:[UIColor blackColor] fontSize:13 rect:rect_2 fontName:CFSTR("Helvetica")];

    CGPDFContextEndPage(myPDFContext);
    
    // 8.释放创建的对象
    CFRelease(pageDictionary);
    CFRelease(myValues[0]);
    CGContextRelease(myPDFContext);
}


- (void)DrawTextWith:(CGContextRef)context string:(NSString *)string color:(UIColor *)color fontSize:(NSInteger)fontSize rect:(CGRect)rect fontName:(CFStringRef)fontName {
    
    CTFontRef ctfont = CTFontCreateWithName(fontName, fontSize, NULL);
    //CTFontRef ctfont = CTFontCreateWithName(CFSTR("STHeitiSC-Medium"), fontSize, NULL);
    CGColorRef ctColor = [color CGColor];
    
    // Create an attributed string
    CFStringRef keys[] = { kCTFontAttributeName,kCTForegroundColorAttributeName };
    CFTypeRef values[] = { ctfont,ctColor};
    CFDictionaryRef attr = CFDictionaryCreate(NULL, (const void **)&keys, (const void **)&values,
                                              sizeof(keys) / sizeof(keys[0]), &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
    
    
    CFStringRef ctStr = CFStringCreateWithCString(nil, [string UTF8String], kCFStringEncodingUTF8);
    CFAttributedStringRef attrString = CFAttributedStringCreate(NULL,ctStr, attr);
    CTLineRef line = CTLineCreateWithAttributedString(attrString);
    CGContextSetTextMatrix(context, CGAffineTransformIdentity);
    CGContextSetTextPosition(context, rect.origin.x, rect.origin.y);
    CTLineDraw(line, context);
    
    CFRelease(line);
    CFRelease(attrString);
    CFRelease(ctStr);
    
    // Clean up
    CFRelease(attr);
    CFRelease(ctfont);
}


//获取字符串宽度
-(CGFloat)getStringWidthFontSize:(NSInteger)sizeFont string:(NSString *)string {
    CGRect rect = [string boundingRectWithSize:CGSizeMake(MAXFLOAT, 14) options:NSStringDrawingTruncatesLastVisibleLine | NSStringDrawingUsesFontLeading | NSStringDrawingUsesLineFragmentOrigin attributes:@{NSFontAttributeName:[UIFont boldSystemFontOfSize:sizeFont]} context:nil];
    return rect.size.width + 5;
}

- (void)drawViewss:(CGContextRef)context width:(CGFloat)view_w height:(CGFloat)view_h array:(NSArray *)array {
    
    // 最大值
    CGFloat max = 1200;
    // 一行中波形绘制的范围（高度）
    CGFloat height = _CellHeight;
    // 每行有多少个点
    CGFloat pointCount = _CellPointCount;
    // 每行之间两个点的间距
    CGFloat gap = view_w / pointCount;
    
    // 行间距
    NSInteger cellGap = _CellGap;
    
    
    CGFloat left = _Left;
    CGFloat right = _Right;
    CGFloat top = _Top;
    
    
    // 绘制网格
    CGFloat k = _GridWidth;
    
    // 横线
    NSInteger cellCount = ceilf(array.count / pointCount);
    
    NSInteger hei = cellCount * height + (cellCount + 1) * cellGap;
    CGFloat start_y = view_h - top;
    
    CGContextSetLineWidth(context, _GridLineWidth);
    CGContextSetStrokeColorWithColor(context, [UIColor redColor].CGColor);
    for (int i = 0; i <= hei / k; i ++) {
        
        CGContextMoveToPoint(context, left , start_y);
        CGContextAddLineToPoint(context, view_w - right, start_y);
        CGContextStrokePath(context);
        
        start_y -= k;
    }
    
    // 竖线
    NSInteger vCount = floorf((view_w - left - right) / k);
    for (int i = 0; i <= vCount; i ++) {
        CGContextMoveToPoint(context, left + k * i, view_h - top);
        CGContextAddLineToPoint(context, left + k * i, start_y + k);
        CGContextStrokePath(context);
    }
    
    
    // 绘制波形
    CGContextSetLineWidth(context, _LineWidth);
    CGContextSetStrokeColorWithColor(context, [UIColor blackColor].CGColor);
    NSInteger count = 0;
    NSInteger cell = 0;
    
    for (int i = 0; i < array.count - 1; i ++) {
        
        NSInteger k_1 = [array[i] integerValue];
        NSInteger k_2 = [array[i + 1] integerValue];
        
        if (k_1 > max) {
            k_1 = max;
        }
        if (k_1 < 0) {
            k_1 = 0;
        }
        
        if (k_2 > max) {
            k_2 = max;
        }
        if (k_2 < 0) {
            k_2 = 0;
        }
        
        CGFloat x_1 = left + count * gap;
        CGFloat kk = (height - k_1 / 1.0 / max * height);
        
        CGFloat y_1 = view_h - (cell * (height + cellGap) + kk) - top - cellGap;
        CGPoint point_1 = CGPointMake(x_1, y_1);
        
        CGFloat x_2 = left + (count + 1) * gap;
        CGFloat y_2 = view_h - (cell * (height + cellGap) + (height - k_2 / 1.0 / max * height)) - top - cellGap;
        CGPoint point_2 = CGPointMake(x_2, y_2);

        CGContextMoveToPoint(context, point_1.x  ,point_1.y);
        CGContextAddLineToPoint(context, point_2.x, point_2.y);
        CGContextStrokePath(context);
        
        count ++;
        
        ///*
        if (x_2 >= view_w - right) {
            count = 0;
            cell ++;
        }
        //*/
        
        /*
        if (count >= pointCount) {
            count = 0;
            cell ++;
        }
        */
    }
    
    
    // 绘制message_3
    CGFloat width_3 = [self getStringWidthFontSize:13 string:_Message_3];
    CGRect rect_3 = CGRectMake((view_w - width_3), view_h - top - 15, width_3, 20);
    [self DrawTextWith:context string:_Message_3 color:[UIColor blackColor] fontSize:13 rect:rect_3 fontName:CFSTR("Helvetica")];
    
    // 绘制message_4
    CGFloat width_4 = [self getStringWidthFontSize:13 string:_Message_4];
    CGRect rect_4 = CGRectMake((view_w - width_4), start_y + 25, width_4, 20);
    [self DrawTextWith:context string:_Message_4 color:[UIColor blackColor] fontSize:13 rect:rect_4 fontName:CFSTR("Helvetica")];
    
}

/*
 * 获取pdf绘图上下文
 * inMediaBox指定pdf页面大小
 * path指定pdf文件保存的路径
 */
CGContextRef MyPDFContextCreate (const CGRect *inMediaBox, CFStringRef path) {
    CGContextRef myOutContext = NULL;
    CFURLRef url;
    CGDataConsumerRef dataConsumer;
    
    url = CFURLCreateWithFileSystemPath (NULL, path, kCFURLPOSIXPathStyle, false);
    
    if (url != NULL) {
        dataConsumer = CGDataConsumerCreateWithURL(url);
        if (dataConsumer != NULL) {
            myOutContext = CGPDFContextCreate (dataConsumer, inMediaBox, NULL);
            CGDataConsumerRelease (dataConsumer);
        }
        CFRelease(url);
    }
    return myOutContext;
}

- (void)setFileName:(NSString *)fileName {
    _fileName = fileName;
}

- (void)setTitle:(NSString *)Title {
    _Title = Title;
}


@end

// 3、.h文件

//
//  CreatePDFImage.h
//  Helo
//
//  Created by 杨峰 on 2017/12/25.
//  Copyright © 2017年 seedmorn. All rights reserved.
//

#import <Foundation/Foundation.h>


@interface CreatePDFImage : NSObject

@property (nonatomic, strong) NSString *Title;

@property (nonatomic, strong) NSString *Message_1;

@property (nonatomic, strong) NSString *Message_2;

@property (nonatomic, strong) NSString *Message_3;

@property (nonatomic, strong) NSString *Message_4;


@property (nonatomic, strong) NSString *fileName;

@property (nonatomic) CGFloat FileWidth;

@property (nonatomic) CGFloat CellHeight;

@property (nonatomic) CGFloat CellPointCount;

@property (nonatomic) CGFloat CellGap;

@property (nonatomic) CGFloat Top;

@property (nonatomic) CGFloat Left;

@property (nonatomic) CGFloat Right;

@property (nonatomic) CGFloat GridWidth;

@property (nonatomic) CGFloat GridLineWidth;

@property (nonatomic) CGFloat LineWidth;


- (void)CreatePdfFile:(NSArray *)array;



@end

// 3、使用 （EcgArray 是绘制波形图的点）

CreatePDFImage *create = [[CreatePDFImage alloc]init];
[create CreatePdfFile:EcgArray];