#include "sequencer_droplet.h"

SequencerDroplet::SequencerDroplet(DaisyPatch* m_patch,
				 DropletState m_state,
				 float sample_rate) :
  Droplet(m_patch,
	  m_state) {
  SetColumns();
  SetControls();
  AdjustSelected(-1);
}

SequencerDroplet::~SequencerDroplet() {}

void SequencerDroplet::Control() {
  Patch()->ProcessAnalogControls();
  Patch()->encoder.Debounce();
  AdjustSelected(Patch()->encoder.Increment());

  if(Patch()->gate_input[0].Trig()) {
    Step();
  }
  if(Patch()->gate_input[1].Trig()) {
    Reset();
  }

  if (control_rate_count == CONTROL_RATE_LIMIT) {
    for (size_t chn = GetChannelMin(); chn < GetChannelMax(); chn++) {
      if (std::abs(control[chn].Process()-last_control_value[chn]) > CONTROL_DEADZONE) {
	if (!InMenu()) {
	  sequence[chn+selected*num_columns] = control[chn].Process();
	} else {
	  if (chn == GetChannelMin()) {
	    sequence_length = std::max(1.0f,control[chn].Process()/4.9f*MAX_SEQUENCE_LENGTH);
	    SetColumns();
	    selected = 0;
	    AdjustSelected(-1);
	  }
	}
    }
      last_control_value[chn] = control[chn].Process();
    }
    control_rate_count = 0;
  }
  control_rate_count++;
}

void SequencerDroplet::Process(AudioHandle::InputBuffer in,
			      AudioHandle::OutputBuffer out,
			      size_t size) {
  for(size_t i = 0; i < size; i++) {
    Patch()->seed.dac.WriteValue(DacHandle::Channel::ONE,
				 sequence[step] * 819.2f);
  }
}

void SequencerDroplet::Draw() {
  int left_padding = 4+GetScreenMin();

  // Active Input
  if (!InMenu()) {
    DrawSolidRect(Patch(),
		  GetScreenMin(),
		  8+selected*8,
		  GetScreenMin()+2,
		  15+selected*8, true);
  }

  // Notes
  for (int i = 0; i < num_columns*NUM_ROWS && i < sequence_length; i++) {
    WriteString(Patch(),
		GetScreenWidth()/num_columns*(i%num_columns)+left_padding,
		8+(std::floor(i/num_columns)*8),
		FloatToString(sequence[i], 2),
		i!=step);
  }

  // Draw info bar
  DrawSolidRect(Patch(),GetScreenMin(),56,GetScreenMax(),63, InMenu());
  length_text = std::to_string(step+1);
  if (sequence_length >= 10) {
    length_text.insert(length_text.begin(), 2 - length_text.length(), '0');
  }
  length_text.append("/"+std::to_string(sequence_length));
  WriteString(Patch(),
	      2+GetScreenMin(),
	      56,
	      length_text,
	      !InMenu());
  DrawName("Sequencer");
}

void SequencerDroplet::UpdateStateCallback() {
  SetColumns();
}

void SequencerDroplet::SetControls() {
    control[0].Init(Patch()->controls[Patch()->CTRL_1],
		    0.0, 5.0f, Parameter::LINEAR);
    control[1].Init(Patch()->controls[Patch()->CTRL_2],
		    0.0, 5.0f, Parameter::LINEAR);
    control[2].Init(Patch()->controls[Patch()->CTRL_3],
		    0.0, 5.0f, Parameter::LINEAR);
    control[3].Init(Patch()->controls[Patch()->CTRL_4],
		    0.0, 5.0f, Parameter::LINEAR);
}

void SequencerDroplet::Step() {
  step = (step + 1) % sequence_length;
}

void SequencerDroplet::Reset() {
  step = 0;
}

void SequencerDroplet::SetColumns() {
  if (GetState() != DropletState::kFull) {
    num_columns = 2;
  } else {
    num_columns = 4;
  }
  num_rows = std::ceil((float)sequence_length/num_columns);
}

void SequencerDroplet::AdjustSelected(int adj) {
  selected = (num_rows+selected+adj+1) % (num_rows+1);
}

bool SequencerDroplet::InMenu() {
  return selected >= num_rows;
}