Add new files for plugin management
This commit is contained in:
manupap1
parent
15532e2a03
commit
199f399caf
|
|
@ -0,0 +1,232 @@
|
|||
#include "zm_detector.h"
|
||||
|
||||
|
||||
|
||||
/*!\fn Detector::Detector(const Detector& source)
|
||||
* \param source is the object to copy
|
||||
*/
|
||||
Detector::Detector(const Detector& source)
|
||||
: m_sDetectionCause(source.m_sDetectionCause),
|
||||
m_fMinAlarmScore(source.m_fMinAlarmScore),
|
||||
m_fMaxAlarmScore(source.m_fMaxAlarmScore),
|
||||
m_fImageScaleFactor(source.m_fImageScaleFactor),
|
||||
m_nNewWidth(source.m_nNewWidth),
|
||||
m_nNewHeight(source.m_nNewHeight),
|
||||
m_sLogPrefix(source.m_sLogPrefix),
|
||||
m_sConfigSectionName(source.m_sConfigSectionName),
|
||||
m_vnPluginZones(source.m_vnPluginZones)
|
||||
{
|
||||
m_bIsPluginEnabled = false;
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*!\fn Detector& ImageAnalyser::Detector::operator=(const ImageAnalyser::Detector& source)
|
||||
* \param source is the object to copy
|
||||
*/
|
||||
Detector& Detector::operator=(const Detector& source)
|
||||
{
|
||||
m_sDetectionCause = source.m_sDetectionCause;
|
||||
m_fMinAlarmScore = source.m_fMinAlarmScore;
|
||||
m_fMaxAlarmScore = source.m_fMaxAlarmScore;
|
||||
m_fImageScaleFactor = source.m_fImageScaleFactor;
|
||||
m_sLogPrefix = source.m_sLogPrefix;
|
||||
m_nNewWidth = source.m_nNewWidth;
|
||||
m_nNewHeight = source.m_nNewHeight;
|
||||
m_sConfigSectionName = source.m_sConfigSectionName;
|
||||
m_vnPluginZones = source.m_vnPluginZones;
|
||||
|
||||
return *this;
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*!\fn Detector::getDetectionCause()
|
||||
* return detection cause as string
|
||||
*/
|
||||
string Detector::getDetectionCause()
|
||||
{
|
||||
return m_sDetectionCause;
|
||||
}
|
||||
|
||||
|
||||
/*!\fn Detector::getConfigSectionName()
|
||||
* return plugin name as string
|
||||
*/
|
||||
string Detector::getPluginName()
|
||||
{
|
||||
return m_sConfigSectionName;
|
||||
}
|
||||
|
||||
|
||||
/*!\fn Detector::EnablePlugin(vector<int> zoneList)
|
||||
* \param vnZoneList is the list of enabled zones for the plugin
|
||||
*/
|
||||
void Detector::EnablePlugin(vector<unsigned int> vnZoneList)
|
||||
{
|
||||
m_vnPluginZones = vnZoneList;
|
||||
m_bIsPluginEnabled = true;
|
||||
}
|
||||
|
||||
|
||||
/*!\fn Detector::getPluginZones()
|
||||
* \return the list of zone which have the plugin enabled
|
||||
*/
|
||||
vector<unsigned int> Detector::getPluginZones()
|
||||
{
|
||||
return m_vnPluginZones;
|
||||
}
|
||||
|
||||
|
||||
/*! \fn Detector::log(int nLogLevel, string sLevel, string sMessage)
|
||||
*/
|
||||
void Detector::log(int nLogLevel, string sLevel, string sMessage)
|
||||
{
|
||||
string sMessageToLog = sLevel + string(" [") + m_sLogPrefix + string(": ") + sMessage + string("]");
|
||||
syslog(nLogLevel, "%s", sMessageToLog.c_str());
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*! \fn int Detector::Detect(const Image &image, Event::StringSet &zoneSet)
|
||||
* \param zmImage is an image to detect faces on
|
||||
* \param zoneSet is set of zone names (see zm_zone.h)
|
||||
* \param score is the detection score
|
||||
* \return true if detection is effective
|
||||
*/
|
||||
bool Detector::Detect(const Image &zmImage, Zone** zones, Event::StringSet &zoneSet, unsigned int &score)
|
||||
{
|
||||
bool alarm = false;
|
||||
char szMessage[100];
|
||||
score = 0;
|
||||
|
||||
if (!m_bIsPluginEnabled) return (alarm);
|
||||
|
||||
// Check preclusive zones first
|
||||
for(std::vector<unsigned int>::iterator it = m_vnPluginZones.begin(); it != m_vnPluginZones.end(); ++it)
|
||||
{
|
||||
Zone *zone = zones[*it];
|
||||
if (!zone->IsPreclusive())
|
||||
continue;
|
||||
if (zone->IsPostProcEnabled() && !zone->IsPostProcInProgress())
|
||||
continue;
|
||||
sprintf(szMessage, "Checking preclusive zone %s", zone->Label());
|
||||
log(LOG_DEBUG, "DEBUG", szMessage);
|
||||
if (checkZone(zone, *it, &zmImage))
|
||||
{
|
||||
alarm = true;
|
||||
score += zone->Score();
|
||||
zoneSet.insert(zone->Text());
|
||||
if (zone->IsPostProcEnabled())
|
||||
{
|
||||
zone->StopPostProcessing();
|
||||
sprintf(szMessage, "Zone is alarmed, zone score = %d (post-processing)", zone->Score());
|
||||
}
|
||||
else
|
||||
{
|
||||
sprintf(szMessage, "Zone is alarmed, zone score = %d", zone->Score());
|
||||
}
|
||||
log(LOG_DEBUG, "DEBUG", szMessage);
|
||||
}
|
||||
}
|
||||
|
||||
if ( alarm )
|
||||
{
|
||||
alarm = false;
|
||||
score = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Find all alarm pixels in active zones
|
||||
for(std::vector<unsigned int>::iterator it = m_vnPluginZones.begin(); it != m_vnPluginZones.end(); ++it)
|
||||
{
|
||||
Zone *zone = zones[*it];
|
||||
if (!zone->IsActive())
|
||||
continue;
|
||||
if (zone->IsPostProcEnabled() && !zone->IsPostProcInProgress())
|
||||
continue;
|
||||
if (checkZone(zone, *it, &zmImage))
|
||||
{
|
||||
alarm = true;
|
||||
score += zone->Score();
|
||||
zoneSet.insert(zone->Text());
|
||||
if (zone->IsPostProcEnabled())
|
||||
{
|
||||
zone->StopPostProcessing();
|
||||
sprintf(szMessage, "Zone is alarmed, zone score = %d (post-processing)", zone->Score());
|
||||
}
|
||||
else
|
||||
{
|
||||
zone->SetAlarm();
|
||||
sprintf(szMessage, "Zone is alarmed, zone score = %d", zone->Score());
|
||||
}
|
||||
log(LOG_DEBUG, "DEBUG", szMessage);
|
||||
}
|
||||
}
|
||||
|
||||
if ( alarm )
|
||||
{
|
||||
// Checking inclusive zones
|
||||
for(std::vector<unsigned int>::iterator it = m_vnPluginZones.begin(); it != m_vnPluginZones.end(); ++it)
|
||||
{
|
||||
Zone *zone = zones[*it];
|
||||
if (!zone->IsInclusive())
|
||||
continue;
|
||||
if (zone->IsPostProcEnabled() && !zone->IsPostProcInProgress())
|
||||
continue;
|
||||
sprintf(szMessage, "Checking inclusive zone %s", zone->Label());
|
||||
log(LOG_DEBUG, "DEBUG", szMessage);
|
||||
if (checkZone(zone, *it, &zmImage))
|
||||
{
|
||||
alarm = true;
|
||||
score += zone->Score();
|
||||
zoneSet.insert(zone->Text());
|
||||
if (zone->IsPostProcEnabled())
|
||||
{
|
||||
zone->StopPostProcessing();
|
||||
sprintf(szMessage, "Zone is alarmed, zone score = %d (post-processing)", zone->Score());
|
||||
}
|
||||
else
|
||||
{
|
||||
zone->SetAlarm();
|
||||
sprintf(szMessage, "Zone is alarmed, zone score = %d", zone->Score());
|
||||
}
|
||||
log(LOG_DEBUG, "DEBUG", szMessage);
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// Find all alarm pixels in exclusive zones
|
||||
for(std::vector<unsigned int>::iterator it = m_vnPluginZones.begin(); it != m_vnPluginZones.end(); ++it)
|
||||
{
|
||||
Zone *zone = zones[*it];
|
||||
if (!zone->IsExclusive())
|
||||
continue;
|
||||
if (zone->IsPostProcEnabled() && !zone->IsPostProcInProgress())
|
||||
continue;
|
||||
sprintf(szMessage, "Checking exclusive zone %s", zone->Label());
|
||||
log(LOG_DEBUG, "DEBUG", szMessage);
|
||||
if (checkZone(zone, *it, &zmImage))
|
||||
{
|
||||
alarm = true;
|
||||
score += zone->Score();
|
||||
zoneSet.insert(zone->Text());
|
||||
if (zone->IsPostProcEnabled())
|
||||
{
|
||||
zone->StopPostProcessing();
|
||||
sprintf(szMessage, "Zone is alarmed, zone score = %d (post-processing)", zone->Score());
|
||||
}
|
||||
else
|
||||
{
|
||||
zone->SetAlarm();
|
||||
sprintf(szMessage, "Zone is alarmed, zone score = %d", zone->Score());
|
||||
}
|
||||
log(LOG_DEBUG, "DEBUG", szMessage);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return alarm;
|
||||
}
|
||||
|
|
@ -0,0 +1,137 @@
|
|||
#ifndef ZM_DETECTOR_H
|
||||
#define ZM_DETECTOR_H
|
||||
|
||||
|
||||
#include <string>
|
||||
#include <algorithm>
|
||||
#include <syslog.h>
|
||||
#include <libgen.h>
|
||||
|
||||
#include "zm_image.h"
|
||||
#include "zm_zone.h"
|
||||
#include "zm_event.h"
|
||||
|
||||
#define DEFAULT_DETECTION_CAUSE "Object Detected"
|
||||
#define DEFAULT_MIN_ALARM_SCORE 1.0
|
||||
#define DEFAULT_MAX_ALARM_SCORE 99.0
|
||||
#define DEFAULT_IMAGE_SCALE_FACTOR 1.0
|
||||
|
||||
#define DEFAULT_LOG_PREFIX "ZM PLUGIN"
|
||||
#define LOG_LEVEL LOG_NOTICE
|
||||
#define DEFAULT_CONFIGFILE_SECTION "libzm_vscvl_plugin"
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
//! Base class for object detectors, defined in plugins.
|
||||
class Detector
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
//! Destructor
|
||||
virtual ~Detector() {}
|
||||
|
||||
//! Default constructor
|
||||
Detector()
|
||||
{
|
||||
m_sLogPrefix = DEFAULT_LOG_PREFIX;
|
||||
m_sDetectionCause = DEFAULT_DETECTION_CAUSE;
|
||||
m_fMinAlarmScore = DEFAULT_MIN_ALARM_SCORE;
|
||||
m_fMaxAlarmScore = DEFAULT_MAX_ALARM_SCORE;
|
||||
m_fImageScaleFactor = DEFAULT_IMAGE_SCALE_FACTOR;
|
||||
m_sConfigSectionName = DEFAULT_CONFIGFILE_SECTION;
|
||||
m_nNewWidth = 0;
|
||||
m_nNewHeight = 0;
|
||||
}
|
||||
|
||||
//! Constructor with section name parameter.
|
||||
Detector(string sPluginFileName)
|
||||
{
|
||||
m_sLogPrefix = DEFAULT_LOG_PREFIX;
|
||||
|
||||
char* szPluginFileName = strdup(sPluginFileName.c_str());
|
||||
|
||||
string sPluginFileNameName = string(basename(szPluginFileName));
|
||||
|
||||
size_t idx = sPluginFileNameName.rfind('.');
|
||||
|
||||
if (idx == string::npos)
|
||||
m_sConfigSectionName = sPluginFileNameName;
|
||||
else
|
||||
m_sConfigSectionName = sPluginFileNameName.substr(0, idx);
|
||||
|
||||
m_sDetectionCause = DEFAULT_DETECTION_CAUSE;
|
||||
m_fMinAlarmScore = DEFAULT_MIN_ALARM_SCORE;
|
||||
m_fMaxAlarmScore = DEFAULT_MAX_ALARM_SCORE;
|
||||
m_fImageScaleFactor = DEFAULT_IMAGE_SCALE_FACTOR;
|
||||
m_nNewWidth = 0;
|
||||
m_nNewHeight = 0;
|
||||
}
|
||||
|
||||
//! Copy constructor
|
||||
Detector(const Detector& source);
|
||||
|
||||
//! Assignment operator
|
||||
Detector& operator=(const Detector& source);
|
||||
|
||||
//! Detect (in an image later)
|
||||
bool Detect(const Image &image, Zone** zones, Event::StringSet &zoneSet, unsigned int &score);
|
||||
|
||||
//! Load detector's parameters.
|
||||
virtual int loadConfig(string sConfigFileName, map<unsigned int,map<string,string> > mapPluginConf) = 0;
|
||||
|
||||
//! Returns detection case string.
|
||||
string getDetectionCause();
|
||||
|
||||
//! Returns plugin name as string.
|
||||
string getPluginName();
|
||||
|
||||
//! Enable the plugin for the given zones.
|
||||
void EnablePlugin(vector<unsigned int> zoneList);
|
||||
|
||||
//! Return the list of enabled zones
|
||||
vector<unsigned int> getPluginZones();
|
||||
|
||||
protected:
|
||||
|
||||
//! Do detection inside one given zone.
|
||||
virtual bool checkZone(Zone *zone, unsigned int n_zone, const Image *zmImage) = 0;
|
||||
|
||||
//! Log messages to the SYSLOG.
|
||||
void log(int, string sLevel, string sMessage);
|
||||
|
||||
//! String to be shown as detection cause for event.
|
||||
string m_sDetectionCause;
|
||||
|
||||
//! Minimum score value to consider frame as to be alarmed.
|
||||
double m_fMinAlarmScore;
|
||||
|
||||
//! Maximum score value to consider frame as to be alarmed.
|
||||
double m_fMaxAlarmScore;
|
||||
|
||||
//! Maximum allowed width of frame image.
|
||||
double m_fImageScaleFactor;
|
||||
|
||||
//! Width of image to resize.
|
||||
int m_nNewWidth;
|
||||
|
||||
//! Height of image to resize.
|
||||
int m_nNewHeight;
|
||||
|
||||
//! String prefix for SYSLOG messages.
|
||||
string m_sLogPrefix;
|
||||
|
||||
//! Name of config file section to search parameters.
|
||||
string m_sConfigSectionName;
|
||||
|
||||
//! List of zones enabled for the plugin
|
||||
vector<unsigned int> m_vnPluginZones;
|
||||
|
||||
//! Plugin status regarding zone settings
|
||||
bool m_bIsPluginEnabled;
|
||||
|
||||
};
|
||||
|
||||
|
||||
#endif // ZM_DETECTOR_H
|
||||
|
|
@ -0,0 +1,105 @@
|
|||
#include "zm_plugin.h"
|
||||
|
||||
|
||||
|
||||
/*!\fn Plugin::Plugin(const std::string &sFilename)
|
||||
* \param sFilename is the name of plugin file to load
|
||||
*/
|
||||
Plugin::Plugin(const std::string &sFilename)
|
||||
: m_sPluginFileName(sFilename),
|
||||
m_hDLL(0),
|
||||
m_pDLLRefCount(0),
|
||||
m_pfnGetEngineVersion(0),
|
||||
m_pfnRegisterPlugin(0)
|
||||
{
|
||||
|
||||
// Try to load the plugin as a dynamic library
|
||||
m_hDLL = dlopen(sFilename.c_str(), RTLD_LAZY|RTLD_GLOBAL);
|
||||
|
||||
if(!m_hDLL) // if library hasn't been loaded successfully
|
||||
{
|
||||
throw runtime_error(string("Could not load '") + sFilename + "'");
|
||||
}
|
||||
|
||||
// Locate the plugin's exported functions
|
||||
try
|
||||
{
|
||||
m_pfnGetEngineVersion = reinterpret_cast<fnGetEngineVersion *>(dlsym(m_hDLL, "getEngineVersion"));
|
||||
m_pfnRegisterPlugin = reinterpret_cast<fnRegisterPlugin *>(dlsym(m_hDLL, "registerPlugin"));
|
||||
|
||||
// If the functions aren't found, we're going to assume this is
|
||||
// a plain simple DLL and not one of our plugins
|
||||
if(!m_pfnGetEngineVersion || ! m_pfnRegisterPlugin)
|
||||
throw runtime_error(string("'") + sFilename + "' is not a valid plugin");
|
||||
|
||||
// Initialize a new DLL reference counter
|
||||
m_pDLLRefCount = new size_t(1);
|
||||
}
|
||||
catch(runtime_error &ex)
|
||||
{
|
||||
dlclose(m_hDLL);
|
||||
throw ex;
|
||||
}
|
||||
catch(...)
|
||||
{
|
||||
dlclose(m_hDLL);
|
||||
throw runtime_error(string("Unknown exception while loading plugin '") + sFilename + string("'"));
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*!\fn Plugin::Plugin(const Plugin &Other)
|
||||
* \param Other is the other plugin instance to copy
|
||||
*/
|
||||
Plugin::Plugin(const Plugin &Other)
|
||||
: m_sPluginFileName(Other.m_sPluginFileName),
|
||||
m_hDLL(Other.m_hDLL),
|
||||
m_pDLLRefCount(Other.m_pDLLRefCount),
|
||||
m_pfnGetEngineVersion(Other.m_pfnGetEngineVersion),
|
||||
m_pfnRegisterPlugin(Other.m_pfnRegisterPlugin)
|
||||
{
|
||||
// Increase DLL reference counter
|
||||
++*m_pDLLRefCount;
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*!\fn Plugin::operator=(const Plugin &Other)
|
||||
* \param Other is the other plugin instance to copy
|
||||
* return copy of object
|
||||
*/
|
||||
Plugin& Plugin::operator=(const Plugin &Other)
|
||||
{
|
||||
m_hDLL = Other.m_hDLL;
|
||||
m_pfnGetEngineVersion = Other.m_pfnGetEngineVersion;
|
||||
m_pfnRegisterPlugin = Other.m_pfnRegisterPlugin;
|
||||
m_pDLLRefCount = Other.m_pDLLRefCount;
|
||||
m_sPluginFileName = Other.m_sPluginFileName;
|
||||
// Increase DLL reference counter
|
||||
++*m_pDLLRefCount;
|
||||
return *this;
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
Plugin::~Plugin()
|
||||
{
|
||||
// Only unload the DLL if there are no more references to it
|
||||
if(!--*m_pDLLRefCount)
|
||||
{
|
||||
delete m_pDLLRefCount;
|
||||
dlclose(m_hDLL);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
/*!\fn Plugin::registerPlugin(PluginManager &K)
|
||||
* \param K is the pointer to plugin manager
|
||||
*/
|
||||
void Plugin::registerPlugin(PluginManager &K)
|
||||
{
|
||||
m_pfnRegisterPlugin(K, m_sPluginFileName);
|
||||
}
|
||||
|
|
@ -0,0 +1,77 @@
|
|||
#ifndef ZM_PLUGIN_H
|
||||
#define ZM_PLUGIN_H
|
||||
|
||||
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <dlfcn.h>
|
||||
|
||||
#include <stdexcept>
|
||||
|
||||
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
|
||||
class PluginManager;
|
||||
|
||||
|
||||
|
||||
//! Signature for the version query function
|
||||
typedef int fnGetEngineVersion();
|
||||
|
||||
//! Signature for the plugin's registration function
|
||||
typedef void fnRegisterPlugin(PluginManager &, string);
|
||||
|
||||
|
||||
|
||||
//! Representation of a plugin.
|
||||
/*! Use for loading plugin's shared library
|
||||
* and registration of it to the PluginManager.
|
||||
*/
|
||||
class Plugin
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
//! Initialize and load plugin
|
||||
Plugin(const std::string &sFilename);
|
||||
|
||||
//! Copy existing plugin instance
|
||||
Plugin(const Plugin &Other);
|
||||
|
||||
//! Operator =.
|
||||
Plugin &operator =(const Plugin &Other);
|
||||
|
||||
//! Unload a plugin
|
||||
~Plugin();
|
||||
|
||||
//! Query the plugin for its expected engine version
|
||||
int getEngineVersion() const { return m_pfnGetEngineVersion();}
|
||||
|
||||
//! Register the plugin to a PluginManager
|
||||
void registerPlugin(PluginManager &K);
|
||||
|
||||
private:
|
||||
|
||||
//! Shared file name.
|
||||
string m_sPluginFileName;
|
||||
|
||||
//! DLL handle
|
||||
void* m_hDLL;
|
||||
|
||||
//! Number of references to the DLL
|
||||
size_t *m_pDLLRefCount;
|
||||
|
||||
//! Version query function
|
||||
fnGetEngineVersion *m_pfnGetEngineVersion;
|
||||
|
||||
//! Plugin registration function
|
||||
fnRegisterPlugin *m_pfnRegisterPlugin;
|
||||
};
|
||||
|
||||
|
||||
|
||||
#endif //ZM_PLUGIN_H
|
||||
|
|
@ -0,0 +1,160 @@
|
|||
#include "zm_plugin_manager.h"
|
||||
|
||||
|
||||
|
||||
/*! \fn file_select(const struct direct *entry)
|
||||
* A functor for selection of files with specified extension.
|
||||
* \param entry is file structure
|
||||
* \return 1 if file match selection criteria and
|
||||
* 0 otherwise.
|
||||
* NOTE: file extension is specified by PluginManager::m_sPluginExt
|
||||
* static variable.
|
||||
*/
|
||||
int file_select(const struct direct *entry)
|
||||
{
|
||||
char *ptr;
|
||||
|
||||
if ((strcmp(entry->d_name, ".")== 0) || (strcmp(entry->d_name, "..") == 0))
|
||||
return 0;
|
||||
|
||||
// Check for filename extensions.
|
||||
ptr = rindex((char*)entry->d_name, '.');
|
||||
if ((ptr != NULL) && (strcmp(ptr, (PluginManager::m_sPluginExt).c_str()) == 0))
|
||||
return 1;
|
||||
else
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
/*! \fn join_paths(const string& p1, const string& p2)
|
||||
* \param p1 is the first part of desired path
|
||||
* \param p2 is the second part of desired path
|
||||
* \return joined path string.
|
||||
*/
|
||||
string join_paths(const string& p1, const string& p2)
|
||||
{
|
||||
char sep = '/';
|
||||
string tmp = p1;
|
||||
|
||||
#ifdef _WIN32
|
||||
sep = '\\';
|
||||
#endif
|
||||
|
||||
if (p1[p1.length()] != sep)
|
||||
{ // Need to add a path separator
|
||||
tmp += sep;
|
||||
return(tmp + p2);
|
||||
}
|
||||
else
|
||||
return(p1 + p2);
|
||||
}
|
||||
|
||||
|
||||
|
||||
string PluginManager::m_sPluginExt = DEFAULT_PLUGIN_EXT;
|
||||
|
||||
|
||||
PluginManager::PluginManager() {}
|
||||
|
||||
|
||||
PluginManager::PluginManager(
|
||||
int nMonitorId
|
||||
) :
|
||||
m_ImageAnalyser( nMonitorId )
|
||||
{}
|
||||
|
||||
|
||||
|
||||
/*!\fn PluginManager::loadPlugin(const string &sFilename))
|
||||
* \param sFilename is the name of plugin file to load
|
||||
*/
|
||||
bool PluginManager::loadPlugin(const string &sFilename)
|
||||
{
|
||||
try
|
||||
{
|
||||
if(m_LoadedPlugins.find(sFilename) == m_LoadedPlugins.end())
|
||||
m_LoadedPlugins.insert(PluginMap::value_type(sFilename, Plugin(sFilename))).first->second.registerPlugin(*this);
|
||||
}
|
||||
catch(runtime_error &ex)
|
||||
{
|
||||
Error("Runtime error: %s", ex.what());
|
||||
return false;
|
||||
}
|
||||
catch(...)
|
||||
{
|
||||
Error("Unknown exception. Could not load %s.", sFilename.c_str());
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
/*!\fn PluginManager::findPlugins(const string &sPath, bool loadPlugins)
|
||||
* \param sPath is the path to folder to search plugins
|
||||
* \param loadPlugins is a flag to allow loading of plugins
|
||||
* \param nNumPlugLoaded is the number of loaded plugins
|
||||
* \return the number of found plugins
|
||||
*/
|
||||
int PluginManager::findPlugins(const string sPath, bool loadPlugins, unsigned int& nNumPlugLoaded)
|
||||
{
|
||||
struct direct **files;
|
||||
int count = scandir(sPath.c_str(), &files, file_select, alphasort);
|
||||
if(count <= 0) count = 0;
|
||||
|
||||
for (int i = 0; i < count; ++i)
|
||||
{
|
||||
string sFileName = string(files[i]->d_name);
|
||||
string sFullPath = join_paths(sPath, sFileName);
|
||||
size_t idx = sFileName.rfind('.');
|
||||
if (idx != string::npos)
|
||||
sFileName = sFileName.substr(0, idx);
|
||||
bool IsPluginRegistered = false;
|
||||
if(config.load_plugins || loadPlugins)
|
||||
{
|
||||
Info("Loading plugin %s ... ", sFullPath.c_str());
|
||||
IsPluginRegistered = loadPlugin(sFullPath);
|
||||
}
|
||||
mapPluginReg.insert( pair<string,bool>(sFileName, IsPluginRegistered) );
|
||||
if (IsPluginRegistered) nNumPlugLoaded++;
|
||||
}
|
||||
return count;
|
||||
}
|
||||
|
||||
|
||||
/*!\fn PluginManager::configurePlugins(string sConfigFileName)
|
||||
* \param sConfigFileName is the path to the configuration file, where parameters for all plugins are given.
|
||||
* \param bDoNativeDet is true if native detection will be performed
|
||||
*/
|
||||
void PluginManager::configurePlugins(string sConfigFileName, bool bDoNativeDet)
|
||||
{
|
||||
m_ImageAnalyser.configurePlugins(sConfigFileName, bDoNativeDet);
|
||||
}
|
||||
|
||||
|
||||
/*!\fn PluginManager::getPluginsGenConf(map<string,pGenConf>& mapPluginGenConf)
|
||||
* \param mapPluginGenConf is the map of general settings for the plugins
|
||||
* \param mapPluginZoneConf is the map of zone settings for the plugins
|
||||
* \return the number of found plugins
|
||||
*/
|
||||
unsigned long PluginManager::getPluginsGenConf(map<string,pGenConf>& mapPluginGenConf)
|
||||
{
|
||||
for (map<string,bool>::iterator it = mapPluginReg.begin() ; it != mapPluginReg.end(); ++it)
|
||||
{
|
||||
pGenConf plugGenConf;
|
||||
m_ImageAnalyser.getRegPluginGenConf( it->first, plugGenConf );
|
||||
plugGenConf.Registered = it->second;
|
||||
mapPluginGenConf.insert( pair<string,pGenConf>(it->first, plugGenConf) );
|
||||
}
|
||||
return mapPluginGenConf.size();
|
||||
}
|
||||
|
||||
|
||||
/*!\fn PluginManager::getPluginZoneConf(string sPluginName, PluginZoneConf& mapPluginZoneConf)
|
||||
* \param mapPluginZoneConf is the map of zone settings for the plugin
|
||||
*/
|
||||
void PluginManager::getPluginZoneConf(string sPluginName, PluginZoneConf& mapPluginZoneConf)
|
||||
{
|
||||
m_ImageAnalyser.getRegPluginZoneConf( sPluginName, mapPluginZoneConf );
|
||||
}
|
||||
|
|
@ -0,0 +1,94 @@
|
|||
#ifndef ZM_PLUGIN_MANAGER_H
|
||||
#define ZM_PLUGIN_MANAGER_H
|
||||
|
||||
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string>
|
||||
#include <fstream>
|
||||
#include <map>
|
||||
#include <sys/types.h>
|
||||
#include <sys/dir.h>
|
||||
#include <sys/param.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "zm_image_analyser.h"
|
||||
#include "zm_detector.h"
|
||||
#include "zm_plugin.h"
|
||||
|
||||
|
||||
|
||||
using namespace std;
|
||||
|
||||
|
||||
#define ZM_ENGINE_VERSION 24
|
||||
|
||||
|
||||
//! Map of plugins by their associated file names.
|
||||
typedef std::map<std::string, Plugin> PluginMap;
|
||||
|
||||
|
||||
//! External function for sorting of files in directory.
|
||||
extern int alphasort();
|
||||
|
||||
|
||||
//! Function to select files with plugins by extension.
|
||||
int file_select(const struct direct *entry);
|
||||
|
||||
|
||||
//! Join two path strings.
|
||||
string join_paths(const string& p1, const string& p2);
|
||||
|
||||
|
||||
|
||||
//! Class for managing all loaded plugins.
|
||||
class PluginManager
|
||||
{
|
||||
public:
|
||||
//! Default constructor.
|
||||
PluginManager();
|
||||
|
||||
//! Constructor with parameters
|
||||
PluginManager(int nMonitorId);
|
||||
|
||||
//! Access the image analyser.
|
||||
ImageAnalyser &getImageAnalyser() {return m_ImageAnalyser;}
|
||||
|
||||
//! Loads a plugin.
|
||||
bool loadPlugin(const string &sFilename);
|
||||
|
||||
//! Find all plugins from given directory, load them if required and
|
||||
//! return the number of found plugins and the number of loaded plugins
|
||||
int findPlugins(const string sPath, bool loadPlugins, unsigned int& nNumPlugLoaded);
|
||||
|
||||
//! Get general settings of plugins
|
||||
unsigned long getPluginsGenConf(map<string,pGenConf>& mapPluginGenConf);
|
||||
|
||||
//! Get zone settings of a plugin
|
||||
void getPluginZoneConf(string sPluginName, PluginZoneConf& mapPluginZoneConf);
|
||||
|
||||
//! Configure all loaded plugins using given configuration file.
|
||||
void configurePlugins(string sConfigFileName, bool bDoNativeDet);
|
||||
|
||||
//! Set plugin extension.
|
||||
void setPluginExt(string sPluginExt) { m_sPluginExt = sPluginExt; }
|
||||
|
||||
//! Extension for zm plugins.
|
||||
static string m_sPluginExt;
|
||||
|
||||
private:
|
||||
|
||||
//! All plugins currently loaded.
|
||||
PluginMap m_LoadedPlugins;
|
||||
|
||||
//! The image analyser.
|
||||
ImageAnalyser m_ImageAnalyser;
|
||||
|
||||
//! Plugin list
|
||||
map<string,bool> mapPluginReg;
|
||||
};
|
||||
|
||||
|
||||
|
||||
#endif //ZM_PLUGIN_MANAGER_H
|
||||
Loading…
Reference in New Issue