\ProvidesExplPackage {luamml-patches-amsmath} {2021-04-23} {0.0.1-alpha}
  {Feel free to add a description here}

\lua_now:n { require'luamml-amsmath' }

% For all of these changes, the redefinitions appear huge.
% But they are almost identical to the original and only
% add luamml commands in appropriate places, so they would
% mostly disappear if there were enough hooks in amsmath.

% aligned and friends
\cs_set:Npn \start@aligned #1#2 {
  \RIfM@
  \else
    \nonmatherr@ { \begin { \@currenvir } }
  \fi
  \savecolumn@ % Assumption: called inside a group
  \luamml_annotate:en {
    nucleus = true, core = false
  } {
    \alignedspace@left
  }
  \ams@start@box {#1} \bgroup
    \maxfields@ #2 \relax
    \ifnum \maxfields@ > \m@ne
      \multiply \maxfields@ \tw@
      \let \math@cr@@@ \math@cr@@@alignedat
      \alignsep@ \z@skip
    \else
      \let \math@cr@@@ \math@cr@@@aligned
      \alignsep@ \minalignsep
    \fi
    \Let@ \chardef \dspbrk@context \@ne
    \default@tag
    \spread@equation % no-op if already called
    \global \column@ \z@
    \ialign \bgroup
      & \column@plus
        \hfil
        \strut@
        $
          \m@th
          \displaystyle
          {##}
          \luamml_flag_save:Nn \displaystyle {mtd}
        $
        \__luamml_amsmath_add_last_to_row:
        \tabskip \z@skip
      & \column@plus
        $
          \m@th
          \displaystyle
          {
            {}
            ##
          }
          \luamml_flag_save:Nn \displaystyle {mtd}
        $
        \__luamml_amsmath_add_last_to_row:
        \hfil
        \tabskip\alignsep@
      \crcr
      \ams@return@opt@arg
}

\cs_set:Npn \endaligned {
      \crcr
      \__luamml_amsmath_save_inner_table:
    \egroup
    \restorecolumn@
  \egroup
  \__luamml_amsmath_finalize_inner_table:
}

% gather
\cs_set:Npn \gather@ #1 {
  \ingather@true
  \let \split \insplit@
  \let \tag \tag@in@align
  \let \label \label@in@display
  \chardef \dspbrk@context \z@
  \intertext@ \displ@y@ \Let@
  \let \math@cr@@@ \math@cr@@@gather
  \gmeasure@ {#1}
  \global \shifttag@false
  \tabskip \z@skip
  \global \row@ \@ne
  \halign to \displaywidth \bgroup
      \strut@
      \setboxz@h {
        $
          \m@th
          \displaystyle
          {##}
          \luamml_flag_save:Nn \displaystyle {mtd}
        $
      }
      \__luamml_amsmath_add_box_to_row:
      \calc@shift@gather
      \set@gather@field
      \tabskip\@centering
    &
      \setboxz@h {
        \strut@
        {##}
      }
      \place@tag@gather
      \__luamml_amsmath_set_tag:
      \tabskip \iftagsleft@
        \gdisplaywidth@
      \else
        \z@skip
      \span \fi
      \crcr
      #1
}
\cs_new_eq:NN \__luamml_amsmath_original_gmeasure:n \gmeasure@
\cs_set:Npn \gmeasure@ #1 {
  \exp_last_unbraced:Nno
    \use_ii_i:nn
    { \luamml_flag_ignore: }
    { \__luamml_amsmath_original_gmeasure:n {#1} }
}

\cs_set:Npn \endgather {
      \math@cr
      \black@ \totwidth@
      \__luamml_amsmath_finalize_table:
  \egroup
  $$
  \ignorespacesafterend
}

% align and friends
\cs_set:Npn \align@preamble {
  &
    \hfil
    \strut@
    \setboxz@h {
      \@lign
      $
        \m@th
        \displaystyle
        {##}
        \ifmeasuring@
          \luamml_flag_ignore:
        \else
          \luamml_flag_save:Nn \displaystyle {mtd}
        \fi
      $
    }
    \ifmeasuring@
      \savefieldlength@
    \else
      \__luamml_amsmath_add_box_to_row:
    \fi
    \set@field
    \tabskip\z@skip
  &
    \setboxz@h {
      \@lign
      $
      \m@th
      \displaystyle
      {
        {}
        ##
      }
      \ifmeasuring@
        \luamml_flag_ignore:
      \else
        \luamml_flag_save:Nn \displaystyle {mtd}
      \fi
      $
    }
    \ifmeasuring@
      \savefieldlength@
    \else
      \__luamml_amsmath_add_box_to_row:
    \fi
    \set@field
    \hfil
    \tabskip\alignsep@
}

\cs_set:Npn \math@cr@@@align {
  \ifst@rred
    \nonumber
  \fi
  \if@eqnsw
    \global \tag@true
  \fi
  \global \advance \row@ \@ne
  \add@amps \maxfields@
  \omit
  \kern -\alignsep@
  \iftag@
    \setboxz@h { 
      \@lign
      \strut@
      { \make@display@tag }
    }
    \place@tag
    \__luamml_amsmath_set_tag:
  \fi
  \ifst@rred
  \else
    \global \@eqnswtrue
  \fi
  \global \lineht@ \z@
  \cr
}

\cs_set:Npn \maketag@@@ #1 {
  \hbox {
    \m@th
    \normalfont
    #1
    \__luamml_amsmath_save_tag:
  }
}

\cs_set:Npn \endalign {
  \math@cr
  \black@ \totwidth@
  \__luamml_amsmath_finalize_table:
  \egroup
  \ifingather@
    \restorealignstate@
    \egroup
    \nonumber
    \ifnum0=‘{\fi\iffalse}\fi
  \else
    $$
  \fi
  \ignorespacesafterend
}

\cs_set:Npn \bBigg@ #1 #2 {
  {
    \ensuremath {
      \Uvextensible height~#1 \big@size axis~exact~#2
    }
  }
}