// Normalize app data func normalizeApp(a *bspb.BatteryStats_App, totalTimeHour float64) *bspb.BatteryStats_App { if a == nil { return nil } res := proto.Clone(a).(*bspb.BatteryStats_App) if norm := normalizeMessage(a.GetForeground(), totalTimeHour); norm != nil { res.Foreground = norm.(*bspb.BatteryStats_App_Foreground) } if norm := normalizeAppApk(a.GetApk(), totalTimeHour); norm != nil { res.Apk = norm } normalizeAppChildren(res.GetChild(), totalTimeHour) if norm := normalizeMessage(a.GetNetwork(), totalTimeHour); norm != nil { res.Network = norm.(*bspb.BatteryStats_App_Network) } if norm := normalizeMessage(a.GetPowerUseItem(), totalTimeHour); norm != nil { res.PowerUseItem = norm.(*bspb.BatteryStats_App_PowerUseItem) } if norm := normalizeRepeatedMessage(a.GetProcess(), totalTimeHour).Interface(); norm != nil { res.Process = norm.([]*bspb.BatteryStats_App_Process) } if norm := normalizeRepeatedMessage(a.GetSensor(), totalTimeHour).Interface(); norm != nil { res.Sensor = norm.([]*bspb.BatteryStats_App_Sensor) } if norm := normalizeMessage(a.GetStateTime(), totalTimeHour); norm != nil { res.StateTime = norm.(*bspb.BatteryStats_App_StateTime) } if norm := normalizeMessage(a.GetVibrator(), totalTimeHour); norm != nil { res.Vibrator = norm.(*bspb.BatteryStats_App_Vibrator) } if norm := normalizeRepeatedMessage(a.GetWakelock(), totalTimeHour).Interface(); norm != nil { res.Wakelock = norm.([]*bspb.BatteryStats_App_Wakelock) } if norm := normalizeRepeatedMessage(a.GetWakeupAlarm(), totalTimeHour).Interface(); norm != nil { res.WakeupAlarm = norm.([]*bspb.BatteryStats_App_WakeupAlarm) } if norm := normalizeMessage(a.GetWifi(), totalTimeHour); norm != nil { res.Wifi = norm.(*bspb.BatteryStats_App_Wifi) } if norm := normalizeRepeatedMessage(a.GetUserActivity(), totalTimeHour).Interface(); norm != nil { res.UserActivity = norm.([]*bspb.BatteryStats_App_UserActivity) } if norm := normalizeRepeatedMessage(a.GetScheduledJob(), totalTimeHour).Interface(); norm != nil { res.ScheduledJob = norm.([]*bspb.BatteryStats_App_ScheduledJob) } return res }
func subtractApp(a1, a2 *bspb.BatteryStats_App) *bspb.BatteryStats_App { // The app name may be empty for one of the inputs, // if that file does not contain any occurence for this particular app // Here we extract the valid name from the 2 protos appName := a1.GetName() if appName == "" { appName = a2.GetName() } a := &bspb.BatteryStats_App{ Name: proto.String(appName), Uid: proto.Int32(a1.GetUid()), VersionName: combineProtoStrings(a1.GetVersionName(), a2.GetVersionName(), connector), VersionCode: a1.VersionCode, } c2map := make(map[string]*bspb.BatteryStats_App_Child) for _, c := range a2.GetChild() { c2map[c.GetName()] = c } changed := false if diff := subtractMessage(a1.GetCpu(), a2.GetCpu()); diff != nil { a.Cpu = diff.(*bspb.BatteryStats_App_Cpu) changed = true } if diff := subtractMessage(a1.GetForeground(), a2.GetForeground()); diff != nil { a.Foreground = diff.(*bspb.BatteryStats_App_Foreground) changed = true } if diff := subtractAppApk(a1.GetApk(), a2.GetApk()); diff != nil { a.Apk = diff changed = true } if diff := subtractController(a1.GetBluetoothController(), a2.GetBluetoothController()); diff != nil { a.BluetoothController = diff changed = true } if diff := subtractController(a1.GetModemController(), a2.GetModemController()); diff != nil { a.ModemController = diff changed = true } if diff := subtractMessage(a1.GetNetwork(), a2.GetNetwork()); diff != nil { a.Network = diff.(*bspb.BatteryStats_App_Network) changed = true } if diff := subtractMessage(a1.GetPowerUseItem(), a2.GetPowerUseItem()); diff != nil { a.PowerUseItem = diff.(*bspb.BatteryStats_App_PowerUseItem) changed = true } if diff := subtractRepeatedMessage(a1.GetProcess(), a2.GetProcess()); !diff.IsNil() { a.Process = diff.Interface().([]*bspb.BatteryStats_App_Process) changed = true } if diff := subtractRepeatedMessage(a1.GetSensor(), a2.GetSensor()); !diff.IsNil() { a.Sensor = diff.Interface().([]*bspb.BatteryStats_App_Sensor) changed = true } if diff := subtractMessage(a1.GetStateTime(), a2.GetStateTime()); diff != nil { a.StateTime = diff.(*bspb.BatteryStats_App_StateTime) changed = true } if diff := subtractMessage(a1.GetVibrator(), a2.GetVibrator()); diff != nil { a.Vibrator = diff.(*bspb.BatteryStats_App_Vibrator) changed = true } if diff := subtractRepeatedMessage(a1.GetWakelock(), a2.GetWakelock()); !diff.IsNil() { a.Wakelock = diff.Interface().([]*bspb.BatteryStats_App_Wakelock) changed = true } if diff := subtractMessage(a1.GetWifi(), a2.GetWifi()); diff != nil { a.Wifi = diff.(*bspb.BatteryStats_App_Wifi) changed = true } if diff := subtractController(a1.GetWifiController(), a2.GetWifiController()); diff != nil { a.WifiController = diff changed = true } if diff := subtractRepeatedMessage(a1.GetSync(), a2.GetSync()); !diff.IsNil() { a.Sync = diff.Interface().([]*bspb.BatteryStats_App_Sync) changed = true } if diff := subtractRepeatedMessage(a1.GetUserActivity(), a2.GetUserActivity()); !diff.IsNil() { a.UserActivity = diff.Interface().([]*bspb.BatteryStats_App_UserActivity) changed = true } if diff := subtractRepeatedMessage(a1.GetScheduledJob(), a2.GetScheduledJob()); !diff.IsNil() { a.ScheduledJob = diff.Interface().([]*bspb.BatteryStats_App_ScheduledJob) changed = true } for _, c := range a1.GetChild() { if c2, ok := c2map[c.GetName()]; ok { if diff := subtractAppChild(c, c2); diff != nil { a.Child = append(a.Child, diff) changed = true } delete(c2map, c.GetName()) } else { // Child doesn't exist in a2 a.Child = append(a.Child, c) changed = true } } for _, c2 := range c2map { // These children aren't in a1 a.Child = append(a.Child, c2) changed = true } // Subtract the head child h1 := a1.GetHeadChild() h2 := a2.GetHeadChild() if h1 != nil || h2 != nil { if h1 == nil { a.HeadChild = h2 changed = true } else if h2 == nil { a.HeadChild = h1 changed = true } else { if diff := subtractAppChild(a1.GetHeadChild(), a2.GetHeadChild()); diff != nil { a.HeadChild = diff changed = true } } } if !changed { return nil } return a }