Monitor.cs

using System;
using System.Collections.Generic;
using System.Text;
using gma.System.Windows;
using System.Windows.Forms;
using System.Drawing;
using System.IO;

namespace Herring
{
    public delegate AppInfo GetAppDelegate(string path);

    class Monitor
    {
        private UserActivityHook actHook;
        private Dictionary<string, AppInfo> apps = new Dictionary<string, AppInfo>();

        private DateTime begin;
        private StringBuilder charsTyped;
        private int keyPressCount;
        private double mouseDistance;
        private int mouseClicks;
        private Point prevMouseLoc;

        public Monitor()
        {
            Reset(DateTime.Now);

            actHook = new UserActivityHook();
            actHook.OnMouseActivity += MouseMoved;
            actHook.KeyDown         += KeyDown;            
            actHook.KeyPress        += KeyPressed;
        }

        public void Start()
        {
            actHook.Start();
        }

        public void MouseMoved(object sender, MouseEventArgs e)
        {
            // Distance
            if (prevMouseLoc.X != -1)
            {
                double dx = e.Location.X - prevMouseLoc.X;
                double dy = e.Location.Y - prevMouseLoc.Y;
                double d = Math.Sqrt(dx * dx + dy * dy);
                mouseDistance += d;
            }
            prevMouseLoc = e.Location;

            // Clicks
            mouseClicks += e.Clicks / 2;
        }

        public void KeyDown(object sender, KeyEventArgs e)
        {
            keyPressCount++;
        }

        public void KeyPressed(object sender, KeyPressEventArgs e)
        {
            if (e.KeyChar < ' ')        // \t, \r, \n etc.
            {
                if (e.KeyChar == '\b')  // backspace
                {
                    if (charsTyped.Length >= 1)
                        charsTyped.Remove(charsTyped.Length - 1, 1);
                }
                else
                {
                    charsTyped.Append(' ');
                }
            }
            else
            {
                charsTyped.Append(e.KeyChar);
            }
        }

        // Takes "now" for strictly precise measurements
        public void Reset(DateTime now)
        {
            begin = now;
            charsTyped = new StringBuilder();
            keyPressCount = 0;
            prevMouseLoc = new Point(-1, -1);
            mouseDistance = 0;
            mouseClicks = 0;
        }

        public AppInfo GetApp(string path)
        {
            AppInfo appInfo;
            if (apps.ContainsKey(path))
            {
                appInfo = apps[path];
            }
            else
            {
                Icon icon = ShellIcon.GetSmallIcon(path);
                appInfo =
                    new AppInfo
                    {
                        Name = Path.GetFileName(path),
                        Path = path,
                        Icon = icon,
                        ColorBins = IconAnalyser.GetColors(icon)
                    };
                apps.Add(path, appInfo);
            }
            return appInfo;
        }

        public ActivitySnapshot GetActivitySnapshot()
        {
            string path = SystemInfo.GetTopWindowPath();
            string windowTitle, applicationTitle;
            SystemInfo.GetTopWindowText(out windowTitle, out applicationTitle);

            // Prepare AppInfo object
            AppInfo appInfo = GetApp(path);

            string doc = ""; ;
            if (appInfo.Name == "chrome.exe")
            {
                doc = SystemInfo.GetChromeUrl();
                if (doc == "(Incognito)")
                {
                    applicationTitle = "(Incognito)";
                    windowTitle = "";
                }
            }


            // Measure the time span since the previous snapshot
            DateTime end = DateTime.Now;
            TimeSpan length = end - begin;

            // Create an activity data
            ActivitySnapshot data =
                new ActivitySnapshot
                {
                    Time = begin,
                    App = appInfo,
                    ApplicationTitle = applicationTitle,
                    WindowTitle = windowTitle,
                    DocumentUrl = doc,
                    KeyboardIntensity = ActivitySnapshot.GetIntensity(keyPressCount, 6, length),
                    MouseIntensity = ActivitySnapshot.GetIntensity((int)mouseDistance, 1000, length)
                };

            Reset(end);

            return data;
        }

    }
}