// Copyright 2013 Olivier Gillet, 2015 Tim Churches
//
// Author: Olivier Gillet (ol.gillet@gmail.com)
// Modifications: Tim Churches (tim.churches@gmail.com)
// Modifications may be determined by examining the differences between the last commit
// by Olivier Gillet (pichenettes) and the HEAD commit at
// https://github.com/timchurches/Mutated-Mutables/tree/master/peaks
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//
// See http://creativecommons.org/licenses/MIT/ for more information.
//
// -----------------------------------------------------------------------------
//
// 808-style snare drum.

#include "peaks/drums/snare_drum.h"

#include <cstdio>

#include "stmlib/utils/dsp.h"
#include "stmlib/utils/random.h"

#include "peaks/resources.h"

namespace peaks {

using namespace stmlib;

void SnareDrum::Init() {
  excitation_1_up_.Init();
  excitation_1_up_.set_delay(0);
  excitation_1_up_.set_decay(1536);

  excitation_1_down_.Init();
  excitation_1_down_.set_delay(1e-3 * 48000);
  excitation_1_down_.set_decay(3072);

  excitation_2_.Init();
  excitation_2_.set_delay(1e-3 * 48000);
  excitation_2_.set_decay(1200);

  excitation_noise_.Init();
  excitation_noise_.set_delay(0);

  body_1_.Init();
  body_2_.Init();

  noise_.Init();
  noise_.set_resonance(2000);

  set_tone(0);
  set_snappy(32768);
  set_decay(32768);
  set_frequency(0);
}

void SnareDrum::Process(
    const GateFlags* gate_flags, int16_t* out, size_t size) {
  while (size--) {
    GateFlags gate_flag = *gate_flags++;
    if (gate_flag & GATE_FLAG_RISING) {
      excitation_1_up_.Trigger(15 * 32768);
      excitation_1_down_.Trigger(-1 * 32768);
      excitation_2_.Trigger(13107);
      excitation_noise_.Trigger(snappy_);
    }

    int32_t excitation_1 = 0;
    excitation_1 += excitation_1_up_.Process();
    excitation_1 += excitation_1_down_.Process();
    excitation_1 += !excitation_1_down_.done() ? 2621 : 0;

    int32_t body_1 = body_1_.Process<SVF_MODE_BP>(excitation_1) + (excitation_1 >> 4);

    int32_t excitation_2 = 0;
    excitation_2 += excitation_2_.Process();
    excitation_2 += !excitation_2_.done() ? 13107 : 0;

    int32_t body_2 = body_2_.Process<SVF_MODE_BP>(excitation_2) + (excitation_2 >> 4);
    int32_t noise_sample = Random::GetSample();
    int32_t noise = noise_.Process<SVF_MODE_BP>(noise_sample);
    int32_t noise_envelope = excitation_noise_.Process();
    int32_t sd = 0;
    sd += body_1 * gain_1_ >> 15;
    sd += body_2 * gain_2_ >> 15;
    sd += noise_envelope * noise >> 15;
    CLIP(sd);
    *out++ = sd;
  }
}

// randomised version
void RandomisedSnareDrum::Init() {
  excitation_1_up_.Init();
  excitation_1_up_.set_delay(0);
  excitation_1_up_.set_decay(1536);

  excitation_1_down_.Init();
  excitation_1_down_.set_delay(1e-3 * 48000);
  excitation_1_down_.set_decay(3072);

  excitation_2_.Init();
  excitation_2_.set_delay(1e-3 * 48000);
  excitation_2_.set_decay(1200);

  excitation_noise_.Init();
  excitation_noise_.set_delay(0);

  body_1_.Init();
  body_2_.Init();

  noise_.Init();
  noise_.set_resonance(2000);

  set_tone(0);
  set_snappy(32768);
  set_decay(32768);
  set_frequency(0);
  base_frequency_ = 0 ;
  last_frequency_ = 0;
  last_random_hit_ = 32768 ;
  randomised_hit_ = 65535 ;
}

void RandomisedSnareDrum::Process(
    const GateFlags* gate_flags, int16_t* out, size_t size) {
  while (size--) {
    GateFlags gate_flag = *gate_flags++;
    if (gate_flag & GATE_FLAG_RISING) {
      // randomise parameters
      // frequency
      uint32_t random_value = stmlib::Random::GetWord() ;
      bool freq_up = (random_value > 2147483647) ? true : false ;
      int32_t randomised_frequency = freq_up ?
                                     (last_frequency_ + (frequency_randomness_ >> 2)) :
                                     (last_frequency_ - (frequency_randomness_ >> 2));
      // Check if we haven't walked out-of-bounds, and if so, reverse direction on last step
      if (randomised_frequency < -32767 || randomised_frequency > 32767) {
        // flip the direction
        freq_up = !freq_up ;
        randomised_frequency = freq_up ?
                                     (last_frequency_ + (frequency_randomness_ >> 2)) :
                                     (last_frequency_ - (frequency_randomness_ >> 2));
      }
      // constrain randomised frequency - probably not necessary
      if (randomised_frequency < -32767) {
        randomised_frequency = -32767;
      } else if (randomised_frequency > 32767) {
        randomised_frequency = 32767;
      }
      // set new random frequency
      set_frequency(randomised_frequency) ;
      last_frequency_ = randomised_frequency ;

      // now randomise the hit
      // bool hit_up = (random_value > 2147483647) ? true : false ;
      // randomised_hit_ = hit_up ?
      //                                (last_random_hit_ - (hit_randomness_ >> 2)) :
      //                               (last_random_hit_ + (hit_randomness_ >> 2));
      // Check if we haven't walked out-of-bounds, and if so, reverse direction on last step
      // if (randomised_hit_ < 0 || randomised_hit_ > 65535) {
      //   // flip the direction
      //   hit_up = !hit_up ;
      //   randomised_hit_ = hit_up ?
      //                                (last_random_hit_ - (hit_randomness_ >> 2)) :
      //                                (last_random_hit_ + (hit_randomness_ >> 2));
      // }

      randomised_hit_ = last_random_hit_ + ((stmlib::Random::GetSample() * hit_randomness_) >> 16);
      // constrain randomised hit
      if (randomised_hit_ < 0) {
        randomised_hit_ = 0;
      } else if (randomised_hit_ > 65535) {
        randomised_hit_ = 65535;
      }
      last_random_hit_ = randomised_hit_;
      set_tone(randomised_hit_);
      set_decay(randomised_hit_);
      excitation_1_up_.Trigger(15 * 32768);
      excitation_1_down_.Trigger(-1 * 32768);
      excitation_2_.Trigger(13107);
      excitation_noise_.Trigger(snappy_);
    }

    int32_t excitation_1 = 0;
    excitation_1 += excitation_1_up_.Process();
    excitation_1 += excitation_1_down_.Process();
    excitation_1 += !excitation_1_down_.done() ? 2621 : 0;

    int32_t body_1 = body_1_.Process<SVF_MODE_BP>(excitation_1) + (excitation_1 >> 4);

    int32_t excitation_2 = 0;
    excitation_2 += excitation_2_.Process();
    excitation_2 += !excitation_2_.done() ? 13107 : 0;

    int32_t body_2 = body_2_.Process<SVF_MODE_BP>(excitation_2) + (excitation_2 >> 4);
    int32_t noise_sample = Random::GetSample();
    int32_t noise = noise_.Process<SVF_MODE_BP>(noise_sample);
    int32_t noise_envelope = excitation_noise_.Process();
    int32_t sd = 0;
    sd += body_1 * gain_1_ >> 15;
    sd += body_2 * gain_2_ >> 15;
    sd += noise_envelope * noise >> 15;

    // sd = (sd * (32767 + (randomised_hit_ >> 1))) >> 16;
    sd = (sd * (16383 + (randomised_hit_ >> 1) + (randomised_hit_ >> 2) )) >> 16;
    CLIP(sd);
    *out++ = sd;
  }
}

}  // namespace peaks
