Generate_Document_Word_Matrix.cpp

#include <RcppArmadillo.h>
//[[Rcpp::depends(RcppArmadillo)]]
using namespace Rcpp;  

// [[Rcpp::export]]  
arma::mat Generate_Document_Word_Matrix(int number_of_docs,
                                        int number_of_unique_words,
                                        std::vector<std::string> unique_words,
                                        List Document_Words,
                                        arma::vec Document_Lengths
){
  arma::mat document_word_matrix = arma::zeros(number_of_docs,number_of_unique_words);  
  
  for(int n = 0; n < number_of_docs; ++n){
    Rcpp::Rcout << "Current Document: " << n << std::endl;
    std::vector<std::string> current = Document_Words[n];
    int length = Document_Lengths[n];
    for(int i = 0; i < length; ++i){
      int already = 0;
      int counter = 0;
      while(already == 0){
        if(counter == number_of_unique_words ){
          already = 1;
        }else{
          if(unique_words[counter] == current[i]){
            document_word_matrix(n,counter) += 1;
            already = 1;
          } 
          counter +=1;
        }
      }
    }
  }    
  return document_word_matrix;     
}