Classifier.h

Go to the documentation of this file.

#ifndef _CLASSIFIER_H
#define _CLASSIFIER_H

#ifdef _CLASSIFIER_EXPORT
# define _FEATURE_EXPORT
#endif
#ifdef _CLASSIFIER_EXPORT_CLASS
# define _FEATURE_EXPORT_CLASS
#endif

#include "../FeatureContainer/FeatureContainer.h"


#define CLASSIFIER_EXPORT FEATURE_EXPORT
#define CLASSIFIER_EXPORT_CLASS FEATURE_EXPORT_CLASS


#define EXPORT_CLASSIFIER(classifier) \
 \
        static classifier *clf; \
        void INIT_EXPORT library_initialize() { clf = NULL; } \
        void FINI_EXPORT library_finalize() { if (clf != NULL) delete clf; } \
 \
        extern "C" CLASSIFIER_EXPORT ClassifierAlgorithm* getClassifier(classifierparams &params) { if (clf == NULL) clf = new classifier(params); return clf; }

typedef parametermap classifierparams;

class CLASSIFIER_EXPORT_CLASS ClassifierAlgorithm {
        public:
                virtual ~ClassifierAlgorithm() {}

                virtual void init(FeatureContainer *fc) = 0;
                virtual membershiplist getClusterMembership(const featurevector *sample) = 0;
                virtual unsigned long nextSample() = 0;
                virtual unsigned long nextSample(const featurevector* sample) = 0;
                
                virtual string serialize() const = 0;

                virtual void unserialize(string data) = 0;

#if _DEBUG_CLASSIFIERS

                virtual string toString() const = 0;
#endif
};

class Classifier : public ClassifierAlgorithm {
        private:
                FeatureContainer *f;
                ClassifierAlgorithm *ca;

#if defined(_WIN32_WCE) || defined (_WINDOWS)
                HMODULE dll;
#else

                void* dlHandle;
#endif

        public:
                Classifier(const string &name, const classifierparams &params);
                virtual ~Classifier();

                virtual void init(FeatureContainer *fc) { f = fc; ca->init(fc); }
                virtual membershiplist getClusterMembership(const featurevector *sample) { return ca->getClusterMembership(sample); }
                virtual unsigned long nextSample() { return ca->nextSample(); }
                virtual unsigned long nextSample(const featurevector* sample) { return ca->nextSample(sample); }
                virtual string serialize() const { return ca->serialize(); }
                virtual void unserialize(string data) { ca->unserialize(data); }

                ClassifierAlgorithm* getCa() const { return ca; }

#if _DEBUG_CLASSIFIERS

                virtual string toString() const { return ca->toString(); }
#endif
};

typedef ClassifierAlgorithm* (*getClassifier_t) (const classifierparams &params);

#endif

---