public static async Task Orchestrate(
            [OrchestrationTrigger] DurableOrchestrationContext ctx)
        {
            var inpaintRequest = ctx.GetInput <InpaintRequest>();

            // TODO: get the setting from parameters
            var settings = new InpaintSettings();

            // TODO: downscale the input image
            // TODO: crop the input image

            var pyramid = await ctx.CallActivityAsync <CloudPyramid>(PyramidsGenerateActivity.Name, inpaintRequest);

            settings.MaxInpaintIterations = 20;
            //settings.PatchMatch.IterationsAmount = 2;
            //var kStep = settings.MeanShift.KDecreaseStep;
            //var minK = settings.MeanShift.MinK;

            for (byte levelIndex = 0; levelIndex < pyramid.LevelsAmount; levelIndex++)
            {
                var imageName   = pyramid.GetImageName(levelIndex);
                var mapping     = pyramid.GetMapping(levelIndex);
                var nnf         = pyramid.GetNnf(levelIndex);
                var inpaintArea = pyramid.GetInpaintArea(levelIndex);
                var mappings    = pyramid.GetSplittedMappings(levelIndex);
                var nnfs        = pyramid.GetSplittedNnfs(levelIndex);

                // TODO: this looks ugly
                var input = NnfInputData.From(nnf, inpaintRequest.Container, imageName, settings, mapping, inpaintArea, false, levelIndex, settings.MeanShift.K, nnfs, mappings);
                await ctx.CallSubOrchestratorAsync(InpaintLevelFunction.Name, input);
            }
        }
Пример #2
0
        public static async Task BuildNnf([OrchestrationTrigger] DurableOrchestrationContext ctx)
        {
            var input       = ctx.GetInput <NnfInputData>();
            var levelIndex  = input.LevelIndex;
            var settings    = input.Settings;
            var mappings    = input.Mappings;
            var nnfs        = input.SplittedNnfNames;
            var inpaintArea = input.InpaintAreaName;
            var imageName   = input.Image;
            var container   = input.Container;

            var tasks = new Task[mappings.Length];

            var isForward = true;

            for (var pmIteration = 0; pmIteration < settings.PatchMatch.IterationsAmount; pmIteration++)
            {
                // process in parallel
                if (mappings.Length > 1)
                {
                    for (int mapIndex = 0; mapIndex < mappings.Length; mapIndex++)
                    {
                        // TODO: this looks ugly
                        var pminput = NnfInputData.From(nnfs[mapIndex], container, imageName,
                                                        settings, mappings[mapIndex], inpaintArea, isForward, levelIndex,
                                                        settings.MeanShift.K, nnfs, mappings);
                        pminput.PatchMatchIteration = pmIteration;

                        tasks[mapIndex] = ctx.CallActivityAsync(NnfBuildActivity.Name, pminput);
                    }

                    await Task.WhenAll(tasks);

                    // TODO: merge nnf into one
                    await ctx.CallActivityAsync(NnfMergeActivity.Name, (nnfs : nnfs, mappings : mappings, resultNnf : input.NnfName, container : input.Container, input.Mapping));
                }
                else
                {
                    var pminput = NnfInputData.From(input.NnfName, container, imageName,
                                                    settings, input.Mapping, inpaintArea, isForward, levelIndex,
                                                    settings.MeanShift.K, nnfs, mappings);
                    pminput.PatchMatchIteration = pmIteration;

                    await ctx.CallActivityAsync(NnfBuildActivity.Name, pminput);
                }

                isForward = !isForward;
            }
        }