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 "nageru_midi_mapping.pb.h"
  23 #include "shared/midi_device.h"
  24 #include "shared/midi_mapper_util.h"
  25 #include "shared/text_proto.h"
  26
  27 using namespace google::protobuf;
  28 using namespace std;
  29 using namespace std::placeholders;
  30
  31 MIDIMapper::MIDIMapper(ControllerReceiver *receiver)
  32         : receiver(receiver), mapping_proto(new MIDIMappingProto), midi_device(this)
  33 {
  34 }
  35
  36 MIDIMapper::~MIDIMapper() {}
  37
  38 bool load_midi_mapping_from_file(const string &filename, MIDIMappingProto *new_mapping)
  39 {
  40         return load_proto_from_file(filename, new_mapping);
  41 }
  42
  43 bool save_midi_mapping_to_file(const MIDIMappingProto &mapping_proto, const string &filename)
  44 {
  45         return save_proto_to_file(mapping_proto, filename);
  46 }
  47
  48 void MIDIMapper::set_midi_mapping(const MIDIMappingProto &new_mapping)
  49 {
  50         lock_guard<mutex> lock(mu);
  51         if (mapping_proto) {
  52                 mapping_proto->CopyFrom(new_mapping);
  53         } else {
  54                 mapping_proto.reset(new MIDIMappingProto(new_mapping));
  55         }
  56
  57         num_controller_banks = min(max(mapping_proto->num_controller_banks(), 1), 5);
  58         current_controller_bank = 0;
  59
  60         receiver->clear_all_highlights();
  61         update_highlights();
  62 }
  63
  64 void MIDIMapper::start_thread()
  65 {
  66         midi_device.start_thread();
  67 }
  68
  69 const MIDIMappingProto &MIDIMapper::get_current_mapping() const
  70 {
  71         lock_guard<mutex> lock(mu);
  72         return *mapping_proto;
  73 }
  74
  75 ControllerReceiver *MIDIMapper::set_receiver(ControllerReceiver *new_receiver)
  76 {
  77         lock_guard<mutex> lock(mu);
  78         swap(receiver, new_receiver);
  79         return new_receiver;  // Now old receiver.
  80 }
  81
  82 void MIDIMapper::controller_received(int controller, int value_int)
  83 {
  84         const float value = map_controller_to_float(controller, value_int);
  85
  86         receiver->controller_changed(controller);
  87
  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));
  95
  96         // Bus controllers.
  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));
 111 }
 112
 113 void MIDIMapper::note_on_received(int note)
 114 {
 115         lock_guard<mutex> lock(mu);
 116         receiver->note_on(note);
 117
 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;
 123                         update_highlights();
 124                         update_lights_lock_held();
 125                 }
 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;
 129                         update_highlights();
 130                         update_lights_lock_held();
 131                 }
 132                 if (bus_mapping.has_select_bank_1() &&
 133                     bus_mapping.select_bank_1().note_number() == note) {
 134                         current_controller_bank = 0;
 135                         update_highlights();
 136                         update_lights_lock_held();
 137                 }
 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;
 142                         update_highlights();
 143                         update_lights_lock_held();
 144                 }
 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;
 149                         update_highlights();
 150                         update_lights_lock_held();
 151                 }
 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;
 156                         update_highlights();
 157                         update_lights_lock_held();
 158                 }
 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;
 163                         update_highlights();
 164                         update_lights_lock_held();
 165                 }
 166         }
 167
 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));
 182         match_button(note, MIDIMappingBusProto::kSwitchVideoChannelFieldNumber, MIDIMappingProto::kSwitchVideoChannelBankFieldNumber,
 183                 bind(&ControllerReceiver::switch_video_channel, receiver, _1));
 184         match_button(note, MIDIMappingBusProto::kApplyTransitionFieldNumber, MIDIMappingProto::kApplyTransitionBankFieldNumber,
 185                 bind(&ControllerReceiver::apply_transition, receiver, _1));
 186         match_button(note, MIDIMappingBusProto::kPrevAudioViewFieldNumber, MIDIMappingProto::kPrevAudioViewBankFieldNumber,
 187                 bind(&ControllerReceiver::prev_audio_view, receiver));
 188         match_button(note, MIDIMappingBusProto::kNextAudioViewFieldNumber, MIDIMappingProto::kNextAudioViewBankFieldNumber,
 189                 bind(&ControllerReceiver::prev_audio_view, receiver));
 190         match_button(note, MIDIMappingBusProto::kBeginNewVideoSegmentFieldNumber, MIDIMappingProto::kBeginNewVideoSegmentBankFieldNumber,
 191                 bind(&ControllerReceiver::begin_new_segment, receiver));
 192         match_button(note, MIDIMappingBusProto::kExitFieldNumber, MIDIMappingProto::kExitBankFieldNumber,
 193                 bind(&ControllerReceiver::exit, receiver));
 194 }
 195
 196 void MIDIMapper::update_num_subscribers(unsigned num_subscribers)
 197 {
 198         num_subscribed_ports = num_subscribers;
 199         update_highlights();
 200 }
 201
 202 void MIDIMapper::match_controller(int controller, int field_number, int bank_field_number, float value, function<void(unsigned, float)> func)
 203 {
 204         if (bank_mismatch(bank_field_number)) {
 205                 return;
 206         }
 207
 208         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 209                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 210                 if (match_controller_helper(bus_mapping, field_number, controller)) {
 211                         func(bus_idx, value);
 212                 }
 213         }
 214 }
 215
 216 void MIDIMapper::match_button(int note, int field_number, int bank_field_number, function<void(unsigned)> func)
 217 {
 218         if (bank_mismatch(bank_field_number)) {
 219                 return;
 220         }
 221
 222         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 223                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 224                 if (match_button_helper(bus_mapping, field_number, note)) {
 225                         func(bus_idx);
 226                 }
 227         }
 228 }
 229
 230 bool MIDIMapper::has_active_controller(unsigned bus_idx, int field_number, int bank_field_number)
 231 {
 232         if (bank_mismatch(bank_field_number)) {
 233                 return false;
 234         }
 235
 236         const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 237         const FieldDescriptor *descriptor = bus_mapping.GetDescriptor()->FindFieldByNumber(field_number);
 238         const Reflection *bus_reflection = bus_mapping.GetReflection();
 239         return bus_reflection->HasField(bus_mapping, descriptor);
 240 }
 241
 242 bool MIDIMapper::bank_mismatch(int bank_field_number)
 243 {
 244         return !match_bank_helper(*mapping_proto, bank_field_number, current_controller_bank);
 245 }
 246
 247 void MIDIMapper::refresh_highlights()
 248 {
 249         receiver->clear_all_highlights();
 250         update_highlights();
 251 }
 252
 253 void MIDIMapper::refresh_lights()
 254 {
 255         lock_guard<mutex> lock(mu);
 256         update_lights_lock_held();
 257 }
 258
 259 void MIDIMapper::update_highlights()
 260 {
 261         if (num_subscribed_ports.load() == 0) {
 262                 receiver->clear_all_highlights();
 263                 return;
 264         }
 265
 266         // Global controllers.
 267         bool highlight_locut = false;
 268         bool highlight_limiter_threshold = false;
 269         bool highlight_makeup_gain = false;
 270         bool highlight_toggle_limiter = false;
 271         bool highlight_toggle_auto_makeup_gain = false;
 272         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 273                 if (has_active_controller(
 274                         bus_idx, MIDIMappingBusProto::kLocutFieldNumber, MIDIMappingProto::kLocutBankFieldNumber)) {
 275                         highlight_locut = true;
 276                 }
 277                 if (has_active_controller(
 278                         bus_idx, MIDIMappingBusProto::kLimiterThresholdFieldNumber, MIDIMappingProto::kLimiterThresholdBankFieldNumber)) {
 279                         highlight_limiter_threshold = true;
 280                 }
 281                 if (has_active_controller(
 282                         bus_idx, MIDIMappingBusProto::kMakeupGainFieldNumber, MIDIMappingProto::kMakeupGainBankFieldNumber)) {
 283                         highlight_makeup_gain = true;
 284                 }
 285                 if (has_active_controller(
 286                         bus_idx, MIDIMappingBusProto::kToggleLimiterFieldNumber, MIDIMappingProto::kToggleLimiterBankFieldNumber)) {
 287                         highlight_toggle_limiter = true;
 288                 }
 289                 if (has_active_controller(
 290                         bus_idx, MIDIMappingBusProto::kToggleAutoMakeupGainFieldNumber, MIDIMappingProto::kToggleAutoMakeupGainBankFieldNumber)) {
 291                         highlight_toggle_auto_makeup_gain = true;
 292                 }
 293         }
 294         receiver->highlight_locut(highlight_locut);
 295         receiver->highlight_limiter_threshold(highlight_limiter_threshold);
 296         receiver->highlight_makeup_gain(highlight_makeup_gain);
 297         receiver->highlight_toggle_limiter(highlight_toggle_limiter);
 298         receiver->highlight_toggle_auto_makeup_gain(highlight_toggle_auto_makeup_gain);
 299
 300         // Per-bus controllers.
 301         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 302                 receiver->highlight_stereo_width(bus_idx, has_active_controller(
 303                         bus_idx, MIDIMappingBusProto::kStereoWidthFieldNumber, MIDIMappingProto::kStereoWidthBankFieldNumber));
 304                 receiver->highlight_treble(bus_idx, has_active_controller(
 305                         bus_idx, MIDIMappingBusProto::kTrebleFieldNumber, MIDIMappingProto::kTrebleBankFieldNumber));
 306                 receiver->highlight_mid(bus_idx, has_active_controller(
 307                         bus_idx, MIDIMappingBusProto::kMidFieldNumber, MIDIMappingProto::kMidBankFieldNumber));
 308                 receiver->highlight_bass(bus_idx, has_active_controller(
 309                         bus_idx, MIDIMappingBusProto::kBassFieldNumber, MIDIMappingProto::kBassBankFieldNumber));
 310                 receiver->highlight_gain(bus_idx, has_active_controller(
 311                         bus_idx, MIDIMappingBusProto::kGainFieldNumber, MIDIMappingProto::kGainBankFieldNumber));
 312                 receiver->highlight_compressor_threshold(bus_idx, has_active_controller(
 313                         bus_idx, MIDIMappingBusProto::kCompressorThresholdFieldNumber, MIDIMappingProto::kCompressorThresholdBankFieldNumber));
 314                 receiver->highlight_fader(bus_idx, has_active_controller(
 315                         bus_idx, MIDIMappingBusProto::kFaderFieldNumber, MIDIMappingProto::kFaderBankFieldNumber));
 316                 receiver->highlight_mute(bus_idx, has_active_controller(
 317                         bus_idx, MIDIMappingBusProto::kToggleMuteFieldNumber, MIDIMappingProto::kToggleMuteBankFieldNumber));
 318                 receiver->highlight_toggle_locut(bus_idx, has_active_controller(
 319                         bus_idx, MIDIMappingBusProto::kToggleLocutFieldNumber, MIDIMappingProto::kToggleLocutBankFieldNumber));
 320                 receiver->highlight_toggle_auto_gain_staging(bus_idx, has_active_controller(
 321                         bus_idx, MIDIMappingBusProto::kToggleAutoGainStagingFieldNumber, MIDIMappingProto::kToggleAutoGainStagingBankFieldNumber));
 322                 receiver->highlight_toggle_compressor(bus_idx, has_active_controller(
 323                         bus_idx, MIDIMappingBusProto::kToggleCompressorFieldNumber, MIDIMappingProto::kToggleCompressorBankFieldNumber));
 324         }
 325 }
 326
 327 void MIDIMapper::update_lights_lock_held()
 328 {
 329         if (global_audio_mixer == nullptr) {
 330                 return;
 331         }
 332
 333         map<MIDIDevice::LightKey, uint8_t> active_lights;  // Desired state.
 334         if (current_controller_bank == 0) {
 335                 activate_lights_all_buses(MIDIMappingBusProto::kBank1IsSelectedFieldNumber, &active_lights);
 336         }
 337         if (current_controller_bank == 1) {
 338                 activate_lights_all_buses(MIDIMappingBusProto::kBank2IsSelectedFieldNumber, &active_lights);
 339         }
 340         if (current_controller_bank == 2) {
 341                 activate_lights_all_buses(MIDIMappingBusProto::kBank3IsSelectedFieldNumber, &active_lights);
 342         }
 343         if (current_controller_bank == 3) {
 344                 activate_lights_all_buses(MIDIMappingBusProto::kBank4IsSelectedFieldNumber, &active_lights);
 345         }
 346         if (current_controller_bank == 4) {
 347                 activate_lights_all_buses(MIDIMappingBusProto::kBank5IsSelectedFieldNumber, &active_lights);
 348         }
 349         if (global_audio_mixer->get_limiter_enabled()) {
 350                 activate_lights_all_buses(MIDIMappingBusProto::kLimiterIsOnFieldNumber, &active_lights);
 351         }
 352         if (global_audio_mixer->get_final_makeup_gain_auto()) {
 353                 activate_lights_all_buses(MIDIMappingBusProto::kAutoMakeupGainIsOnFieldNumber, &active_lights);
 354         }
 355         unsigned num_buses = min<unsigned>(global_audio_mixer->num_buses(), mapping_proto->bus_mapping_size());
 356         for (unsigned bus_idx = 0; bus_idx < num_buses; ++bus_idx) {
 357                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 358                 if (global_audio_mixer->get_mute(bus_idx)) {
 359                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kIsMutedFieldNumber, &active_lights);
 360                 }
 361                 if (global_audio_mixer->get_locut_enabled(bus_idx)) {
 362                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kLocutIsOnFieldNumber, &active_lights);
 363                 }
 364                 if (global_audio_mixer->get_gain_staging_auto(bus_idx)) {
 365                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kAutoGainStagingIsOnFieldNumber, &active_lights);
 366                 }
 367                 if (global_audio_mixer->get_compressor_enabled(bus_idx)) {
 368                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kCompressorIsOnFieldNumber, &active_lights);
 369                 }
 370                 if (has_peaked[bus_idx]) {
 371                         activate_mapped_light(bus_mapping, MIDIMappingBusProto::kHasPeakedFieldNumber, &active_lights);
 372                 }
 373         }
 374
 375         midi_device.update_lights(active_lights);
 376 }
 377
 378 void MIDIMapper::activate_lights_all_buses(int field_number, map<MIDIDevice::LightKey, uint8_t> *active_lights)
 379 {
 380         for (size_t bus_idx = 0; bus_idx < size_t(mapping_proto->bus_mapping_size()); ++bus_idx) {
 381                 const MIDIMappingBusProto &bus_mapping = mapping_proto->bus_mapping(bus_idx);
 382                 activate_mapped_light(bus_mapping, field_number, active_lights);
 383         }
 384 }