337 lines
9.1 KiB
C++
337 lines
9.1 KiB
C++
//
|
|
// ZoneMinder Event Class Implementation, $Date$, $Revision$
|
|
// Copyright (C) 2003 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
//
|
|
|
|
#include <fcntl.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <sys/uio.h>
|
|
#include <getopt.h>
|
|
|
|
#include "zm.h"
|
|
#include "zm_db.h"
|
|
#include "zm_event.h"
|
|
#include "zm_monitor.h"
|
|
|
|
#include "zmf.h"
|
|
|
|
Event::Event( Monitor *p_monitor, struct timeval p_start_time ) : monitor( p_monitor ), start_time( p_start_time )
|
|
{
|
|
static char sql[256];
|
|
static char start_time_str[32];
|
|
|
|
strftime( start_time_str, sizeof(start_time_str), "%Y-%m-%d %H:%M:%S", localtime( &start_time.tv_sec ) );
|
|
sprintf( sql, "insert into Events ( MonitorId, Name, StartTime ) values ( %d, 'Event', '%s' )", monitor->Id(), start_time_str );
|
|
if ( mysql_query( &dbconn, sql ) )
|
|
{
|
|
Error(( "Can't insert event: %s", mysql_error( &dbconn ) ));
|
|
exit( mysql_errno( &dbconn ) );
|
|
}
|
|
id = mysql_insert_id( &dbconn );
|
|
frames = 0;
|
|
alarm_frames = 0;
|
|
tot_score = 0;
|
|
max_score = 0;
|
|
sprintf( path, ZM_DIR_EVENTS "/%s/%d", monitor->Name(), id );
|
|
|
|
struct stat statbuf;
|
|
errno = 0;
|
|
stat( path, &statbuf );
|
|
if ( errno == ENOENT || errno == ENOTDIR )
|
|
{
|
|
if ( mkdir( path, 0755 ) )
|
|
{
|
|
Error(( "Can't make %s: %s", path, strerror(errno)));
|
|
}
|
|
}
|
|
}
|
|
|
|
Event::~Event()
|
|
{
|
|
static char sql[256];
|
|
static char end_time_str[32];
|
|
|
|
struct DeltaTimeval delta_time;
|
|
DELTA_TIMEVAL( delta_time, end_time, start_time, DT_PREC_2 );
|
|
|
|
strftime( end_time_str, sizeof(end_time_str), "%Y-%m-%d %H:%M:%S", localtime( &end_time.tv_sec ) );
|
|
|
|
sprintf( sql, "update Events set Name='Event-%d', EndTime = '%s', Length = %s%ld.%02ld, Frames = %d, AlarmFrames = %d, TotScore = %d, AvgScore = %d, MaxScore = %d where Id = %d", id, end_time_str, delta_time.positive?"":"-", delta_time.sec, delta_time.fsec, frames, alarm_frames, tot_score, (int)(tot_score/alarm_frames), max_score, id );
|
|
if ( mysql_query( &dbconn, sql ) )
|
|
{
|
|
Error(( "Can't update event: %s", mysql_error( &dbconn ) ));
|
|
exit( mysql_errno( &dbconn ) );
|
|
}
|
|
}
|
|
|
|
int Event::sd = -1;
|
|
|
|
bool Event::OpenFrameSocket( int monitor_id )
|
|
{
|
|
if ( sd > 0 )
|
|
{
|
|
close( sd );
|
|
}
|
|
|
|
sd = socket( AF_UNIX, SOCK_STREAM, 0);
|
|
if ( sd < 0 )
|
|
{
|
|
Error(( "Can't create socket: %s", strerror(errno) ));
|
|
return( false );
|
|
}
|
|
|
|
int flags;
|
|
if ( (flags = fcntl( sd, F_GETFL )) < 0 )
|
|
{
|
|
Error(( "Can't get socket flags, error = %s", strerror(errno) ));
|
|
close( sd );
|
|
sd = -1;
|
|
return( false );
|
|
}
|
|
flags |= O_NONBLOCK;
|
|
if ( fcntl( sd, F_SETFL, flags ) < 0 )
|
|
{
|
|
Error(( "Can't set socket flags, error = %s", strerror(errno) ));
|
|
close( sd );
|
|
sd = -1;
|
|
return( false );
|
|
}
|
|
|
|
char sock_path[PATH_MAX] = "";
|
|
sprintf( sock_path, FILE_SOCK_FILE, monitor_id );
|
|
|
|
struct sockaddr_un addr;
|
|
|
|
strcpy( addr.sun_path, sock_path );
|
|
addr.sun_family = AF_UNIX;
|
|
|
|
if ( connect( sd, (struct sockaddr *)&addr, strlen(addr.sun_path)+sizeof(addr.sun_family)) < 0 )
|
|
{
|
|
Warning(( "Can't connect: %s", strerror(errno) ));
|
|
close( sd );
|
|
sd = -1;
|
|
return( false );
|
|
}
|
|
|
|
Info(( "Opened connection to frame server" ));
|
|
return( true );
|
|
}
|
|
|
|
bool Event::ValidateFrameSocket( int monitor_id )
|
|
{
|
|
if ( sd < 0 )
|
|
{
|
|
return( OpenFrameSocket( monitor_id ) );
|
|
}
|
|
return( true );
|
|
}
|
|
|
|
bool Event::SendFrameImage( const Image *image, bool alarm_frame )
|
|
{
|
|
if ( !ValidateFrameSocket( monitor->Id() ) )
|
|
{
|
|
return( false );
|
|
}
|
|
|
|
static int jpg_buffer_size = 0;
|
|
//static unsigned char jpg_buffer[monitor->CameraWidth()*monitor->CameraHeight()];
|
|
static unsigned char jpg_buffer[ZM_MAX_IMAGE_SIZE];
|
|
|
|
image->EncodeJpeg( jpg_buffer, &jpg_buffer_size );
|
|
|
|
static FrameHeader frame_header;
|
|
|
|
frame_header.event_id = id;
|
|
frame_header.frame_id = frames;
|
|
frame_header.alarm_frame = alarm_frame;
|
|
frame_header.image_length = jpg_buffer_size;
|
|
|
|
struct iovec iovecs[2];
|
|
iovecs[0].iov_base = &frame_header;
|
|
iovecs[0].iov_len = sizeof(frame_header);
|
|
iovecs[1].iov_base = jpg_buffer;
|
|
iovecs[1].iov_len = jpg_buffer_size;
|
|
|
|
if ( writev( sd, iovecs, sizeof(iovecs)/sizeof(*iovecs) ) != sizeof(frame_header)+jpg_buffer_size )
|
|
{
|
|
if ( errno == EAGAIN )
|
|
{
|
|
Warning(( "Blocking write detected" ));
|
|
}
|
|
else
|
|
{
|
|
Error(( "Can't write frame: %s", strerror(errno) ));
|
|
close( sd );
|
|
sd = -1;
|
|
}
|
|
return( false );
|
|
}
|
|
Debug( 1, ( "Wrote frame image", jpg_buffer_size ));
|
|
|
|
return( true );
|
|
}
|
|
|
|
bool Event::WriteFrameImage( const Image *image, const char *event_file, bool alarm_frame )
|
|
{
|
|
if ( !ZM_OPT_FRAME_SERVER || !SendFrameImage( image, alarm_frame) )
|
|
{
|
|
image->WriteJpeg( event_file );
|
|
}
|
|
return( true );
|
|
}
|
|
|
|
void Event::AddFrames( int n_frames, struct timeval **timestamps, const Image **images )
|
|
{
|
|
static char sql[4096];
|
|
strcpy( sql, "insert into Frames ( EventId, FrameId, ImagePath, Delta ) values " );
|
|
for ( int i = 0; i < n_frames; i++ )
|
|
{
|
|
frames++;
|
|
|
|
static char event_file[PATH_MAX];
|
|
sprintf( event_file, "%s/capture-%03d.jpg", path, frames );
|
|
|
|
Debug( 1, ( "Writing pre-capture frame %d", frames ));
|
|
WriteFrameImage( images[i], event_file );
|
|
|
|
struct DeltaTimeval delta_time;
|
|
DELTA_TIMEVAL( delta_time, *(timestamps[i]), start_time, DT_PREC_2 );
|
|
|
|
sprintf( sql+strlen(sql), "( %d, %d, '%s', %s%ld.%02ld ), ", id, frames, event_file, delta_time.positive?"":"-", delta_time.sec, delta_time.fsec );
|
|
}
|
|
|
|
Debug( 1, ( "Adding %d frames to DB", n_frames ));
|
|
*(sql+strlen(sql)-2) = '\0';
|
|
if ( mysql_query( &dbconn, sql ) )
|
|
|
|
{
|
|
Error(( "Can't insert frames: %s", mysql_error( &dbconn ) ));
|
|
exit( mysql_errno( &dbconn ) );
|
|
}
|
|
}
|
|
|
|
void Event::AddFrame( struct timeval timestamp, const Image *image, const Image *alarm_image, unsigned int score )
|
|
{
|
|
frames++;
|
|
|
|
static char event_file[PATH_MAX];
|
|
sprintf( event_file, "%s/capture-%03d.jpg", path, frames );
|
|
|
|
Debug( 1, ( "Writing capture frame %d", frames ));
|
|
WriteFrameImage( image, event_file );
|
|
|
|
struct DeltaTimeval delta_time;
|
|
DELTA_TIMEVAL( delta_time, timestamp, start_time, DT_PREC_2 );
|
|
|
|
Debug( 1, ( "Adding frame %d to DB", frames ));
|
|
static char sql[256];
|
|
sprintf( sql, "insert into Frames ( EventId, FrameId, AlarmFrame, ImagePath, Delta, Score ) values ( %d, %d, %d, '%s', %s%ld.%02ld, %d )", id, frames, alarm_image!=0, event_file, delta_time.positive?"":"-", delta_time.sec, delta_time.fsec, score );
|
|
if ( mysql_query( &dbconn, sql ) )
|
|
{
|
|
Error(( "Can't insert frame: %s", mysql_error( &dbconn ) ));
|
|
exit( mysql_errno( &dbconn ) );
|
|
}
|
|
|
|
if ( alarm_image )
|
|
{
|
|
end_time = timestamp;
|
|
|
|
alarm_frames++;
|
|
|
|
sprintf( event_file, "%s/analyse-%03d.jpg", path, frames );
|
|
|
|
Debug( 1, ( "Writing analysis frame %d", frames ));
|
|
WriteFrameImage( alarm_image, event_file, true );
|
|
|
|
tot_score += score;
|
|
if ( score > max_score )
|
|
max_score = score;
|
|
}
|
|
}
|
|
|
|
void Event::StreamEvent( const char *path, int event_id, unsigned long refresh, FILE *fd )
|
|
{
|
|
static char sql[256];
|
|
sprintf( sql, "select Id, EventId, ImagePath, Delta from Frames where EventId = %d order by Id", event_id );
|
|
if ( mysql_query( &dbconn, sql ) )
|
|
{
|
|
Error(( "Can't run query: %s", mysql_error( &dbconn ) ));
|
|
exit( mysql_errno( &dbconn ) );
|
|
}
|
|
|
|
MYSQL_RES *result = mysql_store_result( &dbconn );
|
|
if ( !result )
|
|
{
|
|
Error(( "Can't use query result: %s", mysql_error( &dbconn ) ));
|
|
exit( mysql_errno( &dbconn ) );
|
|
}
|
|
|
|
setbuf( fd, 0 );
|
|
fprintf( fd, "Server: ZoneMinder Stream Server\r\n" );
|
|
fprintf( fd, "Pragma: no-cache\r\n" );
|
|
fprintf( fd, "Cache-Control: no-cache\r\n" );
|
|
fprintf( fd, "Expires: Thu, 01 Dec 1994 16:00:00 GMT\r\n" );
|
|
fprintf( fd, "Content-Type: multipart/x-mixed-replace;boundary=ZoneMinderFrame\r\n\r\n" );
|
|
fprintf( fd, "--ZoneMinderFrame\n" );
|
|
|
|
int n_frames = mysql_num_rows( result );
|
|
Info(( "Got %d frames", n_frames ));
|
|
FILE *fdj = NULL;
|
|
int n_bytes = 0;
|
|
static unsigned char buffer[400000];
|
|
double last_delta = 0;
|
|
for( int i = 0; MYSQL_ROW dbrow = mysql_fetch_row( result ); i++ )
|
|
{
|
|
if ( refresh < 0 )
|
|
{
|
|
if ( i )
|
|
{
|
|
usleep( (int)((1000000*(atof(dbrow[3])-last_delta))/abs(refresh)) );
|
|
}
|
|
last_delta = atof(dbrow[3]);
|
|
}
|
|
char filepath[PATH_MAX];
|
|
sprintf( filepath, "%s/%s", path, dbrow[2] );
|
|
if ( (fdj = fopen( filepath, "r" )) )
|
|
{
|
|
fprintf( fd, "Content-type: image/jpg\n\n" );
|
|
while ( (n_bytes = fread( buffer, 1, sizeof(buffer), fdj )) )
|
|
{
|
|
fwrite( buffer, 1, n_bytes, fd );
|
|
}
|
|
fprintf( fd, "\n--ZoneMinderFrame\n" );
|
|
fclose( fdj );
|
|
}
|
|
else
|
|
{
|
|
Error(( "Can't open %s: %s", filepath, strerror(errno) ));
|
|
}
|
|
if ( refresh > 0 )
|
|
{
|
|
usleep( refresh*1000 );
|
|
}
|
|
}
|
|
if ( mysql_errno( &dbconn ) )
|
|
{
|
|
Error(( "Can't fetch row: %s", mysql_error( &dbconn ) ));
|
|
exit( mysql_errno( &dbconn ) );
|
|
}
|
|
// Yadda yadda
|
|
mysql_free_result( result );
|
|
}
|