Alinhando o eixo dos gráficos incluídos com o texto do documento

Question

Estou tentando alinhar um gráfico pgfplot com a área de texto do documento. Eu crio o gráfico como um pdf independente com a scale only axisopção e width=\textwidthaté agora só consegui alinhá-lo manualmente usando \hspace*{-1.5cm}antes do \includegraphics{figure.pdf}. Isso não só é um pouco tedioso (tenho muitos gráficos!), mas quase nunca é preciso. Abaixo está o que posso fazer até agora:

É possível tornar isso mais automático ou existe alguma outra maneira de fazer isso?

Aqui está o código que tenho até agora:

% --- 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}