ShapeDetection: Eliminate DetectorType enum in ShapeDetector.cpp

third_party/WebKit/Source/modules/shapedetection/ShapeDetector.h

 // Copyright 2016 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.
 
 #ifndef ShapeDetector_h
 #define ShapeDetector_h
 
 #include "bindings/core/v8/ScriptPromise.h"
 #include "bindings/core/v8/ScriptPromiseResolver.h"
 #include "modules/ModulesExport.h"
 #include "modules/canvas2d/CanvasRenderingContext2D.h"
-#include "modules/shapedetection/FaceDetectorOptions.h"
-#include "public/platform/modules/shapedetection/barcodedetection.mojom-blink.h"
-#include "public/platform/modules/shapedetection/facedetection.mojom-blink.h"
 
 namespace blink {
 
 class LocalFrame;
 
 class MODULES_EXPORT ShapeDetector
     : public GarbageCollectedFinalized<ShapeDetector> {
  public:
-  enum class DetectorType {
-    Face,
-    Barcode
-    // TODO(mcasas): Implement TextDetector after
-    // https://github.com/WICG/shape-detection-api/issues/6
-  };
+  // TODO(mcasas): Implement TextDetector after
+  // https://github.com/WICG/shape-detection-api/issues/6
   explicit ShapeDetector(LocalFrame&);
-  ShapeDetector(LocalFrame&, const FaceDetectorOptions&);
   virtual ~ShapeDetector() = default;
 
-  ScriptPromise detectShapes(ScriptState*,
-                             DetectorType,
+  virtual ScriptPromise detect(ScriptState*, const CanvasImageSourceUnion&) = 0;
+  bool preprocessImageSource(ScriptState*,
+                             ScriptPromiseResolver*,
                              const CanvasImageSourceUnion&);
-  DECLARE_VIRTUAL_TRACE();
+  DEFINE_INLINE_VIRTUAL_TRACE() {}
 
  private:
-  ScriptPromise detectShapesOnImageElement(DetectorType,
-                                           ScriptPromiseResolver*,
-                                           const HTMLImageElement*);
-  ScriptPromise detectShapesOnImageBitmap(DetectorType,
-                                          ScriptPromiseResolver*,
-                                          ImageBitmap*);
-  ScriptPromise detectShapesOnVideoElement(DetectorType,
-                                           ScriptPromiseResolver*,
-                                           const HTMLVideoElement*);
+  bool preprocessImageElement(ScriptPromiseResolver*, const HTMLImageElement*);
+  bool preprocessImageBitmap(ScriptPromiseResolver*, ImageBitmap*);
+  bool preprocessVideoElement(ScriptPromiseResolver*, const HTMLVideoElement*);
 
-  ScriptPromise detectShapesOnData(DetectorType,
-                                   ScriptPromiseResolver*,
-                                   uint8_t* data,
-                                   int size,
-                                   int width,
-                                   int height);
-  void onDetectFaces(ScriptPromiseResolver*,
-                     mojom::blink::FaceDetectionResultPtr);
-  void onDetectBarcodes(ScriptPromiseResolver*,
-                        Vector<mojom::blink::BarcodeDetectionResultPtr>);
-
-  // Error handlers for use if mojo service doesn't connect.
-  void onFaceServiceConnectionError();
-  void onBarcodeServiceConnectionError();
-
-  mojom::blink::FaceDetectionPtr m_faceService;
-  mojom::blink::BarcodeDetectionPtr m_barcodeService;
-
-  HeapHashSet<Member<ScriptPromiseResolver>> m_faceServiceRequests;
-  HeapHashSet<Member<ScriptPromiseResolver>> m_barcodeServiceRequests;
+  bool getSharedBufferOnData(ScriptPromiseResolver*,
+                             uint8_t* data,
+                             int size,
+                             int width,
+                             int height);
 
  protected:
-  mojom::blink::FaceDetectorOptionsPtr m_faceDetectorOptions;
+  uint32_t m_imageWidth;
+  uint32_t m_imageHeight;
+  mojo::ScopedSharedBufferHandle m_sharedBufferHandle;
 };
 
 }  // namespace blink
 
 #endif  // ShapeDetector_h