CustomCameraViewController.swift 9.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265
  1. //
  2. // CustomCameraViewController.swift
  3. // MCPlus
  4. //
  5. // Created by seo ha on 31/01/2019.
  6. // Copyright © 2019 KangSH. All rights reserved.
  7. //
  8. import UIKit
  9. import AVFoundation
  10. import ReactiveCocoa
  11. import ReactiveSwift
  12. class CustomCameraViewController: UIViewController {
  13. @IBOutlet weak var patientIdLabel :UILabel!{
  14. didSet{
  15. patientIdLabel.reactive.text <~ PictureViewController.patientId
  16. }
  17. }
  18. @IBOutlet weak var patientNmLabel :UILabel!{
  19. didSet{
  20. patientNmLabel.reactive.text <~ PictureViewController.patientNm
  21. }
  22. }
  23. @IBOutlet weak var ageGenderLabel :UILabel!{
  24. didSet{
  25. ageGenderLabel.reactive.text <~ PictureViewController.ageGender
  26. }
  27. }
  28. @IBOutlet weak var previewView: UIView!
  29. @IBOutlet weak var takePhoto: UIButton!{
  30. didSet{
  31. takePhoto.addTapGestureRecognizer(action: self.takeAction)
  32. }
  33. }
  34. @IBOutlet weak var finishButton: UIButton!{
  35. didSet{
  36. finishButton.addTapGestureRecognizer(action: self.finish)
  37. }
  38. }
  39. @IBOutlet weak var albumButton: UIButton!{
  40. didSet{
  41. albumButton.addTapGestureRecognizer(action: self.albumAction)
  42. }
  43. }
  44. var session: AVCaptureSession?
  45. var stillImageOutput:AVCaptureStillImageOutput{
  46. let stillImageOutput = AVCaptureStillImageOutput()
  47. stillImageOutput.outputSettings = [AVVideoCodecKey: AVVideoCodecJPEG]
  48. return stillImageOutput
  49. }
  50. @available(iOS 11.0, *)
  51. lazy var newStillImageOutput: AVCapturePhotoOutput = AVCapturePhotoOutput()
  52. var videoPreviewLayer: AVCaptureVideoPreviewLayer?
  53. static var photoProcessCommit = MutableProperty<Photo?>(nil)
  54. }
  55. extension CustomCameraViewController{
  56. override func viewDidLoad() {
  57. super.viewDidLoad()
  58. session = AVCaptureSession()
  59. session!.sessionPreset = AVCaptureSession.Preset.photo
  60. let backCamera = AVCaptureDevice.default(for: AVMediaType.video)
  61. var error: NSError?
  62. var input: AVCaptureDeviceInput!
  63. do {
  64. if let camera = backCamera{
  65. input = try AVCaptureDeviceInput(device: camera)
  66. }
  67. } catch let error1 as NSError {
  68. error = error1
  69. input = nil
  70. print(error!.localizedDescription)
  71. }
  72. if error == nil && session!.canAddInput(input) {
  73. session!.addInput(input)
  74. if session!.canAddOutput(stillImageOutput) {
  75. if #available(iOS 11.0, *) {
  76. session!.addOutput(newStillImageOutput)
  77. }else{
  78. session!.addOutput(stillImageOutput)
  79. }
  80. videoPreviewLayer = AVCaptureVideoPreviewLayer(session: session!)
  81. videoPreviewLayer!.videoGravity = AVLayerVideoGravity.resizeAspect
  82. videoPreviewLayer!.connection?.videoOrientation = AVCaptureVideoOrientation.portrait
  83. previewView.layer.addSublayer(videoPreviewLayer!)
  84. if let size = previewView?.bounds.size{
  85. videoPreviewLayer?.frame.size = size
  86. }
  87. videoPreviewLayer?.videoGravity = .resizeAspectFill
  88. session!.startRunning()
  89. }
  90. }
  91. }
  92. override func viewDidAppear(_ animated: Bool) {
  93. super.viewDidAppear(animated)
  94. videoPreviewLayer!.frame = previewView.bounds
  95. }
  96. func finish(){
  97. self.navigationController?.popViewController(animated: true)
  98. }
  99. func albumAction(){
  100. if let parent = self.parent as? PictureViewController{
  101. parent.gallaryAction(action: nil)
  102. }
  103. }
  104. func takeAction(){
  105. if #available(iOS 11.0, *) {
  106. let settings = AVCapturePhotoSettings()
  107. settings.livePhotoVideoCodecType = .jpeg
  108. newStillImageOutput.capturePhoto(with: settings, delegate: self)
  109. }else{
  110. guard let connection = stillImageOutput.connection(with: AVMediaType.video) else {
  111. return
  112. }
  113. stillImageOutput.captureStillImageAsynchronously(from: connection, completionHandler: { [weak self](buffer, error) in
  114. if let _ = error{
  115. return
  116. }
  117. guard let buffer = buffer else{
  118. return
  119. }
  120. if let imageData = AVCaptureStillImageOutput.jpegStillImageNSDataRepresentation(buffer){
  121. let image = UIImage(data: imageData)
  122. self?.processingImage(image: image)
  123. }
  124. })
  125. }
  126. }
  127. func processingImage(image:UIImage?){
  128. if let image = image{
  129. DispatchQueue(label: "background").async{ [weak self] in
  130. guard let `self` = self else{ return }
  131. let datas = [image]
  132. .map(self.resizableTaransform)
  133. .map(self.optimisingTransform)
  134. //도큐먼트의 유저 디렉토리를 찾는다
  135. if let dir = FileManager.default.urls(for: .documentDirectory, in: .userDomainMask).first {
  136. let path = dir.appendingPathComponent("kunkuk")
  137. //해당 폴더가 없으면 생성
  138. if !FileManager.default.fileExists(atPath: path.path){
  139. try? FileManager.default.createDirectory(at: path, withIntermediateDirectories: true, attributes: nil)
  140. }
  141. let thumb = dir.appendingPathComponent("thumb")
  142. //해당 폴더가 없으면 생성
  143. if !FileManager.default.fileExists(atPath: thumb.path){
  144. try? FileManager.default.createDirectory(at: thumb, withIntermediateDirectories: true, attributes: nil)
  145. }
  146. //각 데이터 저장
  147. for data in datas{
  148. let date = Date()
  149. let file = "kunkuk_\(date.fromString(format: "yyyyMMddhhmmss")).jpg"
  150. let fileURL = path.appendingPathComponent(String(file))
  151. let thumbURL = thumb.appendingPathComponent(String(file))
  152. try? data?.write(to: fileURL)
  153. let origin = UIImage(data: data!)
  154. try? self.getThumbnailData(image: origin!)?.write(to: thumbURL)
  155. do {
  156. guard let parent = self.parent as? PictureViewController else{ return }
  157. let date = Date().fromString(format: "yyyy-MM-dd")
  158. let name = parent.user?.patientNm ?? ""
  159. var treatCls = ""
  160. let deptNm = parent.user?.deptNm ?? ""
  161. switch parent.user?.treatCls{
  162. //외래
  163. case "O"?:
  164. treatCls = "외래"
  165. //입원
  166. case "I"?:
  167. treatCls = "입원"
  168. //응급
  169. case "E"?:
  170. treatCls = "응급"
  171. default:
  172. break
  173. }
  174. let photo = Photo()
  175. photo.user = parent.user
  176. photo.file = file
  177. photo.date = date
  178. photo.key = "\(date) \(name) \(treatCls), \(deptNm)"
  179. photo.isSended = false
  180. photo.save()
  181. //노티피케이션 알림
  182. CustomCameraViewController.photoProcessCommit.value = photo
  183. }
  184. }
  185. }
  186. }
  187. }
  188. }
  189. func resizableTaransform(image:UIImage) -> UIImage {
  190. let width = image.size.width
  191. let height = image.size.height
  192. var ratio:CGFloat = 1.0
  193. if width > 640{
  194. ratio = width / 640
  195. }
  196. return image.scaleToSize(newSize: CGSize(width: width / ratio, height: height / ratio))
  197. }
  198. func optimisingTransform(image:UIImage) -> Data? {
  199. return image.jpegData(compressionQuality: 0.7)
  200. }
  201. func getThumbnailData(image:UIImage) -> Data? {
  202. let datas = [image]
  203. .map(resizableTaransform)
  204. .map(optimisingTransform)
  205. return datas.first!
  206. }
  207. }
  208. extension CustomCameraViewController: AVCapturePhotoCaptureDelegate{
  209. @available(iOS 11.0, *)
  210. func photoOutput(_ output: AVCapturePhotoOutput, didFinishProcessingPhoto photo: AVCapturePhoto, error: Error?) {
  211. guard error == nil else{ return }
  212. guard let imageData = photo.fileDataRepresentation() else{ return }
  213. let image = UIImage(data: imageData)
  214. self.processingImage(image: image)
  215. }
  216. }