%module Visualization;

%{

#include <Visualization.h>

#include <Tesselator.h>

#include <Standard.hxx>

%}

%include ../SWIG_files/common/ExceptionCatcher.i

%include "python/std_string.i"

%include "std_vector.i"

%include "typemaps.i"

%template(vector_float) std::vector<float>;

%typemap(out) float [ANY] {

int i;

$result = PyList_New($1_dim0);

for (i = 0; i < $1_dim0; i++) {

PyObject *o = PyFloat_FromFloat((float) $1[i]);

PyList_SetItem($result,i,o);

}

}

enum theTextureMappingRule {

atCube,

atNormal,

atNormalAutoScale

};

%apply int& OUTPUT {int& v1, int& v2, int& v3}

%apply float& OUTPUT {float& x, float& y, float& z}

class Tesselator {

public:

%feature("autodoc", "1");

Tesselator(TopoDS_Shape aShape,

theTextureMappingRule aTxtMapType,

float anAutoScaleSizeOnU,

float anAutoScaleSizeOnV,

float aDeviation,

float aUOrigin,

float aVOrigin,

float aURepeat,

float aVRepeat,

float aScaleU,

float aScaleV,

float aRotationAngle);

%feature("autodoc", "1");

Tesselator(TopoDS_Shape aShape);

%feature("autodoc", "1");

~Tesselator();

%feature("kwargs") Compute;

void Compute(bool uv_coords=true, bool compute_edges=false, float mesh_quality=1.0, bool parallel=false);

void GetVertex(int ivert, float& x, float& y, float& z);

void GetNormal(int inorm, float& x, float& y, float& z);

void GetTriangleIndex(int triangleIdx, int& v1, int& v2, int& v3);

void GetEdgeVertex(int iEdge, int ivert, float& x, float& y, float& z);

float* VerticesList();

int ObjGetTriangleCount();

int ObjGetVertexCount();

int ObjGetNormalCount();

int ObjGetEdgeCount();

int ObjEdgeGetVertexCount(int iEdge);

std::string ExportShapeToX3DIndexedFaceSet();

std::string ExportShapeToThreejsJSONString(char *shape_function_name, bool export_uv=false);

%feature("kwargs") ExportShapeToX3D;

void ExportShapeToX3D(char *filename, int diffR=1, int diffG=0, int diffB=0);

std::vector<float> GetVerticesPositionAsTuple();

std::vector<float> GetNormalsAsTuple();

};

class Display3d {

public:

%feature("autodoc", "1");

Display3d();

%feature("autodoc", "1");

~Display3d();

%feature("autodoc", "1");

void Init(const long handle);

%feature("autodoc", "1");

void SetAnaglyphMode(int mode);

%feature("autodoc", "1");

void ChangeRenderingParams(int Method,

int RaytracingDepth,

bool IsShadowEnabled,

bool IsReflectionEnabled,

bool IsAntialiasingEnabled,

bool IsTransparentShadowEnabled,

int StereoMode,

int AnaglyphFilter,

bool ToReverseStere);

%feature("autodoc", "1");

void EnableVBO();

%feature("autodoc", "1");

void DisableVBO();

%feature("autodoc", "1");

Handle_V3d_View& GetView();

%feature("autodoc", "1");

Handle_V3d_Viewer& GetViewer();

%feature("autodoc", "1");

Handle_AIS_InteractiveContext GetContext();

%feature("autodoc", "1");

void Test();

%feature("autodoc", "1");

bool InitOffscreen(int size_x, int size_y);

%feature("autodoc", "1");

bool SetSize(int size_x, int size_y);

%feature("autodoc", "1");

bool IsOffscreen();

};

%extend Display3d {

PyObject* GetImageData(int bufType = 0) {

const char * data;

size_t size = 0;

Graphic3d_BufferType theBufferType = (Graphic3d_BufferType)bufType;

if ($self->GetImageData(data, size, theBufferType)) {

return PyBytes_FromStringAndSize(data, (Py_ssize_t)size);

}

Py_RETURN_NONE;

}

PyObject* GetSize() {

int size_x;

int size_y;

if ($self->GetSize(size_x, size_y)) {

return Py_BuildValue("ii", size_x, size_y);

}

Py_RETURN_NONE;

}

};