# Visualize combined ReaDDy and Cytosim simulation trajectories


Notebook contains steps for visualizing ReaDDy and Cytosim simulations of a
single actin fiber using [Simularium](https://simularium.allencell.org/).

- [Define visualization settings](#define-visualization-settings)
- [Visualize combined trajectories](#visualize-combined-trajectories)

```python
if __name__ != "__main__":
    raise ImportError("This module is a notebook and is not meant to be imported")
```

```python
from pathlib import Path

from subcell_pipeline.analysis.compression_metrics.compression_metric import (
    CompressionMetric,
)
from subcell_pipeline.visualization.combined_trajectory import (
    visualize_combined_trajectories,
)
```

## Define visualization settings

```python
# S3 buckets for simulation and visualization input and output files
buckets: dict[str, str] = {
    "combined": "s3://subcell-working-bucket",
    "readdy": "s3://readdy-working-bucket",
    "cytosim": "s3://cytosim-working-bucket",
}

# Name of the simulation series
series_name: str = "ACTIN_COMPRESSION_VELOCITY"

# List of condition file keys for each velocity
condition_keys: list[str] = ["0047", "0150", "0470", "1500"]

# Replicate ids for simulations
replicates: list[int] = [1, 2, 3, 4, 5]

# Number of timepoints
n_timepoints = 201

# List of simulators and colors
simulator_colors = {
    "cytosim": "#1cbfaa",
    "readdy": "#ff8f52",
}

# Temporary path to save visualization files
temp_path: Path = Path(__file__).parents[2] / "viz_outputs"
temp_path.mkdir(parents=True, exist_ok=True)

# List of compression metrics to include
metrics = [
    CompressionMetric.NON_COPLANARITY,
    CompressionMetric.PEAK_ASYMMETRY,
    CompressionMetric.AVERAGE_PERP_DISTANCE,
    CompressionMetric.CONTOUR_LENGTH,
    CompressionMetric.COMPRESSION_RATIO,
]
```


## Visualize combined trajectories

Visualize all compression simulations from ReaDDy and Cytosim together in
Simularium.

- Input: `(series_name)/samples/(series_name)_(condition_key)_(replicate).csv`
- Output: `actin_compression_cytosim_readdy.simularium`

```python
visualize_combined_trajectories(
    buckets,
    series_name,
    condition_keys,
    replicates,
    n_timepoints,
    simulator_colors,
    str(temp_path),
    metrics=metrics,
)
```