Dallinger/Dallinger

// Settings
var PPU = 3;      // Pixels per base unit.
var xMax = 100;   // Maximum size of a bar in base units.
var trialIndex = 0;
var stimulusYSize = 0;

var my_node_id;

// Create the agent.
create_agent = function() {
  dallinger.createAgent()
    .done(function (resp) {
      my_node_id = resp.node.id;
      get_info();
    })
    .fail(function (rejection) {
      // A 403 is our signal that it's time to go to the questionnaire
      if (rejection.status === 403) {
        dallinger.allowExit();
        dallinger.goToPage('questionnaire');
      } else {
        dallinger.error(rejection);
      }
    });
};

get_info = function() {
  dallinger.getReceivedInfos(my_node_id).done(function (resp) {
    r = resp.infos[0].contents;
    data = JSON.parse(r);

    // Get training values
    xTrain = data.x;
    yTrain = data.y;

    N = xTrain.length * 2;
    $("#total-trials").html(N);
    yTrainReported = [];

    // Get test values.
    // half are from training the rest are new
    allX = range(1, xMax);
    xTestFromTraining = randomSubset(xTrain, N / 4);
    xTestNew = randomSubset(allX.diff(xTrain), N / 4);
    xTest = shuffle(xTestFromTraining.concat(xTestNew));
    yTest = [];
    drawUserInterface();
    proceedToNextTrial();
  });
};

//
// Draw the user interface.
//
drawUserInterface = function () {
  paper = Raphael(0, 50, 600, 400);

  inset = 1;

  // Draw the X bar background.
  backgroundX = paper.rect(50, 50, 300, 25 - 2 * inset);
  backgroundX.attr("stroke", "#CCCCCC");
  backgroundX.attr("stroke-dasharray", "--");

  // Draw the X bar.
  stimulusX = paper.rect(50, 50 - inset, 0, 25);
  stimulusX.attr("fill", "#0B486B");
  stimulusX.attr("stroke", "none");

  // Draw the Y bar background.
  backgroundY = paper.rect(450, 400 - 300, 25 - 2 * inset, 300);
  backgroundY.attr("stroke", "#CCCCCC");
  backgroundY.attr("stroke-dasharray", "--");

  // Draw the Y bar.
  stimulusY = paper.rect(450 - inset, 400, 25, 0);
  stimulusY.attr("fill", "#C02942");
  stimulusY.attr("stroke", "none");

  // Draw the feedback bar.
  feedback = paper.rect(500, 400, 25, 0);
  feedback.attr("fill", "#CCCCCC");
  feedback.attr("stroke", "none");
  feedback.hide();

  if (trialIndex === 0) {
    // Track the mouse.
    $(document).mousemove( function(e) {
      y = e.pageY - 50;
      stimulusYSize = bounds(400 - y, 1 * PPU, xMax * PPU);
      stimulusY.attr({ y: 400 - stimulusYSize, height: stimulusYSize });
    });

    Mousetrap.bind("space", proceedToNextTrial, "keydown");
    document.addEventListener('click', mousedownEventListener);
  }
};

proceedToNextTrial = function () {
  // Prevent repeat keypresses.
  Mousetrap.pause();

  // Increment the trial counter.
  trialIndex = trialIndex + 1;
  $("#trial-number").html(trialIndex);

  // Set up the stimuli.
  if (trialIndex < N / 2) {
    stimulusXSize = xTrain[trialIndex - 1] * PPU;
  } else {
    stimulusXSize = xTest[trialIndex - N / 2 - 1] * PPU;
  }
  stimulusX.attr({ width: stimulusXSize });
  stimulusX.show();
  stimulusY.show();

  // If this was the last trial, finish up.
  if (trialIndex === N + 1) {
    document.removeEventListener('click', mousedownEventListener);
    paper.remove();

    // Send data back to the server.
    response = JSON.stringify({"x": xTest, "y": yTest});

    dallinger.createInfo(my_node_id, {
      contents: response,
      info_type: "Info"
    }).done(function (resp) {
      create_agent();
    });
  } else {
    clicked = false;
  }
};

//
// Listen for clicks and act accordingly.
//
function mousedownEventListener(event) {
  if (clicked === false) {
    yNow = Math.round(stimulusYSize / PPU);

    // Training phase
    if (trialIndex < N / 2) {
      yTrue = yTrain[trialIndex - 1];
      // if they are wrong show feedback
      yTrainReported.push(yNow);
      feedback.attr({ y: 400 - yTrue * PPU, height: yTrue * PPU });
      feedback.show();
      feedback.animate({fill: "#666"}, 100, "<", function () {
        this.animate({fill: "#CCC"}, 100, ">");
      });
      // Move on to next trial if response is correct.
      if(Math.abs(yNow - yTrue) < 4) {
        clicked = true;
        feedback.hide();
        stimulusX.hide();
        stimulusY.hide();
        Mousetrap.resume();
      }
    // Testing phase
    } else if (trialIndex <= N) {
      clicked = true;
      $("#training-or-testing").html("Testing");
      yTest.push(yNow);
      feedback.hide();
      stimulusX.hide();
      stimulusY.hide();
      Mousetrap.resume();
    }
  }
}