/********************************************************************************

* *

* This file is part of IfcOpenShell. *

* *

* IfcOpenShell is free software: you can redistribute it and/or modify *

* it under the terms of the Lesser GNU General Public License as published by *

* the Free Software Foundation, either version 3.0 of the License, or *

* (at your option) any later version. *

* *

* IfcOpenShell 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 *

* Lesser GNU General Public License for more details. *

* *

* You should have received a copy of the Lesser GNU General Public License *

* along with this program. If not, see <http://www.gnu.org/licenses/>. *

* *

********************************************************************************/

/********************************************************************************

* *

* This file defines materials in .mtl format for several IFC datatypes *

* *

********************************************************************************/

#include <string>

#include <sstream>

class ObjMaterial {

private:

std::string data;

public:

ObjMaterial(const std::string& name,

double Kd_r = 0.7f,double Kd_g = 0.7f,double Kd_b = 0.7f,

double Ks_r = 0.2f,double Ks_g = 0.2f,double Ks_b = 0.2f,

double Ka_r = 0.1f,double Ka_g = 0.1f,double Ka_b = 0.1f,

double Ns = 10.0f, double Tr = 1.0f) {

std::stringstream ss;

ss << "newmtl " << name << std::endl;

ss << "Kd " << Kd_r << " " << Kd_g << " " << Kd_b << std::endl;

ss << "Ks " << Ks_r << " " << Ks_g << " " << Ks_b << std::endl;

ss << "Ka " << Ka_r << " " << Ka_g << " " << Ka_b << std::endl;

ss << "Ns " << Ns << std::endl;

ss << "Tr " << Tr << std::endl;

ss << "d " << Tr << std::endl;

ss << "D " << Tr << std::endl;

data = ss.str();

}

friend ostream& operator<<(ostream& o, const ObjMaterial& m) {o << m.data; return o;}

};

ObjMaterial GetMaterial(const std::string& s) {

if ( s == "IfcSite" ) { return ObjMaterial("IfcSite",0.75f,0.8f,0.65f); }

if ( s == "IfcSlab" ) { return ObjMaterial("IfcSlab",0.4f,0.4f,0.4f); }

if ( s == "IfcWallStandardCase" ) { return ObjMaterial("IfcWallStandardCase",0.9f,0.9f,0.9f); }

if ( s == "IfcWall" ) { return ObjMaterial("IfcWall",0.9f,0.9f,0.9f); }

if ( s == "IfcWindow" ) { return ObjMaterial("IfcWindow",0.75f,0.8f,0.75f,1.0f,1.0f,1.0f,0.0f,0.0f,0.0f,500.0f,0.3f); }

if ( s == "IfcDoor" ) { return ObjMaterial("IfcDoor",0.55f,0.3f,0.15f); }

if ( s == "IfcBeam" ) { return ObjMaterial("IfcBeam",0.75f,0.7f,0.7f); }

if ( s == "IfcRailing" ) { return ObjMaterial("IfcRailing",0.65f,0.6f,0.6f); }

if ( s == "IfcMember" ) { return ObjMaterial("IfcMember",0.65f,0.6f,0.6f); }

if ( s == "IfcPlate" ) { return ObjMaterial("IfcPlate",0.8f,0.8f,0.8f); }

return ObjMaterial(s);

}