// TO DO

// Create another script that can run
// for site in sites:
    // for year in years:
        // register_images()

// VARIABLE DEFINITIONS

// images_to_register        = All images to be registered for a given site and year (no target or seed)
// batch_images_to_register  = Batch images to be registered (no target or seed)
// batch_images_all          = Target, seed and batch images to be registered 

var batch_size = 10;

// Must use forwardslashes in filepath, not backslashes
var batch_download_csv = File("C:/Users/z5079346/OneDrive - UNSW/Projects/Coastsnap_test/CoastSnap_Sites.csv")

// retreive site names from batch_download.csv
var csv_data=[];
batch_download_csv.open('r');
while(!batch_download_csv.eof){   
    var InputLine = batch_download_csv.readln(); 
    if(InputLine.length > 3) csv_data.push(InputLine);
}  
batch_download_csv.close();
var site_names = csv_data.toString().split(",")

// Images parent directory
var parent_folder_path = "C:/Users/z5079346/OneDrive - UNSW/Projects/Coastsnap_test/Images_Test";

var batch_images_to_register = []; // Used in exportLayersToPNG

// Loop through sites
for(var i=5; i<site_names.length; i+=5) {
    var site_name = site_names[i];
    var site_path = parent_folder_path + "/" + site_name;

    // Retrieve target and seed images for the site
    var target_folder = new Folder(site_path + '/Target Image');
    var seed_folder = new Folder(site_path + '/Target Image' + '/Seed Images');
    var target_image = target_folder.getFiles("Target.jpg");
    var seed_images = seed_folder.getFiles("*.jpg");

    // Retrieve processed image years for the site
    var processed_folder = new Folder(site_path + '/Processed');
    var processed_subFolders = processed_folder.getFiles()

    // Loop through years 
    for (var j = 0; j < processed_subFolders.length; j++) {
        
        var year = processed_subFolders[j].name.toString();

        var processed_folder = new Folder(site_path + '/Processed/' + year);
        var photoshop_folder = new Folder(site_path + '/Photoshop/' + year);
        var processed_images_all = processed_folder.getFiles("*.jpg");
        var photoshop_images = photoshop_folder.getFiles("*.jpg");
    
        // Ignore images in processed that have already been registered
        var images_to_register = imagesNotRegistered(processed_images_all, photoshop_images);

        // Register images in batches, to avoid reaching the image limit that causes photoshop to crash
        for (var k = 0; k < images_to_register.length; k += batch_size) {
            batch_register_images(k, site_path, site_name, images_to_register, target_image, seed_images);
        } 
    }
}


// This is the main function, responsible for calling photoshop functions
// in a sequential order to register all_images
function batch_register_images(batchIndex, site_path, site_name, images_to_register, target_image, seed_images) {

    var batch_images_all = [];
    batch_images_to_register = [];
    batch_images_all.push(target_image[0])

    for (var i = 0; i < seed_images.length; i++ ) {
        batch_images_all.push(seed_images[i]);
    }
    
    for (var i = batchIndex*1; i < batchIndex + batch_size; i++) {
        if(i < images_to_register.length) {
            batch_images_all.push(images_to_register[i]);
            batch_images_to_register.push(images_to_register[i].name);

        }
    }

    stackFiles(batch_images_all);
    lockTarget();
    selectAllLayers();
    autoAlign();
    var target_size = cropToTarget();
    var target_width = target_size[0];
    var target_height = target_size[1];

    savePhotoshopDocument(site_path, site_name);
    exportLayersToPNG(target_width, target_height, batch_images_to_register, seed_images); // Won't overwrite images that already exist

    app.activeDocument.close(SaveOptions.DONOTSAVECHANGES)
}


function imagesNotRegistered(site_images_all, photoshop_images) {
    
    var site_images = [];

    // Extract images that haven't been registered 
    for (var i = 0; i < site_images_all.length; i ++) { // For each site image
        var isRegistered = false;
        for (var j = 0; j < photoshop_images.length; j++ ) {  // Check if already exists in photoshop images    
            var site_name_CHECK = site_images_all[i].name.slice(0,-4) + '_CHECK.jpg';
            if((site_images_all[i].name.toString() == photoshop_images[j].name.toString()) || (site_name_CHECK.toString() == photoshop_images[j].name.toString())) {
                isRegistered = true;
                break;
            }
        }
        if(!isRegistered) { // If it doesn't, register it
            site_images.push(site_images_all[i]);
        }
    }  
    return site_images
}

function stackFiles(sFiles){  

    var loadLayersFromScript = true;  
    
    var SCRIPTS_FOLDER =  decodeURI(app.path + '/' + localize('$$$/ScriptingSupport/InstalledScripts=Presets/Scripts')); 
    
    $.evalFile( new File(SCRIPTS_FOLDER +  '/Load Files into Stack.jsx'));   
    
    loadLayers.intoStack(sFiles, false);  
    
};

function selectAllLayers() {

    var desc = new ActionDescriptor();

    var ref = new ActionReference();

    ref.putEnumerated( charIDToTypeID('Lyr '), charIDToTypeID('Ordn'), charIDToTypeID('Trgt') );

    desc.putReference( charIDToTypeID('null'), ref );

    executeAction( stringIDToTypeID('selectAllLayers'), desc, DialogModes.NO );

};

function autoAlign() {

    var desc = new ActionDescriptor();
    
    var ref = new ActionReference();
    
    ref.putEnumerated( charIDToTypeID('Lyr '), charIDToTypeID('Ordn'), charIDToTypeID('Trgt') );
    
    desc.putReference( charIDToTypeID('null'), ref );
    
    desc.putEnumerated( charIDToTypeID('Usng'), charIDToTypeID('ADSt'), stringIDToTypeID('ADSContent') );
    
    desc.putEnumerated( charIDToTypeID('Aply'), stringIDToTypeID('projection'), charIDToTypeID('Auto') );
    
    desc.putBoolean( stringIDToTypeID('vignette'), false );
    
    desc.putBoolean( stringIDToTypeID('radialDistort'), false );
    
    executeAction( charIDToTypeID('Algn'), desc, DialogModes.NO );
        
};

function lockTarget() {
    var layerRef = app.activeDocument.artLayers.getByName("Target.jpg")
    layerRef.allLocked = true;
    app.activeDocument.activeLayer.linkedLayers
}


// CROP 
function cropToTarget() {
    var layerRef = app.activeDocument.artLayers.getByName("Target.jpg")
    app.activeDocument.crop(layerRef.bounds)
    var theBounds = app.activeDocument.activeLayer.bounds;
    var layerWidth = theBounds[2] - theBounds[0];
    var layerHeight = theBounds[3] - theBounds[1];
    return [layerWidth, layerHeight];

}

// TO BE IMPLEMENTED
// function imageSize(layerRef) {

// }

// SAVE PHOTOSHOP DOCUMENT
function savePhotoshopDocument(site_path, site_name) {
    var PSdocumentFile = new File(site_path + '/' + site_name + '.psd');
    app.activeDocument.saveAs(PSdocumentFile)
}



// EXPORT LAYERS TO IMAGES

// $.evalFile(File(app.path + '/Presets/Scripts/Export Layers To Files.jsx'));
function exportLayersToPNG(target_width, target_height, batch_images_to_register, seed_images){
    if(!documents.length) return;
    var doc = activeDocument;
    var oldPath = activeDocument.path;
    
    var outFolder = new Folder(oldPath + "/Photoshop/" + year);
    if (!outFolder.exists) {
        outFolder.create();
    }
    
    scanLayerSets(doc);
    
    function scanLayerSets(el) {

        // // find layer groups
        // for(var a=0;a<el.layerSets.length;a++){
        //     var lname = el.layerSets[a].name;
        //     if (lname.substr(-4) == ".jpg") {
        //         saveLayer(el.layers.getByName(lname), lname, oldPath, true);
        //     } else {
        //         // recursive
        //         scanLayerSets(el.layerSets[a]);
        //     }
        // }
        
        var layerToRegister = true;
        // find plain layers in current group that are site images only (not target or seed)
        for(var j=0; j<el.artLayers.length; j++) { // Loop through photoshop document layers (images)
            var name = el.artLayers[j].name; 
            layerToRegister = true;

            // for (var i = 0; i < seed_images.length; i++ ) {
            //     if((seed_images[i].toString() == name.toString()) || (name.toString() == 'Target.jpg')) {
            //         layerToRegister = false;
            //     }
            // }
            // if(layerToRegister) {
                    
            //     app.activeDocument.activeLayer = el.layers.getByName(name);
            //     var theBounds = app.activeDocument.activeLayer.bounds;
            //     var aligned_image_width = theBounds[2] - theBounds[0];
            //     var aligned_image_height = theBounds[3] - theBounds[1];

            //     var name_updated = name;   
            //     // COMAPRE THE DIMENSIONS OF THE ALIGNED IMAGE WITH THE TARGET
            //     // IF SIGNIFICANT DIFFERENCE (30%), TAG IMAGE WITH '_CHECK.jpg'
            //     if((aligned_image_width/target_width) < 0.7 || (aligned_image_height/target_height) < 0.7) {
            //         name_updated = name.slice(0,-4) + '_CHECK.jpg';
            //     }

            //     saveLayer(el.layers.getByName(name), name_updated, oldPath, false);
            // }

            for (var i = 0; i < batch_images_to_register.length; i++ ) { // Loop through batch_images_to_register 
                if(batch_images_to_register[i].toString() == name.toString()) {
                    
                    app.activeDocument.activeLayer = el.layers.getByName(name);
                    var theBounds = app.activeDocument.activeLayer.bounds;
                    var aligned_image_width = theBounds[2] - theBounds[0];
                    var aligned_image_height = theBounds[3] - theBounds[1];

                    var name_updated = name;   
                    // COMAPRE THE DIMENSIONS OF THE ALIGNED IMAGE WITH THE TARGET
                    // IF SIGNIFICANT DIFFERENCE (30%), TAG IMAGE WITH '_CHECK.jpg'
                    if((aligned_image_width/target_width) < 0.7 || (aligned_image_height/target_height) < 0.7) {
                        name_updated = name.slice(0,-4) + '_CHECK.jpg';
                    }

                    saveLayer(el.layers.getByName(name), name_updated, oldPath, false);
                }
            }

        }
    
    }
    
    function saveLayer(layer, lname, path, shouldMerge) {
        activeDocument.activeLayer = layer;
        dupLayers();
        if (shouldMerge === undefined || shouldMerge === true) {
            activeDocument.mergeVisibleLayers();
        }
        //activeDocument.trim(TrimType.TRANSPARENT,true,true,true,true);
        var saveFile = File(path +"/Photoshop/"+year+"/"+lname);
        SavePNG(saveFile);
        app.activeDocument.close(SaveOptions.DONOTSAVECHANGES);
    }
    
};
    
function dupLayers() { 
    var desc143 = new ActionDescriptor();
    var ref73 = new ActionReference();
    ref73.putClass( charIDToTypeID('Dcmn') );
    desc143.putReference( charIDToTypeID('null'), ref73 );
    desc143.putString( charIDToTypeID('Nm  '), activeDocument.activeLayer.name );
    var ref74 = new ActionReference();
    ref74.putEnumerated( charIDToTypeID('Lyr '), charIDToTypeID('Ordn'), charIDToTypeID('Trgt') );
    desc143.putReference( charIDToTypeID('Usng'), ref74 );
    executeAction( charIDToTypeID('Mk  '), desc143, DialogModes.NO );
};
    
function SavePNG(saveFile){
    var pngOpts = new ExportOptionsSaveForWeb; 
    pngOpts.format = SaveDocumentType.PNG
    pngOpts.PNG8 = false; 
    pngOpts.transparency = true; 
    pngOpts.interlaced = false; 
    pngOpts.quality = 200;
    activeDocument.exportDocument(new File(saveFile),ExportType.SAVEFORWEB,pngOpts); 
}