// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC is free software; you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC.  If not, see <http://www.gnu.org/licenses/>.
//

// Replace the following with your own code.
// WARNING: after doing this, you must prevent this file from
// being overwritten the next time you update BOINC source code.
// You can either:
// 1) write-protect this file, or
// 2) put this in a differently-named file and change the Makefile.am
//    (you need to prevent that from being overwritten too)
// In either case, put your version under source-code control, e.g. SVN

#include "sched_config.h"
#include "sched_msgs.h"
#include "sched_plan.h"

// return the number of usable CPUs, taking prefs into account.
// If prefs limit apply, set bounded to true.
//
static void get_ncpus(SCHEDULER_REQUEST& sreq, int& ncpus, bool& bounded) {
    ncpus = sreq.host.p_ncpus;
    bounded = false;
    if (sreq.global_prefs.max_ncpus_pct && sreq.global_prefs.max_ncpus_pct < 100) {
        bounded = true;
        ncpus = (int)((ncpus*sreq.global_prefs.max_ncpus_pct)/100.);
    }
}

bool app_plan(SCHEDULER_REQUEST& sreq, char* plan_class, HOST_USAGE& hu) {
    if (!strcmp(plan_class, "mt")) {
        // the following is for an app that can use anywhere
        // from 1 to 64 threads, can control this exactly,
        // and whose speedup is .95N
        // (so on a uniprocessor, we'll use a sequential app
        // if one is available)
        //
        int ncpus, nthreads;
        bool bounded;

        get_ncpus(sreq, ncpus, bounded);
        nthreads = ncpus;
        if (nthreads > 64) nthreads = 64;
        hu.avg_ncpus = nthreads;
        hu.max_ncpus = nthreads;
        sprintf(hu.cmdline, "--nthreads %d", nthreads);
        hu.flops = 0.95*sreq.host.p_fpops*nthreads;
        return true;
    } else if (!strcmp(plan_class, "cuda")) {
        // the following is for an app that uses a CUDA GPU
        // and some CPU also, and gets 50 GFLOPS total
        //
        for (unsigned int i=0; i<sreq.coprocs.coprocs.size(); i++) {
            COPROC* cp = sreq.coprocs.coprocs[i];
            if (!strcmp(cp->type, "CUDA")) {
                COPROC_CUDA* cp2 = (COPROC_CUDA*) cp;
                if ((cp2->prop.major)*100 + (cp2->prop.minor) <= 100) {
                    log_messages.printf(MSG_DEBUG, "Host GPU architecture < 1.1");
                    return false;
                } 
                COPROC* cu = new COPROC (cp->type);
                cu->count = 1;
                hu.coprocs.coprocs.push_back(cu);
                double x = 1e9/sreq.host.p_fpops;
                if (x > 1) x = 1;
                hu.avg_ncpus = x;
                hu.max_ncpus = x;
                hu.flops = 5e11;
                return true;
            }
        }
        if (config.debug_version_select) {
            log_messages.printf(MSG_DEBUG,
                "Host lacks CUDA coprocessor for plan class %s\n", plan_class
            );
        }
        return false;
    }
    log_messages.printf(MSG_CRITICAL,
        "Unknown plan class: %s\n", plan_class
    );
    return false;
}