// Copyright (c) Microsoft Corporation. 
// Licensed under the MIT license. 

using System;
using System.IO;
using System.Threading.Tasks;

public class Program
{
    static void RunBillboardCloudVegetationPipeline(Simplygon.ISimplygon sg)
    {
        using (Simplygon.spSceneImporter sgSceneImporter = sg.CreateSceneImporter())
        {
            sgSceneImporter.SetImportFilePath( "../Assets/Tree/Tree.obj" );
            if(!sgSceneImporter.RunImport())
                throw new System.Exception("Failed to load Tree/Tree.obj.");
            Simplygon.spScene sgScene = sgSceneImporter.GetScene();

            // For all materials in the scene set the blend mode to blend (instead of opaque) 
            int materialCount = (int)sgScene.GetMaterialTable().GetMaterialsCount();
            for (int i = 0; i < materialCount; ++i)
            {
                sgScene.GetMaterialTable().GetMaterial(i).SetBlendMode(Simplygon.EMaterialBlendMode.Blend);
            }

            {
            }
            // Create the Impostor processor. 
            using (Simplygon.spBillboardCloudVegetationPipeline sgBillboardCloudVegetationPipeline = sg.CreateBillboardCloudVegetationPipeline())
            {
                using (Simplygon.spBillboardCloudSettings sgBillboardCloudSettings = sgBillboardCloudVegetationPipeline.GetBillboardCloudSettings())
                using (Simplygon.spMappingImageSettings sgMappingImageSettings = sgBillboardCloudVegetationPipeline.GetMappingImageSettings())
                {
                    // Set billboard cloud mode to Foliage. 
                    sgBillboardCloudSettings.SetBillboardMode( Simplygon.EBillboardMode.Foliage );
                    sgBillboardCloudSettings.SetBillboardDensity( 0.5f );
                    sgBillboardCloudSettings.SetGeometricComplexity( 0.9f );
                    sgBillboardCloudSettings.SetMaxPlaneCount( 10 );
                    sgBillboardCloudSettings.SetTwoSided( true );

                    using (Simplygon.spFoliageSettings sgFoliageSettings = sgBillboardCloudSettings.GetFoliageSettings())
                    {
                        // Set the parameters for separating foliage and trunk. 
                        sgFoliageSettings.SetSeparateTrunkAndFoliage( true );
                        sgFoliageSettings.SetSeparateFoliageTriangleRatio( 0.5f );
                        sgFoliageSettings.SetSeparateFoliageTriangleThreshold( 10 );
                        sgFoliageSettings.SetSeparateFoliageAreaThreshold( 0.1f );
                        sgFoliageSettings.SetSeparateFoliageSizeThreshold( 0.1f );
                        sgFoliageSettings.SetTrunkReductionRatio( 0.5f );

                    }
                    sgMappingImageSettings.SetMaximumLayers( 10 );

                    using (Simplygon.spMappingImageOutputMaterialSettings sgOutputMaterialSettings = sgMappingImageSettings.GetOutputMaterialSettings(0))
                    {
                        // Setting the size of the output material for the mapping image. This will be the output size of the 
                        // textures when we do material casting in a later stage. 
                        sgOutputMaterialSettings.SetTextureWidth( 1024 );
                        sgOutputMaterialSettings.SetTextureHeight( 1024 );
                        sgOutputMaterialSettings.SetMultisamplingLevel( 2 );

                    }
                }
                // Add diffuse material caster to pipeline. 
                using (Simplygon.spColorCaster sgDiffuseCaster = sg.CreateColorCaster())
                {
                    using (Simplygon.spColorCasterSettings sgDiffuseCasterSettings = sgDiffuseCaster.GetColorCasterSettings())
                    {
                        sgDiffuseCasterSettings.SetMaterialChannel( "Diffuse" );
                        sgDiffuseCasterSettings.SetOutputImageFileFormat( Simplygon.EImageOutputFormat.PNG );
                        sgDiffuseCasterSettings.SetBakeOpacityInAlpha( false );
                        sgDiffuseCasterSettings.SetOutputPixelFormat( Simplygon.EPixelFormat.R8G8B8 );
                        sgDiffuseCasterSettings.SetDilation( 10 );
                        sgDiffuseCasterSettings.SetFillMode( Simplygon.EAtlasFillMode.Interpolate );
                    }

                    sgBillboardCloudVegetationPipeline.AddMaterialCaster( sgDiffuseCaster, 0 );
                }

                // Add specular material caster to pipeline. 
                using (Simplygon.spColorCaster sgSpecularCaster = sg.CreateColorCaster())
                {
                    using (Simplygon.spColorCasterSettings sgSpecularCasterSettings = sgSpecularCaster.GetColorCasterSettings())
                    {
                        sgSpecularCasterSettings.SetMaterialChannel( "Specular" );
                        sgSpecularCasterSettings.SetOutputImageFileFormat( Simplygon.EImageOutputFormat.PNG );
                        sgSpecularCasterSettings.SetDilation( 10 );
                        sgSpecularCasterSettings.SetFillMode( Simplygon.EAtlasFillMode.Interpolate );
                    }

                    sgBillboardCloudVegetationPipeline.AddMaterialCaster( sgSpecularCaster, 0 );
                }

                // Add normals material caster to pipeline. 
                using (Simplygon.spNormalCaster sgNormalsCaster = sg.CreateNormalCaster())
                {
                    using (Simplygon.spNormalCasterSettings sgNormalsCasterSettings = sgNormalsCaster.GetNormalCasterSettings())
                    {
                        sgNormalsCasterSettings.SetMaterialChannel( "Normals" );
                        sgNormalsCasterSettings.SetGenerateTangentSpaceNormals( true );
                        sgNormalsCasterSettings.SetOutputImageFileFormat( Simplygon.EImageOutputFormat.PNG );
                        sgNormalsCasterSettings.SetDilation( 10 );
                        sgNormalsCasterSettings.SetFillMode( Simplygon.EAtlasFillMode.Interpolate );
                    }

                    sgBillboardCloudVegetationPipeline.AddMaterialCaster( sgNormalsCaster, 0 );
                }

                // Setup and run the opacity material casting. Make sure the there is no dilation or fill. 
                using (Simplygon.spOpacityCaster sgOpacityCaster = sg.CreateOpacityCaster())
                {
                    using (Simplygon.spOpacityCasterSettings sgOpacityCasterSettings = sgOpacityCaster.GetOpacityCasterSettings())
                    {
                        sgOpacityCasterSettings.SetMaterialChannel( "Opacity" );
                        sgOpacityCasterSettings.SetOutputImageFileFormat( Simplygon.EImageOutputFormat.PNG );
                        sgOpacityCasterSettings.SetFillMode( Simplygon.EAtlasFillMode.NoFill );
                        sgOpacityCasterSettings.SetOutputPixelFormat( Simplygon.EPixelFormat.R8 );
                    }

                    sgBillboardCloudVegetationPipeline.AddMaterialCaster( sgOpacityCaster, 0 );
                }

                // Start the impostor pipeline. 
                sgBillboardCloudVegetationPipeline.RunScene(sgScene, Simplygon.EPipelineRunMode.RunInThisProcess);

                // Get the processed scene. 
                using (Simplygon.spScene sgProcessedScene = sgBillboardCloudVegetationPipeline.GetProcessedScene())
                {
                    using (Simplygon.spSceneExporter sgSceneExporter = sg.CreateSceneExporter())
                    {
                        sgSceneExporter.SetScene(sgProcessedScene);
                        sgSceneExporter.SetExportFilePath( "FoliageOutput.obj" );
                        if(!sgSceneExporter.RunExport())
                            throw new System.Exception("Failed to save FoliageOutput.obj.");

                    }
                }
            }
        }
    }
    static int Main(string[] args)
    {
        using (var sg = Simplygon.Loader.InitSimplygon(out var errorCode, out var errorMessage))
        {
            if (errorCode != Simplygon.EErrorCodes.NoError)
                return (int)errorCode;
            RunBillboardCloudVegetationPipeline(sg);
        }

        return 0;
    }
}