// Step 1980: ProjectStackDeclarator
//
//  t1: declare and librariesForDomain returns qualifying libs
//  t2: library with score 0 not returned by librariesForDomain
//  t3: hasDomain checks operationDomains list
//  t4: empty declared set returns empty
//  t5: multiple qualifying libraries for one domain all returned

#include "ProjectStackDeclarator.h"
#include "CUDALibraryProfile.h"
#include "Sprint286IntegrationSummary.h"
#include <iostream>

namespace ws = whetstone;
static int p = 0, f = 0;
#define T(n) { std::cout << "  " << #n << "... "; }
#define P()  { std::cout << "PASS\n"; ++p; }
#define F(m) { std::cout << "FAIL: " << m << "\n"; ++f; }
#define C(c, m) if (!(c)) { F(m); return; }

void t1() {
    T(declare_and_librariesForDomain_returns_qualifying);
    auto ledger = ws::CUDALibraryProfile::buildLedger();
    ws::ProjectStackDeclarator decl;
    decl.projectId = "test";
    decl.declare("cublas");
    decl.declare("cufft");
    decl.declare("thrust");
    auto libs = decl.librariesForDomain("compute.matrix", ledger);
    C(libs.size() == 1,          "1 qualifying lib for compute.matrix");
    C(libs[0] == "cublas",       "cublas qualifies");
    auto fft_libs = decl.librariesForDomain("compute.fft", ledger);
    C(fft_libs.size() == 1,      "1 qualifying lib for compute.fft");
    C(fft_libs[0] == "cufft",    "cufft qualifies");
    P();
}

void t2() {
    T(zero_score_library_not_returned);
    auto ledger = ws::CUDALibraryProfile::buildLedger();
    ws::ProjectStackDeclarator decl;
    decl.declare("cublas");
    decl.declare("cufft");
    // cublas/compute.fft = 0, cufft/compute.fft = 0.99
    auto libs = decl.librariesForDomain("compute.fft", ledger);
    C(libs.size() == 1,       "only cufft qualifies");
    C(libs[0] == "cufft",     "cufft is the only result");
    P();
}

void t3() {
    T(hasDomain_checks_domains_list);
    ws::ProjectStackDeclarator decl;
    decl.addDomain("compute.matrix");
    decl.addDomain("serialization.json");
    C(decl.hasDomain("compute.matrix"),      "compute.matrix present");
    C(decl.hasDomain("serialization.json"),  "serialization.json present");
    C(!decl.hasDomain("compute.fft"),        "compute.fft absent");
    P();
}

void t4() {
    T(empty_declared_returns_empty);
    auto ledger = ws::CUDALibraryProfile::buildLedger();
    ws::ProjectStackDeclarator decl;
    auto libs = decl.librariesForDomain("compute.matrix", ledger);
    C(libs.empty(), "empty declared set -> empty result");
    P();
}

void t5() {
    T(multiple_qualifying_libs_all_returned);
    auto ledger = ws::Sprint286IntegrationSummary::buildSeedLedger();
    ws::ProjectStackDeclarator decl;
    decl.declare("cublas");
    decl.declare("numpy");
    // Both cublas (0.99) and numpy (0.95) qualify for compute.matrix
    auto libs = decl.librariesForDomain("compute.matrix", ledger);
    C(libs.size() == 2, "both cublas and numpy qualify");
    bool hasCublas = false, hasNumpy = false;
    for (const auto& l : libs) {
        if (l == "cublas") hasCublas = true;
        if (l == "numpy")  hasNumpy  = true;
    }
    C(hasCublas, "cublas present");
    C(hasNumpy,  "numpy present");
    P();
}

int main() {
    std::cout << "Step 1980: ProjectStackDeclarator\n";
    t1(); t2(); t3(); t4(); t5();
    std::cout << "\n" << p << "/" << (p + f) << " passed\n";
    return f > 0 ? 1 : 0;
}