git.sesse.net Git - nageru/blob - nageru/midi_mapper.cpp

   1 #include "midi_mapper.h"
   2
   3 #include <alsa/asoundlib.h>
   4 #include <assert.h>
   5 #include <errno.h>
   6 #include <fcntl.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>
  11 #include <pthread.h>
  12 #include <poll.h>
  13 #include <stdint.h>
  14 #include <stdio.h>
  15 #include <sys/eventfd.h>
  16 #include <unistd.h>
  17 #include <algorithm>
  18 #include <functional>
  19 #include <thread>
  20
  21 #include "audio_mixer.h"
  22 #include "midi_mapping.pb.h"
  23 #include "shared/midi_mapper_util.h"
  24 #include "shared/text_proto.h"
  25
  26 using namespace google::protobuf;
  27 using namespace std;
  28 using namespace std::placeholders;
  29
  30 namespace {
  31
  32 double map_controller_to_float(int val)
  33 {
  34         // Slightly hackish mapping so that we can represent exactly 0.0, 0.5 and 1.0.
  35         if (val <= 0) {
  36                 return 0.0;
  37         } else if (val >= 127) {
  38                 return 1.0;
  39         } else {
  40                 return (val + 0.5) / 127.0;
  41         }
  42 }
  43
  44 }  // namespace
  45
  46 MIDIMapper::MIDIMapper(ControllerReceiver *receiver)
  47         : receiver(receiver), mapping_proto(new MIDIMappingProto), midi_device(this)
  48 {
  49 }
  50
  51 MIDIMapper::~MIDIMapper() {}
  52
  53 bool load_midi_mapping_from_file(const string &filename, MIDIMappingProto *new_mapping)
  54 {
  55         return load_proto_from_file(filename, new_mapping);
  56 }
  57
  58 bool save_midi_mapping_to_file(const MIDIMappingProto &mapping_proto, const string &filename)
  59 {
  60         return save_proto_to_file(mapping_proto, filename);
  61 }
  62
  63 void MIDIMapper::set_midi_mapping(const MIDIMappingProto &new_mapping)
  64 {
  65         lock_guard<mutex> lock(mu);
  66         if (mapping_proto) {
  67                 mapping_proto->CopyFrom(new_mapping);
  68         } else {
  69                 mapping_proto.reset(new MIDIMappingProto(new_mapping));
  70         }
  71
  72         num_controller_banks = min(max(mapping_proto->num_controller_banks(), 1), 5);
  73         current_controller_bank = 0;
  74
  75         receiver->clear_all_highlights();
  76         update_highlights();
  77 }
  78
  79 void MIDIMapper::start_thread()
  80 {
  81         midi_device.start_thread();
  82 }
  83
  84 const MIDIMappingProto &MIDIMapper::get_current_mapping() const
  85 {
  86         lock_guard<mutex> lock(mu);
  87         return *mapping_proto;
  88 }
  89
  90 ControllerReceiver *MIDIMapper::set_receiver(ControllerReceiver *new_receiver)
  91 {
  92         lock_guard<mutex> lock(mu);
  93         swap(receiver, new_receiver);
  94         return new_receiver;  // Now old receiver.
  95 }
  96
  97 void MIDIMapper::controller_received(int controller, int value_int)
  98 {
  99         const float value = map_controller_to_float(value_int);
 100
 101         receiver->controller_changed(controller);
 102
 103         // Global controllers.
 104         match_controller(controller, MIDIMappingBusProto::kLocutFieldNumber, MIDIMappingProto::kLocutBankFieldNumber,
 105                 value, bind(&ControllerReceiver::set_locut, receiver, _2));
 106         match_controller(controller, MIDIMappingBusProto::kLimiterThresholdFieldNumber, MIDIMappingProto::kLimiterThresholdBankFieldNumber,
 107                 value, bind(&ControllerReceiver::set_limiter_threshold, receiver, _2));
 108         match_controller(controller, MIDIMappingBusProto::kMakeupGainFieldNumber, MIDIMappingProto::kMakeupGainBankFieldNumber,
 109                 value, bind(&ControllerReceiver::set_makeup_gain, receiver, _2));
 110
 111         // Bus controllers.
 112         match_controller(controller, MIDIMappingBusProto::kStereoWidthFieldNumber, MIDIMappingProto::kStereoWidthBankFieldNumber,
 113                 value, bind(&ControllerReceiver::set_stereo_width, receiver, _1, _2));
 114         match_controller(controller, MIDIMappingBusProto::kTrebleFieldNumber, MIDIMappingProto::kTrebleBankFieldNumber,
 115                 value, bind(&ControllerReceiver::set_treble, receiver, _1, _2));
 116         match_controller(controller, MIDIMappingBusProto::kMidFieldNumber, MIDIMappingProto::kMidBankFieldNumber,
 117                 value, bind(&ControllerReceiver::set_mid, receiver, _1, _2));
 118         match_controller(controller, MIDIMappingBusProto::kBassFieldNumber, MIDIMappingProto::kBassBankFieldNumber,
 119                 value, bind(&ControllerReceiver::set_bass, receiver, _1, _2));
 120         match_controller(controller, MIDIMappingBusProto::kGainFieldNumber, MIDIMappingProto::kGainBankFieldNumber,
 121                 value, bind(&ControllerReceiver::set_gain, receiver, _1, _2));
 122         match_controller(controller, MIDIMappingBusProto::kCompressorThresholdFieldNumber, MIDIMappingProto::kCompressorThresholdBankFieldNumber,
 123                 value, bind(&ControllerReceiver::set_compressor_threshold, receiver, _1, _2));
 124         match_controller(controller, MIDIMappingBusProto::kFaderFieldNumber, MIDIMappingProto::kFaderBankFieldNumber,
 125                 value, bind(&ControllerReceiver::set_fader, receiver, _1, _2));
 126 }
 127
 128 void MIDIMapper::note_on_received(int note)
 129 {
 130         lock_guard<mutex> lock(mu);
 131         receiver->note_on(note);
 132
 133         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 134                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 135                 if (bus_mapping.has_prev_bank() &&
 136                     bus_mapping.prev_bank().note_number() == note) {
 137                         current_controller_bank = (current_controller_bank + num_controller_banks - 1) % num_controller_banks;
 138                         update_highlights();
 139                         update_lights_lock_held();
 140                 }
 141                 if (bus_mapping.has_next_bank() &&
 142                     bus_mapping.next_bank().note_number() == note) {
 143                         current_controller_bank = (current_controller_bank + 1) % num_controller_banks;
 144                         update_highlights();
 145                         update_lights_lock_held();
 146                 }
 147                 if (bus_mapping.has_select_bank_1() &&
 148                     bus_mapping.select_bank_1().note_number() == note) {
 149                         current_controller_bank = 0;
 150                         update_highlights();
 151                         update_lights_lock_held();
 152                 }
 153                 if (bus_mapping.has_select_bank_2() &&
 154                     bus_mapping.select_bank_2().note_number() == note &&
 155                     num_controller_banks >= 2) {
 156                         current_controller_bank = 1;
 157                         update_highlights();
 158                         update_lights_lock_held();
 159                 }
 160                 if (bus_mapping.has_select_bank_3() &&
 161                     bus_mapping.select_bank_3().note_number() == note &&
 162                     num_controller_banks >= 3) {
 163                         current_controller_bank = 2;
 164                         update_highlights();
 165                         update_lights_lock_held();
 166                 }
 167                 if (bus_mapping.has_select_bank_4() &&
 168                     bus_mapping.select_bank_4().note_number() == note &&
 169                     num_controller_banks >= 4) {
 170                         current_controller_bank = 3;
 171                         update_highlights();
 172                         update_lights_lock_held();
 173                 }
 174                 if (bus_mapping.has_select_bank_5() &&
 175                     bus_mapping.select_bank_5().note_number() == note &&
 176                     num_controller_banks >= 5) {
 177                         current_controller_bank = 4;
 178                         update_highlights();
 179                         update_lights_lock_held();
 180                 }
 181         }
 182
 183         match_button(note, MIDIMappingBusProto::kToggleLocutFieldNumber, MIDIMappingProto::kToggleLocutBankFieldNumber,
 184                 bind(&ControllerReceiver::toggle_locut, receiver, _1));
 185         match_button(note, MIDIMappingBusProto::kToggleAutoGainStagingFieldNumber, MIDIMappingProto::kToggleAutoGainStagingBankFieldNumber,
 186                 bind(&ControllerReceiver::toggle_auto_gain_staging, receiver, _1));
 187         match_button(note, MIDIMappingBusProto::kToggleCompressorFieldNumber, MIDIMappingProto::kToggleCompressorBankFieldNumber,
 188                 bind(&ControllerReceiver::toggle_compressor, receiver, _1));
 189         match_button(note, MIDIMappingBusProto::kClearPeakFieldNumber, MIDIMappingProto::kClearPeakBankFieldNumber,
 190                 bind(&ControllerReceiver::clear_peak, receiver, _1));
 191         match_button(note, MIDIMappingBusProto::kToggleMuteFieldNumber, MIDIMappingProto::kClearPeakBankFieldNumber,
 192                 bind(&ControllerReceiver::toggle_mute, receiver, _1));
 193         match_button(note, MIDIMappingBusProto::kToggleLimiterFieldNumber, MIDIMappingProto::kToggleLimiterBankFieldNumber,
 194                 bind(&ControllerReceiver::toggle_limiter, receiver));
 195         match_button(note, MIDIMappingBusProto::kToggleAutoMakeupGainFieldNumber, MIDIMappingProto::kToggleAutoMakeupGainBankFieldNumber,
 196                 bind(&ControllerReceiver::toggle_auto_makeup_gain, receiver));
 197 }
 198
 199 void MIDIMapper::update_num_subscribers(unsigned num_subscribers)
 200 {
 201         num_subscribed_ports = num_subscribers;
 202         update_highlights();
 203 }
 204
 205 void MIDIMapper::match_controller(int controller, int field_number, int bank_field_number, float value, function<void(unsigned, float)> func)
 206 {
 207         if (bank_mismatch(bank_field_number)) {
 208                 return;
 209         }
 210
 211         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 212                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 213                 if (match_controller_helper(bus_mapping, field_number, controller)) {
 214                         func(bus_idx, value);
 215                 }
 216         }
 217 }
 218
 219 void MIDIMapper::match_button(int note, int field_number, int bank_field_number, function<void(unsigned)> func)
 220 {
 221         if (bank_mismatch(bank_field_number)) {
 222                 return;
 223         }
 224
 225         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 226                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 227                 if (match_button_helper(bus_mapping, field_number, note)) {
 228                         func(bus_idx);
 229                 }
 230         }
 231 }
 232
 233 bool MIDIMapper::has_active_controller(unsigned bus_idx, int field_number, int bank_field_number)
 234 {
 235         if (bank_mismatch(bank_field_number)) {
 236                 return false;
 237         }
 238
 239         const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 240         const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
 241         const Reflection *bus_reflection = bus_mapping.GetReflection();
 242         return bus_reflection->HasField(bus_mapping, descriptor);
 243 }
 244
 245 bool MIDIMapper::bank_mismatch(int bank_field_number)
 246 {
 247         return !match_bank_helper(*mapping_proto, bank_field_number, current_controller_bank);
 248 }
 249
 250 void MIDIMapper::refresh_highlights()
 251 {
 252         receiver->clear_all_highlights();
 253         update_highlights();
 254 }
 255
 256 void MIDIMapper::refresh_lights()
 257 {
 258         lock_guard<mutex> lock(mu);
 259         update_lights_lock_held();
 260 }
 261
 262 void MIDIMapper::update_highlights()
 263 {
 264         if (num_subscribed_ports.load() == 0) {
 265                 receiver->clear_all_highlights();
 266                 return;
 267         }
 268
 269         // Global controllers.
 270         bool highlight_locut = false;
 271         bool highlight_limiter_threshold = false;
 272         bool highlight_makeup_gain = false;
 273         bool highlight_toggle_limiter = false;
 274         bool highlight_toggle_auto_makeup_gain = false;
 275         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 276                 if (has_active_controller(
 277                         bus_idx, MIDIMappingBusProto::kLocutFieldNumber, MIDIMappingProto::kLocutBankFieldNumber)) {
 278                         highlight_locut = true;
 279                 }
 280                 if (has_active_controller(
 281                         bus_idx, MIDIMappingBusProto::kLimiterThresholdFieldNumber, MIDIMappingProto::kLimiterThresholdBankFieldNumber)) {
 282                         highlight_limiter_threshold = true;
 283                 }
 284                 if (has_active_controller(
 285                         bus_idx, MIDIMappingBusProto::kMakeupGainFieldNumber, MIDIMappingProto::kMakeupGainBankFieldNumber)) {
 286                         highlight_makeup_gain = true;
 287                 }
 288                 if (has_active_controller(
 289                         bus_idx, MIDIMappingBusProto::kToggleLimiterFieldNumber, MIDIMappingProto::kToggleLimiterBankFieldNumber)) {
 290                         highlight_toggle_limiter = true;
 291                 }
 292                 if (has_active_controller(
 293                         bus_idx, MIDIMappingBusProto::kToggleAutoMakeupGainFieldNumber, MIDIMappingProto::kToggleAutoMakeupGainBankFieldNumber)) {
 294                         highlight_toggle_auto_makeup_gain = true;
 295                 }
 296         }
 297         receiver->highlight_locut(highlight_locut);
 298         receiver->highlight_limiter_threshold(highlight_limiter_threshold);
 299         receiver->highlight_makeup_gain(highlight_makeup_gain);
 300         receiver->highlight_toggle_limiter(highlight_toggle_limiter);
 301         receiver->highlight_toggle_auto_makeup_gain(highlight_toggle_auto_makeup_gain);
 302
 303         // Per-bus controllers.
 304         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 305                 receiver->highlight_stereo_width(bus_idx, has_active_controller(
 306                         bus_idx, MIDIMappingBusProto::kStereoWidthFieldNumber, MIDIMappingProto::kStereoWidthBankFieldNumber));
 307                 receiver->highlight_treble(bus_idx, has_active_controller(
 308                         bus_idx, MIDIMappingBusProto::kTrebleFieldNumber, MIDIMappingProto::kTrebleBankFieldNumber));
 309                 receiver->highlight_mid(bus_idx, has_active_controller(
 310                         bus_idx, MIDIMappingBusProto::kMidFieldNumber, MIDIMappingProto::kMidBankFieldNumber));
 311                 receiver->highlight_bass(bus_idx, has_active_controller(
 312                         bus_idx, MIDIMappingBusProto::kBassFieldNumber, MIDIMappingProto::kBassBankFieldNumber));
 313                 receiver->highlight_gain(bus_idx, has_active_controller(
 314                         bus_idx, MIDIMappingBusProto::kGainFieldNumber, MIDIMappingProto::kGainBankFieldNumber));
 315                 receiver->highlight_compressor_threshold(bus_idx, has_active_controller(
 316                         bus_idx, MIDIMappingBusProto::kCompressorThresholdFieldNumber, MIDIMappingProto::kCompressorThresholdBankFieldNumber));
 317                 receiver->highlight_fader(bus_idx, has_active_controller(
 318                         bus_idx, MIDIMappingBusProto::kFaderFieldNumber, MIDIMappingProto::kFaderBankFieldNumber));
 319                 receiver->highlight_mute(bus_idx, has_active_controller(
 320                         bus_idx, MIDIMappingBusProto::kToggleMuteFieldNumber, MIDIMappingProto::kToggleMuteBankFieldNumber));
 321                 receiver->highlight_toggle_locut(bus_idx, has_active_controller(
 322                         bus_idx, MIDIMappingBusProto::kToggleLocutFieldNumber, MIDIMappingProto::kToggleLocutBankFieldNumber));
 323                 receiver->highlight_toggle_auto_gain_staging(bus_idx, has_active_controller(
 324                         bus_idx, MIDIMappingBusProto::kToggleAutoGainStagingFieldNumber, MIDIMappingProto::kToggleAutoGainStagingBankFieldNumber));
 325                 receiver->highlight_toggle_compressor(bus_idx, has_active_controller(
 326                         bus_idx, MIDIMappingBusProto::kToggleCompressorFieldNumber, MIDIMappingProto::kToggleCompressorBankFieldNumber));
 327         }
 328 }
 329
 330 // Find what MIDI note the given light (as given by field_number) is mapped to, and enable it.
 331 void activate_mapped_light(const MIDIMappingBusProto &bus_mapping, int field_number, set<unsigned> *active_lights)
 332 {
 333         const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
 334         const Reflection *bus_reflection = bus_mapping.GetReflection();
 335         if (!bus_reflection->HasField(bus_mapping, descriptor)) {
 336                 return;
 337         }
 338         const MIDILightProto &light_proto =
 339                 static_cast<const MIDILightProto &>(bus_reflection->GetMessage(bus_mapping, descriptor));
 340         active_lights->insert(light_proto.note_number());
 341 }
 342
 343 void MIDIMapper::update_lights_lock_held()
 344 {
 345         if (global_audio_mixer == nullptr) {
 346                 return;
 347         }
 348
 349         set<unsigned> active_lights;  // Desired state.
 350         if (current_controller_bank == 0) {
 351                 activate_lights_all_buses(MIDIMappingBusProto::kBank1IsSelectedFieldNumber, &active_lights);
 352         }
 353         if (current_controller_bank == 1) {
 354                 activate_lights_all_buses(MIDIMappingBusProto::kBank2IsSelectedFieldNumber, &active_lights);
 355         }
 356         if (current_controller_bank == 2) {
 357                 activate_lights_all_buses(MIDIMappingBusProto::kBank3IsSelectedFieldNumber, &active_lights);
 358         }
 359         if (current_controller_bank == 3) {
 360                 activate_lights_all_buses(MIDIMappingBusProto::kBank4IsSelectedFieldNumber, &active_lights);
 361         }
 362         if (current_controller_bank == 4) {
 363                 activate_lights_all_buses(MIDIMappingBusProto::kBank5IsSelectedFieldNumber, &active_lights);
 364         }
 365         if (global_audio_mixer->get_limiter_enabled()) {
 366                 activate_lights_all_buses(MIDIMappingBusProto::kLimiterIsOnFieldNumber, &active_lights);
 367         }
 368         if (global_audio_mixer->get_final_makeup_gain_auto()) {
 369                 activate_lights_all_buses(MIDIMappingBusProto::kAutoMakeupGainIsOnFieldNumber, &active_lights);
 370         }
 371         unsigned num_buses = min<unsigned>(global_audio_mixer->num_buses(), mapping_proto->bus_mapping_size());
 372         for (unsigned bus_idx = 0; bus_idx < num_buses; ++bus_idx) {
 373                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 374                 if (global_audio_mixer->get_mute(bus_idx)) {
 375                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kIsMutedFieldNumber, &active_lights);
 376                 }
 377                 if (global_audio_mixer->get_locut_enabled(bus_idx)) {
 378                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kLocutIsOnFieldNumber, &active_lights);
 379                 }
 380                 if (global_audio_mixer->get_gain_staging_auto(bus_idx)) {
 381                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kAutoGainStagingIsOnFieldNumber, &active_lights);
 382                 }
 383                 if (global_audio_mixer->get_compressor_enabled(bus_idx)) {
 384                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kCompressorIsOnFieldNumber, &active_lights);
 385                 }
 386                 if (has_peaked[bus_idx]) {
 387                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kHasPeakedFieldNumber, &active_lights);
 388                 }
 389         }
 390
 391         midi_device.update_lights(active_lights);
 392 }
 393
 394 void MIDIMapper::activate_lights_all_buses(int field_number, set<unsigned> *active_lights)
 395 {
 396         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 397                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 398                 activate_mapped_light(bus_mapping, field_number, active_lights);
 399         }
 400 }