From a715804880755ed12bc8ae89409a96ae8f5d2408 Mon Sep 17 00:00:00 2001
From: Bruno Alves de Faria <brunoalvesdefaria@gmail.com>
Date: Mon, 14 Nov 2016 20:30:38 -0200
Subject: [PATCH] LT-291: Allowing non-target removal

---
 .../components/bootstrap/dialog/form.html     |  2 +-
 .../client/compatibility/nonTargetTool.js     |  4 +-
 .../dialog/nonTargetMeasurement.html          | 10 +++-
 .../components/dialog/nonTargetMeasurement.js | 55 ++++++++++++++-----
 .../both/configuration/measurements.js        |  5 ++
 5 files changed, 57 insertions(+), 19 deletions(-)

diff --git a/Packages/ohif-core/client/components/bootstrap/dialog/form.html b/Packages/ohif-core/client/components/bootstrap/dialog/form.html
index eb1c5ddfa..51f4bf9a4 100644
--- a/Packages/ohif-core/client/components/bootstrap/dialog/form.html
+++ b/Packages/ohif-core/client/components/bootstrap/dialog/form.html
@@ -2,7 +2,7 @@
     <div id="{{this.id}}" class="modal fade" tabindex="-1" role="dialog">
         <div class="modal-dialog {{this.dialogClass}}" role="document">
             {{#form class='modal-content' hideValidationBox=true
-                api=instance.api schema=this.schema
+                api=(extend instance.api instance.data.api) schema=this.schema
             }}
                 <div class="modal-header">
                     {{#button class='close' action='cancel'
diff --git a/Packages/ohif-lesiontracker/client/compatibility/nonTargetTool.js b/Packages/ohif-lesiontracker/client/compatibility/nonTargetTool.js
index d3f201a84..a7f9a25f0 100644
--- a/Packages/ohif-lesiontracker/client/compatibility/nonTargetTool.js
+++ b/Packages/ohif-lesiontracker/client/compatibility/nonTargetTool.js
@@ -46,7 +46,7 @@ import { OHIF } from 'meteor/ohif:core';
 
     function changeMeasurementLocationCallback(measurementData, eventData) {
         showLocationDialog({
-            title: 'Select Lesion Location',
+            title: 'Change Lesion Location',
             measurementData,
             eventData,
             edit: true
@@ -241,7 +241,7 @@ import { OHIF } from 'meteor/ohif:core';
 
             // Draw the text
             if (data.measurementNumber) {
-                
+
                 var boundingBox = cornerstoneTools.drawTextBox(context, 'Non-Target ' + data.measurementNumber, canvasTextLocation.x, canvasTextLocation.y, color);
                 data.handles.textBox.boundingBox = boundingBox;
 
diff --git a/Packages/ohif-lesiontracker/client/components/dialog/nonTargetMeasurement.html b/Packages/ohif-lesiontracker/client/components/dialog/nonTargetMeasurement.html
index 13a2a53ff..906265a5c 100644
--- a/Packages/ohif-lesiontracker/client/components/dialog/nonTargetMeasurement.html
+++ b/Packages/ohif-lesiontracker/client/components/dialog/nonTargetMeasurement.html
@@ -1,8 +1,16 @@
 <template name="dialogNonTargetMeasurement">
-    {{#dialogForm (extend this dialogClass='modal-sm' schema=(choose this.schema instance.schema))}}
+    {{#dialogForm (extend this
+        dialogClass='modal-sm'
+        schema=(choose this.schema instance.schema)
+        api=instance.api
+    )}}
         {{>inputSelect labelClass='form-group' key='location' hideSearch=true
             options=(clone placeholder='Choose a lesion location')}}
         {{>inputSelect labelClass='form-group' key='response' hideSearch=true
             options=(clone placeholder='Choose a lesion location response')}}
+        {{#if this.edit}}
+            <hr>
+            {{#button class='btn btn-danger' action='remove'}}Remove{{/button}}
+        {{/if}}
     {{/dialogForm}}
 </template>
diff --git a/Packages/ohif-lesiontracker/client/components/dialog/nonTargetMeasurement.js b/Packages/ohif-lesiontracker/client/components/dialog/nonTargetMeasurement.js
index cfb20f7c4..d6a5439f2 100644
--- a/Packages/ohif-lesiontracker/client/components/dialog/nonTargetMeasurement.js
+++ b/Packages/ohif-lesiontracker/client/components/dialog/nonTargetMeasurement.js
@@ -1,34 +1,64 @@
 import { Template } from 'meteor/templating';
 import { Blaze } from 'meteor/blaze';
 import { SimpleSchema } from 'meteor/aldeed:simple-schema';
-import { _ } from 'meteor/underscore';
+import { OHIF } from 'meteor/ohif:core';
 import { FieldLesionLocation, FieldLesionLocationResponse } from 'meteor/ohif:lesiontracker/both/schema/fields';
 
 Template.dialogNonTargetMeasurement.onCreated(() => {
     const instance = Template.instance();
 
+    instance.measurementTypeId = 'nonTargets';
+
+    const measurementData = instance.data.measurementData;
+
     instance.schema = new SimpleSchema({
         location: FieldLesionLocation,
         response: FieldLesionLocationResponse
     });
+
+    // Remove the measurement from the collection
+    instance.removeMeasurement = () => {
+        instance.viewerData.measurementApi.deleteMeasurements(instance.measurementTypeId, {
+            _id: measurementData._id
+        });
+
+        // Refresh the image with the measurement removed
+        cornerstone.updateImage(instance.cornerstoneElement);
+    };
+
+    // Close the current dialog
+    instance.closeDialog = () => instance.$('.form-action.close').trigger('click');
+
+    instance.api = {
+        // Confirm the deletion of the current non-target measurement
+        remove() {
+            const dialogSettings = {
+                title: 'Remove Measurement',
+                message: 'Are you sure you want to remove this Non-Target measurement?'
+            };
+
+            OHIF.ui.showFormDialog('dialogConfirm', dialogSettings)
+                .then(instance.removeMeasurement);
+
+            instance.closeDialog();
+        }
+    };
 });
 
 Template.dialogNonTargetMeasurement.onRendered(() => {
     const instance = Template.instance();
 
-    const element = instance.data.eventData.element;
+    instance.cornerstoneElement = instance.data.eventData.element;
 
     const form = instance.$('form').data('component');
 
-    const viewerMain = $(element).closest('.viewerMain')[0];
-    const viewerData = Blaze.getData(viewerMain);
-    console.warn('>>>>dialogData/viewerData', instance.data, viewerData);
+    const viewerMain = $(instance.cornerstoneElement).closest('.viewerMain')[0];
+    instance.viewerData = Blaze.getData(viewerMain);
 
-    const measurementApi = viewerData.measurementApi;
+    const measurementApi = instance.viewerData.measurementApi;
 
-    const measurementTypeId = 'nonTargets';
     const measurementData = instance.data.measurementData;
-    const collection = measurementApi[measurementTypeId];
+    const collection = measurementApi[instance.measurementTypeId];
 
     // Get the current inserted measurement from the collection
     const currentMeasurement = collection.findOne({
@@ -42,7 +72,7 @@ Template.dialogNonTargetMeasurement.onRendered(() => {
     measurementData.measurementNumber = currentMeasurement.measurementNumber;
 
     // Refresh the image with the measurement number
-    cornerstone.updateImage(element);
+    cornerstone.updateImage(instance.cornerstoneElement);
 
     // Delete the measurement from collection when dialog is closed and not on edit mode
     instance.data.promise.catch(() => {
@@ -50,12 +80,7 @@ Template.dialogNonTargetMeasurement.onRendered(() => {
             return;
         }
 
-        viewerData.measurementApi.deleteMeasurements('nonTargets', {
-            _id: measurementData._id
-        });
-
-        // Refresh the image with the measurement removed
-        cornerstone.updateImage(element);
+        instance.removeMeasurement();
     });
 
     // Update the location and response after confirming the dialog data
diff --git a/Packages/ohif-measurements/both/configuration/measurements.js b/Packages/ohif-measurements/both/configuration/measurements.js
index c8b9a7d07..893a4f85b 100644
--- a/Packages/ohif-measurements/both/configuration/measurements.js
+++ b/Packages/ohif-measurements/both/configuration/measurements.js
@@ -97,6 +97,11 @@ class MeasurementApi {
         const entries = collection.find(filter).fetch();
         collection.remove(filter);
 
+        // Stop here if no entries were found
+        if (!entries.length) {
+            return;
+        }
+
         // If the filter doesn't have the measurement number, get it from the first entry
         const measurementNumber = filter.measurementNumber || entries[0].measurementNumber;