OSDN Git Service

Doutakuを導入
[kcd/KCD.git] / KCD / CalculateDamageCommand.swift
index 2ab24d6..c72106c 100644 (file)
@@ -9,7 +9,8 @@
 import Cocoa
 import SwiftyJSON
 
-fileprivate enum BattleType {
+enum BattleType {
+    
     case normal
     case combinedAir
     case combinedWater
@@ -17,141 +18,165 @@ fileprivate enum BattleType {
     case eachCombinedWater
     case enemyCombined
 }
-fileprivate enum DamageControlID: Int {
+
+enum DamageControlID: Int {
+    
     case damageControl = 42
     case goddes = 43
 }
-fileprivate enum BattleFleet {
-    case first
-    case second
-    case each
-}
 
-class CalculateDamageCommand: JSONCommand {
-    private let store = TemporaryDataStore.oneTimeEditor()
-    
-    fileprivate var battleType: BattleType = .normal
-    fileprivate var damages: [Damage] {
-        let array = store.sortedDamagesById()
-        if array.count != 12 {
-            buildDamagedEntity()
-            let newDamages = store.sortedDamagesById()
-            guard newDamages.count == 12
-                else {
-                    print("ERROR!!!! CAN NOT CREATE DAMAGE OBJECT")
-                    return []
-            }
-            return newDamages
-        }
-        return array
-    }
-    fileprivate var isCombinedBattle: Bool {
-        switch battleType {
-        case .combinedAir, .combinedWater, .eachCombinedAir, .eachCombinedWater:
-            return true
-        default:
-            return false
-        }
-    }
+final class CalculateDamageCommand: JSONCommand {
     
     override func execute() {
-        guard let battleApi = BattleAPI(rawValue: api)
-            else { return }
         
-        switch battleApi {
+        switch api.endpoint {
+            
         case .battle, .airBattle, .ldAirBattle:
-            calculateBattle()
+            normalBattle(battleType: .normal)
+            
         case .combinedEcBattle:
-            battleType = .enemyCombined
-            calcEnemyCombinedBattle()
+            enemyCombinedBattle(battleType: .enemyCombined)
+            
         case .combinedBattle, .combinedAirBattle:
-            battleType = .combinedAir
-            calcCombinedBattleAir()
+            combinedAirBattle(battleType: .combinedAir)
+            
         case .combinedBattleWater, .combinedLdAirBattle:
-            battleType = .combinedWater
-            calculateBattle()
+            normalBattle(battleType: .combinedWater)
+            
         case .combinedEachBattle:
-            battleType = .eachCombinedAir
-            calcEachBattleAir()
+            eachAirBattle(battleType: .eachCombinedAir)
+            
         case .combinedEachBattleWater:
-            battleType = .eachCombinedWater
-            calculateBattle()
+            normalBattle(battleType: .eachCombinedWater)
+            
+        case .combinedEachNightToDay:
+            eachNightToDayBattle(battleType: .normal)
+            
         case .midnightBattle, .midnightSpMidnight:
-            calculateMidnightBattle()
+            midnightBattle(battleType: .normal)
+            
         case .combinedMidnightBattle, .combinedSpMidnight:
-            battleType = .combinedAir
-            calculateMidnightBattle()
+            midnightBattle(battleType: .combinedAir)
+            
         case .combinedEcMidnightBattle:
-            battleType = .eachCombinedAir
-            calculateMidnightBattle()
+            midnightBattle(battleType: .eachCombinedAir)
+            
         case .battleResult, .combinedBattleResult:
             applyDamage()
             resetDamage()
+            
+        default: return Logger.shared.log("Missing API: \(apiResponse.api)")
         }
     }
     
-    private func resetDamage() {
-        store.damages().forEach { store.delete($0) }
+    func normalBattle(battleType: BattleType) {
+        
+        updateBattleCell()
+        DamageCalculator(json, battleType).calculateBattle()
     }
-    private func applyDamage() {
-        let totalDamages = store.sortedDamagesById()
-        guard totalDamages.count == 12
-            else { return print("Damages count is invalid. count is \(totalDamages.count).") }
-        let aStore = ServerDataStore.oneTimeEditor()
-        totalDamages.forEach {
-            guard let ship = aStore.ship(by: $0.shipID)
-                else { return }
-            
-            if ship.nowhp != $0.hp {
-                Debug.print("\(ship.name)(\(ship.id)),HP \(ship.nowhp) -> \($0.hp)", level: .debug)
-            }
-            
-            ship.nowhp = $0.hp
-            if $0.useDamageControl { removeFirstDamageControl(of: ship) }
-        }
+    
+    func combinedAirBattle(battleType: BattleType) {
+        
+        updateBattleCell()
+        DamageCalculator(json, battleType).calcCombinedBattleAir()
+    }
+    
+    func eachAirBattle(battleType: BattleType) {
+        
+        updateBattleCell()
+        DamageCalculator(json, battleType).calcEachBattleAir()
+    }
+    
+    func eachNightToDayBattle(battleType: BattleType) {
         
-        Debug.print("End Battle ------- ", level: .debug)
+        updateBattleCell()
+        DamageCalculator(json, battleType).calcEachNightToDay()
     }
-    private func buildDamagedEntity() {
-        guard let battle = store.battle()
-            else { return print("Battle is invalid.") }
+    
+    func enemyCombinedBattle(battleType: BattleType) {
+        
+        updateBattleCell()
+        DamageCalculator(json, battleType).calcEnemyCombinedBattle()
+    }
+    
+    func midnightBattle(battleType: BattleType) {
+        
+        DamageCalculator(json, battleType).calcMidnight()
+    }
+}
+
+extension CalculateDamageCommand {
+    
+    func resetDamage() {
+        
+        let store = TemporaryDataStore.oneTimeEditor()
+        
+        store.sync { store.damages().forEach(store.delete) }
+    }
+    
+    func applyDamage() {
+        
+        let store = TemporaryDataStore.oneTimeEditor()
+        
+        let totalDamages = store.sync { store.sortedDamagesById() }
         
         let aStore = ServerDataStore.default
-        var ships: [Any] = []
         
-        // 第一艦隊
-        let firstFleetShips = aStore.ships(byDeckId: battle.deckId)
-        ships += (firstFleetShips as [Any])
-        while ships.count != 6 {
-            ships.append(0)
+        Debug.excute(level: .debug) {
+            
+            print("-------")
+            
+            store.sync {
+                totalDamages.forEach { damage in
+                    
+                    let shipId = damage.shipID
+                    guard let ship = aStore.sync(execute: { aStore.ship(by: shipId) }) else { return }
+                    
+                    let damagedHp = damage.hp
+                    aStore.sync {
+                        if ship.nowhp != damagedHp {
+                            
+                            print("\(ship.name)(\(ship.id)),HP \(ship.nowhp) -> \(damagedHp)")
+                        }
+                    }
+                }
+            }
+            
+            print("------- End Battle")
         }
         
-        // 第二艦隊
-        let secondFleetShips = aStore.ships(byDeckId: 2)
-        ships += (secondFleetShips as [Any])
-        while ships.count != 12 {
-            ships.append(0)
-        }
-        ships.enumerated().forEach {
-            guard let damage = store.createDamage()
-                else { return print("Can not create Damage") }
-            damage.battle = battle
-            damage.id = $0.offset
-            if let ship = $0.element as? Ship {
-                damage.hp = ship.nowhp
-                damage.shipID = ship.id
+        // 第二艦隊単独出撃で正しくデータが反映されるように逆順にして計算
+        store.sync {
+            totalDamages.reversed().forEach { damage in
+                
+                let shipId = damage.shipID
+                guard let ship = aStore.sync(execute: { aStore.ship(by: shipId) }) else { return }
+                
+                let damagedHp = damage.hp
+                aStore.sync { ship.nowhp = damagedHp }
+                
+                if damage.useDamageControl { self.removeFirstDamageControl(of: shipId) }
             }
         }
     }
     
-    fileprivate func updateBattleCell() {
-        guard let battle = store.battle()
-            else { return print("Battle is invalid.") }
-        battle.battleCell = (battle.no == 0 ? nil : battle.no as NSNumber)
+    func updateBattleCell() {
+        
+        let store = TemporaryDataStore.default
+        
+        guard let battle = store.sync(execute: { store.battle() }) else {
+            
+            return Logger.shared.log("Battle is invalid.")
+        }
+        
+        store.sync { battle.battleCell = (battle.no == 0 ? nil : battle.no as NSNumber) }
         
         Debug.excute(level: .debug) {
+            
             print("Enter Cell ------- ")
+            
             if let seiku = json["api_data"]["api_kouku"]["api_stage1"]["api_disp_seiku"].int {
+                
                 switch seiku {
                 case 0: print("制空権 均衡")
                 case 1: print("制空権 確保")
@@ -161,7 +186,9 @@ class CalculateDamageCommand: JSONCommand {
                 default: break
                 }
             }
+            
             if let intercept = json["api_data"]["api_formation"][2].int {
+                
                 switch intercept {
                 case 1: print("交戦形態 同航戦")
                 case 2: print("交戦形態 反航戦")
@@ -172,336 +199,53 @@ class CalculateDamageCommand: JSONCommand {
             }
         }
     }
-}
-// MARK: - Primitive Calclator
-extension CalculateDamageCommand {
-    private func hogekiTargets(_ list: JSON) -> [[Int]]? {
-        guard let targetArraysArray = list
-            .array?
-            .flatMap({ $0.array?.flatMap { $0.int } })
-            else { return nil }
-        guard list.count - 1 == targetArraysArray.count
-            else {
-                print("api_df_list is wrong")
-                return nil
-        }
-        return targetArraysArray
-    }
-    private func hogekiDamages(_ list: JSON) -> [[Int]]? {
-        return list.array?.flatMap { $0.array?.flatMap { $0.int } }
-    }
-    private func enemyFlags(_ list: JSON) -> [Int]? {
-        return list.array?.flatMap { $0.int }.filter { $0 != -1 }
-    }
-    private func isTargetFriend(eFlags: [Int]?, index: Int) -> Bool {
-        if let eFlags = eFlags, 0..<eFlags.count ~= index {
-            return eFlags[index] == 1
-        }
-        return true
-    }
-    private func validTargetPos(_ targetPos: Int, in battleFleet: BattleFleet) -> Bool {
-        let upper = (battleFleet == .each ? 12 : 6)
-        return 1...upper ~= targetPos
-    }
-    private func position(_ pos: Int, in fleet: BattleFleet) -> Int? {
-        let shipOffset = (fleet == .second) ? 6 : 0
-        let damagePos = pos - 1 + shipOffset
-        guard 0..<damages.count ~= damagePos
-            else { return nil }
-        return damagePos
-    }
-    private func calcHP(damage: Damage, receive: Int) {
-        let hp = damage.hp as Int
-        var newHP = hp - receive
-        if newHP <= 0 {
-            let shipId = damage.shipID
-            if let ship = ServerDataStore.default.ship(by: shipId) {
-                let efectiveHP = damageControlIfPossible(nowhp: newHP, ship: ship)
-                if efectiveHP != 0, efectiveHP != newHP {
-                    damage.useDamageControl = true
-                }
-                newHP = efectiveHP
-            }
-        }
-        damage.hp = newHP
-    }
-    fileprivate func calculateHogeki(baseKeyPath: String, _ bf: () -> BattleFleet) {
-        calculateHogeki(baseKeyPath: baseKeyPath, battleFleet: bf())
-    }
-    fileprivate func calculateHogeki(baseKeyPath: String, battleFleet: BattleFleet = .first) {
-        let baseValue = json[baseKeyPath.components(separatedBy: ".")]
-        guard let targetPosLists = hogekiTargets(baseValue["api_df_list"]),
-            let damageLists = hogekiDamages(baseValue["api_damage"])
-            else { return }
-        guard targetPosLists.count == damageLists.count
-            else { return print("api_damage is wrong.") }
-
-        let eFlags = enemyFlags(baseValue["api_at_eflag"])
+    
+    func removeFirstDamageControl(of shipId: Int) {
         
-        Debug.print("Start Hougeki \(baseKeyPath)", level: .debug)
-        zip(targetPosLists, damageLists).enumerated().forEach { (i, list) in
-            if !isTargetFriend(eFlags: eFlags, index: i) { return }
+        let store = ServerDataStore.oneTimeEditor()
+        store.sync {
             
-            zip(list.0, list.1).forEach { (targetPos, damage) in
-                guard validTargetPos(targetPos, in: battleFleet) else { return }
-                
-                guard let damagePos = position(targetPos, in: battleFleet)
-                    else { return print("damage pos is larger than damage count") }
-                calcHP(damage: damages[damagePos], receive: damage)
+            guard let ship = store.ship(by: shipId) else { return }
+            
+            let (item, damageControl) = ship
+                .equippedItem
+                .lazy
+                .compactMap { $0 as? SlotItem }
+                .map { ($0, store.masterSlotItemID(by: $0.id)) }
+                .map { ($0.0, DamageControlID(rawValue: $0.1)) }
+                .filter { $0.1 != nil }
+                .first ?? (nil, nil)
+            
+            if let validDamageControl = damageControl {
                 
-                Debug.excute(level: .debug) {
-                    let shipOffset = (battleFleet == .second) ? 6 : 0
-                    print("Hougeki \(targetPos + shipOffset) -> \(damage)")
+                switch validDamageControl {
+                case .damageControl: break
+                    
+                case .goddes:
+                    ship.fuel = ship.maxFuel
+                    ship.bull = ship.maxBull
                 }
+                
+                guard let equiped = ship.equippedItem.array as? [SlotItem] else { return }
+                
+                ship.equippedItem = NSOrderedSet(array: equiped.filter { $0 != item })
+                
+                return
             }
-        }
-    }
-    fileprivate func calculateFDam(baseKeyPath: String, _ bf: () -> BattleFleet) {
-        calculateFDam(baseKeyPath: baseKeyPath, battleFleet: bf())
-    }
-    fileprivate func calculateFDam(baseKeyPath: String, battleFleet: BattleFleet = .first) {
-        let baseValue = json[baseKeyPath.components(separatedBy: ".")]
-        guard let koukuDamages = baseValue["api_fdam"].arrayObject as? [Int]
-            else { return }
-        
-        Debug.print("Start FDam \(baseKeyPath)", level: .debug)
-        
-        koukuDamages.enumerated().forEach { (idx, damage) in
-            if idx == 0 { return }
             
-            guard let damagePos = position(idx, in: battleFleet)
-                else { return }
-            calcHP(damage: damages[damagePos], receive: damage)
+            // check extra slot
+            let exItemId = store.sync { store.masterSlotItemID(by: ship.slot_ex) }
             
-            Debug.excute(level: .debug) {
-                let shipOffset = (battleFleet == .second) ? 6 : 0
-                print("FDam \(idx + shipOffset) -> \(damage)")
-            }
-        }
-    }
-}
-// MARK: - Battle phase
-extension CalculateDamageCommand {
-    fileprivate func calcKouku() {
-        calculateFDam(baseKeyPath: "api_data.api_kouku.api_stage3")
-        calculateFDam(baseKeyPath: "api_data.api_kouku2.api_stage3")
-        
-        // 艦隊 戦闘艦隊
-        // 連合vs通常(水上) 第2
-        // 連合vs通常(機動) 第2
-        // 連合vs連合(水上) 第2 全体 use kouku nor kouku2
-        // 連合vs連合(機動) 第1 全体 use kouku nor kouku2
-        let bf: () -> BattleFleet = {
-            switch self.battleType {
-            case .combinedWater, .combinedAir,
-                 .eachCombinedWater, .eachCombinedAir:
-                return .second
-            default: return .first
-            }
-        }
-        calculateFDam(baseKeyPath: "api_data.api_kouku.api_stage3_combined", bf)
-        calculateFDam(baseKeyPath: "api_data.api_kouku2.api_stage3_combined", bf)
-        
-    }
-    fileprivate func calcOpeningAttack() {
-        // 艦隊 戦闘艦隊
-        // 連合vs通常(水上) 第2
-        // 連合vs通常(機動) 第2
-        // 連合vs連合(水上) 第2 全体
-        // 連合vs連合(機動) 第2 全体
-        calculateFDam(baseKeyPath: "api_data.api_opening_atack") {
-            switch battleType {
-            case .combinedWater, .combinedAir: return .second
-            case .eachCombinedWater, .eachCombinedAir: return .each
-            default: return .first
-            }
-        }
-    }
-    fileprivate func calcOpeningTaisen() {
-        calculateHogeki(baseKeyPath: "api_data.api_opening_taisen") {
-            isCombinedBattle ? .second : .first
-        }
-    }
-    fileprivate func calcHougeki1() {
-        // 艦隊 戦闘艦隊
-        // 連合vs通常(水上) 第1
-        // 連合vs通常(機動) 第2
-        // 連合vs連合(水上) 第1
-        // 連合vs連合(機動) 第1
-        calculateHogeki(baseKeyPath: "api_data.api_hougeki1") {
-            switch battleType {
-            case .combinedAir: return .second
-            default: return .first
-            }
-        }
-    }
-    fileprivate func calcHougeki2() {
-        // 艦隊 戦闘艦隊
-        // 連合vs通常(水上) 第1
-        // 連合vs通常(機動) 第1
-        // 連合vs連合(水上) 第1 全体
-        // 連合vs連合(機動) 第2
-        calculateHogeki(baseKeyPath: "api_data.api_hougeki2") {
-            switch battleType {
-            case .eachCombinedWater: return .each
-//            case .eachCombinedAir: return .second  // 1~12
-            default: return .first
-            }
-        }
-    }
-    fileprivate func calcHougeki3() {
-        // 艦隊 戦闘艦隊
-        // 連合vs通常(水上) 第2
-        // 連合vs通常(機動) 第1
-        // 連合vs連合(水上) 第2
-        // 連合vs連合(機動) 第1 全体
-        calculateHogeki(baseKeyPath: "api_data.api_hougeki3") {
-            switch battleType {
-            case .combinedWater: return .second
-//            case .eachCombinedWater: return .second  // 1~12
-            case .eachCombinedAir: return .each
-            default: return .first
-            }
-        }
-    }
-    fileprivate func calcRaigeki() {
-        // 艦隊 戦闘艦隊
-        // 連合vs通常(水上) 第2
-        // 連合vs通常(機動) 第2
-        // 連合vs連合(水上) 第2 全体
-        // 連合vs連合(機動) 第2 全体
-        calculateFDam(baseKeyPath: "api_data.api_raigeki") {
-            switch battleType {
-            case .combinedWater, .combinedAir: return .second
-            case .eachCombinedWater, .eachCombinedAir: return .each
-            default: return .first
-            }
-        }
-    }
-    
-    fileprivate func calculateMidnightBattle() {
-        // 艦隊 戦闘艦隊
-        // 連合vs通常(水上) 第2
-        // 連合vs通常(機動) 第2
-        // 連合vs連合(水上) 第2
-        // 連合vs連合(機動) 第2
-        calculateHogeki(baseKeyPath: "api_data.api_hougeki") {
-            isCombinedBattle ? .second : .first
-        }
-    }
-}
-// MARK: - Battle type
-extension CalculateDamageCommand {
-    fileprivate func calculateBattle() {
-        updateBattleCell()
-        
-        calcKouku()
-        calcOpeningTaisen()
-        calcOpeningAttack()
-        calcHougeki1()
-        calcHougeki2()
-        calcHougeki3()
-        calcRaigeki()
-    }
-    fileprivate func calcCombinedBattleAir() {
-        updateBattleCell()
-        
-        calcKouku()
-        calcOpeningTaisen()
-        calcOpeningAttack()
-        calcHougeki1()
-        calcRaigeki()
-        calcHougeki2()
-        calcHougeki3()
-    }
-    fileprivate func calcEachBattleAir() {
-        updateBattleCell()
-        
-        calcKouku()
-        calcOpeningTaisen()
-        calcOpeningAttack()
-        calcHougeki1()
-        calcHougeki2()
-        calcRaigeki()
-        calcHougeki3()
-    }
-    fileprivate func calcEnemyCombinedBattle() {
-        // same phase as combined air
-        calcCombinedBattleAir()
-    }
-}
-// MARK: - Damage control
-extension CalculateDamageCommand {
-    fileprivate func damageControlIfPossible(nowhp: Int, ship: Ship) -> Int {
-        var nowHp = nowhp
-        if nowHp < 0 { nowHp = 0 }
-        let maxhp = ship.maxhp
-        let store = ServerDataStore.default
-        var useDamageControl = false
-        ship.equippedItem.forEach {
-            if useDamageControl { return }
-            guard let master = $0 as? SlotItem else { return }
-            let masterSlotItemId = store.masterSlotItemID(by: master.id)
-            guard let type = DamageControlID(rawValue: masterSlotItemId)
-                else { return }
-            switch type {
-            case .damageControl:
-                nowHp = Int(Double(maxhp) * 0.2)
-                useDamageControl = true
-            case .goddes:
-                nowHp = maxhp
-                useDamageControl = true
-            }
-        }
-        if useDamageControl { return nowHp }
-        // check extra slot
-        let exItemId = store.masterSlotItemID(by: ship.slot_ex)
-        guard let exType = DamageControlID(rawValue: exItemId)
-            else { return nowHp }
-        switch exType {
-        case .damageControl:
-            nowHp = Int(Double(maxhp) * 0.2)
-        case .goddes:
-            nowHp = maxhp
-        }
-        return nowHp
-    }
-    fileprivate func removeFirstDamageControl(of ship: Ship) {
-        let equiped = ship.equippedItem
-        let newEquiped = equiped.array
-        let store = ServerDataStore.default
-        var useDamageControl = false
-        equiped.forEach {
-            if useDamageControl { return }
-            guard let master = $0 as? SlotItem else { return }
-            let masterSlotItemId = store.masterSlotItemID(by: master.id)
-            guard let type = DamageControlID(rawValue: masterSlotItemId)
-                else { return }
-            switch type {
+            guard let exType = DamageControlID(rawValue: exItemId) else { return }
+            
+            switch exType {
+            case .damageControl: break
+                
             case .goddes:
                 ship.fuel = ship.maxFuel
                 ship.bull = ship.maxBull
-                fallthrough
-            case .damageControl:
-                if var equiped = newEquiped as? [SlotItem],
-                    let index = equiped.index(of: master) {
-                    equiped[index...index] = []
-                    ship.equippedItem = NSOrderedSet(array: equiped)
-                    useDamageControl = true
-                }
             }
-        }
-        if useDamageControl { return }
-        // check extra slot
-        let exItemId = store.masterSlotItemID(by: ship.slot_ex)
-        guard let exType = DamageControlID(rawValue: exItemId)
-            else { return }
-        switch exType {
-        case .goddes:
-            ship.fuel = ship.maxFuel
-            ship.bull = ship.maxBull
-            fallthrough
-        case .damageControl:
+            
             ship.slot_ex = -1
         }
     }