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 "midi_mapping.pb.h"
23 #include "shared/text_proto.h"
25 using namespace google::protobuf;
27 using namespace std::placeholders;
31 double map_controller_to_float(int val)
33 // Slightly hackish mapping so that we can represent exactly 0.0, 0.5 and 1.0.
36 } else if (val >= 127) {
39 return (val + 0.5) / 127.0;
45 MIDIMapper::MIDIMapper(ControllerReceiver *receiver)
46 : receiver(receiver), mapping_proto(new MIDIMappingProto), midi_device(this)
50 MIDIMapper::~MIDIMapper() {}
52 bool load_midi_mapping_from_file(const string &filename, MIDIMappingProto *new_mapping)
54 return load_proto_from_file(filename, new_mapping);
57 bool save_midi_mapping_to_file(const MIDIMappingProto &mapping_proto, const string &filename)
59 return save_proto_to_file(mapping_proto, filename);
62 void MIDIMapper::set_midi_mapping(const MIDIMappingProto &new_mapping)
64 lock_guard<mutex> lock(mu);
66 mapping_proto->CopyFrom(new_mapping);
68 mapping_proto.reset(new MIDIMappingProto(new_mapping));
71 num_controller_banks = min(max(mapping_proto->num_controller_banks(), 1), 5);
72 current_controller_bank = 0;
74 receiver->clear_all_highlights();
78 void MIDIMapper::start_thread()
80 midi_device.start_thread();
83 const MIDIMappingProto &MIDIMapper::get_current_mapping() const
85 lock_guard<mutex> lock(mu);
86 return *mapping_proto;
89 ControllerReceiver *MIDIMapper::set_receiver(ControllerReceiver *new_receiver)
91 lock_guard<mutex> lock(mu);
92 swap(receiver, new_receiver);
93 return new_receiver; // Now old receiver.
96 void MIDIMapper::controller_received(int controller, int value_int)
98 const float value = map_controller_to_float(value_int);
100 receiver->controller_changed(controller);
102 // Global controllers.
103 match_controller(controller, MIDIMappingBusProto::kLocutFieldNumber, MIDIMappingProto::kLocutBankFieldNumber,
104 value, bind(&ControllerReceiver::set_locut, receiver, _2));
105 match_controller(controller, MIDIMappingBusProto::kLimiterThresholdFieldNumber, MIDIMappingProto::kLimiterThresholdBankFieldNumber,
106 value, bind(&ControllerReceiver::set_limiter_threshold, receiver, _2));
107 match_controller(controller, MIDIMappingBusProto::kMakeupGainFieldNumber, MIDIMappingProto::kMakeupGainBankFieldNumber,
108 value, bind(&ControllerReceiver::set_makeup_gain, receiver, _2));
111 match_controller(controller, MIDIMappingBusProto::kStereoWidthFieldNumber, MIDIMappingProto::kStereoWidthBankFieldNumber,
112 value, bind(&ControllerReceiver::set_stereo_width, receiver, _1, _2));
113 match_controller(controller, MIDIMappingBusProto::kTrebleFieldNumber, MIDIMappingProto::kTrebleBankFieldNumber,
114 value, bind(&ControllerReceiver::set_treble, receiver, _1, _2));
115 match_controller(controller, MIDIMappingBusProto::kMidFieldNumber, MIDIMappingProto::kMidBankFieldNumber,
116 value, bind(&ControllerReceiver::set_mid, receiver, _1, _2));
117 match_controller(controller, MIDIMappingBusProto::kBassFieldNumber, MIDIMappingProto::kBassBankFieldNumber,
118 value, bind(&ControllerReceiver::set_bass, receiver, _1, _2));
119 match_controller(controller, MIDIMappingBusProto::kGainFieldNumber, MIDIMappingProto::kGainBankFieldNumber,
120 value, bind(&ControllerReceiver::set_gain, receiver, _1, _2));
121 match_controller(controller, MIDIMappingBusProto::kCompressorThresholdFieldNumber, MIDIMappingProto::kCompressorThresholdBankFieldNumber,
122 value, bind(&ControllerReceiver::set_compressor_threshold, receiver, _1, _2));
123 match_controller(controller, MIDIMappingBusProto::kFaderFieldNumber, MIDIMappingProto::kFaderBankFieldNumber,
124 value, bind(&ControllerReceiver::set_fader, receiver, _1, _2));
127 void MIDIMapper::note_on_received(int note)
129 lock_guard<mutex> lock(mu);
130 receiver->note_on(note);
132 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
133 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
134 if (bus_mapping.has_prev_bank() &&
135 bus_mapping.prev_bank().note_number() == note) {
136 current_controller_bank = (current_controller_bank + num_controller_banks - 1) % num_controller_banks;
138 update_lights_lock_held();
140 if (bus_mapping.has_next_bank() &&
141 bus_mapping.next_bank().note_number() == note) {
142 current_controller_bank = (current_controller_bank + 1) % num_controller_banks;
144 update_lights_lock_held();
146 if (bus_mapping.has_select_bank_1() &&
147 bus_mapping.select_bank_1().note_number() == note) {
148 current_controller_bank = 0;
150 update_lights_lock_held();
152 if (bus_mapping.has_select_bank_2() &&
153 bus_mapping.select_bank_2().note_number() == note &&
154 num_controller_banks >= 2) {
155 current_controller_bank = 1;
157 update_lights_lock_held();
159 if (bus_mapping.has_select_bank_3() &&
160 bus_mapping.select_bank_3().note_number() == note &&
161 num_controller_banks >= 3) {
162 current_controller_bank = 2;
164 update_lights_lock_held();
166 if (bus_mapping.has_select_bank_4() &&
167 bus_mapping.select_bank_4().note_number() == note &&
168 num_controller_banks >= 4) {
169 current_controller_bank = 3;
171 update_lights_lock_held();
173 if (bus_mapping.has_select_bank_5() &&
174 bus_mapping.select_bank_5().note_number() == note &&
175 num_controller_banks >= 5) {
176 current_controller_bank = 4;
178 update_lights_lock_held();
182 match_button(note, MIDIMappingBusProto::kToggleLocutFieldNumber, MIDIMappingProto::kToggleLocutBankFieldNumber,
183 bind(&ControllerReceiver::toggle_locut, receiver, _1));
184 match_button(note, MIDIMappingBusProto::kToggleAutoGainStagingFieldNumber, MIDIMappingProto::kToggleAutoGainStagingBankFieldNumber,
185 bind(&ControllerReceiver::toggle_auto_gain_staging, receiver, _1));
186 match_button(note, MIDIMappingBusProto::kToggleCompressorFieldNumber, MIDIMappingProto::kToggleCompressorBankFieldNumber,
187 bind(&ControllerReceiver::toggle_compressor, receiver, _1));
188 match_button(note, MIDIMappingBusProto::kClearPeakFieldNumber, MIDIMappingProto::kClearPeakBankFieldNumber,
189 bind(&ControllerReceiver::clear_peak, receiver, _1));
190 match_button(note, MIDIMappingBusProto::kToggleMuteFieldNumber, MIDIMappingProto::kClearPeakBankFieldNumber,
191 bind(&ControllerReceiver::toggle_mute, receiver, _1));
192 match_button(note, MIDIMappingBusProto::kToggleLimiterFieldNumber, MIDIMappingProto::kToggleLimiterBankFieldNumber,
193 bind(&ControllerReceiver::toggle_limiter, receiver));
194 match_button(note, MIDIMappingBusProto::kToggleAutoMakeupGainFieldNumber, MIDIMappingProto::kToggleAutoMakeupGainBankFieldNumber,
195 bind(&ControllerReceiver::toggle_auto_makeup_gain, receiver));
198 void MIDIMapper::update_num_subscribers(unsigned num_subscribers)
200 num_subscribed_ports = num_subscribers;
204 void MIDIMapper::match_controller(int controller, int field_number, int bank_field_number, float value, function<void(unsigned, float)> 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);
213 const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
214 const Reflection *bus_reflection = bus_mapping.GetReflection();
215 if (!bus_reflection->HasField(bus_mapping, descriptor)) {
218 const MIDIControllerProto &controller_proto =
219 static_cast<const MIDIControllerProto &>(bus_reflection->GetMessage(bus_mapping, descriptor));
220 if (controller_proto.controller_number() == controller) {
221 func(bus_idx, value);
226 void MIDIMapper::match_button(int note, int field_number, int bank_field_number, function<void(unsigned)> func)
228 if (bank_mismatch(bank_field_number)) {
232 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
233 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
235 const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
236 const Reflection *bus_reflection = bus_mapping.GetReflection();
237 if (!bus_reflection->HasField(bus_mapping, descriptor)) {
240 const MIDIButtonProto &button_proto =
241 static_cast<const MIDIButtonProto &>(bus_reflection->GetMessage(bus_mapping, descriptor));
242 if (button_proto.note_number() == note) {
248 bool MIDIMapper::has_active_controller(unsigned bus_idx, int field_number, int bank_field_number)
250 if (bank_mismatch(bank_field_number)) {
254 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
255 const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
256 const Reflection *bus_reflection = bus_mapping.GetReflection();
257 return bus_reflection->HasField(bus_mapping, descriptor);
260 bool MIDIMapper::bank_mismatch(int bank_field_number)
262 const FieldDescriptor *bank_descriptor = mapping_proto->GetDescriptor()->FindFieldByNumber(bank_field_number);
263 const Reflection *reflection = mapping_proto->GetReflection();
264 return (reflection->HasField(*mapping_proto, bank_descriptor) &&
265 reflection->GetInt32(*mapping_proto, bank_descriptor) != current_controller_bank);
268 void MIDIMapper::refresh_highlights()
270 receiver->clear_all_highlights();
274 void MIDIMapper::refresh_lights()
276 lock_guard<mutex> lock(mu);
277 update_lights_lock_held();
280 void MIDIMapper::update_highlights()
282 if (num_subscribed_ports.load() == 0) {
283 receiver->clear_all_highlights();
287 // Global controllers.
288 bool highlight_locut = false;
289 bool highlight_limiter_threshold = false;
290 bool highlight_makeup_gain = false;
291 bool highlight_toggle_limiter = false;
292 bool highlight_toggle_auto_makeup_gain = false;
293 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
294 if (has_active_controller(
295 bus_idx, MIDIMappingBusProto::kLocutFieldNumber, MIDIMappingProto::kLocutBankFieldNumber)) {
296 highlight_locut = true;
298 if (has_active_controller(
299 bus_idx, MIDIMappingBusProto::kLimiterThresholdFieldNumber, MIDIMappingProto::kLimiterThresholdBankFieldNumber)) {
300 highlight_limiter_threshold = true;
302 if (has_active_controller(
303 bus_idx, MIDIMappingBusProto::kMakeupGainFieldNumber, MIDIMappingProto::kMakeupGainBankFieldNumber)) {
304 highlight_makeup_gain = true;
306 if (has_active_controller(
307 bus_idx, MIDIMappingBusProto::kToggleLimiterFieldNumber, MIDIMappingProto::kToggleLimiterBankFieldNumber)) {
308 highlight_toggle_limiter = true;
310 if (has_active_controller(
311 bus_idx, MIDIMappingBusProto::kToggleAutoMakeupGainFieldNumber, MIDIMappingProto::kToggleAutoMakeupGainBankFieldNumber)) {
312 highlight_toggle_auto_makeup_gain = true;
315 receiver->highlight_locut(highlight_locut);
316 receiver->highlight_limiter_threshold(highlight_limiter_threshold);
317 receiver->highlight_makeup_gain(highlight_makeup_gain);
318 receiver->highlight_toggle_limiter(highlight_toggle_limiter);
319 receiver->highlight_toggle_auto_makeup_gain(highlight_toggle_auto_makeup_gain);
321 // Per-bus controllers.
322 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
323 receiver->highlight_stereo_width(bus_idx, has_active_controller(
324 bus_idx, MIDIMappingBusProto::kStereoWidthFieldNumber, MIDIMappingProto::kStereoWidthBankFieldNumber));
325 receiver->highlight_treble(bus_idx, has_active_controller(
326 bus_idx, MIDIMappingBusProto::kTrebleFieldNumber, MIDIMappingProto::kTrebleBankFieldNumber));
327 receiver->highlight_mid(bus_idx, has_active_controller(
328 bus_idx, MIDIMappingBusProto::kMidFieldNumber, MIDIMappingProto::kMidBankFieldNumber));
329 receiver->highlight_bass(bus_idx, has_active_controller(
330 bus_idx, MIDIMappingBusProto::kBassFieldNumber, MIDIMappingProto::kBassBankFieldNumber));
331 receiver->highlight_gain(bus_idx, has_active_controller(
332 bus_idx, MIDIMappingBusProto::kGainFieldNumber, MIDIMappingProto::kGainBankFieldNumber));
333 receiver->highlight_compressor_threshold(bus_idx, has_active_controller(
334 bus_idx, MIDIMappingBusProto::kCompressorThresholdFieldNumber, MIDIMappingProto::kCompressorThresholdBankFieldNumber));
335 receiver->highlight_fader(bus_idx, has_active_controller(
336 bus_idx, MIDIMappingBusProto::kFaderFieldNumber, MIDIMappingProto::kFaderBankFieldNumber));
337 receiver->highlight_mute(bus_idx, has_active_controller(
338 bus_idx, MIDIMappingBusProto::kToggleMuteFieldNumber, MIDIMappingProto::kToggleMuteBankFieldNumber));
339 receiver->highlight_toggle_locut(bus_idx, has_active_controller(
340 bus_idx, MIDIMappingBusProto::kToggleLocutFieldNumber, MIDIMappingProto::kToggleLocutBankFieldNumber));
341 receiver->highlight_toggle_auto_gain_staging(bus_idx, has_active_controller(
342 bus_idx, MIDIMappingBusProto::kToggleAutoGainStagingFieldNumber, MIDIMappingProto::kToggleAutoGainStagingBankFieldNumber));
343 receiver->highlight_toggle_compressor(bus_idx, has_active_controller(
344 bus_idx, MIDIMappingBusProto::kToggleCompressorFieldNumber, MIDIMappingProto::kToggleCompressorBankFieldNumber));
348 void MIDIMapper::update_lights_lock_held()
350 if (global_audio_mixer == nullptr) {
354 set<unsigned> active_lights; // Desired state.
355 if (current_controller_bank == 0) {
356 activate_lights_all_buses(MIDIMappingBusProto::kBank1IsSelectedFieldNumber, &active_lights);
358 if (current_controller_bank == 1) {
359 activate_lights_all_buses(MIDIMappingBusProto::kBank2IsSelectedFieldNumber, &active_lights);
361 if (current_controller_bank == 2) {
362 activate_lights_all_buses(MIDIMappingBusProto::kBank3IsSelectedFieldNumber, &active_lights);
364 if (current_controller_bank == 3) {
365 activate_lights_all_buses(MIDIMappingBusProto::kBank4IsSelectedFieldNumber, &active_lights);
367 if (current_controller_bank == 4) {
368 activate_lights_all_buses(MIDIMappingBusProto::kBank5IsSelectedFieldNumber, &active_lights);
370 if (global_audio_mixer->get_limiter_enabled()) {
371 activate_lights_all_buses(MIDIMappingBusProto::kLimiterIsOnFieldNumber, &active_lights);
373 if (global_audio_mixer->get_final_makeup_gain_auto()) {
374 activate_lights_all_buses(MIDIMappingBusProto::kAutoMakeupGainIsOnFieldNumber, &active_lights);
376 unsigned num_buses = min<unsigned>(global_audio_mixer->num_buses(), mapping_proto->bus_mapping_size());
377 for (unsigned bus_idx = 0; bus_idx < num_buses; ++bus_idx) {
378 if (global_audio_mixer->get_mute(bus_idx)) {
379 activate_lights(bus_idx, MIDIMappingBusProto::kIsMutedFieldNumber, &active_lights);
381 if (global_audio_mixer->get_locut_enabled(bus_idx)) {
382 activate_lights(bus_idx, MIDIMappingBusProto::kLocutIsOnFieldNumber, &active_lights);
384 if (global_audio_mixer->get_gain_staging_auto(bus_idx)) {
385 activate_lights(bus_idx, MIDIMappingBusProto::kAutoGainStagingIsOnFieldNumber, &active_lights);
387 if (global_audio_mixer->get_compressor_enabled(bus_idx)) {
388 activate_lights(bus_idx, MIDIMappingBusProto::kCompressorIsOnFieldNumber, &active_lights);
390 if (has_peaked[bus_idx]) {
391 activate_lights(bus_idx, MIDIMappingBusProto::kHasPeakedFieldNumber, &active_lights);
395 midi_device.update_lights(active_lights);
398 void MIDIMapper::activate_lights(unsigned bus_idx, int field_number, set<unsigned> *active_lights)
400 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
402 const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
403 const Reflection *bus_reflection = bus_mapping.GetReflection();
404 if (!bus_reflection->HasField(bus_mapping, descriptor)) {
407 const MIDILightProto &light_proto =
408 static_cast<const MIDILightProto &>(bus_reflection->GetMessage(bus_mapping, descriptor));
409 active_lights->insert(light_proto.note_number());
412 void MIDIMapper::activate_lights_all_buses(int field_number, set<unsigned> *active_lights)
414 for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
415 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
417 const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
418 const Reflection *bus_reflection = bus_mapping.GetReflection();
419 if (!bus_reflection->HasField(bus_mapping, descriptor)) {
422 const MIDILightProto &light_proto =
423 static_cast<const MIDILightProto &>(bus_reflection->GetMessage(bus_mapping, descriptor));
424 active_lights->insert(light_proto.note_number());