Выравнивание оси вставленной графики с текстом в документе

Question

Я пытаюсь выровнять график pgfplot с текстовой областью документа. Я создаю график как отдельный pdf с опцией scale only axisи width=\textwidthи пока мне удалось выровнять его вручную только с помощью \hspace*{-1.5cm}перед \includegraphics{figure.pdf}. Это не только немного утомительно (у меня довольно много графиков!), но и вряд ли когда-либо точно. Ниже приведено то, что я могу сделать на данный момент:

Можно ли сделать это более автоматическим или есть какой-то другой способ?

Вот код, который у меня есть на данный момент:

% --- main.tex ---
\documentclass{article}
\usepackage{graphicx}
\usepackage{float}
\usepackage{showframe}

\begin{document} 

Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aliquam cursus pharetra tristique. Curabitur sagittis libero sit amet ante fringilla, nec luctus neque laoreet. Morbi imperdiet convallis velit et elementum. 

\begin{figure}[h!]
\hspace*{-1.5cm}\includegraphics{figure.pdf}
\caption{Pellentesque molestie fermentum tellus ut posuere. Vestibulum euismod lectus quis augue commodo, imperdiet tincidunt felis viverra. Nulla facilisi.}
\end{figure}

Suspendisse id facilisis massa. Phasellus leo sapien, porttitor ac urna non, imperdiet maximus orci. Phasellus vel neque dui. Phasellus iaculis dictum dolor id iaculis. 

\end{document}

% --- figure.tex ---
\documentclass{standalone}

\usepackage{tikz}
\usepackage{pgfplots}

\begin{document}
\begin{tikzpicture}
\begin{axis}[
    xlabel=$x$,
    ylabel={$f(x)$},
    colorbar,
    width=\textwidth,
    height=2cm,
    scale only axis,
]
\addplot {x^2};
\end{axis}
\end{tikzpicture}
\end{document}