swift-duct-calc/Sources/ManualDCore/DuctSizing.swift

import Dependencies
import Foundation

public enum DuctSizing {

  public struct RectangularDuct: Codable, Equatable, Identifiable, Sendable {

    public let id: UUID
    public let register: Int?
    public let height: Int

    public init(
      id: UUID = .init(),
      register: Int? = nil,
      height: Int,
    ) {
      self.id = id
      self.register = register
      self.height = height
    }

  }

  public struct SizeContainer: Codable, Equatable, Sendable {

    public let rectangularID: RectangularDuct.ID?
    public let designCFM: DesignCFM
    public let roundSize: Double
    public let finalSize: Int
    public let velocity: Int
    public let flexSize: Int
    public let height: Int?
    public let width: Int?

    public init(
      rectangularID: RectangularDuct.ID? = nil,
      designCFM: DuctSizing.DesignCFM,
      roundSize: Double,
      finalSize: Int,
      velocity: Int,
      flexSize: Int,
      height: Int? = nil,
      width: Int? = nil
    ) {
      self.rectangularID = rectangularID
      self.designCFM = designCFM
      self.roundSize = roundSize
      self.finalSize = finalSize
      self.velocity = velocity
      self.flexSize = flexSize
      self.height = height
      self.width = width
    }
  }

  // TODO: Uses SizeContainer

  @dynamicMemberLookup
  public struct RoomContainer: Codable, Equatable, Sendable {

    public let roomID: Room.ID
    public let roomName: String
    public let roomRegister: Int
    public let heatingLoad: Double
    public let coolingLoad: Double
    public let heatingCFM: Double
    public let coolingCFM: Double
    public let ductSize: SizeContainer

    public init(
      roomID: Room.ID,
      roomName: String,
      roomRegister: Int,
      heatingLoad: Double,
      coolingLoad: Double,
      heatingCFM: Double,
      coolingCFM: Double,
      ductSize: SizeContainer
    ) {
      self.roomID = roomID
      self.roomName = roomName
      self.roomRegister = roomRegister
      self.heatingLoad = heatingLoad
      self.coolingLoad = coolingLoad
      self.heatingCFM = heatingCFM
      self.coolingCFM = coolingCFM
      self.ductSize = ductSize
    }

    // public init(
    //   roomID: Room.ID,
    //   roomName: String,
    //   roomRegister: Int,
    //   heatingLoad: Double,
    //   coolingLoad: Double,
    //   heatingCFM: Double,
    //   coolingCFM: Double,
    //   designCFM: DesignCFM,
    //   roundSize: Double,
    //   finalSize: Int,
    //   velocity: Int,
    //   flexSize: Int,
    //   rectangularSize: RectangularDuct? = nil,
    //   rectangularWidth: Int? = nil
    // ) {
    //   self.roomID = roomID
    //   self.roomName = roomName
    //   self.roomRegister = roomRegister
    //   self.heatingLoad = heatingLoad
    //   self.coolingLoad = coolingLoad
    //   self.heatingCFM = heatingCFM
    //   self.coolingCFM = coolingCFM
    //   self.ductSize = .init(
    //     rectangularID: rectangularSize?.id,
    //     designCFM: designCFM,
    //     roundSize: roundSize,
    //     finalSize: finalSize,
    //     velocity: velocity,
    //     flexSize: flexSize,
    //     height: rectangularSize?.height,
    //     width: rectangularWidth
    //   )
    // }

    public subscript<T>(dynamicMember keyPath: KeyPath<DuctSizing.SizeContainer, T>) -> T {
      ductSize[keyPath: keyPath]
    }
  }

  public enum DesignCFM: Codable, Equatable, Sendable {
    case heating(Double)
    case cooling(Double)

    public init(heating: Double, cooling: Double) {
      if heating >= cooling {
        self = .heating(heating)
      } else {
        self = .cooling(cooling)
      }
    }

    public var value: Double {
      switch self {
      case .heating(let value): return value
      case .cooling(let value): return value
      }
    }
  }
}

extension DuctSizing {

  // Represents the database model that the duct sizes have been calculated
  // for.
  @dynamicMemberLookup
  public struct TrunkContainer: Codable, Equatable, Identifiable, Sendable {
    public var id: TrunkSize.ID { trunk.id }

    public let trunk: TrunkSize
    public let ductSize: SizeContainer

    public init(
      trunk: TrunkSize,
      ductSize: SizeContainer
    ) {
      self.trunk = trunk
      self.ductSize = ductSize
    }

    public subscript<T>(dynamicMember keyPath: KeyPath<DuctSizing.TrunkSize, T>) -> T {
      trunk[keyPath: keyPath]
    }

    public subscript<T>(dynamicMember keyPath: KeyPath<DuctSizing.SizeContainer, T>) -> T {
      ductSize[keyPath: keyPath]
    }
  }

  // TODO: Add an optional label that the user can set.

  // Represents the database model.
  public struct TrunkSize: Codable, Equatable, Identifiable, Sendable {

    public let id: UUID
    public let projectID: Project.ID
    public let type: TrunkType
    public let rooms: [RoomProxy]
    public let height: Int?
    public let name: String?

    public init(
      id: UUID,
      projectID: Project.ID,
      type: DuctSizing.TrunkSize.TrunkType,
      rooms: [DuctSizing.TrunkSize.RoomProxy],
      height: Int? = nil,
      name: String? = nil
    ) {
      self.id = id
      self.projectID = projectID
      self.type = type
      self.rooms = rooms
      self.height = height
      self.name = name
    }
  }

}

extension DuctSizing.TrunkSize {
  public struct Create: Codable, Equatable, Sendable {

    public let projectID: Project.ID
    public let type: TrunkType
    public let rooms: [Room.ID: [Int]]
    public let height: Int?
    public let name: String?

    public init(
      projectID: Project.ID,
      type: DuctSizing.TrunkSize.TrunkType,
      rooms: [Room.ID: [Int]],
      height: Int? = nil,
      name: String? = nil
    ) {
      self.projectID = projectID
      self.type = type
      self.rooms = rooms
      self.height = height
      self.name = name
    }
  }

  public struct Update: Codable, Equatable, Sendable {

    public let type: TrunkType?
    public let rooms: [Room.ID: [Int]]?
    public let height: Int?
    public let name: String?

    public init(
      type: DuctSizing.TrunkSize.TrunkType? = nil,
      rooms: [Room.ID: [Int]]? = nil,
      height: Int? = nil,
      name: String? = nil
    ) {
      self.type = type
      self.rooms = rooms
      self.height = height
      self.name = name
    }
  }

  // TODO: Make registers non-optional
  public struct RoomProxy: Codable, Equatable, Identifiable, Sendable {

    public var id: Room.ID { room.id }
    public let room: Room
    public let registers: [Int]

    public init(room: Room, registers: [Int]) {
      self.room = room
      self.registers = registers
    }
  }

  public enum TrunkType: String, CaseIterable, Codable, Equatable, Sendable {
    case `return`
    case supply

    public static let allCases = [Self.supply, .return]
  }
}