git.sesse.net Git - nageru/blob - futatabi/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 <math.h>
  12 #include <pthread.h>
  13 #include <poll.h>
  14 #include <stdint.h>
  15 #include <stdio.h>
  16 #include <sys/eventfd.h>
  17 #include <unistd.h>
  18 #include <algorithm>
  19 #include <functional>
  20 #include <map>
  21 #include <thread>
  22
  23 #include "defs.h"
  24 #include "futatabi_midi_mapping.pb.h"
  25 #include "shared/midi_mapper_util.h"
  26 #include "shared/text_proto.h"
  27
  28 using namespace google::protobuf;
  29 using namespace std;
  30 using namespace std::placeholders;
  31
  32 MIDIMapper::MIDIMapper(ControllerReceiver *receiver)
  33         : receiver(receiver), mapping_proto(new MIDIMappingProto), midi_device(this)
  34 {
  35 }
  36
  37 MIDIMapper::~MIDIMapper() {}
  38
  39 bool load_midi_mapping_from_file(const string &filename, MIDIMappingProto *new_mapping)
  40 {
  41         return load_proto_from_file(filename, new_mapping);
  42 }
  43
  44 bool save_midi_mapping_to_file(const MIDIMappingProto &mapping_proto, const string &filename)
  45 {
  46         return save_proto_to_file(mapping_proto, filename);
  47 }
  48
  49 void MIDIMapper::set_midi_mapping(const MIDIMappingProto &new_mapping)
  50 {
  51         lock_guard<mutex> lock(mu);
  52         if (mapping_proto) {
  53                 mapping_proto->CopyFrom(new_mapping);
  54         } else {
  55                 mapping_proto.reset(new MIDIMappingProto(new_mapping));
  56         }
  57
  58         num_controller_banks = min(max(mapping_proto->num_controller_banks(), 1), 5);
  59         current_controller_bank = 0;
  60 }
  61
  62 void MIDIMapper::start_thread()
  63 {
  64         midi_device.start_thread();
  65 }
  66
  67 const MIDIMappingProto &MIDIMapper::get_current_mapping() const
  68 {
  69         lock_guard<mutex> lock(mu);
  70         return *mapping_proto;
  71 }
  72
  73 ControllerReceiver *MIDIMapper::set_receiver(ControllerReceiver *new_receiver)
  74 {
  75         lock_guard<mutex> lock(mu);
  76         swap(receiver, new_receiver);
  77         return new_receiver;  // Now old receiver.
  78 }
  79
  80 void MIDIMapper::controller_received(int controller, int value_int)
  81 {
  82         int delta_value = value_int - 64;  // For infinite controllers such as jog.
  83         float value = map_controller_to_float(controller, value_int);
  84
  85         receiver->controller_changed(controller);
  86
  87         match_controller(controller, MIDIMappingProto::kJogFieldNumber, MIDIMappingProto::kJogBankFieldNumber,
  88                 delta_value, bind(&ControllerReceiver::jog, receiver, _1));
  89
  90         // Speed goes from 0.0 to 2.0 (the receiver will clamp).
  91         match_controller(controller, MIDIMappingProto::kMasterSpeedFieldNumber, MIDIMappingProto::kMasterSpeedBankFieldNumber,
  92                 value * 2.0, bind(&ControllerReceiver::set_master_speed, receiver, _1));
  93 }
  94
  95 void MIDIMapper::note_on_received(int note)
  96 {
  97         lock_guard<mutex> lock(mu);
  98         receiver->note_on(note);
  99
 100         if (mapping_proto->has_prev_bank() &&
 101             mapping_proto->prev_bank().note_number() == note) {
 102                 current_controller_bank = (current_controller_bank + num_controller_banks - 1) % num_controller_banks;
 103                 update_lights_lock_held();
 104         }
 105         if (mapping_proto->has_next_bank() &&
 106             mapping_proto->next_bank().note_number() == note) {
 107                 current_controller_bank = (current_controller_bank + 1) % num_controller_banks;
 108                 update_lights_lock_held();
 109         }
 110         if (mapping_proto->has_select_bank_1() &&
 111             mapping_proto->select_bank_1().note_number() == note) {
 112                 current_controller_bank = 0;
 113                 update_lights_lock_held();
 114         }
 115         if (mapping_proto->has_select_bank_2() &&
 116             mapping_proto->select_bank_2().note_number() == note &&
 117             num_controller_banks >= 2) {
 118                 current_controller_bank = 1;
 119                 update_lights_lock_held();
 120         }
 121         if (mapping_proto->has_select_bank_3() &&
 122             mapping_proto->select_bank_3().note_number() == note &&
 123             num_controller_banks >= 3) {
 124                 current_controller_bank = 2;
 125                 update_lights_lock_held();
 126         }
 127         if (mapping_proto->has_select_bank_4() &&
 128             mapping_proto->select_bank_4().note_number() == note &&
 129             num_controller_banks >= 4) {
 130                 current_controller_bank = 3;
 131                 update_lights_lock_held();
 132         }
 133         if (mapping_proto->has_select_bank_5() &&
 134             mapping_proto->select_bank_5().note_number() == note &&
 135             num_controller_banks >= 5) {
 136                 current_controller_bank = 4;
 137                 update_lights_lock_held();
 138         }
 139
 140         match_button(note, MIDIMappingProto::kPreviewFieldNumber, MIDIMappingProto::kPreviewBankFieldNumber,
 141                 bind(&ControllerReceiver::preview, receiver));
 142         match_button(note, MIDIMappingProto::kQueueFieldNumber, MIDIMappingProto::kQueueBankFieldNumber,
 143                 bind(&ControllerReceiver::queue, receiver));
 144         match_button(note, MIDIMappingProto::kPlayFieldNumber, MIDIMappingProto::kPlayBankFieldNumber,
 145                 bind(&ControllerReceiver::play, receiver));
 146         match_button(note, MIDIMappingProto::kToggleLockFieldNumber, MIDIMappingProto::kToggleLockBankFieldNumber,
 147                 bind(&ControllerReceiver::toggle_lock, receiver));
 148
 149         unsigned num_cameras = std::min(MAX_STREAMS, mapping_proto->camera_size());
 150         for (unsigned camera_idx = 0; camera_idx < num_cameras; ++camera_idx) {
 151                 const CameraMIDIMappingProto &camera = mapping_proto->camera(camera_idx);
 152                 if (match_bank_helper(camera, CameraMIDIMappingProto::kBankFieldNumber, current_controller_bank) &&
 153                     match_button_helper(camera, CameraMIDIMappingProto::kButtonFieldNumber, note)) {
 154                         receiver->switch_camera(camera_idx);
 155                 }
 156         }
 157
 158         match_button(note, MIDIMappingProto::kCueInFieldNumber, MIDIMappingProto::kCueInBankFieldNumber,
 159                 bind(&ControllerReceiver::cue_in, receiver));
 160         match_button(note, MIDIMappingProto::kCueOutFieldNumber, MIDIMappingProto::kCueOutBankFieldNumber,
 161                 bind(&ControllerReceiver::cue_out, receiver));
 162 }
 163
 164 void MIDIMapper::match_controller(int controller, int field_number, int bank_field_number, float value, function<void(float)> func)
 165 {
 166         if (bank_mismatch(bank_field_number)) {
 167                 return;
 168         }
 169
 170         if (match_controller_helper(*mapping_proto, field_number, controller)) {
 171                 func(value);
 172         }
 173 }
 174
 175 void MIDIMapper::match_button(int note, int field_number, int bank_field_number, function<void()> func)
 176 {
 177         if (bank_mismatch(bank_field_number)) {
 178                 return;
 179         }
 180
 181         if (match_button_helper(*mapping_proto, field_number, note)) {
 182                 func();
 183         }
 184 }
 185
 186 bool MIDIMapper::has_active_controller(int field_number, int bank_field_number)
 187 {
 188         if (bank_mismatch(bank_field_number)) {
 189                 return false;
 190         }
 191
 192         const FieldDescriptor *descriptor = mapping_proto->GetDescriptor()->FindFieldByNumber(field_number);
 193         const Reflection *reflection = mapping_proto->GetReflection();
 194         return reflection->HasField(*mapping_proto, descriptor);
 195 }
 196
 197 bool MIDIMapper::bank_mismatch(int bank_field_number)
 198 {
 199         return !match_bank_helper(*mapping_proto, bank_field_number, current_controller_bank);
 200 }
 201
 202 void MIDIMapper::refresh_lights()
 203 {
 204         lock_guard<mutex> lock(mu);
 205         update_lights_lock_held();
 206 }
 207
 208 void MIDIMapper::update_lights_lock_held()
 209 {
 210         map<MIDIDevice::LightKey, uint8_t> active_lights;  // Desired state.
 211         if (current_controller_bank == 0) {
 212                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kBank1IsSelectedFieldNumber, &active_lights);
 213         }
 214         if (current_controller_bank == 1) {
 215                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kBank2IsSelectedFieldNumber, &active_lights);
 216         }
 217         if (current_controller_bank == 2) {
 218                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kBank3IsSelectedFieldNumber, &active_lights);
 219         }
 220         if (current_controller_bank == 3) {
 221                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kBank4IsSelectedFieldNumber, &active_lights);
 222         }
 223         if (current_controller_bank == 4) {
 224                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kBank5IsSelectedFieldNumber, &active_lights);
 225         }
 226         if (preview_enabled_light == On) {  // Playing.
 227                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kPreviewPlayingFieldNumber, &active_lights);
 228         } else if (preview_enabled_light == Blinking) {  // Preview ready.
 229                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kPreviewReadyFieldNumber, &active_lights);
 230         }
 231         if (queue_enabled_light) {
 232                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kQueueEnabledFieldNumber, &active_lights);
 233         }
 234         if (play_enabled_light == On) {  // Playing.
 235                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kPlayingFieldNumber, &active_lights);
 236         } else if (play_enabled_light == Blinking) {  // Play ready.
 237                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kPlayReadyFieldNumber, &active_lights);
 238         }
 239         if (locked_light == On) {
 240                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kLockedFieldNumber, &active_lights);
 241         } else if (locked_light == Blinking) {
 242                 activate_mapped_light(*mapping_proto, MIDIMappingProto::kLockedBlinkingFieldNumber, &active_lights);
 243         }
 244         if (current_highlighted_camera >= 0 && current_highlighted_camera < mapping_proto->camera_size()) {
 245                 const CameraMIDIMappingProto &camera = mapping_proto->camera(current_highlighted_camera);
 246                 activate_mapped_light(camera, CameraMIDIMappingProto::kIsCurrentFieldNumber, &active_lights);
 247         }
 248
 249         // Master speed light.
 250         if (mapping_proto->has_master_speed_light()) {
 251                 unsigned controller = mapping_proto->master_speed_light().controller_number();
 252                 unsigned min = mapping_proto->master_speed_light_min();
 253                 unsigned max = mapping_proto->master_speed_light_max();
 254                 int speed_light_value = lrintf((max - min - 1) * current_speed / 2.0f) + min;
 255                 active_lights[MIDIDevice::LightKey{MIDIDevice::LightKey::CONTROLLER, controller}] = speed_light_value;
 256         }
 257
 258         // These are always enabled right now.
 259         activate_mapped_light(*mapping_proto, MIDIMappingProto::kCueInEnabledFieldNumber, &active_lights);
 260         activate_mapped_light(*mapping_proto, MIDIMappingProto::kCueOutEnabledFieldNumber, &active_lights);
 261
 262         midi_device.update_lights(active_lights);
 263 }