Support other types of files Perfetto already supports including .gz compressed traces, Chromium F12 DevTools Perf Profile Export, and legacy chrome://tracing .json files (#89)

PerfettoCds/Pipeline/PerfettoDataSource.cs

@@ -8,6 +8,27 @@ using Microsoft.Performance.SDK.Processing;
 
 namespace PerfettoCds
 {
+    public static class PerfettoCds
+    {
+        public static string[] PerfettoCopyrightLicense = new string[]
+        {
+            "Built using Google Perfetto 4\n" +
+            "Copyright (C) 2020 The Android Open Source Project\n" +
+            "\n" +
+            "Licensed under the Apache License, Version 2.0 (the \"License\");\n" +
+            "you may not use this file except in compliance with the License.\n" +
+            "You may obtain a copy of the License at\n" +
+            "\n" +
+            "http://www.apache.org/licenses/LICENSE-2.0\n" +
+            "\n" +
+            "Unless required by applicable law or agreed to in writing, software\n" +
+            "distributed under the License is distributed on an \"AS IS\" BASIS,\n" +
+            "WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n" +
+            "See the License for the specific language governing permissions and\n" +
+            "limitations under the License.\n"
+        };
+    }
+
     [ProcessingSource("9fc8515e-9206-4690-b14a-3e7b54745c5f", "PerfettoTraceDataSource", "Processes Perfetto trace files")]
     [FileDataSource(".perfetto-trace", "Perfetto trace files")]
     public sealed class PerfettoDataSource : ProcessingSource
@@ -30,23 +51,7 @@ namespace PerfettoCds
             {
                 ProjectInfo = new ProjectInfo() { Uri = "https://aka.ms/linuxperftools" },
                 CopyrightNotice = "Copyright (C) " + DateTime.UtcNow.Year,
-                AdditionalInformation = new[]
-                {
-                    "Built using Google Perfetto 4\n" +
-                    "Copyright (C) 2020 The Android Open Source Project\n" +
-                    "\n" +
-                    "Licensed under the Apache License, Version 2.0 (the \"License\");\n" +
-                    "you may not use this file except in compliance with the License.\n" +
-                    "You may obtain a copy of the License at\n" +
-                    "\n" +
-                    "http://www.apache.org/licenses/LICENSE-2.0\n" +
-                    "\n" +
-                    "Unless required by applicable law or agreed to in writing, software\n" +
-                    "distributed under the License is distributed on an \"AS IS\" BASIS,\n" +
-                    "WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n" +
-                    "See the License for the specific language governing permissions and\n" +
-                    "limitations under the License.\n"
-                },
+                AdditionalInformation = PerfettoCds.PerfettoCopyrightLicense,
             };
         }
 
@@ -90,23 +95,7 @@ namespace PerfettoCds
             {
                 ProjectInfo = new ProjectInfo() { Uri = "https://aka.ms/linuxperftools" },
                 CopyrightNotice = "Copyright (C) " + DateTime.UtcNow.Year,
-                AdditionalInformation = new[]
-                {
-                    "Built using Google Perfetto 4\n" +
-                    "Copyright (C) 2020 The Android Open Source Project\n" +
-                    "\n" +
-                    "Licensed under the Apache License, Version 2.0 (the \"License\");\n" +
-                    "you may not use this file except in compliance with the License.\n" +
-                    "You may obtain a copy of the License at\n" +
-                    "\n" +
-                    "http://www.apache.org/licenses/LICENSE-2.0\n" +
-                    "\n" +
-                    "Unless required by applicable law or agreed to in writing, software\n" +
-                    "distributed under the License is distributed on an \"AS IS\" BASIS,\n" +
-                    "WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n" +
-                    "See the License for the specific language governing permissions and\n" +
-                    "limitations under the License.\n"
-                },
+                AdditionalInformation = PerfettoCds.PerfettoCopyrightLicense,
             };
         }
 
@@ -127,4 +116,95 @@ namespace PerfettoCds
             this.applicationEnvironment = applicationEnvironment;
         }
     }
+
+    [ProcessingSource("DAAF538C-8482-4CDA-A477-7A4BD5F1EFEA", "ChromiumDevToolsExportDataSource", "Processes .json Chromium DevTools Perf profile export")]
+    [FileDataSource(".json", "Chromium DevTools Perf profile export")]
+    public sealed class ChromiumDevToolsExportDataSource : ProcessingSource
+    {
+        private IApplicationEnvironment applicationEnvironment;
+
+        protected override ICustomDataProcessor CreateProcessorCore(IEnumerable<IDataSource> dataSources, IProcessorEnvironment processorEnvironment, ProcessorOptions options)
+        {
+            var filePath = dataSources.First().Uri.LocalPath;
+            var parser = new PerfettoSourceParser(filePath);
+            return new PerfettoDataProcessor(parser,
+                                            options,
+                                            this.applicationEnvironment,
+                                            processorEnvironment);
+        }
+
+        public override ProcessingSourceInfo GetAboutInfo()
+        {
+            return new ProcessingSourceInfo()
+            {
+                ProjectInfo = new ProjectInfo() { Uri = "https://aka.ms/linuxperftools" },
+                CopyrightNotice = "Copyright (C) " + DateTime.UtcNow.Year,
+                AdditionalInformation = PerfettoCds.PerfettoCopyrightLicense,
+            };
+        }
+
+        protected override bool IsDataSourceSupportedCore(IDataSource dataSource)
+        {
+            if (dataSource.IsDirectory())
+            {
+                return false;
+            }
+
+            var ext = Path.GetExtension(dataSource.Uri.LocalPath);
+
+            return dataSource.IsFile() && StringComparer.OrdinalIgnoreCase.Equals(".json", ext);
+        }
+
+        protected override void SetApplicationEnvironmentCore(IApplicationEnvironment applicationEnvironment)
+        {
+            this.applicationEnvironment = applicationEnvironment;
+        }
+    }
+
+    [ProcessingSource("E352CC69-5991-45E0-B329-4C4071BA7CFD", "GZipPerfettoDataSource", "Processes .gz json from chrome://tracing legacy UI")]
+    [FileDataSource(".gz", ".gz chrome://tracing legacy UI or other Perfetto .gz")]
+    public sealed class GZipPerfettoDataSource : ProcessingSource
+    {
+        private IApplicationEnvironment applicationEnvironment;
+
+        protected override ICustomDataProcessor CreateProcessorCore(IEnumerable<IDataSource> dataSources, IProcessorEnvironment processorEnvironment, ProcessorOptions options)
+        {
+            var filePath = dataSources.First().Uri.LocalPath;
+            var parser = new PerfettoSourceParser(filePath);
+            return new PerfettoDataProcessor(parser,
+                                            options,
+                                            this.applicationEnvironment,
+                                            processorEnvironment);
+        }
+
+        public override ProcessingSourceInfo GetAboutInfo()
+        {
+            return new ProcessingSourceInfo()
+            {
+                ProjectInfo = new ProjectInfo() { Uri = "https://aka.ms/linuxperftools" },
+                CopyrightNotice = "Copyright (C) " + DateTime.UtcNow.Year,
+                AdditionalInformation = PerfettoCds.PerfettoCopyrightLicense,
+            };
+        }
+
+        protected override bool IsDataSourceSupportedCore(IDataSource dataSource)
+        {
+            if (dataSource.IsDirectory())
+            {
+                return false;
+            }
+
+            var ext = Path.GetExtension(dataSource.Uri.LocalPath);
+
+            return dataSource.IsFile() && StringComparer.OrdinalIgnoreCase.Equals(".gz", ext) &&
+                ( dataSource.Uri.LocalPath.EndsWith(".json.gz") ||
+                  dataSource.Uri.LocalPath.EndsWith("pftrace.gz") ||
+                  dataSource.Uri.LocalPath.EndsWith("perfetto-trace.gz") );
+        }
+
+        protected override void SetApplicationEnvironmentCore(IApplicationEnvironment applicationEnvironment)
+        {
+            this.applicationEnvironment = applicationEnvironment;
+        }
+    }
 }