#ifndef org_apache_lucene_classification_document_SimpleNaiveBayesDocumentClassifier_H
#define org_apache_lucene_classification_document_SimpleNaiveBayesDocumentClassifier_H

#include "org/apache/lucene/classification/SimpleNaiveBayesClassifier.h"

namespace java {
  namespace lang {
    class String;
    class Class;
  }
  namespace util {
    class List;
    class Map;
  }
  namespace io {
    class IOException;
  }
}
namespace org {
  namespace apache {
    namespace lucene {
      namespace analysis {
        class Analyzer;
      }
      namespace document {
        class Document;
      }
      namespace classification {
        class ClassificationResult;
        namespace document {
          class DocumentClassifier;
        }
      }
      namespace util {
        class BytesRef;
      }
      namespace index {
        class IndexReader;
      }
      namespace search {
        class Query;
      }
    }
  }
}
template<class T> class JArray;

namespace org {
  namespace apache {
    namespace lucene {
      namespace classification {
        namespace document {

          class SimpleNaiveBayesDocumentClassifier : public ::org::apache::lucene::classification::SimpleNaiveBayesClassifier {
           public:
            enum {
              mid_init$_07c0d851d30fdfde,
              mid_assignClass_987e4a0d2eb5f5bf,
              mid_getClasses_0182824bbffa26b6,
              mid_getClasses_424985777d4acd23,
              mid_getTokenArray_f9bcce6e27cce930,
              max_mid
            };

            static ::java::lang::Class *class$;
            static jmethodID *mids$;
            static bool live$;
            static jclass initializeClass(bool);

            explicit SimpleNaiveBayesDocumentClassifier(jobject obj) : ::org::apache::lucene::classification::SimpleNaiveBayesClassifier(obj) {
              if (obj != NULL && mids$ == NULL)
                env->getClass(initializeClass);
            }
            SimpleNaiveBayesDocumentClassifier(const SimpleNaiveBayesDocumentClassifier& obj) : ::org::apache::lucene::classification::SimpleNaiveBayesClassifier(obj) {}

            SimpleNaiveBayesDocumentClassifier(const ::org::apache::lucene::index::IndexReader &, const ::org::apache::lucene::search::Query &, const ::java::lang::String &, const ::java::util::Map &, const JArray< ::java::lang::String > &);

            ::org::apache::lucene::classification::ClassificationResult assignClass(const ::org::apache::lucene::document::Document &) const;
            ::java::util::List getClasses(const ::org::apache::lucene::document::Document &) const;
            ::java::util::List getClasses(const ::org::apache::lucene::document::Document &, jint) const;
          };
        }
      }
    }
  }
}

#include <Python.h>

namespace org {
  namespace apache {
    namespace lucene {
      namespace classification {
        namespace document {
          extern PyType_Def PY_TYPE_DEF(SimpleNaiveBayesDocumentClassifier);
          extern PyTypeObject *PY_TYPE(SimpleNaiveBayesDocumentClassifier);

          class t_SimpleNaiveBayesDocumentClassifier {
          public:
            PyObject_HEAD
            SimpleNaiveBayesDocumentClassifier object;
            static PyObject *wrap_Object(const SimpleNaiveBayesDocumentClassifier&);
            static PyObject *wrap_jobject(const jobject&);
            static void install(PyObject *module);
            static void initialize(PyObject *module);
          };
        }
      }
    }
  }
}

#endif