src/Haarcascades.h

Go to the documentation of this file.

#ifndef _HAARCASCADES_H_
#define _HAARCASCADES_H_

#include <string>

#include "LibFaceConfig.h"

namespace libface
{

// A Cascade structure has :
// a name (for easy reference),
// a filename (to load the actual cascade from)
// a CvHaarClassifierCascade object pointer

typedef struct CascadeStruct
{
    std::string              name;
    CvHaarClassifierCascade* haarcasc;
} Cascade;

class FACEAPI Haarcascades
{

public:

    Haarcascades(const std::string& path);

    ~Haarcascades();

    void addCascade(const Cascade& newCascade, int newWeight);

    void addCascade(const std::string& name, int weight);

    void removeCascade(const std::string& name);

    void removeCascade(int index);

    void setWeight(const std::string& name, int weight);

    void setWeight(int index, int weight);

    int getWeight(const std::string& name) const;

    int getWeight(int index) const;

    bool hasCascade(const std::string& name) const;

    const Cascade& getCascade(const std::string& name) const;

    const Cascade& getCascade(int index) const;

    int getSize() const;

    void clear();

private:

    class HaarcascadesPriv;
    HaarcascadesPriv* const d;
};

} // namespace libface

#endif // _HAARCASCADES_H_