Chart Setting - import seaborn, add default setting

minjk-bl · minjk-bl · commit 5d3e5397f30c · 2022-05-06T12:50:29.000+09:00
diff --git a/html/m_visualize/chartSetting.html b/html/m_visualize/chartSetting.html
@@ -1,5 +1,5 @@
 <body>
-    <div class="vp-grid-border-box vp-grid-col-95">
+    <div class="vp-chart-setting-body vp-grid-border-box vp-grid-col-95">
         <label for="figureWidth" class="">Figure size</label>
         <div>
             <input type="number" id="figureWidth" class="vp-input m vp-state" placeholder="width" value="12">
@@ -12,4 +12,8 @@
         <label for="fontSize" class="">Font size</label>
         <input type="number" id="fontSize" class="vp-input vp-state" placeholder="size" value="10">
     </div>
+    <label class="mt5">
+        <input type="checkbox" id="setDefault">
+        <span title="Set chart setting to default.">Set Default</span>
+    </label>
 </body>
diff --git a/js/m_visualize/ChartSetting.js b/js/m_visualize/ChartSetting.js
@@ -42,7 +42,18 @@ define([
             super._bindEvent();
 
             let that = this;
-            
+            // setting popup - set default
+            $(this.wrapSelector('#setDefault')).on('change', function() {
+                let checked = $(this).prop('checked');
+
+                if (checked) {
+                    // disable input
+                    $(that.wrapSelector('.vp-chart-setting-body input')).prop('disabled', true);
+                } else {
+                    // enable input
+                    $(that.wrapSelector('.vp-chart-setting-body input')).prop('disabled', false);
+                }
+            });
         }
 
         templateForBody() {
@@ -113,31 +124,40 @@ define([
         generateImportCode() {
             var code = new com_String();
             code.appendLine('import matplotlib.pyplot as plt');
-            code.append('import seaborn as sns');
+            code.appendLine('import seaborn as sns');
+            code.append('%matplotlib inline');
             return [code.toString()];
         }
 
         generateCode() {
             var code = new com_String();
     
             // get parameters
-            let { figureWidth, figureHeight, styleSheet, fontName, fontSize } = this.state;
-    
-            code.appendLine('import matplotlib.pyplot as plt');
-            code.appendFormatLine("plt.rc('figure', figsize=({0}, {1}))", figureWidth, figureHeight);
-            if (styleSheet && styleSheet.length > 0) {
-                code.appendFormatLine("plt.style.use('{0}')", styleSheet);
-            }
-            code.appendLine();
-    
-            code.appendLine('from matplotlib import rcParams');
-            if (fontName && fontName.length > 0) {
-                code.appendFormatLine("rcParams['font.family'] = '{0}'", fontName);
-            }
-            if (fontSize && fontSize.length > 0) {
-                code.appendFormatLine("rcParams['font.size'] = {0}", fontSize);
+            let setDefault = $(this.wrapSelector('#setDefault')).prop('checked');
+            if (setDefault == true) {
+                code.appendLine('from matplotlib import rcParams, rcParamsDefault');
+                code.append('rcParams.update(rcParamsDefault)');
+            } else {
+                // get parameters
+                let { figureWidth, figureHeight, styleSheet, fontName, fontSize } = this.state;
+        
+                code.appendLine('import matplotlib.pyplot as plt');
+                code.appendLine('import seaborn as sns');
+                code.appendFormatLine("plt.rc('figure', figsize=({0}, {1}))", figureWidth, figureHeight);
+                if (styleSheet && styleSheet.length > 0) {
+                    code.appendFormatLine("plt.style.use('{0}')", styleSheet);
+                }
+                code.appendLine();
+        
+                code.appendLine('from matplotlib import rcParams');
+                if (fontName && fontName.length > 0) {
+                    code.appendFormatLine("rcParams['font.family'] = '{0}'", fontName);
+                }
+                if (fontSize && fontSize.length > 0) {
+                    code.appendFormatLine("rcParams['font.size'] = {0}", fontSize);
+                }
+                code.append("rcParams['axes.unicode_minus'] = False");
             }
-            code.append("rcParams['axes.unicode_minus'] = False");
     
             return code.toString();
         }
diff --git a/js/m_visualize/Seaborn.js b/js/m_visualize/Seaborn.js
@@ -491,6 +491,7 @@ define([
                 var fontSize = $(this.wrapSelector('#fontSize')).val();
         
                 code.appendLine('import matplotlib.pyplot as plt');
+                code.appendLine('import seaborn as sns');
                 code.appendFormatLine("plt.rc('figure', figsize=({0}, {1}))", figWidth, figHeight);
                 if (styleName && styleName.length > 0) {
                     code.appendFormatLine("plt.style.use('{0}')", styleName);
