deploy: Penn-Electric-Racing/SuboptimumG@d174c0f

Author: alex-yang-upenn

.doctrees/_autosummary/suboptimumg.plotting.internal_data_plots.doctree

@@ -48,10 +48,11 @@
     "awd_comp_data = michigan2023_awd.to_data()\n",
     "\n",
     "# We can also optionally load parameters from published Google Sheets\n",
-    "michigan2023_revX_alt = load_competition_from_gdrive(\n",
-    "    car_config_url=\"https://docs.google.com/spreadsheets/d/e/2PACX-1vQRPeSToFhhcMYo4J12FvPuIoCv0xAVCvcOlb-SlRxRleQQJNwq7Jiw1H0Hwr6oYOFBcCD_422CosnJ/pub?output=csv\",\n",
-    "    comp_config_url=\"https://docs.google.com/spreadsheets/d/e/2PACX-1vSPaKIHa64Buiqw_2yk4P3iP3PlsNjuE-GjYSHQDenn6dFYG_EelNSg7oqwH5mEvXzSNfB7j1p42H12/pub?output=csv\",\n",
-    ")"
+    "# WARNING: Sheets are not up to date at this moment\n",
+    "# michigan2023_revX_alt = load_competition_from_gdrive(\n",
+    "#     car_config_url=\"https://docs.google.com/spreadsheets/d/e/2PACX-1vQRPeSToFhhcMYo4J12FvPuIoCv0xAVCvcOlb-SlRxRleQQJNwq7Jiw1H0Hwr6oYOFBcCD_422CosnJ/pub?output=csv\",\n",
+    "#     comp_config_url=\"https://docs.google.com/spreadsheets/d/e/2PACX-1vSPaKIHa64Buiqw_2yk4P3iP3PlsNjuE-GjYSHQDenn6dFYG_EelNSg7oqwH5mEvXzSNfB7j1p42H12/pub?output=csv\",\n",
+    "# )"
    ]
   },
   {
@@ -102,16 +103,20 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from suboptimumg.plotting.internal_data_plots import plot_per_seed, Axis\n",
-    "\n",
-    "# endu_res = comp_results.endurance\n",
+    "from suboptimumg.plotting import plot_per_seed, plot_internal_keys, Axis\n",
     "\n",
-    "plot_per_seed(\n",
+    "plot_internal_keys(\n",
     "    results=endu_res.lapsim_results,\n",
+    "    keys=[\"v_max_profile\", \"seed_idx_list\"],\n",
     "    x_axis=Axis.DISTANCE,\n",
-    "    include_indices=range(4),\n",
-    "    include_power=True,\n",
-    ")"
+    ")\n",
+    "\n",
+    "# plot_per_seed(\n",
+    "#     results=endu_res.lapsim_results,\n",
+    "#     x_axis=Axis.DISTANCE,\n",
+    "#     include_indices=range(4),\n",
+    "#     include_power=True,\n",
+    "# )"
    ]
   },
   {

.doctrees/_autosummary/suboptimumg.track.track.doctree

@@ -505,6 +505,8 @@
 - Internal Data top-level keys: <code class="docutils literal notranslate"><span class="pre">v_max_profile</span></code>, <code class="docutils literal notranslate"><span class="pre">seed_idx_list</span></code>, <code class="docutils literal notranslate"><span class="pre">cumulative_dist</span></code>
 - LapsimResults top-level keys: <code class="docutils literal notranslate"><span class="pre">lap_t</span></code>, <code class="docutils literal notranslate"><span class="pre">lap_dxs</span></code>, <code class="docutils literal notranslate"><span class="pre">lap_vels</span></code>, <code class="docutils literal notranslate"><span class="pre">lap_accs</span></code>, <code class="docutils literal notranslate"><span class="pre">lap_powers</span></code>, <code class="docutils literal notranslate"><span class="pre">lap_eff_motor_torques</span></code>
 - Per-seed keys in the form <code class="docutils literal notranslate"><span class="pre">{field}_seed_{n}</span></code> (e.g. <code class="docutils literal notranslate"><span class="pre">v_proposal_seed_2</span></code>)</p>
+<p>Special case: <code class="docutils literal notranslate"><span class="pre">seed_idx_list</span></code> ignores the x_axis parameter and always plots
+v_max_profile vs cumulative_dist with red dots overlaid at seed index positions.</p>
 </dd></dl>
 
 <dl class="py function">

.doctrees/environment.pickle

@@ -502,7 +502,7 @@
 
 <dl class="py method">
 <dt class="sig sig-object py" id="suboptimumg.track.track.Track.plot">
-<span class="sig-name descname"><span class="pre">plot</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">rotate_by_deg</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0.0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">theme</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">font_config</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">layout_config</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/suboptimumg/track/track.html#Track.plot"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#suboptimumg.track.track.Track.plot" title="Link to this definition">#</a></dt>
+<span class="sig-name descname"><span class="pre">plot</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">rotate_by_deg</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0.0</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">theme</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">'suspension'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">font_config</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">layout_config</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/suboptimumg/track/track.html#Track.plot"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#suboptimumg.track.track.Track.plot" title="Link to this definition">#</a></dt>
 <dd><p>Visualizes the track.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>

.doctrees/getting_started/notebook.doctree

@@ -832,6 +832,67 @@ <h1>Source code for suboptimumg.plotting.internal_data_plots</h1><div class="hig
 
 
 
+<span class="k">def</span><span class="w"> </span><span class="nf">_plot_seed_idx_list</span><span class="p">(</span>
+    <span class="n">results</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">LapsimResults</span><span class="p">],</span>
+    <span class="n">x_axis</span><span class="p">:</span> <span class="n">Axis</span><span class="p">,</span>
+    <span class="n">row_idx</span><span class="p">:</span> <span class="nb">int</span><span class="p">,</span>
+    <span class="n">fig</span><span class="p">:</span> <span class="n">go</span><span class="o">.</span><span class="n">Figure</span><span class="p">,</span>
+<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
+<span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Special plotting for seed_idx_list: shows v_max_profile vs cumulative_dist</span>
+<span class="sd">    with red dots overlaid at seed index positions.</span>
+
+<span class="sd">    Parameters</span>
+<span class="sd">    ----------</span>
+<span class="sd">    results : List[LapsimResults]</span>
+<span class="sd">        List of lapsim results to plot</span>
+<span class="sd">    x_axis : Axis</span>
+<span class="sd">        Axis to use for the x-axis (distance or time)</span>
+<span class="sd">    row_idx : int</span>
+<span class="sd">        Row index for the subplot</span>
+<span class="sd">    fig : go.Figure</span>
+<span class="sd">        Figure to add traces to</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="k">for</span> <span class="n">run_idx</span><span class="p">,</span> <span class="n">r</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">results</span><span class="p">):</span>
+        <span class="n">x_vals</span> <span class="o">=</span> <span class="n">_get_x_vals</span><span class="p">(</span><span class="n">r</span><span class="p">,</span> <span class="n">x_axis</span><span class="p">)</span>
+
+        <span class="n">v_max_profile</span> <span class="o">=</span> <span class="n">r</span><span class="o">.</span><span class="n">internal_data</span><span class="o">.</span><span class="n">v_max_profile</span>
+        <span class="n">seed_idx_list</span> <span class="o">=</span> <span class="n">r</span><span class="o">.</span><span class="n">internal_data</span><span class="o">.</span><span class="n">seed_idx_list</span>
+
+        <span class="c1"># Plot v_max_profile as the base line</span>
+        <span class="n">fig</span><span class="o">.</span><span class="n">add_trace</span><span class="p">(</span>
+            <span class="n">go</span><span class="o">.</span><span class="n">Scatter</span><span class="p">(</span>
+                <span class="n">x</span><span class="o">=</span><span class="n">x_vals</span><span class="p">,</span>
+                <span class="n">y</span><span class="o">=</span><span class="n">v_max_profile</span><span class="p">,</span>
+                <span class="n">mode</span><span class="o">=</span><span class="s2">&quot;lines&quot;</span><span class="p">,</span>
+                <span class="n">name</span><span class="o">=</span><span class="sa">f</span><span class="s2">&quot;run_</span><span class="si">{</span><span class="n">run_idx</span><span class="si">}</span><span class="s2">:v_max_profile&quot;</span><span class="p">,</span>
+                <span class="n">line</span><span class="o">=</span><span class="nb">dict</span><span class="p">(</span><span class="n">color</span><span class="o">=</span><span class="s2">&quot;blue&quot;</span><span class="p">),</span>
+            <span class="p">),</span>
+            <span class="n">row</span><span class="o">=</span><span class="n">row_idx</span><span class="p">,</span>
+            <span class="n">col</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
+        <span class="p">)</span>
+
+        <span class="c1"># Overlay red dots at seed indices</span>
+        <span class="n">seed_x</span> <span class="o">=</span> <span class="n">x_vals</span><span class="p">[</span><span class="n">seed_idx_list</span><span class="p">]</span>
+        <span class="n">seed_y</span> <span class="o">=</span> <span class="n">v_max_profile</span><span class="p">[</span><span class="n">seed_idx_list</span><span class="p">]</span>
+
+        <span class="n">fig</span><span class="o">.</span><span class="n">add_trace</span><span class="p">(</span>
+            <span class="n">go</span><span class="o">.</span><span class="n">Scatter</span><span class="p">(</span>
+                <span class="n">x</span><span class="o">=</span><span class="n">seed_x</span><span class="p">,</span>
+                <span class="n">y</span><span class="o">=</span><span class="n">seed_y</span><span class="p">,</span>
+                <span class="n">mode</span><span class="o">=</span><span class="s2">&quot;markers&quot;</span><span class="p">,</span>
+                <span class="n">name</span><span class="o">=</span><span class="sa">f</span><span class="s2">&quot;run_</span><span class="si">{</span><span class="n">run_idx</span><span class="si">}</span><span class="s2">:seed_indices&quot;</span><span class="p">,</span>
+                <span class="n">marker</span><span class="o">=</span><span class="nb">dict</span><span class="p">(</span><span class="n">color</span><span class="o">=</span><span class="s2">&quot;red&quot;</span><span class="p">,</span> <span class="n">size</span><span class="o">=</span><span class="n">MARKER_SIZE_LARGE</span><span class="p">,</span> <span class="n">symbol</span><span class="o">=</span><span class="s2">&quot;circle&quot;</span><span class="p">),</span>
+            <span class="p">),</span>
+            <span class="n">row</span><span class="o">=</span><span class="n">row_idx</span><span class="p">,</span>
+            <span class="n">col</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
+        <span class="p">)</span>
+        <span class="n">fig</span><span class="o">.</span><span class="n">update_yaxes</span><span class="p">(</span>
+            <span class="n">title_text</span><span class="o">=</span><span class="s2">&quot;v_max_profile w/ seeds highlighted&quot;</span><span class="p">,</span> <span class="n">row</span><span class="o">=</span><span class="n">row_idx</span><span class="p">,</span> <span class="n">col</span><span class="o">=</span><span class="mi">1</span>
+        <span class="p">)</span>
+        <span class="n">fig</span><span class="o">.</span><span class="n">update_xaxes</span><span class="p">(</span><span class="n">title_text</span><span class="o">=</span><span class="n">Axis</span><span class="o">.</span><span class="n">DISTANCE</span><span class="o">.</span><span class="n">value</span><span class="p">,</span> <span class="n">row</span><span class="o">=</span><span class="n">row_idx</span><span class="p">,</span> <span class="n">col</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span>
+
+
 <div class="viewcode-block" id="plot_internal_keys">
 <a class="viewcode-back" href="../../../_autosummary/suboptimumg.plotting.internal_data_plots.html#suboptimumg.plotting.internal_data_plots.plot_internal_keys">[docs]</a>
 <span class="k">def</span><span class="w"> </span><span class="nf">plot_internal_keys</span><span class="p">(</span>
@@ -874,19 +935,32 @@ <h1>Source code for suboptimumg.plotting.internal_data_plots</h1><div class="hig
 <span class="sd">    - Internal Data top-level keys: ``v_max_profile``, ``seed_idx_list``, ``cumulative_dist``</span>
 <span class="sd">    - LapsimResults top-level keys: ``lap_t``, ``lap_dxs``, ``lap_vels``, ``lap_accs``, ``lap_powers``, ``lap_eff_motor_torques``</span>
 <span class="sd">    - Per-seed keys in the form ``{field}_seed_{n}`` (e.g. ``v_proposal_seed_2``)</span>
+
+<span class="sd">    Special case: ``seed_idx_list`` ignores the x_axis parameter and always plots</span>
+<span class="sd">    v_max_profile vs cumulative_dist with red dots overlaid at seed index positions.</span>
 <span class="sd">    &quot;&quot;&quot;</span>
+    <span class="c1"># Normalize inputs to lists</span>
+    <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">results</span><span class="p">,</span> <span class="nb">list</span><span class="p">):</span>
+        <span class="n">results</span> <span class="o">=</span> <span class="p">[</span><span class="n">results</span><span class="p">]</span>
+
     <span class="n">fig</span> <span class="o">=</span> <span class="n">make_subplots</span><span class="p">(</span>
         <span class="n">rows</span><span class="o">=</span><span class="nb">len</span><span class="p">(</span><span class="n">keys</span><span class="p">),</span>
         <span class="n">cols</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
-        <span class="n">shared_xaxes</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
+        <span class="n">shared_xaxes</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
         <span class="n">subplot_titles</span><span class="o">=</span><span class="n">keys</span><span class="p">,</span>
     <span class="p">)</span>
 
     <span class="k">for</span> <span class="n">idx</span><span class="p">,</span> <span class="n">key</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">keys</span><span class="p">):</span>
         <span class="n">row_idx</span> <span class="o">=</span> <span class="n">idx</span> <span class="o">+</span> <span class="mi">1</span>
+        <span class="n">true_axis</span> <span class="o">=</span> <span class="n">x_axis</span>
+
+        <span class="c1"># Special case for seed_idx_list</span>
+        <span class="k">if</span> <span class="n">key</span> <span class="o">==</span> <span class="s2">&quot;seed_idx_list&quot;</span><span class="p">:</span>
+            <span class="n">_plot_seed_idx_list</span><span class="p">(</span><span class="n">results</span><span class="p">,</span> <span class="n">x_axis</span><span class="p">,</span> <span class="n">row_idx</span><span class="p">,</span> <span class="n">fig</span><span class="p">)</span>
+            <span class="k">continue</span>
 
         <span class="k">for</span> <span class="n">run_idx</span><span class="p">,</span> <span class="n">r</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">results</span><span class="p">):</span>
-            <span class="n">x_vals</span> <span class="o">=</span> <span class="n">_get_x_vals</span><span class="p">(</span><span class="n">r</span><span class="p">,</span> <span class="n">axis</span><span class="p">)</span>
+            <span class="n">x_vals</span> <span class="o">=</span> <span class="n">_get_x_vals</span><span class="p">(</span><span class="n">r</span><span class="p">,</span> <span class="n">true_axis</span><span class="p">)</span>
             <span class="n">y_vals</span> <span class="o">=</span> <span class="n">_extract_data_by_key</span><span class="p">(</span><span class="n">key</span><span class="p">,</span> <span class="n">r</span><span class="p">)</span>
 
             <span class="n">fig</span><span class="o">.</span><span class="n">add_trace</span><span class="p">(</span>
@@ -900,8 +974,8 @@ <h1>Source code for suboptimumg.plotting.internal_data_plots</h1><div class="hig
                 <span class="n">col</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span>
             <span class="p">)</span>
         <span class="n">fig</span><span class="o">.</span><span class="n">update_yaxes</span><span class="p">(</span><span class="n">title_text</span><span class="o">=</span><span class="n">key</span><span class="p">,</span> <span class="n">row</span><span class="o">=</span><span class="n">row_idx</span><span class="p">,</span> <span class="n">col</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span>
+        <span class="n">fig</span><span class="o">.</span><span class="n">update_xaxes</span><span class="p">(</span><span class="n">title_text</span><span class="o">=</span><span class="n">true_axis</span><span class="o">.</span><span class="n">value</span><span class="p">,</span> <span class="n">row</span><span class="o">=</span><span class="n">row_idx</span><span class="p">,</span> <span class="n">col</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span>
 
-    <span class="n">fig</span><span class="o">.</span><span class="n">update_xaxes</span><span class="p">(</span><span class="n">title_text</span><span class="o">=</span><span class="n">x_axis</span><span class="o">.</span><span class="n">value</span><span class="p">)</span>
     <span class="n">fig</span><span class="o">.</span><span class="n">update_layout</span><span class="p">(</span>
         <span class="n">height</span><span class="o">=</span><span class="n">layout_config</span><span class="o">.</span><span class="n">height</span> <span class="o">*</span> <span class="nb">len</span><span class="p">(</span><span class="n">keys</span><span class="p">),</span>
         <span class="n">title</span><span class="o">=</span><span class="p">{</span>