theme-my-login-custom.php with custom template dir

theme-my-login-custom.php

<?php
/**
 * theme-my-login-custom.php
 */


/**
 * Cambiamos las plantillas por unas propias
 * @param  string $useTemplate Plantilla encontrada
 * @param  array $templates    Plantillas buscadas
 * @return string              Plantilla a usar
 */
function tml_template( $useTemplate, $templates ) {
	foreach( $templates as $template) {
		if (file_exists(  TEMPLATEPATH . '/login/' . $template )) {
			$useTemplate =  TEMPLATEPATH . '/login/' . $template  ;
			break;
		}
	}
	return $useTemplate ;

}
add_filter( 'tml_template', 'tml_template', 10, 3 );