[RPP] Correctly map profile data to original sources

The original code was now using the resolved UILocation's line/col in
the profile data, instead always using the raw location no matter if the
resolved UISourceCode was mapped via source maps.

Now the UILocation line/col is used. Additionally, we continue to
provide profile data for the generated file, as it's expected users may
switch between the two views of the same script when getting deep into
performance issues.

Bug: 462212096
Change-Id: I8e7ee303020f7ba28e78434c679229ec75872749
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/7173541
Auto-Submit: Connor Clark <cjamcl@chromium.org>
Commit-Queue: Connor Clark <cjamcl@chromium.org>
Reviewed-by: Paul Irish <paulirish@chromium.org>

Author: Connor Clark

front_end/ui/legacy/components/perf_ui/LineLevelProfile.ts

@@ -231,18 +231,21 @@ export class Helper {
                         String(scriptIdOrUrl) as Protocol.Runtime.ScriptId, zeroBasedLine, zeroBasedColumn || 0);
                 if (rawLocation) {
                   pending.push(workspaceBinding.rawLocationToUILocation(rawLocation).then(uiLocation => {
-                    if (uiLocation) {
-                      let lineMap = decorationsBySource.get(uiLocation.uiSourceCode);
-                      if (!lineMap) {
-                        lineMap = new Map();
-                        decorationsBySource.set(uiLocation.uiSourceCode, lineMap);
-                      }
-                      let columnMap = lineMap.get(lineNumber);
-                      if (!columnMap) {
-                        columnMap = new Map();
-                        lineMap.set(lineNumber, columnMap);
+                    if (!uiLocation) {
+                      return;
+                    }
+
+                    this.addLineColumnData(
+                        decorationsBySource, uiLocation.uiSourceCode, uiLocation.lineNumber + 1,
+                        (uiLocation.columnNumber ?? 0) + 1, data);
+
+                    // If the above was a source mapped UILocation, then we also need to add it to the generated UILocation.
+                    if (uiLocation.uiSourceCode.contentType().isFromSourceMap()) {
+                      const script = rawLocation.script();
+                      const uiSourceCode = script ? workspaceBinding.uiSourceCodeForScript(script) : null;
+                      if (uiSourceCode) {
+                        this.addLineColumnData(decorationsBySource, uiSourceCode, lineNumber, columnNumber, data);
                       }
-                      columnMap.set((zeroBasedColumn || 0) + 1, data);
                     }
                   }));
                 }
@@ -267,4 +270,22 @@ export class Helper {
       }
     }
   }
+
+  private addLineColumnData(
+      decorationsBySource: ProfileDataMap, uiSourceCode: Workspace.UISourceCode.UISourceCode, lineOneIndexed: number,
+      columnOneIndexed: number, data: number): void {
+    let lineMap = decorationsBySource.get(uiSourceCode);
+    if (!lineMap) {
+      lineMap = new Map();
+      decorationsBySource.set(uiSourceCode, lineMap);
+    }
+
+    let columnMap = lineMap.get(lineOneIndexed);
+    if (!columnMap) {
+      columnMap = new Map();
+      lineMap.set(lineOneIndexed, columnMap);
+    }
+
+    columnMap.set(columnOneIndexed, (columnMap.get(columnOneIndexed) ?? 0) + data);
+  }
 }