267 lines
7.5 KiB
C++
267 lines
7.5 KiB
C++
//
|
|
// ZoneMinder Core Interfaces, $Date$, $Revision$
|
|
// Copyright (C) 2001-2008 Philip Coombes
|
|
//
|
|
// This program is free software; you can redistribute it and/or
|
|
// modify it under the terms of the GNU General Public License
|
|
// as published by the Free Software Foundation; either version 2
|
|
// of the License, or (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program; if not, write to the Free Software
|
|
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
//
|
|
|
|
#ifndef ZM_EVENT_H
|
|
#define ZM_EVENT_H
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
#include <time.h>
|
|
#include <sys/time.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <mysql/mysql.h>
|
|
|
|
#include <set>
|
|
#include <map>
|
|
|
|
#include "zm.h"
|
|
#include "zm_image.h"
|
|
#include "zm_stream.h"
|
|
#include "zm_video.h"
|
|
|
|
class Zone;
|
|
class Monitor;
|
|
|
|
#define MAX_PRE_ALARM_FRAMES 16 // Maximum number of prealarm frames that can be stored
|
|
|
|
//
|
|
// Class describing events, i.e. captured periods of activity.
|
|
//
|
|
class Event {
|
|
friend class EventStream;
|
|
|
|
protected:
|
|
static bool initialised;
|
|
static char capture_file_format[PATH_MAX];
|
|
static char analyse_file_format[PATH_MAX];
|
|
static char general_file_format[PATH_MAX];
|
|
static char video_file_format[PATH_MAX];
|
|
|
|
protected:
|
|
static int sd;
|
|
|
|
public:
|
|
typedef std::set<std::string> StringSet;
|
|
typedef std::map<std::string,StringSet> StringSetMap;
|
|
|
|
protected:
|
|
typedef enum { NORMAL, BULK, ALARM } FrameType;
|
|
|
|
struct PreAlarmData {
|
|
Image *image;
|
|
struct timeval timestamp;
|
|
unsigned int score;
|
|
Image *alarm_frame;
|
|
};
|
|
|
|
static int pre_alarm_count;
|
|
static PreAlarmData pre_alarm_data[MAX_PRE_ALARM_FRAMES];
|
|
|
|
protected:
|
|
unsigned int id;
|
|
Monitor *monitor;
|
|
struct timeval start_time;
|
|
struct timeval end_time;
|
|
std::string cause;
|
|
StringSetMap noteSetMap;
|
|
bool videoEvent;
|
|
int frames;
|
|
int alarm_frames;
|
|
unsigned int tot_score;
|
|
unsigned int max_score;
|
|
char path[PATH_MAX];
|
|
VideoWriter* videowriter;
|
|
FILE* timecodes_fd;
|
|
char video_name[PATH_MAX];
|
|
char video_file[PATH_MAX];
|
|
char timecodes_name[PATH_MAX];
|
|
char timecodes_file[PATH_MAX];
|
|
|
|
protected:
|
|
int last_db_frame;
|
|
|
|
protected:
|
|
static void Initialise() {
|
|
if ( initialised )
|
|
return;
|
|
|
|
snprintf( capture_file_format, sizeof(capture_file_format), "%%s/%%0%dd-capture.jpg", config.event_image_digits );
|
|
snprintf( analyse_file_format, sizeof(analyse_file_format), "%%s/%%0%dd-analyse.jpg", config.event_image_digits );
|
|
snprintf( general_file_format, sizeof(general_file_format), "%%s/%%0%dd-%%s", config.event_image_digits );
|
|
snprintf( video_file_format, sizeof(video_file_format), "%%s/%%s");
|
|
|
|
initialised = true;
|
|
}
|
|
|
|
void createNotes( std::string ¬es );
|
|
|
|
public:
|
|
static bool OpenFrameSocket( int );
|
|
static bool ValidateFrameSocket( int );
|
|
|
|
public:
|
|
Event( Monitor *p_monitor, struct timeval p_start_time, const std::string &p_cause, const StringSetMap &p_noteSetMap, bool p_videoEvent=false );
|
|
~Event();
|
|
|
|
int Id() const { return( id ); }
|
|
const std::string &Cause() { return( cause ); }
|
|
int Frames() const { return( frames ); }
|
|
int AlarmFrames() const { return( alarm_frames ); }
|
|
|
|
const struct timeval &StartTime() const { return( start_time ); }
|
|
const struct timeval &EndTime() const { return( end_time ); }
|
|
struct timeval &EndTime() { return( end_time ); }
|
|
|
|
bool SendFrameImage( const Image *image, bool alarm_frame=false );
|
|
bool WriteFrameImage( Image *image, struct timeval timestamp, const char *event_file, bool alarm_frame=false );
|
|
bool WriteFrameVideo( const Image *image, const struct timeval timestamp, VideoWriter* videow );
|
|
|
|
void updateNotes( const StringSetMap &stringSetMap );
|
|
|
|
void AddFrames( int n_frames, Image **images, struct timeval **timestamps );
|
|
void AddFrame( Image *image, struct timeval timestamp, int score=0, Image *alarm_frame=NULL );
|
|
|
|
private:
|
|
void AddFramesInternal( int n_frames, int start_frame, Image **images, struct timeval **timestamps );
|
|
|
|
public:
|
|
static const char *getSubPath( struct tm *time ) {
|
|
static char subpath[PATH_MAX] = "";
|
|
snprintf( subpath, sizeof(subpath), "%02d/%02d/%02d/%02d/%02d/%02d", time->tm_year-100, time->tm_mon+1, time->tm_mday, time->tm_hour, time->tm_min, time->tm_sec );
|
|
return( subpath );
|
|
}
|
|
static const char *getSubPath( time_t *time ) {
|
|
return( Event::getSubPath( localtime( time ) ) );
|
|
}
|
|
|
|
char* getEventFile(void) {
|
|
return video_file;
|
|
}
|
|
|
|
public:
|
|
static int PreAlarmCount() {
|
|
return( pre_alarm_count );
|
|
}
|
|
static void EmptyPreAlarmFrames() {
|
|
if ( pre_alarm_count > 0 ) {
|
|
for ( int i = 0; i < MAX_PRE_ALARM_FRAMES; i++ ) {
|
|
delete pre_alarm_data[i].image;
|
|
delete pre_alarm_data[i].alarm_frame;
|
|
}
|
|
memset( pre_alarm_data, 0, sizeof(pre_alarm_data) );
|
|
}
|
|
pre_alarm_count = 0;
|
|
}
|
|
static void AddPreAlarmFrame( Image *image, struct timeval timestamp, int score=0, Image *alarm_frame=NULL ) {
|
|
pre_alarm_data[pre_alarm_count].image = new Image( *image );
|
|
pre_alarm_data[pre_alarm_count].timestamp = timestamp;
|
|
pre_alarm_data[pre_alarm_count].score = score;
|
|
if ( alarm_frame ) {
|
|
pre_alarm_data[pre_alarm_count].alarm_frame = new Image( *alarm_frame );
|
|
}
|
|
pre_alarm_count++;
|
|
}
|
|
void SavePreAlarmFrames() {
|
|
for ( int i = 0; i < pre_alarm_count; i++ ) {
|
|
AddFrame( pre_alarm_data[i].image, pre_alarm_data[i].timestamp, pre_alarm_data[i].score, pre_alarm_data[i].alarm_frame );
|
|
}
|
|
EmptyPreAlarmFrames();
|
|
}
|
|
};
|
|
|
|
class EventStream : public StreamBase {
|
|
public:
|
|
typedef enum { MODE_SINGLE, MODE_ALL, MODE_ALL_GAPLESS } StreamMode;
|
|
|
|
protected:
|
|
struct FrameData {
|
|
//unsigned long id;
|
|
time_t timestamp;
|
|
time_t offset;
|
|
double delta;
|
|
bool in_db;
|
|
};
|
|
|
|
struct EventData {
|
|
unsigned long event_id;
|
|
unsigned long monitor_id;
|
|
unsigned long frame_count;
|
|
time_t start_time;
|
|
double duration;
|
|
char path[PATH_MAX];
|
|
int n_frames;
|
|
FrameData *frames;
|
|
char video_file[PATH_MAX];
|
|
};
|
|
|
|
protected:
|
|
static const int STREAM_PAUSE_WAIT = 250000; // Microseconds
|
|
|
|
static const StreamMode DEFAULT_MODE = MODE_SINGLE;
|
|
|
|
protected:
|
|
StreamMode mode;
|
|
bool forceEventChange;
|
|
|
|
protected:
|
|
int curr_frame_id;
|
|
double curr_stream_time;
|
|
|
|
EventData *event_data;
|
|
|
|
protected:
|
|
bool loadEventData( int event_id );
|
|
bool loadInitialEventData( int init_event_id, unsigned int init_frame_id );
|
|
bool loadInitialEventData( int monitor_id, time_t event_time );
|
|
|
|
void checkEventLoaded();
|
|
void processCommand( const CmdMsg *msg );
|
|
bool sendFrame( int delta_us );
|
|
|
|
public:
|
|
EventStream() {
|
|
mode = DEFAULT_MODE;
|
|
|
|
forceEventChange = false;
|
|
|
|
curr_frame_id = 0;
|
|
curr_stream_time = 0.0;
|
|
|
|
event_data = 0;
|
|
}
|
|
void setStreamStart( int init_event_id, unsigned int init_frame_id=0 ) {
|
|
loadInitialEventData( init_event_id, init_frame_id );
|
|
loadMonitor( event_data->monitor_id );
|
|
}
|
|
void setStreamStart( int monitor_id, time_t event_time ) {
|
|
loadInitialEventData( monitor_id, event_time );
|
|
loadMonitor( monitor_id );
|
|
}
|
|
void setStreamMode( StreamMode p_mode ) {
|
|
mode = p_mode;
|
|
}
|
|
void runStream();
|
|
};
|
|
|
|
#endif // ZM_EVENT_H
|