Merge pull request #1010 from WolframResearch/bugfix/more-template-box-serialization

More template box serialization

Author: rhennigan

.vscode/settings.json

@@ -11,6 +11,7 @@
         "Checkmarked",
         "CICD",
         "clicktocopy",
+        "CMYK",
         "Componentwise",
         "Connor",
         "datarepository",

Source/Chatbook/Serialization.wl

@@ -199,15 +199,23 @@ $boxOp = <| SuperscriptBox -> "^", SubscriptBox -> "_" |>;
 
 (* How to choose TemplateBox arguments for serialization *)
 $templateBoxRules = <|
-    "AssistantMessageBox"       -> First,
-    "ConditionalExpression"     -> makeExpressionString,
-    "GrayLink"                  -> First,
-    "HyperlinkDefault"          -> First,
-    "Key0"                      -> First,
-    "Key1"                      -> (Riffle[ #, "-" ] &),
-    "RowDefault"                -> Identity,
-    "TransferFunctionModelFull" -> makeExpressionString,
-    "UserMessageBox"            -> First
+    "AssistantMessageBox"          -> First,
+    "ConditionalExpression"        -> makeExpressionString,
+    "GrayLink"                     -> First,
+    "HyperlinkDefault"             -> First,
+    "Key0"                         -> First,
+    "Key1"                         -> (Riffle[ #, "-" ] &),
+    "RowDefault"                   -> Identity,
+    "TransferFunctionModelFull"    -> makeExpressionString,
+    "UserMessageBox"               -> First,
+    "CMYKColorSwatchTemplate"      -> inputFormString @* Lookup[ "color" ],
+    "GrayLevelColorSwatchTemplate" -> inputFormString @* Lookup[ "color" ],
+    "HueColorSwatchTemplate"       -> inputFormString @* Lookup[ "color" ],
+    "LABColorSwatchTemplate"       -> inputFormString @* Lookup[ "color" ],
+    "LCHColorSwatchTemplate"       -> inputFormString @* Lookup[ "color" ],
+    "LUVColorSwatchTemplate"       -> inputFormString @* Lookup[ "color" ],
+    "RGBColorSwatchTemplate"       -> inputFormString @* Lookup[ "color" ],
+    "XYZColorSwatchTemplate"       -> inputFormString @* Lookup[ "color" ]
 |>;
 
 (* ::**************************************************************************************************************:: *)
@@ -232,6 +240,7 @@ $$graphicsBox = Alternatives[
 (* Serialize the first argument of these and ignore the rest *)
 $stringStripHeads = Alternatives[
     ActionMenuBox,
+    AdjustmentBox,
     ButtonBox,
     CellGroupData,
     FrameBox,
@@ -1300,6 +1309,10 @@ fasterCellToString0[ TemplateBox[ args: { _, _, str_String, ___ }, "MessageTempl
     fasterCellToString0 @ str
 );
 
+(* Percent References *)
+fasterCellToString0[ TemplateBox[ KeyValuePattern[ "OutNumber" -> n_Integer ], "PercentRef", ___ ] ] :=
+    "%" <> ToString @ n;
+
 (* Large Outputs *)
 fasterCellToString0[ TemplateBox[ KeyValuePattern[ "shortenedBoxes" -> boxes_ ], "OutputSizeLimitTemplate", ___ ] ] :=
     fasterCellToString0 @ boxes;
@@ -1487,12 +1500,6 @@ fasterCellToString0[ box: TemplateBox[ args_, name_String, ___ ] ] :=
         fasterCellToString0 @ f @ args /; ! MissingQ @ f && f =!= makeExpressionString
     ];
 
-fasterCellToString0[ TemplateBox[ { args___ }, ___, InterpretationFunction -> f_, ___ ] ] :=
-    fasterCellToString0 @ f @ args;
-
-fasterCellToString0[ TemplateBox[ args_, ___, InterpretationFunction -> f_, ___ ] ] :=
-    fasterCellToString0 @ f @ args;
-
 fasterCellToString0[ OverlayBox[ { a_, ___ }, ___ ] ] :=
     fasterCellToString0 @ a;
 
@@ -2303,6 +2310,64 @@ fasterCellToString0[ DynamicModuleBox[ a___ ] ] /; ! TrueQ @ $CellToStringDebug
     "DynamicModule[\[LeftSkeleton]" <> ToString @ Length @ HoldComplete @ a <> "\[RightSkeleton]]"
 );
 
+(* ::**************************************************************************************************************:: *)
+(* ::Subsubsubsection::Closed:: *)
+(*Unhandled TemplateBoxes*)
+fasterCellToString0[ box: TemplateBox[ args_, ___ ] ] :=
+    With[ { f = getTemplateBoxFunction @ box },
+        fasterCellToString0 @ applyTemplateBoxDisplayFunction[ f, args ] /; ! MissingQ @ f
+    ];
+
+(* ::**************************************************************************************************************:: *)
+(* ::Subsubsubsubsection::Closed:: *)
+(*getTemplateBoxFunction*)
+getTemplateBoxFunction // beginDefinition;
+getTemplateBoxFunction[ TemplateBox[ __, InterpretationFunction -> f: Except[ $$unspecified ], ___ ] ] := f;
+getTemplateBoxFunction[ TemplateBox[ __, DisplayFunction -> f: Except[ $$unspecified ], ___ ] ] := f;
+getTemplateBoxFunction[ TemplateBox[ _, name_String, ___, InterpretationFunction -> Automatic, ___ ] ] := name;
+getTemplateBoxFunction[ TemplateBox[ _, name_String, ___ ] ] := getTemplateBoxFunction @ name;
+getTemplateBoxFunction[ name_String ] := Lookup[ $templateBoxCache, name, getTemplateBoxFunction0 @ name ];
+getTemplateBoxFunction // endDefinition;
+
+
+getTemplateBoxFunction0 // beginDefinition;
+
+getTemplateBoxFunction0[ name_String ] :=
+    getTemplateBoxFunction0[ name, usingFrontEnd @ CurrentValue @ { StyleDefinitions, name, TemplateBoxOptions } ];
+
+getTemplateBoxFunction0[
+    name_,
+    KeyValuePattern[ InterpretationFunction|"InterpretationFunction" -> f: Except[ $$unspecified ] ]
+] := $templateBoxCache[ name ] = f;
+
+getTemplateBoxFunction0[
+    name_,
+    KeyValuePattern[ InterpretationFunction|"InterpretationFunction" -> Automatic ]
+] := $templateBoxCache[ name ] = name;
+
+getTemplateBoxFunction0[
+    name_,
+    KeyValuePattern[ DisplayFunction|"DisplayFunction" -> f: Except[ $$unspecified ] ]
+] := $templateBoxCache[ name ] = f;
+
+getTemplateBoxFunction0[ name_, _ ] :=
+    $templateBoxCache[ name ] = Missing[ "NotFound" ];
+
+getTemplateBoxFunction0 // endDefinition;
+
+
+$templateBoxCache = <| |>;
+
+(* ::**************************************************************************************************************:: *)
+(* ::Subsubsubsubsection::Closed:: *)
+(*applyTemplateBoxDisplayFunction*)
+applyTemplateBoxDisplayFunction // beginDefinition;
+applyTemplateBoxDisplayFunction[ f_, TemplateBox[ args_, ___ ] ] := applyTemplateBoxDisplayFunction[ f, args ];
+applyTemplateBoxDisplayFunction[ f_String, a_List ] := f <> "[" <> StringRiffle[ fasterCellToString0 /@ a, ", " ] <> "]";
+applyTemplateBoxDisplayFunction[ f_, { args___ } ] := f @ args;
+applyTemplateBoxDisplayFunction[ f_, args___ ] := f @ args;
+applyTemplateBoxDisplayFunction // endDefinition;
+
 (* ::**************************************************************************************************************:: *)
 (* ::Subsubsubsection::Closed:: *)
 (*Ignored/Skipped*)