Tempo desde a última ocorrência de superfície de água

// DADOS PARA A ANÁLISE:
var assetIdWaterSurface = 'projects/mapbiomas-public/assets/brazil/water/collection5/mapbiomas_brazil_collection5_water_annual_v4';
var targetClass = 1; // Classe que representa superfície de água.
var startYear = 1985; // 1º ano da série temporal
var endYear = 2025; // último ano da série
var years = ee.List.sequence(startYear, endYear, 1);  // Gera os anos da série (1985 - 2025)
var bandPrefix = 'classification_';  // Prefixo das bandas. Exemplo: classification_1985 | Cada banda representa a classificação de um ano da série

// FUNÇÃO: ÚLTIMO ANO DE OCORRÊNCIA
/* Esta função responde: "Qual foi o último ano em que cada pixel foi classificado como a classe de interesse?" 
   Exemplo:
   1985 → água
   ...
   2020 → água
   2021 → não água
  ...
  2025 → não água
  Resultado: pixel = 2020
*/
var getClassLastYear = function(assetID, targetClass, years, bandPrefix) {
    var image = ee.Image(assetID); // Carrega a imagem multibanda.
    
    var imageYears = years.map(  // Para cada ano, cria uma imagem contendo:  ano → se o pixel pertence à classe alvo;  0   → caso contrário
        function(year) {
            year = ee.Number(year);
            var bandName = ee.String(bandPrefix).cat(year.format('%.0f'));

            var imageYear =
                image
                .select(bandName) // Seleciona cada banda (classificação anual)

                .eq(targetClass) // Verifica se o pixel pertence à classe alvo.

                .multiply(year) // Se pertence à classe, guarda o ano. Caso contrário, o valor será 0.

                .rename('year');

            return imageYear;
        }
    );
    var imageCollection = ee.ImageCollection.fromImages(imageYears);

    var lastYear = imageCollection.max().selfMask();  // Obtém o maior ano para cada pixel. O maior ano corresponde à última ocorrência da classe.
    return lastYear;
};

// ============================================================
// FUNÇÃO: TEMPO DESDE A ÚLTIMA OCORRÊNCIA
// ============================================================
// Calcula: ano final da série - último ano de ocorrência
// Exemplo:
//
// último ano de água = 2018
// ano final da série = 2025
//
// 2025 - 2018 = 7 anos
//
// Portanto: pixel = 7
// ============================================================
var timeSinceLastYear = function(lastYear, finalYear){
    var timeSinceImage =
        ee.Image
        .constant(finalYear)
        .subtract(lastYear)
        .selfMask();
    return timeSinceImage;
};