DemoManMonitor/DemoManMonitor.cpp at master · tdicola/DemoManMonitor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
// Demolition Man Verbal Morality Statute Monitor
// Created by Tony DiCola (tony@tonydicola.com)
// Released under an MIT license (http://opensource.org/licenses/MIT).

#include "DemoManMonitor.h"

#include <iostream>
#include <unistd.h>

using namespace std;

DemoManMonitor::DemoManMonitor(size_t bufferSize,
							   Adafruit_Thermal* printer,
							   AudioSource* audioSource,
							   AudioSink* audioSink,
							   KeywordSpotter* spotter,
							   std::vector<uint8_t>* alarm,
							   std::function<void(bool enable)> light):
	_printer(printer),
	_audioSource(audioSource),
	_audioSink(audioSink),
	_spotter(spotter),
	_alarm(alarm),
	_buffer(bufferSize),
	_ticketSteps(),
	_quietMode(false),
	_light(light)
{
	setTicketSteps();
}

DemoManMonitor::~DemoManMonitor()
{}

void DemoManMonitor::update() {
	// Grab a buffer of audio.
	if (_audioSource->record(_buffer)) {
		// Look for a keyword.
		string keyword = _spotter->process(_buffer);
		if (keyword != "") {
			// Keyword was spotted, sound alarm.
			cout << "KEYWORD SPOTTED: " << keyword << endl;
			raiseAlarm(keyword);
		}
	}
	else {
		// There was a problem getting data from the microphone.
		// Print an error and try again with next update.
		cerr << "Failed to get a full buffer of microphone data!" << endl;
	}
}

// Define a macro for adding a ticket step in the setTicketSteps
// function further below.
#define ADD_TICKET_STEP(command) \
  { _ticketSteps.push_back([](Adafruit_Thermal* printer) { \
  	command; \
  }); }

void DemoManMonitor::setTicketSteps() {
	// Define the steps for printing the fine ticket.
	// Because each printer command will block until the
	// printer is ready, an array of commands is defined
	// as a continuation that will allow execution of
	// printer commands while audio plays.
	_ticketSteps.clear();
	ADD_TICKET_STEP(printer->upsideDownOn());
	ADD_TICKET_STEP(printer->justify('C'));
	ADD_TICKET_STEP(printer->doubleWidthOn());
	ADD_TICKET_STEP(printer->boldOn());
	ADD_TICKET_STEP(printer->println("***** FINE *****"));
	ADD_TICKET_STEP(printer->boldOff());
	ADD_TICKET_STEP(printer->doubleWidthOff());
	ADD_TICKET_STEP(printer->feed(2));
	ADD_TICKET_STEP(printer->println("YOU ARE FINED:"));
	ADD_TICKET_STEP(printer->feed(1));
	ADD_TICKET_STEP(printer->doubleWidthOn());
	ADD_TICKET_STEP(printer->boldOn());
	ADD_TICKET_STEP(printer->println("ONE CREDIT"));
	ADD_TICKET_STEP(printer->boldOff());
	ADD_TICKET_STEP(printer->doubleWidthOff());
	ADD_TICKET_STEP(printer->feed(1));
	ADD_TICKET_STEP(printer->println("FOR VIOLATION OF THE"));
	ADD_TICKET_STEP(printer->println("VERBAL MORALITY STATUTES"));
	ADD_TICKET_STEP(printer->feed(2));
	ADD_TICKET_STEP(printer->doubleWidthOn());
	ADD_TICKET_STEP(printer->boldOn());
	ADD_TICKET_STEP(printer->println("***** FINE *****"));
	ADD_TICKET_STEP(printer->feed(6));
}

void DemoManMonitor::raiseAlarm(const std::string& keyword) {
	// Don't play audio or print ticket while in quiet mode.
	if (_quietMode) {
		return;
	}
	// Turn the light on.
	_light(true);
	// Stop audio recording while the alarm is raised.
	_audioSource->pause();
	// Play audio and print the ticket at the same time.
	// Since the Pi only has one core and timing is somewhat
	// critical (for smooth audio playback), a tight loop to
	// update audio and ticket printing state will be executed.
	size_t step = 0;
	_audioSink->resume();
	_audioSink->playAsync(*_alarm);
	bool playing = true;
	while (playing || step < _ticketSteps.size()) {
		// Check if the printer is ready for a new command.
		if (step < _ticketSteps.size() && _printer->ready()) {
			// Execute the current step and increment to the next one.
			_ticketSteps[step](_printer);
			step++;
		}
		playing = _audioSink->asyncUpdate();
	}
	// Stop audio playback.
	_audioSink->pause();
	// Enable recording again.
	_audioSource->resume();
	// Turn the light off.
	_light(false);
}

void DemoManMonitor::setQuietMode(bool quietMode) {
	_quietMode = quietMode;
}