週番号表示カレンダー

投稿者: | 2016年2月11日

さて、カレンダープログラムに週番号表示機能を追加しましょうか。引数チェック部分をクラス化したのでこれをメインプログラムから追い出して、57から62行目のように引数チェックの結果のみ取り込みます。

#! /usr/bin/swift
import Foundation
let month_odr: [String] = ["January", "Febrary", "March", "April", "May", "June", "July", "August", "September", "October", "November", "December"]
let week_odr: [[String]] = [["Su", "Mo", "Tu", "We", "Th", "Fr", "Sa"], ["日","月","火","水","木","金","土"]]
let esc = (fg_bk:"\u{001B}[0;30m", fg_mg:"\u{001B}[0;35m", bg_wt:"\u{001B}[1;47m", bg_gy:"\u{001B}[0;47m")
var curr = (year:0, month:0, day:0, week_m:0, week_y:0)
var d_flag = (raws:0, cols:0, week:false, year:0, month:0, je:0)
// シェル変数名を指定すると値を返す関数
func getEnvironmentVar(name: String) -> String? {
	guard let rawValue = getenv(name) else { return nil }
	return String(utf8String: rawValue)
}
// Date型の引数を受け取り、DateComponents型を返す関数
func getCalComp(date: Date) -> DateComponents {
    let cal: NSCalendar = NSCalendar(calendarIdentifier: NSCalendar.Identifier.gregorian)!
    let components: DateComponents = cal.components([.year, .month, .day, .weekday,
                                                        .weekOfYear, .weekOfMonth], from:date)
    return components
}
// 一日始まりのDateComponents型の引数を受け取り、その月の配列を返す関数
func getCalendar(components: DateComponents) -> (cal_seq: [String], dayorder: Int, totaldays: Int) {
    var endofmonth: [Int] = [0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    var cal_seq: [String] = week_odr[d_flag.je]
    if (components.year! % 4) == 0 && (components.year! % 100) != 0 || (components.year! % 400) == 0 {
        endofmonth[2]+=1
    }
    for _ in 0..<42 {
        cal_seq.append("  ")
    }
    var datestring: String
    for var i in 1..<endofmonth[components.month!]+1 {
        if i < 10 {
            datestring = " \(i)"
        } else {
            datestring = "\(i)"
        }
        if components.year! == curr.year && components.month! == curr.month && i == curr.day {
            datestring = "\(esc.bg_gy)\(datestring)\(esc.bg_wt)"
        }
        cal_seq[ (6 + i) + (components.weekday! - 1) ] = datestring
    }
    var dayorder = curr.day
    var totaldays = 0
    for var i in 0..<curr.month {
        dayorder += endofmonth[i]
    }
    for var i in 1..<12+1 {
        totaldays += endofmonth[i]
    }
    return (cal_seq, dayorder, totaldays)
}
// 実質プログラムの始まり
if getEnvironmentVar(name:"LANG") == "ja_JP.UTF-8" {
	d_flag.je = 1
}
let arguments: [String] = CommandLine.arguments
var chkopt = checkOption(arguments: arguments)
d_flag.raws = chkopt.getCalendarRaws()
d_flag.cols = chkopt.getCalendarCols()
d_flag.week = chkopt.getWeekFormat()
d_flag.year = chkopt.getStartYear()
d_flag.month = chkopt.getStartMonth()
let now: Date = Date()
var cal_comp: DateComponents = getCalComp(date:now)
curr = (cal_comp.year!, cal_comp.month!, cal_comp.day!, cal_comp.weekOfMonth!, cal_comp.weekOfYear!)
if d_flag.year != 0 {
    cal_comp.year = d_flag.year
}
if d_flag.month > 0 {
    cal_comp.month = d_flag.month
} else if d_flag.month != 0 {
    cal_comp.month! += d_flag.month
}
cal_comp.day = 1
var sequence: [String] = []
var header: String = ""
var header_y: String = ""
var header_m: [String] = ["", "", "", "", "", "", "", "", "", "", "", ""]
var footer: String = ""
var limit = d_flag.raws * d_flag.cols
for var i in 0..<limit {
    var calen: Date = Calendar.current.date(from:cal_comp)!
    cal_comp = getCalComp(date:calen)
    var thismonth = getCalendar(components:cal_comp)
    sequence += thismonth.cal_seq
    footer += "(\(thismonth.dayorder)/\(thismonth.totaldays))\n"
    if (d_flag.cols == 3 && i == 1) || (d_flag.cols == 1 && i == 0) {
        header_y += "\(cal_comp.year!)"
    }
    if d_flag.je == 1 {
        header_m[i] = "\(cal_comp.month!)月"
    } else {
        header_m[i] = "\(month_odr[cal_comp.month! - 1])"
    }
    cal_comp.month!+=1
}
var length: Int
if d_flag.cols == 3 {
    length = (65 - header_y.characters.count) / 2
    for var i in 0..<length+1 {
        header_y = " " + header_y
    }
    print("\(header_y)\n")
}
var hol: [Int] = [0, 6]
for var l in 0..<d_flag.raws {
    if d_flag.cols == 3 {
        for var i in 0..<d_flag.cols {
            length = (21 - header_m[i + 3 * l].characters.count - d_flag.je) / 2
            for var j in 0..<length {
                header_m[i + 3 * l] = " " + header_m[i + 3 * l]
            }
            length = 22 - header_m[i + 3 * l].characters.count - d_flag.je
            for var j in 0..<length {
                header_m[i + 3 * l] = header_m[i + 3 * l] + " "
            }
            header += header_m[i + 3 * l]
        }
    } else {
        header = "\(header_m[0]) \(header_y)"
        length = (21 - header.characters.count - d_flag.je) / 2
        for var j in 0..<length {
            header = " " + header
        }
    }
    print(header)
    header = ""
    for var i in 0..<7 {
        if !(d_flag.week) || ((d_flag.week) && ((i == 0) || (i == curr.week_m))) {
            for var j in 0..<d_flag.cols {
                for var k in 0..<7 {
                    for l in hol{
                        if l == k {
                            print(esc.fg_mg, terminator:"")
                        }
                    }
                    print("\(sequence[i * 7 + ((j + 3 * l) * 49 + k)])\(esc.fg_bk) ", terminator:"")
                }
                print(" ", terminator:"")
            }
            print("\n", terminator:"")
        }
    }
    if (d_flag.week) {
        print("US Week \(curr.week_y) \(footer)")
    }
}

月表示と週表示を比較すると、

  • 年月ヘッダはどちらも同じ表示
  • 曜日ヘッダも同じ
  • 日付部分については月表示ではすべて表示するが、週表示では今週のみ

となるので、129行目を差し込みました。

もともとカレンダープログラムの月表示は曜日ヘッダを含む日付部分6週分を7個づつ7行表示するという動作です。したがって週表示の時は最初の行である曜日ヘッダと今週部分のみ表示するように判定させればOK。使いどころのわからなかったweekOfMonthが役に立ちました。

そして日付表示後に週番号も表示しますので、17行目のようにDateComponentsにweekOfYearとweekOfMonthを追加しました。

週表示カレンダー

で、オマケなんですが週表示時のみ年初からの日数と今年の総日数を表示するため、42から49行目のように本日の日付と各月の日数から必要な情報を算出して145行目でこれを表示しています。

この変更によりgetCalendar()関数の戻り値が増えたので(しかも型がバラバラ)配列を含むタプルで値を返すように変更しました。

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です