// NYU ITP

// Try mashing up 2 of the data sources.

​

// Array of words

let words = [];

​

// Word chain

let wc = {};

​

// Current word

let current_word = 0;

​

// New string

let new_string = '';

​

// Paragraph to display text

let p;

​

function preload() {

  loadStrings('virgin.txt', count);

}

​

function setup() {

  noCanvas();

  p = createP();

}

​

function draw() {

  // Pick next word every half-second

  if(frameCount%30 ==0) {

    // Add current word to new string

    new_string += ' ' + current_word;

    // All possible next words (pnw)

    print(current_word);

    let possible_next_word = wc[current_word];

    // Pick new current word (cw) of them at random

    let random_possible_next_word = floor(random(possible_next_word.length));

    current_word = possible_next_word[random_possible_next_word];

    // Display string in parapgraph element

    p.html(new_string);

  }

}

​

function count(data) {

  // Go line by line

  for (let d of data) {

    // Turn each line into an array of words

    let line = splitTokens(d);

    // Add it to 1 big array

    words = words.concat(line);

  }

​

  // Clean up all the words

  for (let w in words) {

    let word = words[w];

    // Remove punctuation

    word = word.replace(/[-_:;.,!?\(\)]/g, "");

    // Make it all lowercase

    word = word.toLowerCase();

    // Get rid of whitespace around the word

    word = word.trim();

    // If nothing is left, get rid of it

    if (word.length < 1) words.splice(w, 1);

    // Otherwise put cleaned up word back in array

    else words[w] = word;

  }

  // Collect all possible next words (pnws) for each word

  for (let w = 0; w < words.length; w++) {

    let word = words[w];

    // If we're on the last word, call it quits

    if(w >= words.length-1) break;

    // Get the next word

    let nextword = words[w+1];

    // If this word has already been indexed, 

    // add the next word to its list of possible next words

    if (word in wc) wc[word].push(nextword);

    // Otherwise, start a new list

    else wc[word] = [nextword];

  }

  // Print word count object

  console.log(wc);

  // Start with first word

  current_word = words[0];

}