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