#include "TH.h"
#include "luaT.h"
#include "omp.h"

#define torch_(NAME) TH_CONCAT_3(torch_, Real, NAME)
#define torch_string_(NAME) TH_CONCAT_STRING_3(torch., Real, NAME)
#define nn_(NAME) TH_CONCAT_3(nn_, Real, NAME)

static const void* torch_FloatTensor_id = NULL;
static const void* torch_DoubleTensor_id = NULL;

#include "generic/Abs.c"
#include "THGenerateFloatTypes.h"

#include "generic/HardShrink.c"
#include "THGenerateFloatTypes.h"

#include "generic/SpatialLinear.c"
#include "THGenerateFloatTypes.h"

#include "generic/SpatialUpSampling.c"
#include "THGenerateFloatTypes.h"

#include "generic/SpatialReSampling.c"
#include "THGenerateFloatTypes.h"

#include "generic/SparseCriterion.c"
#include "THGenerateFloatTypes.h"

#include "generic/SpatialSparseCriterion.c"
#include "THGenerateFloatTypes.h"

#include "generic/SpatialMSECriterion.c"
#include "THGenerateFloatTypes.h"

#include "generic/SpatialClassNLLCriterion.c"
#include "THGenerateFloatTypes.h"

#include "generic/Threshold.c"
#include "THGenerateFloatTypes.h"

#include "generic/SpatialGraph.c"
#include "THGenerateFloatTypes.h"

#include "generic/DataSetLabelMe.c"
#include "THGenerateFloatTypes.h"

DLL_EXPORT int luaopen_libnnx(lua_State *L)
{
  torch_FloatTensor_id = luaT_checktypename2id(L, "torch.FloatTensor");
  torch_DoubleTensor_id = luaT_checktypename2id(L, "torch.DoubleTensor");

  nn_FloatSpatialLinear_init(L);
  nn_FloatHardShrink_init(L);
  nn_FloatAbs_init(L);
  nn_FloatThreshold_init(L);
  nn_FloatSpatialUpSampling_init(L);
  nn_FloatSpatialReSampling_init(L);
  nn_FloatSparseCriterion_init(L);
  nn_FloatSpatialSparseCriterion_init(L);
  nn_FloatSpatialMSECriterion_init(L);
  nn_FloatSpatialClassNLLCriterion_init(L);
  nn_FloatSpatialGraph_init(L);
  nn_FloatDataSetLabelMe_init(L);

  nn_DoubleSpatialLinear_init(L);
  nn_DoubleHardShrink_init(L);
  nn_DoubleAbs_init(L);
  nn_DoubleThreshold_init(L);
  nn_DoubleSpatialUpSampling_init(L);
  nn_DoubleSpatialReSampling_init(L);
  nn_DoubleSparseCriterion_init(L);
  nn_DoubleSpatialSparseCriterion_init(L);
  nn_DoubleSpatialMSECriterion_init(L);
  nn_DoubleSpatialClassNLLCriterion_init(L);
  nn_DoubleSpatialGraph_init(L);
  nn_DoubleDataSetLabelMe_init(L);

  return 1;
}