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];