1 files changed, 72 insertions, 0 deletions
diff --git a/tesseract/src/classify/tessclassifier.h b/tesseract/src/classify/tessclassifier.h
new file mode 100644
index 00000000..a8b3f753
--- /dev/null
+++ b/tesseract/src/classify/tessclassifier.h
@@ -0,0 +1,72 @@
+// Copyright 2011 Google Inc. All Rights Reserved.
+// Author: rays@google.com (Ray Smith)
+///////////////////////////////////////////////////////////////////////
+// File:        tessclassifier.h
+// Description: Tesseract implementation of a ShapeClassifier.
+// Author:      Ray Smith
+// Created:     Tue Nov 22 14:10:45 PST 2011
+//
+// (C) Copyright 2011, Google Inc.
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+///////////////////////////////////////////////////////////////////////
+
+#ifndef THIRD_PARTY_TESSERACT_CLASSIFY_TESSCLASSIFIER_H_
+#define THIRD_PARTY_TESSERACT_CLASSIFY_TESSCLASSIFIER_H_
+
+#include "shapeclassifier.h"
+
+namespace tesseract {
+
+class Classify;
+class TrainingSample;
+
+// Tesseract implementation of a ShapeClassifier.
+// Due to limitations in the content of TrainingSample, this currently
+// only works for the static classifier and only works if the ShapeTable
+// in classify is not nullptr.
+class TESS_API TessClassifier : public ShapeClassifier {
+ public:
+  TessClassifier(bool pruner_only, tesseract::Classify* classify)
+    : pruner_only_(pruner_only), classify_(classify) {}
+  ~TessClassifier() override = default;
+
+  // Classifies the given [training] sample, writing to results.
+  // See ShapeClassifier for a full description.
+  int UnicharClassifySample(const TrainingSample& sample, Pix* page_pix,
+                                    int debug, UNICHAR_ID keep_this,
+                                    std::vector<UnicharRating>* results) override;
+  // Provides access to the ShapeTable that this classifier works with.
+  const ShapeTable* GetShapeTable() const override;
+  // Provides access to the UNICHARSET that this classifier works with.
+  // Only needs to be overridden if GetShapeTable() can return nullptr.
+  const UNICHARSET& GetUnicharset() const override;
+
+  // Displays classification as the given shape_id. Creates as many windows
+  // as it feels fit, using index as a guide for placement. Adds any created
+  // windows to the windows output and returns a new index that may be used
+  // by any subsequent classifiers. Caller waits for the user to view and
+  // then destroys the windows by clearing the vector.
+  int DisplayClassifyAs(const TrainingSample& sample, Pix* page_pix,
+                                int unichar_id, int index,
+                                PointerVector<ScrollView>* windows) override;
+
+ private:
+  // Indicates that this classifier is to use just the ClassPruner, or the
+  // full classifier if false.
+  bool pruner_only_;
+  // Borrowed pointer to the actual Tesseract classifier.
+  tesseract::Classify* classify_;
+};
+
+}  // namespace tesseract
+
+#endif /* THIRD_PARTY_TESSERACT_CLASSIFY_TESSCLASSIFIER_H_ */