1 #include "midi_mapper.h"
3 #include <alsa/asoundlib.h>
7 #include <google/protobuf/descriptor.h>
8 #include <google/protobuf/io/zero_copy_stream_impl.h>
9 #include <google/protobuf/message.h>
10 #include <google/protobuf/text_format.h>
15 #include <sys/eventfd.h>
21 #include "audio_mixer.h"
22 #include "nageru_midi_mapping.pb.h"
23 #include "shared/midi_device.h"
24 #include "shared/midi_mapper_util.h"
25 #include "shared/text_proto.h"
27 using namespace google::protobuf;
29 using namespace std::placeholders;
31 MIDIMapper::MIDIMapper(ControllerReceiver *receiver)
32 : receiver(receiver), mapping_proto(new MIDIMappingProto), midi_device(this)
36 MIDIMapper::~MIDIMapper() {}
38 bool load_midi_mapping_from_file(const string &filename, MIDIMappingProto *new_mapping)
40 return load_proto_from_file(filename, new_mapping);
43 bool save_midi_mapping_to_file(const MIDIMappingProto &mapping_proto, const string &filename)
45 return save_proto_to_file(mapping_proto, filename);
48 void MIDIMapper::set_midi_mapping(const MIDIMappingProto &new_mapping)
50 lock_guard<mutex> lock(mu);
52 mapping_proto->CopyFrom(new_mapping);
54 mapping_proto.reset(new MIDIMappingProto(new_mapping));
57 num_controller_banks = min(max(mapping_proto->num_controller_banks(), 1), 5);
58 current_controller_bank = 0;
60 receiver->clear_all_highlights();
64 void MIDIMapper::start_thread()
66 midi_device.start_thread();
69 const MIDIMappingProto &MIDIMapper::get_current_mapping() const
71 lock_guard<mutex> lock(mu);
72 return *mapping_proto;
75 ControllerReceiver *MIDIMapper::set_receiver(ControllerReceiver *new_receiver)
77 lock_guard<mutex> lock(mu);
78 swap(receiver, new_receiver);
79 return new_receiver; // Now old receiver.
82 void MIDIMapper::controller_received(int controller, int value_int)
84 const float value = map_controller_to_float(controller, value_int);
86 receiver->controller_changed(controller);
88 // Global controllers.
89 match_controller(controller, MIDIMappingBusProto::kLocutFieldNumber, MIDIMappingProto::kLocutBankFieldNumber,
90 value, bind(&ControllerReceiver::set_locut, receiver, _2));
91 match_controller(controller, MIDIMappingBusProto::kLimiterThresholdFieldNumber, MIDIMappingProto::kLimiterThresholdBankFieldNumber,
92 value, bind(&ControllerReceiver::set_limiter_threshold, receiver, _2));
93 match_controller(controller, MIDIMappingBusProto::kMakeupGainFieldNumber, MIDIMappingProto::kMakeupGainBankFieldNumber,
94 value, bind(&ControllerReceiver::set_makeup_gain, receiver, _2));
97 match_controller(controller, MIDIMappingBusProto::kStereoWidthFieldNumber, MIDIMappingProto::kStereoWidthBankFieldNumber,
98 value, bind(&ControllerReceiver::set_stereo_width, receiver, _1, _2));
99 match_controller(controller, MIDIMappingBusProto::kTrebleFieldNumber, MIDIMappingProto::kTrebleBankFieldNumber,
100 value, bind(&ControllerReceiver::set_treble, receiver, _1, _2));
101 match_controller(controller, MIDIMappingBusProto::kMidFieldNumber, MIDIMappingProto::kMidBankFieldNumber,
102 value, bind(&ControllerReceiver::set_mid, receiver, _1, _2));
103 match_controller(controller, MIDIMappingBusProto::kBassFieldNumber, MIDIMappingProto::kBassBankFieldNumber,
104 value, bind(&ControllerReceiver::set_bass, receiver, _1, _2));
105 match_controller(controller, MIDIMappingBusProto::kGainFieldNumber, MIDIMappingProto::kGainBankFieldNumber,
106 value, bind(&ControllerReceiver::set_gain, receiver, _1, _2));
107 match_controller(controller, MIDIMappingBusProto::kCompressorThresholdFieldNumber, MIDIMappingProto::kCompressorThresholdBankFieldNumber,
108 value, bind(&ControllerReceiver::set_compressor_threshold, receiver, _1, _2));
109 match_controller(controller, MIDIMappingBusProto::kFaderFieldNumber, MIDIMappingProto::kFaderBankFieldNumber,
110 value, bind(&ControllerReceiver::set_fader, receiver, _1, _2));
113 void MIDIMapper::note_on_received(int note)
115 lock_guard<mutex> lock(mu);
116 receiver->note_on(note);
118 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
119 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
120 if (bus_mapping.has_prev_bank() &&
121 bus_mapping.prev_bank().note_number() == note) {
122 current_controller_bank = (current_controller_bank + num_controller_banks - 1) % num_controller_banks;
124 update_lights_lock_held();
126 if (bus_mapping.has_next_bank() &&
127 bus_mapping.next_bank().note_number() == note) {
128 current_controller_bank = (current_controller_bank + 1) % num_controller_banks;
130 update_lights_lock_held();
132 if (bus_mapping.has_select_bank_1() &&
133 bus_mapping.select_bank_1().note_number() == note) {
134 current_controller_bank = 0;
136 update_lights_lock_held();
138 if (bus_mapping.has_select_bank_2() &&
139 bus_mapping.select_bank_2().note_number() == note &&
140 num_controller_banks >= 2) {
141 current_controller_bank = 1;
143 update_lights_lock_held();
145 if (bus_mapping.has_select_bank_3() &&
146 bus_mapping.select_bank_3().note_number() == note &&
147 num_controller_banks >= 3) {
148 current_controller_bank = 2;
150 update_lights_lock_held();
152 if (bus_mapping.has_select_bank_4() &&
153 bus_mapping.select_bank_4().note_number() == note &&
154 num_controller_banks >= 4) {
155 current_controller_bank = 3;
157 update_lights_lock_held();
159 if (bus_mapping.has_select_bank_5() &&
160 bus_mapping.select_bank_5().note_number() == note &&
161 num_controller_banks >= 5) {
162 current_controller_bank = 4;
164 update_lights_lock_held();
168 match_button(note, MIDIMappingBusProto::kToggleLocutFieldNumber, MIDIMappingProto::kToggleLocutBankFieldNumber,
169 bind(&ControllerReceiver::toggle_locut, receiver, _1));
170 match_button(note, MIDIMappingBusProto::kToggleAutoGainStagingFieldNumber, MIDIMappingProto::kToggleAutoGainStagingBankFieldNumber,
171 bind(&ControllerReceiver::toggle_auto_gain_staging, receiver, _1));
172 match_button(note, MIDIMappingBusProto::kToggleCompressorFieldNumber, MIDIMappingProto::kToggleCompressorBankFieldNumber,
173 bind(&ControllerReceiver::toggle_compressor, receiver, _1));
174 match_button(note, MIDIMappingBusProto::kClearPeakFieldNumber, MIDIMappingProto::kClearPeakBankFieldNumber,
175 bind(&ControllerReceiver::clear_peak, receiver, _1));
176 match_button(note, MIDIMappingBusProto::kToggleMuteFieldNumber, MIDIMappingProto::kClearPeakBankFieldNumber,
177 bind(&ControllerReceiver::toggle_mute, receiver, _1));
178 match_button(note, MIDIMappingBusProto::kToggleLimiterFieldNumber, MIDIMappingProto::kToggleLimiterBankFieldNumber,
179 bind(&ControllerReceiver::toggle_limiter, receiver));
180 match_button(note, MIDIMappingBusProto::kToggleAutoMakeupGainFieldNumber, MIDIMappingProto::kToggleAutoMakeupGainBankFieldNumber,
181 bind(&ControllerReceiver::toggle_auto_makeup_gain, receiver));
184 void MIDIMapper::update_num_subscribers(unsigned num_subscribers)
186 num_subscribed_ports = num_subscribers;
190 void MIDIMapper::match_controller(int controller, int field_number, int bank_field_number, float value, function<void(unsigned, float)> func)
192 if (bank_mismatch(bank_field_number)) {
196 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
197 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
198 if (match_controller_helper(bus_mapping, field_number, controller)) {
199 func(bus_idx, value);
204 void MIDIMapper::match_button(int note, int field_number, int bank_field_number, function<void(unsigned)> func)
206 if (bank_mismatch(bank_field_number)) {
210 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
211 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
212 if (match_button_helper(bus_mapping, field_number, note)) {
218 bool MIDIMapper::has_active_controller(unsigned bus_idx, int field_number, int bank_field_number)
220 if (bank_mismatch(bank_field_number)) {
224 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
225 const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
226 const Reflection *bus_reflection = bus_mapping.GetReflection();
227 return bus_reflection->HasField(bus_mapping, descriptor);
230 bool MIDIMapper::bank_mismatch(int bank_field_number)
232 return !match_bank_helper(*mapping_proto, bank_field_number, current_controller_bank);
235 void MIDIMapper::refresh_highlights()
237 receiver->clear_all_highlights();
241 void MIDIMapper::refresh_lights()
243 lock_guard<mutex> lock(mu);
244 update_lights_lock_held();
247 void MIDIMapper::update_highlights()
249 if (num_subscribed_ports.load() == 0) {
250 receiver->clear_all_highlights();
254 // Global controllers.
255 bool highlight_locut = false;
256 bool highlight_limiter_threshold = false;
257 bool highlight_makeup_gain = false;
258 bool highlight_toggle_limiter = false;
259 bool highlight_toggle_auto_makeup_gain = false;
260 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
261 if (has_active_controller(
262 bus_idx, MIDIMappingBusProto::kLocutFieldNumber, MIDIMappingProto::kLocutBankFieldNumber)) {
263 highlight_locut = true;
265 if (has_active_controller(
266 bus_idx, MIDIMappingBusProto::kLimiterThresholdFieldNumber, MIDIMappingProto::kLimiterThresholdBankFieldNumber)) {
267 highlight_limiter_threshold = true;
269 if (has_active_controller(
270 bus_idx, MIDIMappingBusProto::kMakeupGainFieldNumber, MIDIMappingProto::kMakeupGainBankFieldNumber)) {
271 highlight_makeup_gain = true;
273 if (has_active_controller(
274 bus_idx, MIDIMappingBusProto::kToggleLimiterFieldNumber, MIDIMappingProto::kToggleLimiterBankFieldNumber)) {
275 highlight_toggle_limiter = true;
277 if (has_active_controller(
278 bus_idx, MIDIMappingBusProto::kToggleAutoMakeupGainFieldNumber, MIDIMappingProto::kToggleAutoMakeupGainBankFieldNumber)) {
279 highlight_toggle_auto_makeup_gain = true;
282 receiver->highlight_locut(highlight_locut);
283 receiver->highlight_limiter_threshold(highlight_limiter_threshold);
284 receiver->highlight_makeup_gain(highlight_makeup_gain);
285 receiver->highlight_toggle_limiter(highlight_toggle_limiter);
286 receiver->highlight_toggle_auto_makeup_gain(highlight_toggle_auto_makeup_gain);
288 // Per-bus controllers.
289 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
290 receiver->highlight_stereo_width(bus_idx, has_active_controller(
291 bus_idx, MIDIMappingBusProto::kStereoWidthFieldNumber, MIDIMappingProto::kStereoWidthBankFieldNumber));
292 receiver->highlight_treble(bus_idx, has_active_controller(
293 bus_idx, MIDIMappingBusProto::kTrebleFieldNumber, MIDIMappingProto::kTrebleBankFieldNumber));
294 receiver->highlight_mid(bus_idx, has_active_controller(
295 bus_idx, MIDIMappingBusProto::kMidFieldNumber, MIDIMappingProto::kMidBankFieldNumber));
296 receiver->highlight_bass(bus_idx, has_active_controller(
297 bus_idx, MIDIMappingBusProto::kBassFieldNumber, MIDIMappingProto::kBassBankFieldNumber));
298 receiver->highlight_gain(bus_idx, has_active_controller(
299 bus_idx, MIDIMappingBusProto::kGainFieldNumber, MIDIMappingProto::kGainBankFieldNumber));
300 receiver->highlight_compressor_threshold(bus_idx, has_active_controller(
301 bus_idx, MIDIMappingBusProto::kCompressorThresholdFieldNumber, MIDIMappingProto::kCompressorThresholdBankFieldNumber));
302 receiver->highlight_fader(bus_idx, has_active_controller(
303 bus_idx, MIDIMappingBusProto::kFaderFieldNumber, MIDIMappingProto::kFaderBankFieldNumber));
304 receiver->highlight_mute(bus_idx, has_active_controller(
305 bus_idx, MIDIMappingBusProto::kToggleMuteFieldNumber, MIDIMappingProto::kToggleMuteBankFieldNumber));
306 receiver->highlight_toggle_locut(bus_idx, has_active_controller(
307 bus_idx, MIDIMappingBusProto::kToggleLocutFieldNumber, MIDIMappingProto::kToggleLocutBankFieldNumber));
308 receiver->highlight_toggle_auto_gain_staging(bus_idx, has_active_controller(
309 bus_idx, MIDIMappingBusProto::kToggleAutoGainStagingFieldNumber, MIDIMappingProto::kToggleAutoGainStagingBankFieldNumber));
310 receiver->highlight_toggle_compressor(bus_idx, has_active_controller(
311 bus_idx, MIDIMappingBusProto::kToggleCompressorFieldNumber, MIDIMappingProto::kToggleCompressorBankFieldNumber));
315 void MIDIMapper::update_lights_lock_held()
317 if (global_audio_mixer == nullptr) {
321 map<MIDIDevice::LightKey, uint8_t> active_lights; // Desired state.
322 if (current_controller_bank == 0) {
323 activate_lights_all_buses(MIDIMappingBusProto::kBank1IsSelectedFieldNumber, &active_lights);
325 if (current_controller_bank == 1) {
326 activate_lights_all_buses(MIDIMappingBusProto::kBank2IsSelectedFieldNumber, &active_lights);
328 if (current_controller_bank == 2) {
329 activate_lights_all_buses(MIDIMappingBusProto::kBank3IsSelectedFieldNumber, &active_lights);
331 if (current_controller_bank == 3) {
332 activate_lights_all_buses(MIDIMappingBusProto::kBank4IsSelectedFieldNumber, &active_lights);
334 if (current_controller_bank == 4) {
335 activate_lights_all_buses(MIDIMappingBusProto::kBank5IsSelectedFieldNumber, &active_lights);
337 if (global_audio_mixer->get_limiter_enabled()) {
338 activate_lights_all_buses(MIDIMappingBusProto::kLimiterIsOnFieldNumber, &active_lights);
340 if (global_audio_mixer->get_final_makeup_gain_auto()) {
341 activate_lights_all_buses(MIDIMappingBusProto::kAutoMakeupGainIsOnFieldNumber, &active_lights);
343 unsigned num_buses = min<unsigned>(global_audio_mixer->num_buses(), mapping_proto->bus_mapping_size());
344 for (unsigned bus_idx = 0; bus_idx < num_buses; ++bus_idx) {
345 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
346 if (global_audio_mixer->get_mute(bus_idx)) {
347 activate_mapped_light(bus_mapping, MIDIMappingBusProto::kIsMutedFieldNumber, &active_lights);
349 if (global_audio_mixer->get_locut_enabled(bus_idx)) {
350 activate_mapped_light(bus_mapping, MIDIMappingBusProto::kLocutIsOnFieldNumber, &active_lights);
352 if (global_audio_mixer->get_gain_staging_auto(bus_idx)) {
353 activate_mapped_light(bus_mapping, MIDIMappingBusProto::kAutoGainStagingIsOnFieldNumber, &active_lights);
355 if (global_audio_mixer->get_compressor_enabled(bus_idx)) {
356 activate_mapped_light(bus_mapping, MIDIMappingBusProto::kCompressorIsOnFieldNumber, &active_lights);
358 if (has_peaked[bus_idx]) {
359 activate_mapped_light(bus_mapping, MIDIMappingBusProto::kHasPeakedFieldNumber, &active_lights);
363 midi_device.update_lights(active_lights);
366 void MIDIMapper::activate_lights_all_buses(int field_number, map<MIDIDevice::LightKey, uint8_t> *active_lights)
368 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
369 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
370 activate_mapped_light(bus_mapping, field_number, active_lights);