Add spellcheck and word suggestion to the prediction service

services/keyboard_native/view_observer_delegate.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <math.h>
 
 #include "services/keyboard_native/view_observer_delegate.h"
 
 #include "base/bind.h"
 #include "base/sys_info.h"
 #include "mojo/gpu/gl_texture.h"
 #include "mojo/public/cpp/application/application_impl.h"
 #include "mojo/public/cpp/application/connect.h"
+#include "mojo/services/prediction/public/interfaces/prediction.mojom.h"
 #include "mojo/skia/ganesh_surface.h"
 #include "services/keyboard_native/clip_animation.h"
 #include "services/keyboard_native/keyboard_service_impl.h"
+#include "services/keyboard_native/predictor.h"
 #include "skia/ext/refptr.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 #include "third_party/skia/include/core/SkTypeface.h"
 #include "ui/gfx/geometry/rect_f.h"
 #include "ui/gfx/shadow_value.h"
 #include "ui/gfx/skia_util.h"
 #include "ui/gfx/vector2d.h"
 
 namespace keyboard {
 
 static const base::TimeDelta kAnimationDurationMs =
     base::TimeDelta::FromMilliseconds(1000);
 
 mojo::Size ToSize(const mojo::Rect& rect) {
   mojo::Size size;
   size.width = rect.width;
   size.height = rect.height;
   return size;
 }
 
 ViewObserverDelegate::ViewObserverDelegate()
     : keyboard_service_impl_(nullptr),
+      predictor_(nullptr),
       view_(nullptr),
       id_namespace_(0u),
       surface_id_(1u),
       key_layout_(),
       needs_draw_(false),
       frame_pending_(false),
       floating_key_height_(0.0f),
       floating_key_width_(0.0f),
       weak_factory_(this) {
   key_layout_.SetTextCallback(
       base::Bind(&ViewObserverDelegate::OnText, weak_factory_.GetWeakPtr()));
   key_layout_.SetDeleteCallback(
       base::Bind(&ViewObserverDelegate::OnDelete, weak_factory_.GetWeakPtr()));
+  key_layout_.SetSuggestTextCallback(base::Bind(
+      &ViewObserverDelegate::OnSuggestText, weak_factory_.GetWeakPtr()));
   submit_frame_callback_ = base::Bind(&ViewObserverDelegate::OnFrameComplete,
                                       weak_factory_.GetWeakPtr());
 }
 
 ViewObserverDelegate::~ViewObserverDelegate() {
 }
 
 void ViewObserverDelegate::SetKeyboardServiceImpl(
     KeyboardServiceImpl* keyboard_service_impl) {
   keyboard_service_impl_ = keyboard_service_impl;
@@ skipping 23 matching lines @@
   gr_context_.reset(new mojo::GaneshContext(gl_context_));
   mojo::ServiceProviderPtr surfaces_service_provider;
   shell->ConnectToApplication("mojo:surfaces_service",
                               mojo::GetProxy(&surfaces_service_provider),
                               nullptr);
   mojo::ConnectToService(surfaces_service_provider.get(), &surface_);
   surface_->SetResourceReturner(resource_returner.Pass());
   surface_->CreateSurface(surface_id_);
   surface_->GetIdNamespace(base::Bind(&ViewObserverDelegate::SetIdNamespace,
                                       base::Unretained(this)));
+  predictor_ = new Predictor(shell);
+  predictor_->SetUpdateCallback(base::Bind(
+      &ViewObserverDelegate::OnUpdateSuggestion, weak_factory_.GetWeakPtr()));
+  key_layout_.SetPredictor(predictor_);
   IssueDraw();
 }
 
 void ViewObserverDelegate::OnText(const std::string& text) {
   keyboard_service_impl_->OnKey(text.c_str());
 }
 
 void ViewObserverDelegate::OnDelete() {
   keyboard_service_impl_->OnDelete();
 }
 
+void ViewObserverDelegate::OnSuggestText(const std::string& text) {
+  std::string text_with_space = text + " ";
+  keyboard_service_impl_->OnKey(text_with_space.c_str());
+}
+
+void ViewObserverDelegate::OnUpdateSuggestion() {
+  IssueDraw();
+}
+
 void ViewObserverDelegate::UpdateState(int32 pointer_id,
                                        int action,
                                        const gfx::PointF& touch_point) {
   // Ignore touches outside of key area.
   if (!key_area_.Contains(touch_point)) {
     return;
   }
 
   base::TimeTicks current_ticks = base::TimeTicks::Now();
 
@@ skipping 243 matching lines @@
 
 void ViewObserverDelegate::OnViewDestroyed(mojo::View* view) {
   if (view_ == view) {
     view_->RemoveObserver(this);
     view_ = nullptr;
     gl_context_->Destroy();
   }
 }
 
 }  // namespace keyboard