binding.gyp

##
# @license
# Copyright 2018 Google LLC. All Rights Reserved.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# =============================================================================

# Node.js TensorFlow Binding config:
{
  'variables' : {
    'tensorflow_include_dir' : '<(module_root_dir)/deps/include',
    'tensorflow_headers' : [
      '<@(tensorflow_include_dir)/tensorflow/c/c_api.h',
      '<@(tensorflow_include_dir)/tensorflow/c/eager/c_api.h',
    ],
    'tensorflow-library-action': 'move'
  },
  'targets' : [{
    'target_name' : 'tfjs_binding',
    'sources' : [
      'binding/tfjs_backend.cc',
      'binding/tfjs_binding.cc'
    ],
    'include_dirs' : [ '..', '<(tensorflow_include_dir)' ],
    'conditions' : [
      [
        'OS=="linux"', {
          'libraries' : [
            '-Wl,-rpath,\$$ORIGIN/../../deps/lib',
            '-ltensorflow',
            '-ltensorflow_framework',
          ],
          'library_dirs' : ['<(module_root_dir)/deps/lib'],
        }
      ],
      [
        'OS=="mac"', {
          'libraries' : [
            '<(module_root_dir)/deps/lib/libtensorflow.2.dylib',
            '<(module_root_dir)/deps/lib/libtensorflow_framework.2.dylib',
          ],
          'postbuilds': [
            {
              'postbuild_name': 'Adjust libtensorflow load path',
              'action': [
                'install_name_tool',
                "-change",
                "@rpath/libtensorflow.2.dylib",
                "@loader_path/../../deps/lib/libtensorflow.2.dylib",
                "<(PRODUCT_DIR)/tfjs_binding.node"
              ]
            },
            {
              'postbuild_name': 'Adjust libtensorflow_framework load path',
              'action': [
                'install_name_tool',
                "-change",
                "@rpath/libtensorflow_framework.2.dylib",
                "@loader_path/../../deps/lib/libtensorflow_framework.2.dylib",
                "<(PRODUCT_DIR)/tfjs_binding.node"
              ]
            }
          ],
        }
      ],
      [
        'OS=="win"', {
          'defines': ['COMPILER_MSVC'],
          'libraries': ['tensorflow'],
          'library_dirs' : ['<(module_root_dir)/deps/lib'],
          'variables': {
            'tensorflow-library-target': 'windows'
          },
          'msvs_disabled_warnings': [
            # Warning	C4190: 'TF_NewWhile' has C-linkage specified, but returns
            # UDT 'TF_WhileParams' which is incompatible with C.
            # (in include/tensorflow/c/c_api.h)
            4190
          ]
        },
      ]
    ],
  }
  , {
      "target_name": "action_after_build",
      "type": "none",
      "dependencies": [ "<(module_name)" ],
      "copies": [
        {
          "files": [ "<(PRODUCT_DIR)/<(module_name).node" ],
          "destination": "<(module_path)"
        }
      ]
    }
    ],
  "defines": [
      "NAPI_VERSION=<(napi_build_version)"
  ]
}