From fcdefda5861ca81cf319b4d0e0c6d01341b6db54 Mon Sep 17 00:00:00 2001
From: Bruno Alves de Faria <brunoalvesdefaria@gmail.com>
Date: Wed, 8 Jun 2016 15:23:14 -0300
Subject: [PATCH] LT-247: Creating modalities summary

---
 .../studyTimepointBrowser/studyTimepoint.html |  2 +-
 .../studyTimepointBrowser/studyTimepoint.js   |  9 -----
 .../studyTimepointBrowser.html                |  5 ++-
 .../studyTimepointBrowser.js                  | 35 +++++++++++++++++++
 .../studyTimepointStudy.html                  |  8 +++--
 5 files changed, 44 insertions(+), 15 deletions(-)

diff --git a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepoint.html b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepoint.html
index aa5a777e7..35b94f658 100644
--- a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepoint.html
+++ b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepoint.html
@@ -1,7 +1,7 @@
 <template name="studyTimepoint">
     <div class="studyTimepointWrapper">
         <div class="studyTimepoint">
-            {{#each study in (studies this.timepoint)}}
+            {{#each study in this.studies}}
                 {{>studyTimepointStudy study=study}}
             {{/each}}
         </div>
diff --git a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepoint.js b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepoint.js
index ab8dcb386..ccf061b8c 100644
--- a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepoint.js
+++ b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepoint.js
@@ -47,12 +47,3 @@ Template.studyTimepoint.events({
         $wrapper.css('max-height', $timepoint.height());
     }
 });
-
-Template.studyTimepoint.helpers({
-    studies: function(timepoint) {
-        const query = {
-            selected: true
-        };
-        return ViewerStudies.find(query);
-    }
-});
diff --git a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointBrowser.html b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointBrowser.html
index a4b281cd9..582c7ea9b 100644
--- a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointBrowser.html
+++ b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointBrowser.html
@@ -15,10 +15,9 @@
                                     </div>
                                     <div class="timepointDate pull-right m-r-1">{{formatDA timepoint.earliestDate 'D-MMM-YYYY'}}</div>
                                 </div>
-                                <!-- TODO: [design] find out where to get this information -->
-                                <div class="timepointModalities">2 MR, 2 CT</div>
+                                <div class="timepointModalities">{{modalitiesSummary timepoint}}</div>
                             </div>
-                            {{>studyTimepoint timepoint=timepoint index=@index}}
+                            {{>studyTimepoint studies=(studies timepoint) index=@index}}
                         </div>
                         <hr class="m-y-1">
                     {{/if}}
diff --git a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointBrowser.js b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointBrowser.js
index 4cfc5bc4d..0d7b5c5f8 100644
--- a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointBrowser.js
+++ b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointBrowser.js
@@ -1,3 +1,15 @@
+Template.studyTimepointBrowser.onCreated(() => {
+    const instance = Template.instance();
+
+    // Get the studies for a specific timepoint
+    instance.getStudies = timepoint => {
+        const query = {
+            selected: true
+        };
+        return ViewerStudies.find(query);
+    };
+});
+
 Template.studyTimepointBrowser.onRendered(() => {
     const instance = Template.instance();
     instance.autorun(() => {
@@ -37,6 +49,10 @@ Template.studyTimepointBrowser.helpers({
         // Returns all timepoints with sorting
         return Timepoints.find({}, sort);
     },
+    // Get the studies for a specific timepoint
+    studies(timepoint) {
+        return Template.instance().getStudies(timepoint);
+    },
     // Decides if a timepoint should be shown or omitted
     shouldShowTimepoint(timepoint, index) {
         const instance = Template.instance();
@@ -64,5 +80,24 @@ Template.studyTimepointBrowser.helpers({
         const followUp = total - index - 1;
         const parenthesis = states[index] || '';
         return `Follow-up ${followUp} ${parenthesis}`;
+    },
+    // Build the modalities summary for all timepoint's studies
+    modalitiesSummary(timepoint) {
+        const instance = Template.instance();
+
+        const studies = instance.getStudies(timepoint);
+
+        const modalities = {};
+        studies.forEach(study => {
+            const modality = study.modalities || 'UN';
+            modalities[modality] = modalities[modality] + 1 || 1;
+        });
+
+        const result = [];
+        _.each(modalities, (count, modality) => {
+            result.push(`${count} ${modality}`);
+        });
+
+        return result.join(', ');
     }
 });
diff --git a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointStudy.html b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointStudy.html
index 12edb7158..3a7a42063 100644
--- a/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointStudy.html
+++ b/Packages/lesiontracker/client/components/studyTimepointBrowser/studyTimepointStudy.html
@@ -2,7 +2,11 @@
     <div class="studyTimepointStudy">
         <div class="studyModality">
             <div class="studyModalityBox">
-                {{this.study.modalities}}
+                {{#if this.study.modalities}}
+                    {{this.study.modalities}}
+                {{else}}
+                    UN
+                {{/if}}
             </div>
             <div class="studyModalityText">
                 <div class="studyModalityDate">{{formatDA this.study.studyDate 'D-MMM-YYYY'}}</div>
@@ -11,7 +15,7 @@
         </div>
         <div class="studyTimepointThumbnails">
             {{#each thumbnail in (thumbnails this.study)}}
-            {{>thumbnailEntry thumbnail}}
+                {{>thumbnailEntry thumbnail}}
             {{/each}}
         </div>
     </div>