Exemple #1
0
    static void RunMeshroomCompute(
        string MeshroomComputeBin,
        string ImagesDir,
        string graph,
        ComputeProgress ComputeProgressCB)
    {
        /* remove previous cache dir, if it exists */
        Utils.RemoveDir(Meshroom.GetCacheDir(ImagesDir));

        /* start meshroom compute process */
        var proc      = Utils.Run(MeshroomComputeBin, graph);
        var graphName = Path.GetFileNameWithoutExtension(graph);

        var stepsPoller = MeshroomProgress.GetPoller(ImagesDir);

        while (!proc.HasExited)
        {
            var done  = stepsPoller.PollStepsDone();
            var total = stepsPoller.TotalSteps;
            ComputeProgressCB(graphName, (float)done / (float)total);
            Thread.Sleep(1300);
        }

        var exitCode = proc.ExitCode;

        if (exitCode != 0)
        {
            var err = string.Format("{0} failed, exit code {1}", MeshroomComputeBin, exitCode);
            throw new Exception(err);
        }

        CopyCamerasSFMFile(ImagesDir, graphName);
    }
 static IEnumerable <string> NodeCacheDirs(string CacheDir)
 {
     foreach (var nodeRoot in Directory.GetDirectories(CacheDir))
     {
         yield return(Meshroom.GetNodeCacheDir(nodeRoot));
     }
 }
    static void WaitForCamInitDir(string path)
    {
        var camInitRoot = Path.Combine(path, "CameraInit");

        while (!Directory.Exists(camInitRoot))
        {
            Thread.Sleep(500);
        }

        string nodeDir;

        while ((nodeDir = Meshroom.GetNodeCacheDir(camInitRoot)) == null)
        {
            Thread.Sleep(1500);
        }

        string statusFile = Path.Combine(nodeDir, "status");

        while (!File.Exists(statusFile))
        {
            Thread.Sleep(500);
        }

        while (!Status.isDone(statusFile))
        {
            Thread.Sleep(500);
        }
    }
    public static ComputeProgress GetPoller(string path)
    {
        var CacheDir = Meshroom.GetCacheDir(path);

        WaitForCamInitDir(CacheDir);
        return(new ComputeProgress(StatusFiles(CacheDir)));
    }
Exemple #5
0
    static void CopyCamerasSFMFile(string ImagesDir, string graphName)
    {
        var MeshroomCacheDir = Meshroom.GetCacheDir(ImagesDir);
        var SfmCacheDir      =
            Meshroom.GetNodeCacheDir(
                Path.Combine(MeshroomCacheDir, "StructureFromMotion"));

        var camsSfmFile = Path.Combine(SfmCacheDir, "cameras.sfm");
        var destFile    = Path.Combine(ImagesDir, $"{graphName}.sfm");

        File.Copy(camsSfmFile, destFile, true);
    }
Exemple #6
0
    static void RemoveCacheDir(string ImagesDir, int chunkNum)
    {
        var cacheDir = Meshroom.GetCacheDir(ImagesDir);

        if (SAVE_MESHROOM_CACHE && Directory.Exists(cacheDir))
        {
            var destDir = Path.Combine(ImagesDir, $"MeshroomCache{chunkNum}");
            Log.Msg($"saving '{cacheDir}' to '{destDir}'");
            Directory.Move(cacheDir, destDir);
        }
        Utils.RemoveDir(cacheDir);
    }
Exemple #7
0
    public static void PhotogrammImages(
        string MeshroomComputeBin,
        string SensorDatabase, string VocTree, string ImagesDir,
        TimeBase TimeBase, uint LastFrame,
        ComputeProgress ComputeProgressCB)
    {
        var chunks = Chunks.GetChunks(TimeBase, LastFrame);
        var graphs = MeshroomGraphs(SensorDatabase, VocTree, ImagesDir, chunks);

        foreach (var graph in graphs)
        {
            RunMeshroomCompute(MeshroomComputeBin, ImagesDir, graph, ComputeProgressCB);
        }

        /* clean-up last meshroom cache directory */
        Utils.RemoveDir(Meshroom.GetCacheDir(ImagesDir));
    }