<?php

  /* add ensemble/copyright/ucd header here */

  //Author: Fred Barrett
  //Janata Lab
  //Center for Mind and Brain
  //Developed for the Ensemble Project

  //this is the beginning of a function set that will
  //interface with EchoNest to assist with:
  // - creating and populating EchoNest catalogs
  // - playlist generation, based upon Ensemble EchoNest catalogs
  // - playlist generation, based upon responses to the STOMP

  /*
   FUNCTIONS
  */

  // this function converts enum responses to real data
function enum2data($enum) {
  return log($enum,2)+1;
}

// this function retrieves either the most or least preferred
// genres endorsed by the current participant, and returns an
// array of the genre names matching the requested preference
//
// $preferred = TRUE/false - get the preferred or the disliked genres from STOMP

function get_stomp_genres($preferred) {
  $oh = fopen('/home/fbarrett/data/logs/echonest_ensemble/ensemble_echonest_api_get_stomp_genres.log','a+');
  fwrite($oh,"\n\ndate: " . date('MdYHis') . "\n\n");

  // initialize static variables for this function
  $stomp_formid = 395;
  $stomp_qid = 954;
  $genres = array('alternative','bluegrass','blues','classical','country','dance','electronica',
		  'folk','funk','gospel','heavy+metal','hip+hop','international','jazz','new+age',
		  'oldies','opera','pop','punk','rap','reggae','religious','rock','soul',
		  'soundtracks');
  
  // get this ppt's STOMP responses
  $responses = array();

  $mysql_str = sprintf("SELECT response_enum FROM %s WHERE subject_id=%s AND session_id=%d ".
		       "AND form_id=%d AND question_id=%d ORDER BY subquestion",
		       $_SESSION['response_table'],
		       GetSQLValueString($_SESSION['subject_id'],'text'),
		       GetSQLValueString($_SESSION['session_id'],'int'),
		       GetSQLValueString($stomp_formid,'int'),
		       GetSQLValueString($stomp_qid,'int'));
  fwrite($oh,"mysql_str: {$mysql_str}\n");
  $results = mysql_query($mysql_str) or die(mysql_error());
  $nrows = mysql_num_rows($results);
  fwrite($oh,"nrows: $nrows\n");

  while ($row = mysql_fetch_assoc($results)) {
    array_push($responses,enum2data($row['response_enum']));
  }

  // get, return target (most or least preferred) genres
  $target_genres = array();
  for ($i=0; $i<count($responses); $i++) {
    if ($preferred) {
      if ($responses[$i] == max($responses)) {
	fwrite($oh,"max response ({max($responses)}: {$genres[$i]}\n");
	array_push($target_genres,$genres[$i]);
      } elseif ((max($responses) == 7) && $responses[$i] == 6) {
	fwrite($oh,"response of 6: {$genres[$i]}\n");
	array_push($target_genres,$genres[$i]);
      }
    } else {
      if ($responses[$i] == min($responses)) {
	fwrite($oh,"min response ({min($responses)}: {$genres[$i]}\n");
	array_push($target_genres,$genres[$i]);
      } elseif ((min($responses) == 1) && $responses[$i] == 2) {
	fwrite($oh,"response of 1: {$genres[$i]}\n");
	array_push($target_genres,$genres[$i]);
      }
    }
  }

  fclose($oh);

  return $target_genres;
}

// this function returns an array of genres rated in the STOMP_v2_EchoNest form
// and weights each genre given the preference ratings given in that form
//
// $preferred = TRUE/false - weights stimuli either as rated (TRUE) or inverse 
//   to what was rated (false)

function get_stomp_weighted_genres($preferred) {
  $oh = fopen('/home/fbarrett/data/logs/echonest_ensemble/ensemble_echonest_api_get_stomp_genres.log','a+');
  fwrite($oh,"\n\ndate: " . date('MdYHis') . "\n\n");

  // initialize static variables for this function
  $stomp_formid = 395;
  $stomp_qid = 954;
  $genres = array('alternative','bluegrass','blues','classical','country','dance','electronica',
		  'folk','funk','gospel','heavy+metal','hip+hop','international','jazz','new+age',
		  'oldies','opera','pop','punk','rap','reggae','religious','rock','soul',
		  'soundtracks');
  
  // get this ppt's STOMP responses
  $responses = array();

  $mysql_str = sprintf("SELECT response_enum FROM %s WHERE subject_id=%s AND session_id=%d ".
		       "AND form_id=%d AND question_id=%d ORDER BY subquestion",
		       $_SESSION['response_table'],
		       GetSQLValueString($_SESSION['subject_id'],'text'),
		       GetSQLValueString($_SESSION['session_id'],'int'),
		       GetSQLValueString($stomp_formid,'int'),
		       GetSQLValueString($stomp_qid,'int'));
  fwrite($oh,"mysql_str: {$mysql_str}\n");
  $results = mysql_query($mysql_str) or die(mysql_error());
  $nrows = mysql_num_rows($results);
  fwrite($oh,"nrows: $nrows\n");

  while ($row = mysql_fetch_assoc($results)) {
    array_push($responses,enum2data($row['response_enum']));
  }

  // get, return target (most or least preferred) genres
  $target_genres = array();
  for ($i=0; $i<count($responses); $i++) {
    if ($preferred) {
      $target_genres[$genres[$i]] = $responses[$i]/4;
    } else {
      $target_genres[$genres[$i]] = 2-$responses[$i]/4;
    }
    fwrite($oh,"{$genres[$i]}: {$target_genres[$genres[$i]]}\n");
  }

  fclose($oh);

  return $target_genres;
}

// this function uses http 'GET' to interface with EchoNest
function get_echonest_data($api,$params) {

  $oh = fopen('/home/fbarrett/data/logs/echonest_ensemble/ensemble_echonest_api_get_en_data.log','a+');
  fwrite($oh,"\n\ndate: " . date('MdYHis') . "\n\n");

  foreach ($params as $key=>$val) {
    fwrite($oh,"param: $key - $val\n");
  }

  // construct the URL
  $url = "http://developer.echonest.com/api/v4/{$api}?";

  // concatenate key=>value pairs from params, delimit with '&'
  foreach ($params as $key=>$value) {
    if (is_array($value)) {
      // nested array: probably description
      foreach ($value as $nkey => $nval) {
	if (is_array($nval)) {
	  foreach ($nval as $lnval) {
	    //$url .= '&' . urlencode($key) . '=' . urlencode($nkey) . ':' . urlencode($lnval);
	    if (strcmp($nkey,'style') == 0) {
	      $url .= '&' . $key . '=' . $nkey . '^' . $lnval;
	    } else {
	      $url .= '&' . $key . '=' . $nkey . ':' . $lnval;
	    }
	  }
	} else {
	  if (empty($nval)) {
	    //$url .= '&' . urlencode($key) . '=' . urlencode($nkey);
	    $url .= '&' . $key . '=' . $nkey;
	  } else {
	    //$url .= '&' . urlencode($key) . '=' . urlencode($nkey) . ':' . urlencode($nval);
	    if (strcmp($key,'style') == 0) {
	      $url .= '&' . $key . '=' . $nkey . '^' . $nval;
	    } else {
	      $url .= '&' . $key . '=' . $nkey . ':' . $nval;
	    }
	  }
	}
      }
    } else {
      //$url .= '&' . urlencode($key) . '=' . urlencode($value);
      $url .= '&' . $key . '=' . $value;
    }
  }

  // get the playlist
  fwrite($oh,"url: $url\n");
  $json_obj = file_get_contents($url);

  // check status
  if (is_bool($json_obj)) {
    fwrite($oh,"error fetching url\n");
    return '';
  } else {
    fwrite($oh,"json: {$json_obj}\n");
    $jdata = json_decode($json_obj);
  }
  $jstat = $jdata->{'response'}->{'status'};
  if ($jstat->{'code'} > 0) {
    fwrite($oh,"error: {$jstat->{'message'}}\n");
    die('ERROR: ' . $jstat->{'message'});
  }

  fclose($oh);
  return $jdata->{'response'};
}

// this function uses http 'POST' to interface with EchoNest
function post_echonest_data($api,$params) {
  $url = "http://developer.echonest.com/api/v4/$api";

  $ch = curl_init($url);
  curl_setopt($ch,CURLOPT_POST,1);

  if (!empty($params)) {
    //    $postflds = '';
    //foreach ($params as $key=>$value){
    //  $postflds .= urlencode($key) . '=' . urlencode($value) . '&';
    //}
    //$postflds = substr($postflds,0,strlen($postflds)-1);
    //curl_setopt($ch,CURLOPT_POSTFIELDS,$postflds);
    curl_setopt($ch,CURLOPT_POSTFIELDS,$params);
    //print "post fields: $postflds\n";
  }
  curl_setopt($ch,CURLOPT_RETURNTRANSFER,true);

  $data = curl_exec($ch);
  curl_close($ch);
  
  $jdata = json_decode($data);
  return $jdata->{'response'};
}

// this function creates and/or finds the id for an EchoNest catalog
// must define: $params['api_key'], $params['type'], $params['name']
function get_echonest_catalog($params) {
  $catdata = post_echonest_data('catalog/create',$params);

  $statcode = $catdata->{'status'}->{'code'};
  if ($statcode) {
    print $catdata->{'status'}->{'message'} . "\n\n";
    print "Trying to fetch a current catalog ID\n";
    
    // fetch current catalog ID for the given name
    unset($params['type']);
    
    $data = get_echonest_data('catalog/profile',$params);
    
    $statcode = $data->{'status'}->{'code'};
    if ($statcode) {
      print $data->{'status'}->{'message'} . "\n\n";
      return;
    } else {
      $en_cat_id = $data->{'catalog'}->{'id'};
    }
  } else {
    $en_cat_id = $catdata->{'id'};
  }
  return $en_cat_id;
}