iOS - 截取tableView长图

- (UIImage *)snapshotScreen
{
if ([[UIScreen mainScreen] respondsToSelector:@selector(scale)]){
    UIGraphicsBeginImageContextWithOptions(self.tableView.contentSize, NO, [UIScreen mainScreen].scale);
 
} else {
    UIGraphicsBeginImageContext(self.tableView.contentSize);
}
//先保存原来frame 和 偏移量
CGPoint savedContentOffset = self.tableView.contentOffset;
CGRect savedFrame = self.tableView.frame;
CGSize contentSize = self.tableView.contentSize;
CGRect oldBounds = self.tableView.layer.bounds;

if(@available(iOS 13, *)){
    //iOS 13 系统截屏需要改变tableview 的bounds
     [self.tableView.layer setBounds:CGRectMake(oldBounds.origin.x, oldBounds.origin.y, contentSize.width, contentSize.height)];
}

//偏移量归零
self.tableView.contentOffset = CGPointZero;
//frame变为contentSize
self.tableView.frame = CGRectMake(0, 0, self.tableView.contentSize.width, self.tableView.contentSize.height);
//截图
 [self.tableView.layer renderInContext:UIGraphicsGetCurrentContext()];
if(@available(iOS 13,*)){
    [self.tableView.layer setBounds:oldBounds];
}
UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
UIGraphicsEndImageContext();
//还原frame 和 偏移量
self.tableView.contentOffset = savedContentOffset;
self.tableView.frame = savedFrame;


//    UIImageWriteToSavedPhotosAlbum(image, nil, nil, nil);
return image;

}