LayoutConstraintItem.swift 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  1. //
  2. // SnapKit
  3. //
  4. // Copyright (c) 2011-Present SnapKit Team - https://github.com/SnapKit
  5. //
  6. // Permission is hereby granted, free of charge, to any person obtaining a copy
  7. // of this software and associated documentation files (the "Software"), to deal
  8. // in the Software without restriction, including without limitation the rights
  9. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10. // copies of the Software, and to permit persons to whom the Software is
  11. // furnished to do so, subject to the following conditions:
  12. //
  13. // The above copyright notice and this permission notice shall be included in
  14. // all copies or substantial portions of the Software.
  15. //
  16. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  22. // THE SOFTWARE.
  23. #if os(iOS) || os(tvOS)
  24. import UIKit
  25. #else
  26. import AppKit
  27. #endif
  28. public protocol LayoutConstraintItem: AnyObject {
  29. }
  30. @available(iOS 9.0, OSX 10.11, *)
  31. extension ConstraintLayoutGuide : LayoutConstraintItem {
  32. }
  33. extension ConstraintView : LayoutConstraintItem {
  34. }
  35. extension LayoutConstraintItem {
  36. internal func prepare() {
  37. if let view = self as? ConstraintView {
  38. view.translatesAutoresizingMaskIntoConstraints = false
  39. }
  40. }
  41. internal var superview: ConstraintView? {
  42. if let view = self as? ConstraintView {
  43. return view.superview
  44. }
  45. if #available(iOS 9.0, OSX 10.11, *), let guide = self as? ConstraintLayoutGuide {
  46. return guide.owningView
  47. }
  48. return nil
  49. }
  50. internal var constraints: [Constraint] {
  51. return self.constraintsSet.allObjects as! [Constraint]
  52. }
  53. internal func add(constraints: [Constraint]) {
  54. let constraintsSet = self.constraintsSet
  55. for constraint in constraints {
  56. constraintsSet.add(constraint)
  57. }
  58. }
  59. internal func remove(constraints: [Constraint]) {
  60. let constraintsSet = self.constraintsSet
  61. for constraint in constraints {
  62. constraintsSet.remove(constraint)
  63. }
  64. }
  65. private var constraintsSet: NSMutableSet {
  66. let constraintsSet: NSMutableSet
  67. if let existing = objc_getAssociatedObject(self, &constraintsKey) as? NSMutableSet {
  68. constraintsSet = existing
  69. } else {
  70. constraintsSet = NSMutableSet()
  71. objc_setAssociatedObject(self, &constraintsKey, constraintsSet, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
  72. }
  73. return constraintsSet
  74. }
  75. }
  76. private var constraintsKey: UInt8 = 0