Alinear el eje de los gráficos incluidos con el texto del documento

Question

Estoy intentando alinear un gráfico pgfplot con el área de texto del documento. Creo el gráfico como un pdf independiente con la scale only axisopción y width=\textwidthhasta ahora solo logré alinearlo manualmente usando \hspace*{-1.5cm}antes del archivo \includegraphics{figure.pdf}. Esto no sólo es un poco tedioso (¡tengo bastantes gráficos!), sino que casi nunca es preciso. A continuación se muestra lo que puedo hacer hasta ahora:

¿Es posible hacer esto más automático o hay alguna otra forma de hacerlo?

Aquí está el código que tengo hasta ahora:

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