query.hpp

Go to the documentation of this file.

/*************************************************************************
 * Copyright (C) 2011 by Saleh Dindar and the Swarm-NG Development Team  *
 *                                                                       *
 * This program is free software; you can redistribute it and/or modify  *
 * it under the terms of the GNU General Public License as published by  *
 * the Free Software Foundation; either version 3 of the License.        *
 *                                                                       *
 * This program is distributed in the hope that it will be useful,       *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 * GNU General Public License for more details.                          *
 *                                                                       *
 * You should have received a copy of the GNU General Public License     *
 * along with this program; if not, write to the                         *
 * Free Software Foundation, Inc.,                                       *
 * 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ************************************************************************/

#pragma once

#include "common.hpp"
#include "log/io.hpp"
#include <boost/regex.hpp>
#include <boost/program_options.hpp>

namespace swarm { namespace query {

template<typename T>
T arg_parse(const std::string &s)
{
        if(s == "MIN") { return MIN; }
        if(s == "MAX") { return MAX; }
        return boost::lexical_cast<T>(s);
}

template<typename T>
void validate(boost::any& v,
        const std::vector<std::string>& values,
        range<T>* target_type, int)
{
        using namespace boost::program_options;
        typedef range<T> rangeT;

        // Make sure no previous assignment to 'a' was made.
        validators::check_first_occurrence(v);

        // Extract the first string from 'values'. If there is more than
        // one string, it's an error, and exception will be thrown.
        const std::string& s = validators::get_single_string(values);

        if(s == "ALL")
        {
                v = boost::any(rangeT(ALL));
                return;
        }

        static boost::regex r("(.+?)(?:\\.\\.(.+))?");
        boost::smatch match;
        if (boost::regex_match(s, match, r)) {
                //for(int i=0; i != match.size(); i++) { std::cerr << match[i] << "\n"; }
                if(match[2] == "")
                {
                        v = boost::any(rangeT(arg_parse<T>(match[1])));
                }
                else
                {
                        v = boost::any(rangeT(arg_parse<T>(match[1]), arg_parse<T>(match[2])));
                }
        } else {
#if BOOST_VERSION  < 104200
                throw validation_error("invalid value");
#else
                throw validation_error(validation_error::invalid_option_value);
#endif
        }
}

template<typename T>
std::ostream &operator<<(std::ostream &out, const range<T> &r)
{
        if(r.first == T(MIN) && r.last == T(MAX)) { return out << "ALL"; }
        if(r.first == r.last) { return out << r.first; }

        if(r.first == T(MIN)) { out << "MIN"; } else { out << r.first; }
        out << "..";
        if(r.last == T(MAX)) { out << "MAX"; } else { out << r.last; }

        return out;
}
        
void execute(const std::string &dbfile, time_range_t T, sys_range_t sys, body_range_t bod = body_range_t() );

std::ostream &output_record(std::ostream &out, gpulog::logrecord &lr, const body_range_t &bod = body_range_t() );

enum planets_coordinate_system_t {
  astrocentric, barycentric, jacobi, origin
};

void set_cartesian_output(const planets_coordinate_system_t& coordinate_system = origin) ;
void set_keplerian_output(const planets_coordinate_system_t& coordinate_system = jacobi) ;
void set_coordinate_system(const planets_coordinate_system_t& coordinate_system);  

} } // namespace swarm query