stage_plan.h

Go to the documentation of this file.

 
#ifndef STAGE_PLAN_H
#define STAGE_PLAN_H
 
#include <sstream>
#include <string>
 
struct StagePlan {
    // Step D sub-steps
    bool match = true;  
    bool track = true;  
    bool format = true; 
    // Step E / F
    bool recon = true;  
    bool deform = true; 
 
    // Optional narrowing for embarrassingly-parallel array tasks.
    int only_pair = 0; 
    int only_cam = 0;  
 
    bool anyStepD() const { return match || track || format; }
 
    static bool parse(const std::string& spec, StagePlan& out, std::string* err) {
        out.match = out.track = out.format = out.recon = out.deform = false;
        std::stringstream ss(spec);
        std::string tok;
        while (std::getline(ss, tok, ',')) {
            // trim + lowercase
            size_t b = tok.find_first_not_of(" \t");
            size_t e = tok.find_last_not_of(" \t");
            if (b == std::string::npos) continue;
            tok = tok.substr(b, e - b + 1);
            for (char& c : tok) c = static_cast<char>(::tolower(c));
 
            if (tok == "all") {
                out.match = out.track = out.format = out.recon = out.deform = true;
            } else if (tok == "d") {
                out.match = out.track = out.format = true;
            } else if (tok == "match" || tok == "matching") {
                out.match = true;
            } else if (tok == "track" || tok == "tracking") {
                out.track = true;
            } else if (tok == "format") {
                out.format = true;
            } else if (tok == "e" || tok == "recon") {
                out.recon = true;
            } else if (tok == "f" || tok == "deform") {
                out.deform = true;
            } else {
                if (err) *err = "unknown stage token: '" + tok + "'";
                return false;
            }
        }
        return true;
    }
};
 
#endif // STAGE_PLAN_H