From 97a66c4d205630f470ec30af3b2b9a6542ad5535 Mon Sep 17 00:00:00 2001
From: Curtis Lisle <curtislisle@knowledgevis.com>
Date: Wed, 1 Jun 2022 16:20:39 -0400
Subject: [PATCH] Window presets (#515)

* window level preset buttons

* add CT to allowed scan types

* Add migration for new scan type option

* Lint fix

Co-authored-by: curtislisle <clisle@knowledgevis.com>
Co-authored-by: Anne Haley <anne.haley@kitware.com>
---
 client/src/components/ControlPanel.vue    | 40 +++++++++++++++++++++++
 miqa/core/migrations/0034_CT_scan_type.py | 40 +++++++++++++++++++++++
 miqa/core/models/scan.py                  |  1 +
 3 files changed, 81 insertions(+)
 create mode 100644 miqa/core/migrations/0034_CT_scan_type.py

diff --git a/client/src/components/ControlPanel.vue b/client/src/components/ControlPanel.vue
index 4e40f458..29d5765a 100644
--- a/client/src/components/ControlPanel.vue
+++ b/client/src/components/ControlPanel.vue
@@ -175,6 +175,20 @@ export default {
       this.representation.setWindowWidth(this.currentWindowWidth);
       this.representation.setWindowLevel(this.currentWindowLevel);
     },
+
+    // this is a preset value assigned when the user clicks a W/L preset button icon
+    setWindowLevelToHighContrast() {
+      this.setCurrentWindowWidth(1500);
+      this.setCurrentWindowLevel(-500);
+      this.representation.setWindowWidth(this.currentWindowWidth);
+      this.representation.setWindowLevel(this.currentWindowLevel);
+    },
+    setWindowLevelToLowContrast() {
+      this.setCurrentWindowWidth(6000);
+      this.setCurrentWindowLevel(500);
+      this.representation.setWindowWidth(this.currentWindowWidth);
+      this.representation.setWindowLevel(this.currentWindowLevel);
+    },
     navigateToFrame(frameId) {
       if (frameId && frameId !== this.$route.params.frameId) {
         this.$router
@@ -577,6 +591,32 @@ export default {
                         </v-slider>
                       </v-col>
                     </v-row>
+
+                    <v-row>
+                      <v-col cols="4">
+                        Window/Level Presets
+                      </v-col>
+                      <v-col cols="4">
+                        <v-btn
+                          small
+                          depressed
+                          @click="setWindowLevelToHighContrast"
+                        >
+                          High Contrast
+                        </v-btn>
+                      </v-col>
+
+                      <v-col cols="4">
+                        <v-btn
+                          small
+                          depressed
+                          @click="setWindowLevelToLowContrast"
+                        >
+                          Low Contrast
+                        </v-btn>
+                      </v-col>
+                    </v-row>
+
                     <v-row class="mx-0">
                       <v-col
                         cols="12"
diff --git a/miqa/core/migrations/0034_CT_scan_type.py b/miqa/core/migrations/0034_CT_scan_type.py
new file mode 100644
index 00000000..08d96ae9
--- /dev/null
+++ b/miqa/core/migrations/0034_CT_scan_type.py
@@ -0,0 +1,40 @@
+# Generated by Django 3.2.13 on 2022-06-01 20:01
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0033_project_anatomy_orientation'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='scan',
+            name='scan_type',
+            field=models.CharField(
+                choices=[
+                    ('T1', 'T1'),
+                    ('T2', 'T2'),
+                    ('FMRI', 'FMRI'),
+                    ('MRA', 'MRA'),
+                    ('PD', 'PD'),
+                    ('DTI', 'DTI'),
+                    ('DWI', 'DWI'),
+                    ('ncanda-t1spgr-v1', 'ncanda-t1spgr-v1'),
+                    ('ncanda-mprage-v1', 'ncanda-mprage-v1'),
+                    ('ncanda-t2fse-v1', 'ncanda-t2fse-v1'),
+                    ('ncanda-dti6b500pepolar-v1', 'ncanda-dti6b500pepolar-v1'),
+                    ('ncanda-dti30b400-v1', 'ncanda-dti30b400-v1'),
+                    ('ncanda-dti60b1000-v1', 'ncanda-dti60b1000-v1'),
+                    ('ncanda-grefieldmap-v1', 'ncanda-grefieldmap-v1'),
+                    ('ncanda-rsfmri-v1', 'ncanda-rsfmri-v1'),
+                    ('PET', 'PET'),
+                    ('CT', 'CT'),
+                ],
+                default='T1',
+                max_length=25,
+            ),
+        ),
+    ]
diff --git a/miqa/core/models/scan.py b/miqa/core/models/scan.py
index e03b1412..6517a069 100644
--- a/miqa/core/models/scan.py
+++ b/miqa/core/models/scan.py
@@ -22,6 +22,7 @@
     ('ncanda-grefieldmap-v1', 'ncanda-grefieldmap-v1'),
     ('ncanda-rsfmri-v1', 'ncanda-rsfmri-v1'),
     ('PET', 'PET'),
+    ('CT', 'CT'),
 ]