zoneminder/src/zm_event.cpp

//
// ZoneMinder Event Class Implementation, $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.
//

#include <fcntl.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/un.h>
#include <sys/uio.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <getopt.h>
#include <arpa/inet.h>
#include <glob.h>

#include "zm.h"
#include "zm_db.h"
#include "zm_time.h"
#include "zm_signal.h"
#include "zm_event.h"
#include "zm_monitor.h"

//#define USE_PREPARED_SQL 1

const char * Event::frame_type_names[3] = { "Normal", "Bulk", "Alarm" };

int Event::pre_alarm_count = 0;

Event::PreAlarmData Event::pre_alarm_data[MAX_PRE_ALARM_FRAMES] = { { 0 } };

Event::Event( Monitor *p_monitor, struct timeval p_start_time, const std::string &p_cause, const StringSetMap &p_noteSetMap ) :
  monitor( p_monitor ),
  start_time( p_start_time ),
  cause( p_cause ),
  noteSetMap( p_noteSetMap )
{

  std::string notes;
  createNotes( notes );

  bool untimedEvent = false;
  if ( !start_time.tv_sec ) {
    untimedEvent = true;
    gettimeofday( &start_time, 0 );
  }

  Storage * storage = monitor->getStorage();

  unsigned int state_id = 0;
  zmDbRow dbrow;
  if ( dbrow.fetch("SELECT Id FROM States WHERE IsActive=1") ) {
    state_id = atoi(dbrow[0]);
  }

  static char sql[ZM_SQL_MED_BUFSIZ];
  snprintf( sql, sizeof(sql), "INSERT INTO Events ( MonitorId, StorageId, Name, StartTime, Width, Height, Cause, Notes, StateId, Orientation, Videoed, DefaultVideo, SaveJPEGs ) values ( %d, %d, 'New Event', from_unixtime( %ld ), %d, %d, '%s', '%s', %d, %d, %d, '', %d )",
      monitor->Id(), 
      storage->Id(),
      start_time.tv_sec,
      monitor->Width(),
      monitor->Height(),
      cause.c_str(),
      notes.c_str(), 
      state_id,
      monitor->getOrientation(),
      ( monitor->GetOptVideoWriter() != 0 ? 1 : 0 ),
      monitor->GetOptSaveJPEGs()
      );
  if ( mysql_query( &dbconn, sql ) ) {
    Error( "Can't insert event: %s. sql was (%s)", mysql_error( &dbconn ), sql );
    exit( mysql_errno( &dbconn ) );
  }
  id = mysql_insert_id( &dbconn );
  if ( untimedEvent ) {
    Warning( "Event %d has zero time, setting to current", id );
  }
  end_time.tv_sec = 0;
  frames = 0;
  alarm_frames = 0;
  tot_score = 0;
  max_score = 0;
  have_video_keyframe = false;

  struct stat statbuf;
  char id_file[PATH_MAX];
  struct tm *stime = localtime( &start_time.tv_sec );

  if ( config.use_deep_storage ) {
    char *path_ptr = path;
    path_ptr += snprintf( path_ptr, sizeof(path), "%s/%d", storage->Path(), monitor->Id() );

    int dt_parts[6];
    dt_parts[0] = stime->tm_year-100;
    dt_parts[1] = stime->tm_mon+1;
    dt_parts[2] = stime->tm_mday;
    dt_parts[3] = stime->tm_hour;
    dt_parts[4] = stime->tm_min;
    dt_parts[5] = stime->tm_sec;

    char date_path[PATH_MAX] = "";
    char time_path[PATH_MAX] = "";
    char *time_path_ptr = time_path;
    for ( unsigned int i = 0; i < sizeof(dt_parts)/sizeof(*dt_parts); i++ ) {
      path_ptr += snprintf( path_ptr, sizeof(path)-(path_ptr-path), "/%02d", dt_parts[i] );

      errno = 0;
      // Do we really need to stat it?  Perhaps we could do that on error, instead
      if ( stat( path, &statbuf ) ) {
        if ( errno == ENOENT || errno == ENOTDIR ) {
          if ( mkdir( path, 0755 ) ) {
            // FIXME This should not be fatal.  Should probably move to a different storage area.
            Fatal( "Can't mkdir %s: %s", path, strerror(errno));
          }
        } else {
          Warning( "Error stat'ing %s, may be fatal. error is %s", path, strerror(errno));
        }
      }
      if ( i == 2 )
        strncpy( date_path, path, sizeof(date_path) );
      else if ( i >= 3 )
        time_path_ptr += snprintf( time_path_ptr, sizeof(time_path)-(time_path_ptr-time_path), "%s%02d", i>3?"/":"", dt_parts[i] );
    }
    // Create event id symlink
    snprintf( id_file, sizeof(id_file), "%s/.%d", date_path, id );
    if ( symlink( time_path, id_file ) < 0 )
      Error( "Can't symlink %s -> %s: %s", id_file, path, strerror(errno));
  } else {
    // Shallow Storage
    snprintf( path, sizeof(path), "%s/%d/%d", storage->Path(), monitor->Id(), id );

    errno = 0;
    stat( path, &statbuf );
    if ( errno == ENOENT || errno == ENOTDIR ) {
      if ( mkdir( path, 0755 ) ) {
        Error( "Can't mkdir %s: %s", path, strerror(errno));
      }
    }

    // Create empty id tag file
    snprintf( id_file, sizeof(id_file), "%s/.%d", path, id );
    if ( FILE *id_fp = fopen( id_file, "w" ) )
      fclose( id_fp );
    else
      Error( "Can't fopen %s: %s", id_file, strerror(errno));
  } // deep storage or not

  last_db_frame = 0;

  video_name[0] = 0;

  /* Save as video */

  if ( monitor->GetOptVideoWriter() != 0 ) {
    std::string container = monitor->OutputContainer();
    if ( container == "auto" || container == "" ) {
      if ( monitor->OutputCodec() == "h264" ) {
        container = "mp4";
      } else {
        container = "mkv";
      }
    }
        
    snprintf( video_name, sizeof(video_name), "%d-%s.%s", id, "video", container.c_str() );
    snprintf( video_file, sizeof(video_file), staticConfig.video_file_format, path, video_name );
    Debug(1,"Writing video file to %s", video_file );
    Camera * camera = monitor->getCamera();
    videoStore = new VideoStore(
        video_file,
        container.c_str(),
        camera->get_VideoStream(),
        ( monitor->RecordAudio() ? camera->get_AudioStream() : NULL ),
        monitor );

    if ( ! videoStore->open() ) {
      delete videoStore;
      videoStore = NULL;
    } 
  }

} // Event::Event( Monitor *p_monitor, struct timeval p_start_time, const std::string &p_cause, const StringSetMap &p_noteSetMap, bool p_videoEvent )

Event::~Event() {
  static char sql[ZM_SQL_MED_BUFSIZ];
  struct DeltaTimeval delta_time;
  DELTA_TIMEVAL( delta_time, end_time, start_time, DT_PREC_2 );

  if ( frames > last_db_frame ) {
    Debug( 1, "Adding closing frame %d to DB", frames );
    snprintf( sql, sizeof(sql), 
        "insert into Frames ( EventId, FrameId, TimeStamp, Delta ) values ( %d, %d, from_unixtime( %ld ), %s%ld.%02ld )",
        id, frames, end_time.tv_sec, delta_time.positive?"":"-", delta_time.sec, delta_time.fsec );
    if ( mysql_query( &dbconn, sql ) ) {
      Error( "Can't insert frame: %s", mysql_error( &dbconn ) );
      exit( mysql_errno( &dbconn ) );
    }
  }

  /* Close the video file */
  if ( videoStore ) {
    delete videoStore;
    videoStore = NULL;
  }

  snprintf( sql, sizeof(sql), "UPDATE Events SET Name='%s%d', EndTime = from_unixtime( %ld ), Length = %s%ld.%02ld, Frames = %d, AlarmFrames = %d, TotScore = %d, AvgScore = %d, MaxScore = %d, DefaultVideo = '%s' where Id = %d", monitor->EventPrefix(), id, end_time.tv_sec, delta_time.positive?"":"-", delta_time.sec, delta_time.fsec, frames, alarm_frames, tot_score, (int)(alarm_frames?(tot_score/alarm_frames):0), max_score, video_name, id );
  if ( mysql_query( &dbconn, sql ) ) {
    Error( "Can't update event: %s", mysql_error( &dbconn ) );
    exit( mysql_errno( &dbconn ) );
  }
} // ~Event

void Event::createNotes( std::string &notes ) {
  notes.clear();
  for ( StringSetMap::const_iterator mapIter = noteSetMap.begin(); mapIter != noteSetMap.end(); ++mapIter ) {
    notes += mapIter->first;
    notes += ": ";
    const StringSet &stringSet = mapIter->second;
    for ( StringSet::const_iterator setIter = stringSet.begin(); setIter != stringSet.end(); ++setIter ) {
      if ( setIter != stringSet.begin() )
        notes += ", ";
      notes += *setIter;
    }
  }
}

int Event::sd = -1;

bool Event::WriteFrameImage( Image *image, struct timeval timestamp, const char *event_file, bool alarm_frame ) {

  int thisquality = ( alarm_frame && (config.jpeg_alarm_file_quality > config.jpeg_file_quality) ) ? config.jpeg_alarm_file_quality : 0 ;   // quality to use, zero is default
  bool rc;
Debug(3, "Writing image to %s", event_file );

  if ( !config.timestamp_on_capture ) {
    // stash the image we plan to use in another pointer regardless if timestamped.
    Image *ts_image = new Image(*image);
    monitor->TimestampImage( ts_image, &timestamp );
    rc = ts_image->WriteJpeg( event_file, thisquality, (monitor->Exif() ? timestamp : (timeval){0,0}) ); // exif is only timestamp at present this switches on or off for write
    delete(ts_image);
  } else {
    rc = image->WriteJpeg( event_file, thisquality, (monitor->Exif() ? timestamp : (timeval){0,0}) ); // exif is only timestamp at present this switches on or off for write
  }

  return rc;
} // end Event::WriteFrameImage( Image *image, struct timeval timestamp, const char *event_file, bool alarm_frame )

bool Event::WritePacket( ZMPacket &packet ) {
  
  if ( videoStore->writePacket( &packet ) < 0 )
    return false;
  return true;
}

void Event::updateNotes( const StringSetMap &newNoteSetMap ) {
  bool update = false;

  //Info( "Checking notes, %d <> %d", noteSetMap.size(), newNoteSetMap.size() );
  if ( newNoteSetMap.size() > 0 ) {
    if ( noteSetMap.size() == 0 ) {
      noteSetMap = newNoteSetMap;
      update = true;
    } else {
      for ( StringSetMap::const_iterator newNoteSetMapIter = newNoteSetMap.begin(); newNoteSetMapIter != newNoteSetMap.end(); ++newNoteSetMapIter ) {
        const std::string &newNoteGroup = newNoteSetMapIter->first;
        const StringSet &newNoteSet = newNoteSetMapIter->second;
        //Info( "Got %d new strings", newNoteSet.size() );
        if ( newNoteSet.size() > 0 ) {
          StringSetMap::iterator noteSetMapIter = noteSetMap.find( newNoteGroup );
          if ( noteSetMapIter == noteSetMap.end() ) {
            //Info( "Can't find note group %s, copying %d strings", newNoteGroup.c_str(), newNoteSet.size() );
            noteSetMap.insert( StringSetMap::value_type( newNoteGroup, newNoteSet ) );
            update = true;
          } else {
            StringSet &noteSet = noteSetMapIter->second;
            //Info( "Found note group %s, got %d strings", newNoteGroup.c_str(), newNoteSet.size() );
            for ( StringSet::const_iterator newNoteSetIter = newNoteSet.begin(); newNoteSetIter != newNoteSet.end(); ++newNoteSetIter ) {
              const std::string &newNote = *newNoteSetIter;
              StringSet::iterator noteSetIter = noteSet.find( newNote );
              if ( noteSetIter == noteSet.end() ) {
                noteSet.insert( newNote );
                update = true;
              }
            } // end for
          } // end if ( noteSetMap.size() == 0
        } // end if newNoteSetupMap.size() > 0
      } // end foreach newNoteSetMap
    } // end if have old notes
  } // end if have new notes

  if ( update ) {
    std::string notes;
    createNotes( notes );

    Debug( 2, "Updating notes for event %d, '%s'", id, notes.c_str() );
    static char sql[ZM_SQL_MED_BUFSIZ];
#if USE_PREPARED_SQL
    static MYSQL_STMT *stmt = 0;

    char notesStr[ZM_SQL_MED_BUFSIZ] = "";
    unsigned long notesLen = 0;

    if ( !stmt ) {
      const char *sql = "update Events set Notes = ? where Id = ?";

      stmt = mysql_stmt_init( &dbconn );
      if ( mysql_stmt_prepare( stmt, sql, strlen(sql) ) ) {
        Fatal( "Unable to prepare sql '%s': %s", sql, mysql_stmt_error(stmt) );
      }

      /* Get the parameter count from the statement */
      if ( mysql_stmt_param_count( stmt ) != 2 ) {
        Fatal( "Unexpected parameter count %ld in sql '%s'", mysql_stmt_param_count( stmt ), sql );
      }

      MYSQL_BIND  bind[2];
      memset(bind, 0, sizeof(bind));

      /* STRING PARAM */
      bind[0].buffer_type = MYSQL_TYPE_STRING;
      bind[0].buffer = (char *)notesStr;
      bind[0].buffer_length = sizeof(notesStr);
      bind[0].is_null = 0;
      bind[0].length = &notesLen;

      bind[1].buffer_type= MYSQL_TYPE_LONG;
      bind[1].buffer= (char *)&id;
      bind[1].is_null= 0;
      bind[1].length= 0;

      /* Bind the buffers */
      if ( mysql_stmt_bind_param( stmt, bind ) ) {
        Fatal( "Unable to bind sql '%s': %s", sql, mysql_stmt_error(stmt) );
      }
    }

    strncpy( notesStr, notes.c_str(), sizeof(notesStr) );

    if ( mysql_stmt_execute( stmt ) ) {
      Fatal( "Unable to execute sql '%s': %s", sql, mysql_stmt_error(stmt) );
    }
#else
    static char escapedNotes[ZM_SQL_MED_BUFSIZ];

    mysql_real_escape_string( &dbconn, escapedNotes, notes.c_str(), notes.length() );

    snprintf( sql, sizeof(sql), "update Events set Notes = '%s' where Id = %d", escapedNotes, id );
    if ( mysql_query( &dbconn, sql ) ) {
      Error( "Can't insert event: %s", mysql_error( &dbconn ) );
    }
#endif
  }
}

void Event::AddFrames( int n_frames, Image **images, struct timeval **timestamps ) {
  for (int i = 0; i < n_frames; i += ZM_SQL_BATCH_SIZE) {
    AddFramesInternal(n_frames, i, images, timestamps);
  }
}

void Event::AddFramesInternal( int n_frames, int start_frame, Image **images, struct timeval **timestamps ) {
  static char sql[ZM_SQL_LGE_BUFSIZ];
  strncpy( sql, "insert into Frames ( EventId, FrameId, TimeStamp, Delta ) values ", sizeof(sql) );
  int frameCount = 0;
  for ( int i = start_frame; i < n_frames && i - start_frame < ZM_SQL_BATCH_SIZE; i++ ) {
    if ( timestamps[i]->tv_sec <= 0 ) {
      Debug( 1, "Not adding pre-capture frame %d, zero or less than 0 timestamp", i );
      continue;
    } else if ( timestamps[i]->tv_sec < 0 ) {
      Warning( "Not adding pre-capture frame %d, negative timestamp", i );
      continue;
    } else {
      Debug( 3, "Adding pre-capture frame %d, timestamp = (%d), start_time=(%d)", i, timestamps[i]->tv_sec, start_time.tv_sec );
    }

    frames++;

    static char event_file[PATH_MAX];
    snprintf( event_file, sizeof(event_file), staticConfig.capture_file_format, path, frames );
    if ( monitor->GetOptSaveJPEGs() & 4 ) {
      //If this is the first frame, we should add a thumbnail to the event directory
      // ICON: We are working through the pre-event frames so this snapshot won't 
      // neccessarily be of the motion.  But some events are less than 10 frames, 
      // so I am changing this to 1, but we should overwrite it later with a better snapshot.
      if ( frames == 1 ) {
        std::string snapshot_file = std::string(path) + "/snapshot.jpg";
        WriteFrameImage( images[i], *(timestamps[i]), snapshot_file.c_str() );
      }
    }
    if ( monitor->GetOptSaveJPEGs() & 1 ) {
      Debug( 1, "Writing pre-capture frame %d", frames );
      WriteFrameImage( images[i], *(timestamps[i]), event_file );
    }

    struct DeltaTimeval delta_time;
    DELTA_TIMEVAL( delta_time, *(timestamps[i]), start_time, DT_PREC_2 );

    int sql_len = strlen(sql);
    snprintf( sql+sql_len, sizeof(sql)-sql_len, "( %d, %d, from_unixtime(%ld), %s%ld.%02ld ), ", id, frames, timestamps[i]->tv_sec, delta_time.positive?"":"-", delta_time.sec, delta_time.fsec );

    frameCount++;
  }

  if ( frameCount ) {
    Debug( 1, "Adding %d/%d frames to DB", frameCount, n_frames );
    *(sql+strlen(sql)-2) = '\0';
    if ( mysql_query( &dbconn, sql ) ) {
      Error( "Can't insert frames: %s, sql was (%s)", mysql_error( &dbconn ), sql );
      exit( mysql_errno( &dbconn ) );
    }
    last_db_frame = frames;
  } else {
    Debug( 1, "No valid pre-capture frames to add" );
  }
}

void Event::AddPacket( ZMPacket *packet, int score, Image *alarm_image ) {

  have_video_keyframe = have_video_keyframe || ( ( packet->codec_type == AVMEDIA_TYPE_VIDEO ) && packet->keyframe );
  if ( videoStore ) {
    if ( have_video_keyframe )  {
      videoStore->writePacket( packet );
    } else {
      Debug(2, "No video keyframe yet, not writing");
    }
    //FIXME if it fails, we should write a jpeg
  } else {
    Debug(2,"AddPacket but no videostore?!");
  }
  if ( have_video_keyframe && ( packet->codec_type == AVMEDIA_TYPE_VIDEO ) ) {
    AddFrame( packet->image, *packet->timestamp, score, alarm_image );
  } // end if is video
  return;
}

void Event::AddFrame( Image *image, struct timeval timestamp, int score, Image *alarm_image ) {
  if ( !timestamp.tv_sec ) {
    Debug( 1, "Not adding new frame, zero timestamp" );
    return;
  }

  frames++;

  static char event_file[PATH_MAX];
  snprintf( event_file, sizeof(event_file), staticConfig.capture_file_format, path, frames );

  if ( monitor->GetOptSaveJPEGs() & 4 ) {
    // Only snapshots
    //If this is the first frame, we should add a thumbnail to the event directory
    if ( frames == 10 || frames == 1 ) {
      std::string snapshot_file = std::string(path) + "/snapshot.jpg";
      WriteFrameImage( image, timestamp, snapshot_file.c_str() );
    }
  }
  if ( monitor->GetOptSaveJPEGs() & 1 ) {
    Debug( 1, "Writing capture frame %d to %s", frames, event_file );
    if ( ! WriteFrameImage( image, timestamp, event_file ) ) {
      Error("Failed to write frame image");
    }
  }

  struct DeltaTimeval delta_time;
  DELTA_TIMEVAL( delta_time, timestamp, start_time, DT_PREC_2 );

  FrameType frame_type = score>0?ALARM:(score<0?BULK:NORMAL);
  // < 0 means no motion detection is being done.
  if ( score < 0 )
    score = 0;

  bool db_frame = ( frame_type != BULK ) || ((frames%config.bulk_frame_interval)==0) || !frames;
  if ( db_frame ) {

    Debug( 1, "Adding frame %d of type \"%s\" to DB", frames, Event::frame_type_names[frame_type] );
    static char sql[ZM_SQL_MED_BUFSIZ];
    snprintf( sql, sizeof(sql), "INSERT INTO Frames ( EventId, FrameId, Type, TimeStamp, Delta, Score ) VALUES ( %d, %d, '%s', from_unixtime( %ld ), %s%ld.%02ld, %d )", id, frames, frame_type_names[frame_type], timestamp.tv_sec, delta_time.positive?"":"-", delta_time.sec, delta_time.fsec, score );
    if ( mysql_query(&dbconn, sql) ) {
      Error("Can't insert frame: (%s) reason:%s", sql, mysql_error(&dbconn));
      zmDbClose();
      if ( ! zmDbConnect() )
        exit(mysql_errno(&dbconn));
      if ( mysql_query(&dbconn, sql) ) {
        Error("Can't insert frame: (%s) reason:%s", sql, mysql_error(&dbconn));
        exit(mysql_errno(&dbconn));
      }
    }
    last_db_frame = frames;

    // We are writing a Bulk frame
    if ( frame_type == BULK ) {
      snprintf( sql, sizeof(sql), "update Events set Length = %s%ld.%02ld, Frames = %d, AlarmFrames = %d, TotScore = %d, AvgScore = %d, MaxScore = %d where Id = %d", 
          ( delta_time.positive?"":"-" ),
          delta_time.sec, delta_time.fsec,
          frames, 
          alarm_frames,
          tot_score,
          (int)(alarm_frames?(tot_score/alarm_frames):0),
          max_score,
          id
          );
      if ( mysql_query( &dbconn, sql ) ) {
        Error( "Can't update event: %s", mysql_error( &dbconn ) );
        exit( mysql_errno( &dbconn ) );
      }
    }
  } // end if db_frame

  end_time = timestamp;

  // We are writing an Alarm frame
  if ( frame_type == ALARM ) {
    alarm_frames++;

    tot_score += score;
    if ( score > (int)max_score )
      max_score = score;

    if ( alarm_image ) {
      snprintf( event_file, sizeof(event_file), staticConfig.analyse_file_format, path, frames );

      Debug( 1, "Writing analysis frame %d", frames );
      if ( monitor->GetOptSaveJPEGs() & 2 ) {
        WriteFrameImage(alarm_image, timestamp, event_file, true);
      }
    }
  }
}