// CIDnetworks C++ code snippets.

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
double logLikeSourceC (int edge1, int edge2, int edgeval,
		       NumericMatrix logitb,
		       IntegerVector group,
		       NumericVector covariateRow,
		       NumericVector beta) {

  int jj;
  double out;
  double xbeta = 0;

  for (jj=0; jj<beta.size(); ++jj) xbeta += covariateRow[jj]*beta[jj];
  out = logitb(group[edge1-1]-1, group[edge2-1]-1) + xbeta;

  out = edgeval*out - log(1+exp(out));
  return(out);
}


// [[Rcpp::export]]
NumericVector logLikeVectorC (IntegerMatrix nby3,
		       NumericMatrix logitb,
		       IntegerVector group,
		       NumericMatrix covariateMatrix,
		       NumericVector beta) {

  NumericVector out(covariateMatrix.rows());
  double xbeta;
  int jj, rr;
  //Rcout << covariateMatrix.rows() << " " << covariateMatrix.cols() << " " << beta.size() << std::endl;

  for (rr=0; rr<covariateMatrix.rows(); ++rr) {
    //Rcout << rr << std::endl;
    xbeta = 0;
    for (jj=0; jj<beta.size(); ++jj) xbeta += covariateMatrix(rr,jj)*beta[jj];
    out[rr] = logitb(group[nby3(rr,0)-1]-1, group[nby3(rr,1)-1]-1) + xbeta;
    out[rr] = nby3(rr,2)*out[rr] - log(1+exp(out[rr]));
  }
  return(out);
}