1/*
2 *  Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
3 *
4 *  Use of this source code is governed by a BSD-style license
5 *  that can be found in the LICENSE file in the root of the source
6 *  tree. An additional intellectual property rights grant can be found
7 *  in the file PATENTS.  All contributing project authors may
8 *  be found in the AUTHORS file in the root of the source tree.
9 */
10
11#include "webrtc/modules/audio_processing/typing_detection.h"
12
13namespace webrtc {
14
15TypingDetection::TypingDetection()
16    : time_active_(0),
17      time_since_last_typing_(0),
18      penalty_counter_(0),
19      counter_since_last_detection_update_(0),
20      detection_to_report_(false),
21      new_detection_to_report_(false),
22      time_window_(10),
23      cost_per_typing_(100),
24      reporting_threshold_(300),
25      penalty_decay_(1),
26      type_event_delay_(2),
27      report_detection_update_period_(1) {
28}
29
30TypingDetection::~TypingDetection() {}
31
32bool TypingDetection::Process(bool key_pressed, bool vad_activity) {
33  if (vad_activity)
34    time_active_++;
35  else
36    time_active_ = 0;
37
38  // Keep track if time since last typing event
39  if (key_pressed)
40    time_since_last_typing_ = 0;
41  else
42    ++time_since_last_typing_;
43
44  if (time_since_last_typing_ < type_event_delay_ &&
45      vad_activity &&
46      time_active_ < time_window_) {
47    penalty_counter_ += cost_per_typing_;
48    if (penalty_counter_ > reporting_threshold_)
49      new_detection_to_report_ = true;
50  }
51
52  if (penalty_counter_ > 0)
53    penalty_counter_ -= penalty_decay_;
54
55  if (++counter_since_last_detection_update_ ==
56      report_detection_update_period_) {
57    detection_to_report_ = new_detection_to_report_;
58    new_detection_to_report_ = false;
59    counter_since_last_detection_update_ = 0;
60  }
61
62  return detection_to_report_;
63}
64
65int TypingDetection::TimeSinceLastDetectionInSeconds() {
66  // Round to whole seconds.
67  return (time_since_last_typing_ + 50) / 100;
68}
69
70void TypingDetection::SetParameters(int time_window,
71                                    int cost_per_typing,
72                                    int reporting_threshold,
73                                    int penalty_decay,
74                                    int type_event_delay,
75                                    int report_detection_update_period) {
76  if (time_window) time_window_ = time_window;
77
78  if (cost_per_typing) cost_per_typing_ = cost_per_typing;
79
80  if (reporting_threshold) reporting_threshold_ = reporting_threshold;
81
82  if (penalty_decay) penalty_decay_ = penalty_decay;
83
84  if (type_event_delay) type_event_delay_ = type_event_delay;
85
86  if (report_detection_update_period)
87    report_detection_update_period_ = report_detection_update_period;
88}
89
90}  // namespace webrtc
91