Commit: eda377c223cfc0460e3294c95771313f8d4388c3
Author: Gaia Clary
Date:   Tue Jun 4 14:31:22 2019 +0200
Branches: master
https://developer.blender.org/rBeda377c223cfc0460e3294c95771313f8d4388c3

update Collada Importer: reworked export and import of Materials

- added import of transparency and emission into principled BSDF Shader
- added support for importing all default collada material parameters

  * diffuse
  * emission
  * index_of_refraction
  * shininess (mapped to BSDF Roughness)
  * reflectivity (mapped to BSDF Metallic)
  * transparency + transparent mapped to BSDF Alpha)
  * ambient (creates unconnected texture node)
  * specular (creates unconnected texture node)
  * reflective(creates unconnected texture node)

- added support for exporting collada material parameters:

  * diffuse
  * emission
  * index_of_refraction
  * shininess (mapped to BSDF Roughness)
  * reflectivity (mapped to BSDF Metallic)
  * transparency + transparent mapped to BSDF Alpha)

- prepared support for exporting the following parameters
  but currently commented out:

  * ambient (creates unconnected texture node)
  * specular (creates unconnected texture node)
  * reflective(creates unconnected texture node)

  Problem: For now we only allow export of principled BSDF based
  materials. I am not sure from where to get ambient, specular
  and reflective as those values are not included in the
  principled BSDF Shader (wip).

===================================================================

M       source/blender/collada/BlenderTypes.h
M       source/blender/collada/DocumentImporter.cpp
M       source/blender/collada/EffectExporter.cpp
M       source/blender/collada/EffectExporter.h
M       source/blender/collada/Materials.cpp
M       source/blender/collada/Materials.h
M       source/blender/collada/collada_utils.cpp
M       source/blender/collada/collada_utils.h

===================================================================

diff --git a/source/blender/collada/BlenderTypes.h 
b/source/blender/collada/BlenderTypes.h
index 22ad9b8c1e0..0e024be2374 100644
--- a/source/blender/collada/BlenderTypes.h
+++ b/source/blender/collada/BlenderTypes.h
@@ -23,6 +23,7 @@
 
 typedef float(Vector)[3];
 typedef float(Quat)[4];
+typedef float(Color)[4];
 typedef float(Matrix)[4][4];
 typedef double(DMatrix)[4][4];
 
diff --git a/source/blender/collada/DocumentImporter.cpp 
b/source/blender/collada/DocumentImporter.cpp
index 1c241671c81..06f6682f401 100644
--- a/source/blender/collada/DocumentImporter.cpp
+++ b/source/blender/collada/DocumentImporter.cpp
@@ -796,14 +796,21 @@ bool DocumentImporter::writeMaterial(const 
COLLADAFW::Material *cmat)
 void DocumentImporter::write_profile_COMMON(COLLADAFW::EffectCommon *ef, 
Material *ma)
 {
   MaterialNode matNode = MaterialNode(mContext, ef, ma, uid_image_map);
-  matNode.set_reflectivity(ef->getReflectivity().getFloatValue());
-  matNode.set_ior(ef->getIndexOfRefraction().getFloatValue());
-  matNode.set_diffuse(ef->getDiffuse(), "Diffuse");
-  matNode.set_ambient(ef->getAmbient(), "Ambient");
-  matNode.set_specular(ef->getSpecular(), "Specular");
-  matNode.set_reflective(ef->getReflective(), "Reflective");
-  matNode.set_emission(ef->getEmission(), "Emission");
-  matNode.set_opacity(ef->getOpacity(), "Opacity");
+
+  /* Direct mapping to principled BSDF Shader */
+  matNode.set_diffuse(ef->getDiffuse());
+  matNode.set_emission(ef->getEmission());
+  matNode.set_ior(ef->getIndexOfRefraction());
+  matNode.set_alpha(ef->getOpaqueMode(), ef->getTransparent(), 
ef->getTransparency());
+
+  /* following mapping still needs to be verified */
+  matNode.set_shininess(ef->getShininess());
+  matNode.set_reflectivity(ef->getReflectivity());
+
+  /* not supported by principled BSDF */
+  matNode.set_ambient(ef->getAmbient());
+  matNode.set_specular(ef->getSpecular());
+  matNode.set_reflective(ef->getReflective());
 }
 
 /** When this method is called, the writer must write the effect.
diff --git a/source/blender/collada/EffectExporter.cpp 
b/source/blender/collada/EffectExporter.cpp
index a45cdccecf3..a0ad3740e45 100644
--- a/source/blender/collada/EffectExporter.cpp
+++ b/source/blender/collada/EffectExporter.cpp
@@ -104,6 +104,7 @@ void 
EffectsExporter::set_transparency(COLLADASW::EffectProfile &ep, Material *m
   double alpha = bc_get_alpha(ma);
   ep.setTransparency(alpha, false, "alpha");
 }
+
 void EffectsExporter::set_diffuse_color(COLLADASW::EffectProfile &ep, Material 
*ma)
 {
   // get diffuse color
@@ -111,6 +112,25 @@ void 
EffectsExporter::set_diffuse_color(COLLADASW::EffectProfile &ep, Material *
   ep.setDiffuse(cot, false, "diffuse");
 }
 
+void EffectsExporter::set_ambient(COLLADASW::EffectProfile &ep, Material *ma)
+{
+  // get diffuse color
+  COLLADASW::ColorOrTexture cot = bc_get_ambient(ma);
+  ep.setAmbient(cot, false, "ambient");
+}
+void EffectsExporter::set_specular(COLLADASW::EffectProfile &ep, Material *ma)
+{
+  // get diffuse color
+  COLLADASW::ColorOrTexture cot = bc_get_specular(ma);
+  ep.setSpecular(cot, false, "specular");
+}
+void EffectsExporter::set_reflective(COLLADASW::EffectProfile &ep, Material 
*ma)
+{
+  // get diffuse color
+  COLLADASW::ColorOrTexture cot = bc_get_reflective(ma);
+  ep.setReflective(cot, false, "reflective");
+}
+
 void EffectsExporter::set_reflectivity(COLLADASW::EffectProfile &ep, Material 
*ma)
 {
   double reflectivity = bc_get_reflectivity(ma);
@@ -123,6 +143,18 @@ void 
EffectsExporter::set_emission(COLLADASW::EffectProfile &ep, Material *ma)
   ep.setEmission(cot, false, "emission");
 }
 
+void EffectsExporter::set_ior(COLLADASW::EffectProfile &ep, Material *ma)
+{
+  double alpha = bc_get_ior(ma);
+  ep.setIndexOfRefraction(alpha, false, "ior");
+}
+
+void EffectsExporter::set_shininess(COLLADASW::EffectProfile &ep, Material *ma)
+{
+  double shininess = bc_get_shininess(ma);
+  ep.setShininess(shininess, false, "shininess");
+}
+
 void EffectsExporter::get_images(Material *ma, KeyImageMap &material_image_map)
 {
   if (!ma->use_nodes) {
@@ -180,10 +212,17 @@ void EffectsExporter::operator()(Material *ma, Object *ob)
 
   COLLADASW::ColorOrTexture cot;
 
-  set_transparency(ep, ma);
   set_diffuse_color(ep, ma);
-  set_reflectivity(ep, ma);
   set_emission(ep, ma);
+  set_ior(ep, ma);
+  set_shininess(ep, ma);
+  set_reflectivity(ep, ma);
+  set_transparency(ep, ma);
+
+  /* TODO: from where to get ambient, specular and reflective? */
+  // set_ambient(ep, ma);
+  // set_specular(ep, ma);
+  // set_reflective(ep, ma);
 
   get_images(ma, material_image_map);
   std::string active_uv(getActiveUVLayerName(ob));
diff --git a/source/blender/collada/EffectExporter.h 
b/source/blender/collada/EffectExporter.h
index 75511800081..57df844233c 100644
--- a/source/blender/collada/EffectExporter.h
+++ b/source/blender/collada/EffectExporter.h
@@ -55,10 +55,17 @@ class EffectsExporter : COLLADASW::LibraryEffects {
 
  private:
   void set_shader_type(COLLADASW::EffectProfile &ep, Material *ma);
-  void set_transparency(COLLADASW::EffectProfile &ep, Material *ma);
+
   void set_diffuse_color(COLLADASW::EffectProfile &ep, Material *ma);
-  void set_reflectivity(COLLADASW::EffectProfile &ep, Material *ma);
   void set_emission(COLLADASW::EffectProfile &ep, Material *ma);
+  void set_ior(COLLADASW::EffectProfile &ep, Material *ma);
+  void set_shininess(COLLADASW::EffectProfile &ep, Material *ma);
+  void set_reflectivity(COLLADASW::EffectProfile &ep, Material *ma);
+  void set_transparency(COLLADASW::EffectProfile &ep, Material *ma);
+  void set_ambient(COLLADASW::EffectProfile &ep, Material *ma);
+  void set_specular(COLLADASW::EffectProfile &ep, Material *ma);
+  void set_reflective(COLLADASW::EffectProfile &ep, Material *ma);
+
   void get_images(Material *ma, KeyImageMap &uid_image_map);
   void create_image_samplers(COLLADASW::EffectProfile &ep,
                              KeyImageMap &uid_image_map,
diff --git a/source/blender/collada/Materials.cpp 
b/source/blender/collada/Materials.cpp
index 4313e634b23..becbe49a55e 100644
--- a/source/blender/collada/Materials.cpp
+++ b/source/blender/collada/Materials.cpp
@@ -21,6 +21,9 @@ MaterialNode::MaterialNode(bContext *C, Material *ma, 
KeyImageMap &key_image_map
 {
   ntree = prepare_material_nodetree();
   setShaderType();
+  shader_node = add_node(SH_NODE_BSDF_PRINCIPLED, 0, 300, "");
+  output_node = add_node(SH_NODE_OUTPUT_MATERIAL, 300, 300, "");
+  add_link(shader_node, 0, output_node, 0);
 }
 
 MaterialNode::MaterialNode(bContext *C,
@@ -118,25 +121,64 @@ void MaterialNode::add_link(bNode *from_node, int 
from_index, bNode *to_node, in
   nodeAddLink(ntree, from_node, from_socket, to_node, to_socket);
 }
 
-void MaterialNode::set_reflectivity(float val)
+void MaterialNode::set_reflectivity(COLLADAFW::FloatOrParam &val)
+{
+  float reflectivity = val.getFloatValue();
+  bNodeSocket *socket = nodeFindSocket(shader_node, SOCK_IN, "Metallic");
+  ((bNodeSocketValueFloat *)socket->default_value)->value = reflectivity;
+
+  material->metallic = reflectivity;
+}
+
+void MaterialNode::set_shininess(COLLADAFW::FloatOrParam &val)
+{
+  float roughness = val.getFloatValue();
+  bNodeSocket *socket = nodeFindSocket(shader_node, SOCK_IN, "Roughness");
+  ((bNodeSocketValueFloat *)socket->default_value)->value = roughness;
+}
+
+void MaterialNode::set_ior(COLLADAFW::FloatOrParam &val)
 {
-  material->metallic = val;
-  bNodeSocket *socket = (bNodeSocket *)BLI_findlink(&shader_node->inputs, 
BC_PBR_METALLIC);
-  *(float *)socket->default_value = val;
+  float ior = val.getFloatValue();
+  if (ior < 0) {
+    fprintf(stderr,
+            "IOR of negative value is not allowed for materials (using Blender 
default value "
+            "instead)");
+    return;
+  }
+
+  bNodeSocket *socket = nodeFindSocket(shader_node, SOCK_IN, "IOR");
+  ((bNodeSocketValueFloat *)socket->default_value)->value = ior;
 }
 
-void MaterialNode::set_ior(float val)
+void MaterialNode::set_alpha(COLLADAFW::EffectCommon::OpaqueMode mode,
+                             COLLADAFW::ColorOrTexture &cot,
+                             COLLADAFW::FloatOrParam &val)
 {
-  bNodeSocket *socket = (bNodeSocket *)BLI_findlink(&shader_node->inputs, 
BC_PBR_IOR);
-  *(float *)socket->default_value = val;
+  if (effect == nullptr) {
+    return;
+  }
+
+  if (cot.isColor() || !cot.isValid()) {
+    COLLADAFW::Color col = (cot.isValid()) ? cot.getColor() : 
COLLADAFW::Color(1, 1, 1, 1);
+    float alpha = val.getFloatValue() * col.getAlpha();  // Assuming A_ONE 
opaque mode
+
+    bNodeSocket *socket = nodeFindSocket(shader_node, SOCK_IN, "Alpha");
+    ((bNodeSocketValueFloat *)socket->default_value)->value = alpha;
+  }
+  else if (cot.isTexture()) {
+    int locy = -300 * (node_map.size() - 2);
+    add_texture_node(cot, -300, locy, "Alpha");
+    // TODO: Connect node
+  }
 }
 
-void MaterialNode::set_diffuse(COLLADAFW::ColorOrTexture &cot, std::string 
label)
+void MaterialNode::set_diffuse(COLLADAFW::ColorOrTexture &cot)
 {
   int locy = -300 * (node_map.size() - 2);
   if (cot.isColor()) {
     COLLADAFW::Color col = cot.getColor();
-    bNodeSocket *socket = (bNodeSocket *)BLI_findlink(&shader_node->inputs, 
BC_PBR_DIFFUSE);
+    bNodeSocket *socket = nodeFindSocket(shader_node, SOCK_IN, "Base Color");
     float *fcol = (float *)socket->default_value;
 
     fcol[0] = material->r = col.getRed();
@@ -145,7 +187,7 @@ void MaterialNode::set_diffuse(COLLADAFW::ColorOrTexture 
&cot, std::string label
     fcol[3] = material->a = col.getAlpha();
   }
   else if (cot.isTexture()) {
-    bNode *texture_node = add_texture_node(cot, -300, locy, label);
+    bNode *texture_node = add_texture_node(cot, -300, locy, "Base Color");
     if (texture_node != NULL) {
       add_link(texture_node, 0, shader_node, 0);
     }
@@ -159,7 +201,7 @@ Image *MaterialNode::get_diffuse_image()
     return nullptr;
   }
 
-  bNodeSocket *in_socket = (bNodeSocket *)BLI_findlink(&shader->inputs, 
BC_PBR_DIFFUSE);
+  bNodeSocket *in_socket = nodeFindSocket(shader_node, SOCK_IN, "Base Color");
   if (in_socket == nullptr) {
     return nullptr;
   }
@@ -193,54 +235,60 @@ static bNodeSocket *set_color(bNode *node, 
COLLADAFW::Color co

@@ Diff output truncated at 10240 characters. @@

_______________________________________________
Bf-blender-cvs mailing list
Bf-blender-cvs@blender.org
https://lists.blender.org/mailman/listinfo/bf-blender-cvs

Reply via email to