extensionの書き方。extensionに引数を渡して処理を分岐したい。

Question

いつもお世話になります。見当違いというか違う方法のほうがよい場合はご指摘ください。

doubleをStringに変換する処理をdoubleのextensionを記述して行いたいのですが。
引数を渡して返ってくるStringを変えるにはどのように書けばよいでしょうか？

↓とりあえず書いたもの

import Foundation

extension Double {
    var displayTimeStr: String {
        let sec = Int(self)
        let msec = Int((self - Double(sec)) * 100)
        let timeStr = NSString(format: "%02d:%02d:%02d.%02d", sec / 3600, sec % 3600 / 60, sec % 60, msec) as String
    return timeStr
    }
}

//呼びだし
let timeLabel.text = DoubleTime.displayTimeStr

↓こんな感じにしたいのですが、簡単にできるでしょうか？

import Foundation

// 引数に渡したいenum
enum timeStrType {
          case A
          case B
          case C
          case D
      }

// 引数を受け取りたいが(timeStrType: timeStrType)をどこに書いたらよいかイメージが浮かばない。
extension Double {
    var displayTimeStr: String {
        let sec = Int(self)
        let msec = Int((self - Double(sec)) * 100)
        
        let timeStr: String     
        switch timeStrType {
            case .A :
                timeStr = NSString(format: "%02d:%02d:%02d.%02d", sec / 3600, sec % 3600 / 60, sec % 60, msec) as String
            case .B :
                timeStr = NSString//省略
            case .C :
                timeStr = NSString//省略
            case .D :
                timeStr = NSString//省略
       }

    return timeStr
    }
}

//こんなふうに呼びだしたい
let timeLabel.text = DoubleTime.displayTimeStr(timeStrType: .A)

Answer 1

プロパティじゃなくメソッドで。

extension Double {
    func displayTimeStr(timeStrType: timeStrType) -> String {
        let sec = Int(self)
        let msec = Int((self - Double(sec)) * 100)
        
        let timeStr: String
        switch timeStrType {
        case .A :
            timeStr = String(format: "%02d:%02d:%02d.%02d", sec / 3600, sec % 3600 / 60, sec % 60, msec)
        case .B :
            timeStr = "..."
        case .C :
            timeStr = "..."
        case .D :
            timeStr = "..."
        }
        
        return timeStr
    }
}

Comments

[退会済みユーザー]

ありがとうございます。秒殺×2の回答を思いう浮かばない自分が悲しいです。

Answer 2

ん？

func displayTimeStr(timeStrType: TimeStrType) -> String {

※enumの名前変えてます

Comments

[退会済みユーザー]

回答ありがとうございます。
回答内容がほぼ同じで・回答時間全く同じなので…上に表示されているdaisuke7さんを便宜上ベストアンサーとさせていただきます。申し訳ないです。