File: trigger_policies.cc

package info (click to toggle)
chromium 138.0.7204.183-1~deb12u1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm-proposed-updates
  • size: 6,080,960 kB
  • sloc: cpp: 34,937,079; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,954; asm: 946,768; xml: 739,971; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,811; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (145 lines) | stat: -rw-r--r-- 5,059 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/tabs/organization/trigger_policies.h"

#include <cmath>
#include <numbers>

#include "base/metrics/histogram_functions.h"
#include "base/time/time.h"
#include "chrome/browser/metrics/desktop_session_duration/desktop_session_duration_tracker.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/tabs/organization/prefs.h"
#include "components/prefs/pref_service.h"

UsageTickClock::UsageTickClock(const base::TickClock* base_clock)
    : base_clock_(base_clock), start_time_(base_clock_->NowTicks()) {
  if (metrics::DesktopSessionDurationTracker::IsInitialized()) {
    auto* const tracker = metrics::DesktopSessionDurationTracker::Get();
    tracker->AddObserver(this);
    if (tracker->in_session()) {
      current_usage_session_start_time_ = start_time_;
    }
  }
}

UsageTickClock::~UsageTickClock() {
  if (metrics::DesktopSessionDurationTracker::IsInitialized()) {
    metrics::DesktopSessionDurationTracker::Get()->RemoveObserver(this);
  }
}

base::TimeTicks UsageTickClock::NowTicks() const {
  if (!metrics::DesktopSessionDurationTracker::IsInitialized()) {
    return base_clock_->NowTicks();
  }

  const base::TimeTicks completed_session_time =
      start_time_ + usage_time_in_completed_sessions_;
  if (current_usage_session_start_time_.has_value()) {
    return completed_session_time + (base_clock_->NowTicks() -
                                     current_usage_session_start_time_.value());
  }
  return completed_session_time;
}

void UsageTickClock::OnSessionStarted(base::TimeTicks session_start) {
  DCHECK(!current_usage_session_start_time_.has_value());

  // Ignore `session_start`; it doesn't come from `base_clock_`.
  current_usage_session_start_time_ = base_clock_->NowTicks();
}

void UsageTickClock::OnSessionEnded(base::TimeDelta session_length,
                                    base::TimeTicks session_end) {
  DCHECK(current_usage_session_start_time_.has_value());

  // Ignore `session_length`/`session_end`; they don't come from `base_clock_`.
  usage_time_in_completed_sessions_ +=
      base_clock_->NowTicks() - current_usage_session_start_time_.value();
  current_usage_session_start_time_ = std::nullopt;
}

ProfilePrefBackoffLevelProvider::ProfilePrefBackoffLevelProvider(
    content::BrowserContext* context)
    : prefs_(Profile::FromBrowserContext(context)->GetPrefs()) {}

ProfilePrefBackoffLevelProvider::~ProfilePrefBackoffLevelProvider() = default;

unsigned int ProfilePrefBackoffLevelProvider::Get() const {
  return prefs_->GetInteger(
      tab_organization_prefs::kTabOrganizationNudgeBackoffCount);
}

void ProfilePrefBackoffLevelProvider::Increment() {
  prefs_->SetInteger(tab_organization_prefs::kTabOrganizationNudgeBackoffCount,
                     Get() + 1);
}

void ProfilePrefBackoffLevelProvider::Decrement() {
  prefs_->SetInteger(tab_organization_prefs::kTabOrganizationNudgeBackoffCount,
                     std::max(1u, Get()) - 1);
}

TargetFrequencyTriggerPolicy::TargetFrequencyTriggerPolicy(
    std::unique_ptr<base::TickClock> clock,
    base::TimeDelta base_period,
    float backoff_base,
    BackoffLevelProvider* backoff_level_provider)
    : clock_(std::move(clock)),
      base_period_(base_period),
      backoff_base_(backoff_base),
      backoff_level_provider_(backoff_level_provider),
      cycle_start_time_(clock_->NowTicks()) {}

TargetFrequencyTriggerPolicy::~TargetFrequencyTriggerPolicy() = default;

bool TargetFrequencyTriggerPolicy::ShouldTrigger(float score) {
  const base::TimeTicks current_time = clock_->NowTicks();
  const base::TimeDelta period =
      base_period_ * std::pow(backoff_base_, backoff_level_provider_->Get());

  // Restart the cycle if `period_` has elapsed.
  if (current_time > cycle_start_time_ + period) {
    cycle_start_time_ += period;
    best_score = std::nullopt;
    base::UmaHistogramBoolean("Tab.Organization.Trigger.TriggeredInPeriod",
                              has_triggered_);
    has_triggered_ = false;
  }

  // Update the best score if we're in the observation phase.
  const base::TimeDelta observation_period = period / std::numbers::e_v<float>;
  if (current_time < cycle_start_time_ + observation_period) {
    best_score =
        best_score.has_value() ? std::max(best_score.value(), score) : score;
    return false;
  }

  // Trigger if we haven't triggered yet and have a new high score.
  if (!has_triggered_ && best_score.has_value() && score > best_score) {
    best_score = std::nullopt;
    has_triggered_ = true;
    return true;
  }

  return false;
}

void TargetFrequencyTriggerPolicy::OnTriggerSucceeded() {
  backoff_level_provider_->Decrement();
}

void TargetFrequencyTriggerPolicy::OnTriggerFailed() {
  backoff_level_provider_->Increment();
}

bool NeverTriggerPolicy::ShouldTrigger(float score) {
  return false;
}

bool DemoTriggerPolicy::ShouldTrigger(float score) {
  return true;
}