#include <iostream>

#include "vtkSmartPointer.h"
#include "vtkOSPRayMaterialLibrary.h"

void printSet(const std::set<std::string> &container);

void printVector(const std::vector<std::string> &container);

void printVectorDouble(const std::vector<double> &container);

int main() {
    const char *materialFile = "data/geometries.mtl";

    vtkSmartPointer<vtkOSPRayMaterialLibrary> matlib = vtkSmartPointer<vtkOSPRayMaterialLibrary>::New();
    cout << "Open " << materialFile << endl;
    matlib->ReadFile(materialFile);

    cout << "Parsed file OK, now check for expected contents." << endl;
    std::set<std::string> mats = matlib->GetMaterialNames();

    cout << "Materials are:" << endl;
    printSet(mats);

    auto ka = matlib->GetDoubleShaderVariable("Red", "Ka");
    cout<<"ka of Red"<< endl;
    printVectorDouble(ka);

    auto kd = matlib->GetDoubleShaderVariable("Red", "Kd");
    cout<<"kd of Red"<< endl;
    printVectorDouble(ka);


}

void printSet(const std::set<std::string> &container) {
    auto it = container.begin();
    while (it != container.end()) {
        cout << *it << "\t";
        ++it;
    }
    cout<<endl;
}

void printVector(const std::vector<std::string> &container) {
    auto it = container.begin();
    while (it != container.end()) {
        cout << *it << "\t";
        ++it;
    }
    cout<<endl;
}

void printVectorDouble(const std::vector<double> &container) {
    auto it = container.begin();
    while (it != container.end()) {
        cout << *it << "\t";
        ++it;
    }
    cout<<endl;
}