zoneminder/src/zm_event.h

//
// 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=0, BULK, ALARM } FrameType;
    static const char * frame_type_names[3];

    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 &notes );

  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 &StartTime() { return( start_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   storage_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();
    Image *getImage();
};

#endif // ZM_EVENT_H