source/blender/nodes/intern/type_callbacks.cc

   1 /*
   2  * This program is free software; you can redistribute it and/or
   3  * modify it under the terms of the GNU General Public License
   4  * as published by the Free Software Foundation; either version 2
   5  * of the License, or (at your option) any later version.
   6  *
   7  * This program is distributed in the hope that it will be useful,
   8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  * GNU General Public License for more details.
  11  *
  12  * You should have received a copy of the GNU General Public License
  13  * along with this program; if not, write to the Free Software Foundation,
  14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  15  */
  16
  17 #include "NOD_node_tree_multi_function.hh"
  18 #include "NOD_type_callbacks.hh"
  19
  20 namespace blender::nodes {
  21
  22 const CPPType *socket_cpp_type_get(const bNodeSocketType &stype)
  23 {
  24   if (stype.get_cpp_type != nullptr) {
  25     return stype.get_cpp_type();
  26   }
  27   return nullptr;
  28 }
  29
  30 std::optional<MFDataType> socket_mf_type_get(const bNodeSocketType &stype)
  31 {
  32   const CPPType *cpp_type = socket_cpp_type_get(stype);
  33   if (cpp_type != nullptr) {
  34     return MFDataType::ForSingle(*cpp_type);
  35   }
  36   return {};
  37 }
  38
  39 bool socket_is_mf_data_socket(const bNodeSocketType &stype)
  40 {
  41   if (!socket_mf_type_get(stype).has_value()) {
  42     return false;
  43   }
  44   if (stype.expand_in_mf_network == nullptr && stype.get_cpp_value == nullptr) {
  45     return false;
  46   }
  47   return true;
  48 }
  49
  50 bool socket_cpp_value_get(const bNodeSocket &socket, void *r_value)
  51 {
  52   if (socket.typeinfo->get_cpp_value != nullptr) {
  53     socket.typeinfo->get_cpp_value(socket, r_value);
  54     return true;
  55   }
  56   return false;
  57 }
  58
  59 void socket_expand_in_mf_network(SocketMFNetworkBuilder &builder)
  60 {
  61   bNodeSocket &socket = builder.bsocket();
  62   if (socket.typeinfo->expand_in_mf_network != nullptr) {
  63     socket.typeinfo->expand_in_mf_network(builder);
  64   }
  65   else if (socket.typeinfo->get_cpp_value != nullptr) {
  66     const CPPType &type = *socket_cpp_type_get(*socket.typeinfo);
  67     void *buffer = builder.resources().linear_allocator().allocate(type.size(), type.alignment());
  68     socket.typeinfo->get_cpp_value(socket, buffer);
  69     builder.set_constant_value(type, buffer);
  70   }
  71   else {
  72     BLI_assert(false);
  73   }
  74 }
  75
  76 }  // namespace blender::nodes