---
output: 
  rmarkdown::html_vignette:
    toc: true
    keep_md: true
vignette: >
  %\VignetteIndexEntry{Manual for the RCM pacakage}
  %\VignetteEngine{knitr::rmarkdown}
  %\VignetteEncoding{UTF-8}
---

# RCM package: vignette

\setcounter{tocdepth}{5}
\tableofcontents

# Introduction

The RCM package combines unconstrained and constrained ordination of microbiome 
read count data into a single package. The package functions allow fitting 
of the ordination model, plotting it and performing diagnostic checks.

# Publication

The underlying method of the RCM package is described in detail in the following article: 
["A unified framework for unconstrained and constrained ordination of microbiome read count data"](https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0205474).

# Installation

The package can be installed and loaded using the following commands:

```{r load-packages, warning=FALSE, message=FALSE, echo=FALSE}
knitr::opts_chunk$set(cache = TRUE, autodep = TRUE,
                      warning = FALSE, message = FALSE, echo = TRUE, eval = TRUE, 
                      tidy = TRUE, fig.width = 9, fig.height = 6, purl = TRUE, 
                      fig.show = "hold", cache.lazy = FALSE)
palStore = palette()
#Load all fits, to avoid refitting every time rebuilding the vignette
load(system.file("fits", "zellerFits.RData", package = "RCM"))
```

```{r install, eval = FALSE}
library(BiocManager)
install("RCM", update = FALSE)
```

```{r loadRCMpackage}
suppressPackageStartupMessages(library(RCM))
cat("RCM package version", as.character(packageVersion("RCM")), "\n")
```

## Dataset

As example data we use a study on the microbiome of colorectal cancer patients
"Potential of fecal microbiota for early-stage detection of colorectal cancer"
(2014) by Zeller _et al._.

```{r loadZellerData}
data(Zeller)
library(phyloseq)
```

## Unconstrained RCM

### Fitting the unconstrained RCM

The unconstrained RC(M) method represents all variability present in the data, 
regardless of covariate information. It should be used as a first step 
in an exploratory analysis. The _RCM_ function is the front end function 
for this purpose, behind it is the *RCM_NB* function that does the hard work
but requires numeric matrix imputs. We first fit a model with two dimensions,

```{r fitUnconstrainedRCM, eval = FALSE}
ZellerRCM2 = RCM(Zeller, k=2, round = TRUE)
```

which took `r round(ZellerRCM2$runtime,1)` minutes. 
Here we supplied a phyloseq object to the RCM function which is preferable. 
Alternatively one can also provide a count matrix with samples in the rows
and taxa in the columns, but then all plotting variables should be supplied
manually as vectors.

#### Adding dimensions

By default, only 2 dimensions are fitted, 
but more can be requested throught the _k_ argument.

```{r ThreeDimensions, eval = FALSE}
ZellerRCM3 = RCM(Zeller, k = 3, round = TRUE)
```

The total runtime for all dimensions combined was 
`r round(ZellerRCM3$runtime,1)` minutes.

#### Conditioning

In order to condition on certain variables, supply their names to the
_confounders_ argument. Here we condition on the _country_ variable

```{r condition, eval = FALSE}
ZellerRCM2cond = RCM(Zeller, k = 2, round = TRUE, confounders = c("Country"))
```

Conditioning can be applied for unconstrained as well as constrained RCM,
the ensuing plots have exactly the same interpretation as before, 
only now the variability due to the confounding variable has been filtered out.

### Plotting the uconstrained RCM

#### Monoplots

To plot only samples or taxa, specify the _plotType_ argument
in the generic _plot_ function.

```{r plotUnconstrainedRCMsam}
plot(ZellerRCM2, plotType = "samples")
```

No clear signal is present at first sight. Note also that the plot
is rectangular according to the values of the importance parameters $\psi$.
In order to truthfully represent the distances between samples all axis
must be on the same scale. We can add a colour code for the cancer diagnosis
contained in the phyloseq object.

```{r plotUnconstrainedRCMsamCol}
plot(ZellerRCM2, plotType = "samples", samColour = "Diagnosis")
```

It is clear that some of the variability of the samples is explained
by the cancer status of the patients.

We can also add a richness measure as a colur,
see ?phyloseq::estimate_richness for a list of available richness measures.
Here we plot the Shannon diversity

```{r plotUnconstrainedRCMsamColShannon}
plot(ZellerRCM2, plotType = "samples", samColour = "Shannon")
```

In order to plot only the species, modify the _plotType_ argument.

```{r plotUnconstrainedRCMspec}
plot(ZellerRCM2, plotType = "species")
```

The researchers found that species from the Fusobacteria genus
are associated with cancer. We can plot only these species using
a regular expression.

```{r plotUnconstrainedRCMspec2}
plot(ZellerRCM2, plotType = "species", taxRegExp = "Fusobacter", 
     taxLabels = TRUE)
```

It is clear that these Fusobacterium species behave very differently 
between the species. We can also colour the species plots by phylogenetic level,
e.g. order level, if available in the dataset.

```{r plotUnconstrainedRCMspec3}
plot(ZellerRCM2, plotType = "species", taxLabels = TRUE, taxCol = "Order")
```

#### Biplots

Finally we can combine both plots into an interpretable biplot,
which is the default for unconstrained RC(M). 
To avoid overplotting we only show the taxa
with the 10 most important departures from independence.

```{r plotUnconstrainedRCMall}
plot(ZellerRCM2, taxNum = 10, samColour = "Diagnosis")
```

Samples are represented by dots, taxa by arrows. 
Both represent vectors with the origin as starting point.

Valid interpretatations are the following:

 - Samples (endpoints of sample vectors, the red dots) close together
 depart from independence in a similar way
 - The orthogonal projection of the taxon arrows on the sample arrows
 are proportional to the departure from independence of that taxon
 in that sample on the log scale, in the first two dimensions. 
 For example Fusobacterium mortiferum is more abundant than average
 in samples on the left side of the plot, and more abundant in samples
 on the right side.
 - The importance parameters $\psi$ shown for every axis reflect
 the relative importance of the dimensions
 
Distances between endpoints of taxon vectors are meaningless.

##### Adding projections

We can also graphically highlight the departure from independence
for a particular taxon and sample using the _addOrthProjection_ function:

```{r plotUnconstrainedRCMhighlight}
tmpPlot = plot(ZellerRCM2, taxNum = 10, samColour = "Diagnosis",
               returnCoords = TRUE)
addOrthProjection(tmpPlot, species = "Alloprevotella tannerae",
                  sample = c(-1.2,1.5))
```

The projection of the species vector is graphically shown here, 
the orange bar representing the extent of the departure from independence. 
Note that we providid the exact taxon name, 
and approximate sample coordinates visually derived from the graph, 
but any combination of both is possible.

We can then plot any combination of two dimensions we want,
e.g. the first and the third.

```{r plotAddedDimension}
plot(ZellerRCM3, Dim = c(1,3), samColour = "Diagnosis", taxNum = 6)
```

The third dimension also correlates with a separation of cancer patients vs. 
healthy and small adenoma patients.

### Assessing the goodness of fit

Some taxa (or samples) may not follow a negative binomial distribution,
or their departures from independence may not be appropriately represented
in low dimensions. We visualize these taxa and samples through their deviances,
which are the squared sums of their deviance residuals. 
This allows us to graphically identify taxa and samples that are poorly
represented in the current ordination.

The deviances per sample can be plotted by just supplying the argument
"Deviance" to _samColour_

```{r plotUnconstrainedRCMsamColDev}
plot(ZellerRCM2, plotType = "samples", samColour = "Deviance", samSize = 2.5)
```

Samples with the largest scores exhibit the poorest fit.
This may indicate that samples with strong departures from independence
acquire large scores, but still are not well represented in lower dimensions. 
Especially the one bottom right may be a problematic case.

The same principle can be applied to the taxa

```{r plotUnconstrainedRCMtaxDev}
plot(ZellerRCM3, plotType = "species", taxCol = "Deviance", samSize = 2.5,
     Dim = c(1,2), arrowSize = 0.5)
```

For the taxa it appears to be the taxa with smaller scores are the more poorly
fitted ones. Note that since the count table is not square, we cannot compare
sample and taxon deviances. They have not been calculated based
on the same number of taxa. Also, one cannot do chi-squared tests
based on the deviances since this is not a classical regression model,
but an overparametrized one.

## Constrained RCM

In this second step we look for the variability in the dataset explained
by linear combinations of covariates that maximally separate the niches
of the species. This should be done in a second step, and preferably only
with variables that are believed to have an impact on the species' abundances.
Here we used the variables age, gender, BMI, country and diagnosis
in the gradient. 
In this analysis all covariates values of a sample are projected
onto a single scalar, the environmental score of this sample. 
The projection vector is called the environmental gradient,
the magnitude of its components reveals the importance of each variable. 
The taxon-wise response functions then describe how the 
logged mean abundance depends on the environmental score.

### Fitting the constrained RCM model

In order to request a constrained RCM fit it suffises to supply
the names of the constraining variables to the _covariates_ argument.
The shape of the response function can be either "linear", "quadratic"
or "nonparametric" and must be provided to the _responseFun_ argument.
Here we illustrate the use of linear and nonparametric response functions.
Linear response functions may be too simplistic, they have the advantage
of being easy to interpret (and plot). Non-parametric ones (based on splines)
are more flexible but are harder to plot.

```{r constrLinAndNP, eval = FALSE}
#Linear
ZellerRCM2constr = RCM(Zeller, k = 2, round = TRUE, 
                       covariates = c("Age", "Gender","BMI","Country",
                                      "Diagnosis"), responseFun = "linear")
#Nonparametric
ZellerRCM2constrNonParam = RCM(Zeller, round = TRUE, k = 2,
                               covariates = c("Age","Gender","BMI","Country",
                                              "Diagnosis"),
                               responseFun = "nonparametric")
```

### Plotting the constrained RCM model

#### Monoplots

As before we can make monoplots of only samples or taxa. 
Starting with the samples

```{r constrLinPlot}
plot(ZellerRCM2constr, plotType = c("samples"))
```

we clearly see three groups of samples appearing.

```{r constrLinPlot2}
plot(ZellerRCM2constr, plotType = c("samples"), samColour = "Diagnosis", 
     samShape = "Country")
```

One group are the healthy patients, the other two are cancer patients.
The cancer patients are separated by country.
Note that from Germany there are only cancer patients in this dataset. 
In case of non-parametric response functions,
the samples mononoplot is not interpretable.

Unique to the constrained ordinations are monoplots of the variables. 
Do note however, that the axes have not been scaled,
and trends on the x-and y-axis should be interpreted independently.

```{r plotLinVar}
plot(ZellerRCM2constr, plotType = "variables")
```

Variables far away from the origin have a strong role
in shaping the environmental gradient.

```{r plotnonParamVar}
plot(ZellerRCM2constrNonParam, plotType = "variables")
```

The envrionmental gradients are quite different from the case with the linear
response functions. Especially age is an important driver of the environmental
gradient here. Still the results for cancer diagnosis,
country and gender are similar to before.

#### Biplots

In the constrained case two different biplots are meaningful:
sample-taxon biplots and variable-taxon biplots.

##### Sample-taxon biplot

```{r plotlin2cor}
plot(ZellerRCM2constr, plotType = c("species", "samples"))
```

The interpretation is similar as before: the orthogonal projection
of a taxon's arrow on a sample represents the departure from independence
for that taxon in that sample, _explained by environmental variables_.
New is also that the taxa arrows do not start from the origin,
but all have their own starting point. This starting point represents
the environmental scores for which there is no departure from independence.
The direction of the arrow then represents in which direction
of the environmental gradient its expected abundance increases.
Again we can show this projection visually:

```{r plotlin2corVis}
tmpPlot2 = plot(ZellerRCM2constr, plotType = c("species", "samples"), 
                returnCoords = TRUE)
addOrthProjection(tmpPlot2, species = "Pseudomonas fluorescens", 
                  sample = c(-12,7))
```

Note that the projection bar does not start from the origin in this case either.

##### Variable-taxon biplot

```{r plotlin3}
plot(ZellerRCM2constr, plotType = c("species", "variables"))
```

The projection of species arrows on environmental variables
(starting from the origin) represents the sensitivity of this taxon
to changes in this variables. Note that the fact that the arrows
of BMI and gender are of similar length indicates that one _standard deviation_ 
in BMI has a similar effect to gender.

Also this interpretation we can show visually on the graph

```{r plotlin3Vis}
tmpPlot3 = plot(ZellerRCM2constr, plotType = c("species", "variables"), 
                returnCoords = TRUE)
addOrthProjection(tmpPlot3, species = "Pseudomonas fluorescens", 
                  variable = "DiagnosisSmall_adenoma")
```

We observe that country and diagnosis are the main drivers
of the environmental gradient. We also see that the healthy status
has a very similar to the small adenoma status, 
but that they are very different from cancer patients.

#### Triplot

Triplots combine all the three components in a single plot. 
This is the default behaviour of the _plot_ function.

```{r plotlin3Triplot}
plot(ZellerRCM2constr)
```

Note that the samples and the environmental variables cannot be related 
to each other, but the sample-taxon and variable-taxon relationships
discussed before are still valid on the triplot.

For the non-parametric response functions we can only plot one dimensional
triplots, whereby the shape of the response functions of the most strongly 
reacting taxa is shown. For this we use the _plotRespFun_ function.

```{r plotNPTriplot}
plotRespFun(ZellerRCM2constrNonParam, taxa = NULL ,subdivisions = 50L,
            yLocVar = c(-30, -50,-75,-62.5,-30,-62.5,-70,-50,-30)*0.225, 
            Palette = "Set1", angle = 90, yLocSam = -20,  axisTitleSize = 16, 
            axisLabSize = 11, legendTitleSize = 18, legendLabSize = 12, 
            samShape = "Diagnosis", labSize = 5)
```

The first dimension is shown by default, the environmental scores of the samples
are shown below as dots, the y-axis shows the response functions of the 8 most
reacting species, whereby the x-axis represents the independence model.

 Let us have a look at how _Fusobacterium_ species react to this gradient,
 as the authors found them to be more abundant in cancer patients
 \cite{Zeller2014}.

```{r plotnonParamRespFunFuso}
FusoSpecies = grep("Fusobacterium",value = TRUE,
                   taxa_names(ZellerRCM2constrNonParam$physeq))
plotRespFun(ZellerRCM2constrNonParam, Dim = 1, taxa = FusoSpecies, 
            samShape = "Diagnosis")
```

### Assessing the goodness of fit

Also for constrained ordination it can be interesting
to use the deviance residuals. We can use them in the unconstrained case
by summing over taxa or samples, or plot them versus the environmental gradient
to detect lack of fit for the shape of the response function. 
For this latter goal we provide two procedures: a diagnostic plot of the taxa
with the strongest response to the environmental gradient, or an automatic trend
detection using the runs test statistic by Ward and Wolfowitz.
Checking the linearity (or Gaussian) assumption of the response function
is crucial: excessive departure invalidate the interpretation
of the ordination plot.

A deviance residual plot for the strongest responders,
made with the _residualPlot_ function, specified through the _numTaxa_ argument

```{r plotDevResp}
residualPlot(ZellerRCM2constr, whichTaxa = "response", numTaxa = 6)
```

The same taxa but with Pearson residuals 

```{r plotPearResp}
residualPlot(ZellerRCM2constr, whichTaxa = "response",
             resid = "Pearson", numTaxa = 6)
```

Most species do not exhibit any obvious pattern, although departures
seem to increase with environmental scores.

The runs test is a test that automatically attempts to detect non randomness
in the sequence of positive and negative residuals. We plot the residual plots
for the taxa with the largest run statistic (since the sample sizes
and null distributions are equal this corresponds with the smallest p-values,
but we do not do inference here because of the non-classical framework). 

```{r plotDevRuns}
residualPlot(ZellerRCM2constr, whichTaxa = "runs", resid = "Deviance", 
             numTaxa = 6)
```

For these 6 taxa we see no signal at all.

### Identifying influential observations

It can also be interesting to identify samples have the strongest influence
on the environmental gradient. For this aim we can colour the samples
by influence by provinding the "inflVar" argument:

```{r inflAge}
plot(ZellerRCM2constr, plotType = c("variables", "samples"), inflVar = "Age")
```

Below we see a couple of very influential observations. They may be very young?

```{r inflAgeCol}
plot(ZellerRCM2constr, plotType = c("variables", "samples"), samColour = "Age")
```

Indeed, a few young subjects affect the estimation of the age parameter most. 
One should always be wary of these kind of influential observations 
that strongly affect the ordination. The age coefficient is largest
in the second dimension actually, let's look at the influence on that component

```{r inflAgeFast2}
plot(ZellerRCM2constr, plotType = c("variables", "samples"), inflVar = "Age",
     Influence = TRUE, inflDim = 2)
```

When looking at categorical variable, one looks at one dummy at the time:

```{r inflDiag}
plot(ZellerRCM2constr, plotType = c("variables", "samples"), inflVar = "DiagnosisNormal", samShape ="Diagnosis" )
```

In the unconstrained case, the "psi" variables seems a logical, overall choice

```{r inflPsi}
plot(ZellerRCM2, plotType = "samples", samSize = 2.5, inflVar = "psi")
```

although it is also possible in the constrained case:

```{r inflPsiConstr}
plot(ZellerRCM2constr, plotType = "samples", samSize = 2.5, inflVar = "psi")
```

## Importance of dimensions

There are many quantities that reflect the (relative) importance
of the different dimensions.

### Importance parameters \psi

The simplest axis labels are simply the importance parameters $\psi_m$
from the RC(M)-model. They directly reflect the impact on the mean counts
in the respective dimension, as the factors to which they are multplied
are normalized. Hence the $\psi_m$'s are the only components that can grow
in size to add more weight to a dimension.

On the other hand, the sample scores _have_ already been scaled with respect
to these parameters, and they do not add addtional information to the plot
itself. Also, their order is often but not necessarily strictly decreasing.
Only the ratios of the different $\psi_m$'s are direclty interpretable. 
Still plotting the $\psi_m$'s is the default option of the _RCM_ package.

```{r plotUnconstrainedRCMpsis, fig.height = 8, fig.width = 9}
plot(ZellerRCM2cond, plotPsi = "psi")
```

### Log-likelihoods

Another way to quantify the importance of the dimensions is to compare 
their differences in log-likelihoods of model m ($ll_m$) with respect
to the saturated model ("deviances"). The log-likelihood saturared model
($ll_{sat}$) is calculated using the Poisson density, setting mean and variance
equal to the observed counts. These differences in log-likelihood
are then normalized with respect to the difference in loglikelihood between
the independence model ($ll_{independence}$) and the saturated model. 
The terms obtained for dimension $m$ is then:

$$\frac{ll_{m}-ll^*}{ll_{sat}-ll_{independence}}$$

, with $ll^*$ the log-likelihood of the lower dimension,
which can be the independence model, the model after filtering on confounders
or simply the lower dimension of the RC(M) model.

This approach has the advantages of also providing a measure of importance
for the confounders. Also it uses the saturated model as a reference
and thus provides a fraction of "total variability".

Disadvantages are the difficulty in intepreting log-likelihoods, 
and the fact that in some corner case the log-likelihood drops with higher
dimensions. This is because the estimation of the dispersions and environmental
gradients is not full maximum likelihood.

```{r plotUnconstrainedRCMlogliks, fig.height = 8, fig.width = 9}
plot(ZellerRCM2cond, plotPsi = "loglik")
```

### Inertia

As with correspondence analysis, the fraction of total inertia explained
by the different dimensions can be plotted on the axes. The inertia is defined
as the sum of squared Pearson residuals, or

$$\sum_{i=1}^n\sum_{j=1}^p \frac{(x_{ij}-e_{ijm})^2}{e_{ijm}}$$

with $x_{ij}$ the observed count and $e_{ijm}$ the expected count
under the model with $m$ dimensions. The inertia has the advantage
that also the variance explained by the filtering on confouders step
can be plotted, and that there is a measure of residual variance.

On the other hand, as we argue in the manuscript, the inertia is not a good
 measure of variability for overdispersed data, as it implicitly assumes
 the mean to equal the variance. Hence this criterion of dimension importance
 should be interpreted with caution.

```{r plotUnconstrainedRCMinertia, fig.height = 8, fig.width = 9}
plot(ZellerRCM2cond, plotPsi = "inertia")
```

## Advanced plotting

### Extracting coodinates

First of all it is important to note that the plotting coordinates can easily be extracted. This may be useful when making plots with third party software.

```{r extractCoord}
zellerCoords = extractCoord(ZellerRCM2)
str(zellerCoords)
```

Also, this can be used to obtain a list of taxa contributing to the separation of the samples, sorted from most to least important.

```{r taxaStrongestSignal}
taxaSignals = rowSums(zellerCoords$species[, c("end1","end2")]^2)
sortedTaxa = taxa_names(ZellerRCM2$physeq)[order(taxaSignals, decreasing = TRUE)]
sortedTaxa[1:10]
```

These most important taxa are the ones that are plotted by default by the $plot.RCM()$ function.

### Non-squared plots

The $RCM$ package provides the option to make plots with non squared axis. This may lead to better proportioned plots, but as it also misrepresents the ordination __WE DO NOT ADVOCATE NON-SQUARED PLOTS__!

```{r NonsquaredPlots}
plot(ZellerRCM2, axesFixed = FALSE)
```

# Session info

This vignette was generated with following version of R:

```{r sessionInfo}
sessionInfo()
```
