cloud_masks_test

/**** Start of imports. If edited, may not auto-convert in the playground. ****/
var imageCollection = ee.ImageCollection("MODIS/006/MOD09GA");
/***** End of imports. If edited, may not auto-convert in the playground. *****/
/***
 * Tests for cloud mask functions
 * 
 * Author: Rodrigo E. Principe
 * email: fitoprincipe82@gmail.com
 * License: MIT
 */
 
 var s2mask = require('users/fitoprincipe/geetools:cloud_masks').sentinel2
 var landsatSR = require('users/fitoprincipe/geetools:cloud_masks').landsatSR
 var cloud = require('users/fitoprincipe/geetools:cloud_masks')
 
 
 var show = function(colid, cloud_prop, viz, name, func, params, cloud_percent) {
   var percent = cloud_percent || 40;
   var cent = Map.getCenter();
   var col = ee.ImageCollection(colid)
             .filterBounds(cent)
             .filterMetadata(cloud_prop, 'greater_than', percent)
             //.filterDate('2017-01-01', '2017-03-01')
             .sort(cloud_prop);
   
   if (col.size().getInfo() !== 0) {
     var img = ee.Image(col.first());
     //var vis = {bands:['B8', 'B12', 'B4'], min:0, max:5000}
     var masked = func(params)(img)
     Map.addLayer(masked, viz, name+' masked');
     Map.addLayer(img, viz, name+' NOT masked', false);
     Map.centerObject(img)
   }
   else {
     print('Collection is empty')
   }
 }
 
 var sentinel2 = function(options) {
   var opt = options || null
   show('COPERNICUS/S2', 
        'CLOUDY_PIXEL_PERCENTAGE', 
        {bands:['B8', 'B12', 'B4'], min:0, max:5000}, 
        'SENTINEL 2', 
        cloud.make.sentinel2, 
        opt)
 }
 
 var landsat8SR = function(options) {
   var opt = options || null
   show('LANDSAT/LC08/C01/T1_SR', 
        'CLOUD_COVER', 
        {bands:['B5', 'B7', 'B4'], min:0, max:5000}, 
        'Landsat 8 SR', 
        cloud.make.landsatSR, 
        opt)
 }
 
 var landsat5SR = function(options) {
   var opt = options || null
   show('LANDSAT/LT05/C01/T1_SR', 
        'CLOUD_COVER', 
        {bands:['B4', 'B5', 'B3'], min:0, max:5000}, 
        'Landsat 5 SR', 
        cloud.make.landsatSR, 
        opt)
 }
 
 var landsat7SR = function(options) {
   var opt = options || null
   show('LANDSAT/LE07/C01/T1_SR', 
        'CLOUD_COVER', 
        {bands:['B4', 'B5', 'B3'], min:0, max:5000}, 
        'Landsat 7 SR', 
        cloud.make.landsatSR, 
        opt)
 }
 
 var landsat4SR = function(options) {
   var opt = options || null
   show('LANDSAT/LT04/C01/T1_SR', 
        'CLOUD_COVER', 
        {bands:['B4', 'B5', 'B3'], min:0, max:5000}, 
        'Landsat 4 SR', 
        cloud.make.landsatSR, 
        opt)
 }
 
 var landsat4TOA = function(options) {
   var opt = options || null
   show('LANDSAT/LT04/C01/T1_TOA', 
        'CLOUD_COVER',
        {bands:['B4', 'B5', 'B3'], min:0, max:0.5},
        'Landsat 4 TOA',
        cloud.make.landsatTOA,
        opt)
 }
 
 var landsat5TOA = function(options) {
   var opt = options || null
   show('LANDSAT/LT05/C01/T1_TOA', 
        'CLOUD_COVER',
        {bands:['B4', 'B5', 'B3'], min:0, max:0.5},
        'Landsat 5 TOA',
        cloud.make.landsatTOA,
        opt)
 }
 
 var landsat7TOA = function(options) {
   var opt = options || null
   show('LANDSAT/LE07/C01/T1_TOA', 
        'CLOUD_COVER',
        {bands:['B4', 'B5', 'B3'], min:0, max:0.5},
        'Landsat 7 TOA',
        cloud.make.landsatTOA,
        opt)
 }
 
 var landsat8TOA = function(options) {
   var opt = options || null
   show('LANDSAT/LC08/C01/T1_TOA', 
        'CLOUD_COVER',
        {bands:['B5', 'B7', 'B4'], min:0, max:0.5},
        'Landsat 8 TOA',
        cloud.make.landsatTOA,
        opt)
 }
 
 var modis = function(date, options, name) {
   var opt = options || null
   var col = ee.ImageCollection('MODIS/006/MOD09GA')
   var vis = {bands:['sur_refl_b02', 'sur_refl_b06', 'sur_refl_b01'], min:0, max:5000}
   var d = date || '2017-03-01'
   var eedate = ee.Date(d)
   var n = name || 'MODIS TERRA 500m'
   
   var filtered = col.filterDate(eedate, eedate.advance(1, 'day'))
   if (filtered.size().getInfo() !== 0) {
     var i = ee.Image(filtered.first())
     var masked = cloud.make.modis(opt)(i)
     Map.addLayer(masked, vis, n+' masked');
     Map.addLayer(i, vis, n+' NOT masked', false);
   }
   else {
     print('Collection is empty')
   }
 }
 
 exports.sentinel2 = sentinel2
 exports.landsat8SR = landsat8SR
 exports.landsat5SR = landsat5SR
 exports.landsat4SR = landsat4SR
 exports.landsat7SR = landsat7SR
 exports.landsat4TOA = landsat4TOA
 exports.modis = modis