From 72bda33f39be9df6894e764a9f9bec52cc249e08 Mon Sep 17 00:00:00 2001
From: andy <agoldschmidt11@gmail.com>
Date: Tue, 25 Jun 2024 21:50:10 -0500
Subject: [PATCH] add cnames for partial get_comps

---
 src/methods_named_trajectory.jl | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/methods_named_trajectory.jl b/src/methods_named_trajectory.jl
index aaa5c35..3c0d3e5 100644
--- a/src/methods_named_trajectory.jl
+++ b/src/methods_named_trajectory.jl
@@ -71,16 +71,17 @@ end
 
 
 """
-    components(::NamedTrajectory)
+    get_components(::NamedTrajectory)
 
 Returns a NamedTuple containing the names and corresponding data matrices of the trajectory.
 """
-function get_components(traj::NamedTrajectory)
-    data = [traj[comp] for comp ∈ traj.names]
-    return NamedTuple(zip(traj.names, data))
+function get_components(cnames::Union{Tuple, AbstractVector}, traj::NamedTrajectory)
+    symbs = Tuple(c for c in cnames)
+    vals = [traj[c] for c ∈ cnames]
+    return NamedTuple{symbs}(vals)
 end
 
-
+get_components(traj::NamedTrajectory) = get_components(traj.names, traj)
 
 
 """