python_symmetry.h File Reference

#include "Python.h"

Include dependency graph for python_symmetry.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	SymListObject

Functions
PyObject *	SymList_new (PyTypeObject *type, PyObject *args, PyObject *kwargs)
void	SymList_dealloc (SymListObject *self)
PyObject *	SymList_readSymmetryFile (PyObject *obj, PyObject *args, PyObject *kwargs)
PyObject *	SymList_computeDistance (PyObject *obj, PyObject *args, PyObject *kwargs)
PyObject *	SymList_getSymmetryMatrices (PyObject *obj, PyObject *args, PyObject *kwargs)

Variables
PyObject *	PyXmippError
PyMethodDef	SymList_methods []
PyTypeObject	SymListType

Function Documentation

SymList_computeDistance()

PyObject* SymList_computeDistance	(	PyObject *	obj,
		PyObject *	args,
		PyObject *	kwargs
	)

Definition at line 140 of file python_symmetry.cpp.

{

    PyObject *pyMd = nullptr;
    auto *pyProjdirMode = Py_False;
    auto *pyCheckMirrors = Py_False;
    auto *pyObjectRotation = Py_False;

    if (PyArg_ParseTuple(args, "O|OOO", &pyMd,
                         &pyProjdirMode,
                         &pyCheckMirrors,
                         &pyObjectRotation))
    {
        if (!MetaData_Check(pyMd))
            PyErr_SetString(PyExc_TypeError,
                            "Expected MetaData as first argument");
        try
        {
            bool projdir_mode    = false;
            bool check_mirrors   = false;
            bool object_rotation = false;
            if (PyBool_Check(pyProjdirMode))
                projdir_mode = (pyProjdirMode == Py_True);
            if (PyBool_Check(pyCheckMirrors))
                check_mirrors = (pyCheckMirrors == Py_True);
            if (PyBool_Check(pyObjectRotation))
                object_rotation = (pyObjectRotation == Py_True);
            auto *self = (SymListObject*) obj;
            self->symlist->computeDistance(MetaData_Value(pyMd),projdir_mode,
                                                                check_mirrors,
                                                                object_rotation);
            Py_RETURN_NONE;
        }
        catch (XmippError &xe)
        {
            PyErr_SetString(PyXmippError, xe.what());
        }
    }
    return nullptr;
}

SymList_dealloc()

void SymList_dealloc

(

SymListObject *

self

)

Definition at line 49 of file python_symmetry.cpp.

{
    delete self->symlist;
    Py_TYPE(self)->tp_free(self);
}

SymList_getSymmetryMatrices()

PyObject* SymList_getSymmetryMatrices	(	PyObject *	obj,
		PyObject *	args,
		PyObject *	kwargs
	)

Definition at line 87 of file python_symmetry.cpp.

{
    Matrix2D<double> L(4, 4);
    Matrix2D<double> R(4, 4);
    PyObject * symMatrices;
    PyObject * symMatrix;
    PyObject * row;
    char * str = nullptr;

    if (PyArg_ParseTuple(args, "s", &str))
    {
        try
        {
            auto *self = (SymListObject*) obj;
            
            //create symmetry object
            self->symlist->readSymmetryFile(str);
            symMatrices = PyList_New(self->symlist->symsNo()+1);
            symMatrix   = PyList_New(3);
            
            //add identity matrix to results
            row = Py_BuildValue("[fff]", 1., 0., 0.);
            PyList_SetItem(symMatrix,0,row);
            row = Py_BuildValue("[fff]", 0., 1., 0.);
            PyList_SetItem(symMatrix,1,row);
            row = Py_BuildValue("[fff]", 0., 0., 1.);
            PyList_SetItem(symMatrix,2,row);
            PyList_SetItem(symMatrices,0,symMatrix);
            
            //copy each symmetry matrix to a python list
            for (int i=0; i < self->symlist->true_symNo; ++i)
            {
                symMatrix   = PyList_New(3);
                self->symlist->getMatrices(i, L, R);
                for (int j=0; j < 3; ++j)
                    {
                    row = Py_BuildValue("[fff]", R(j,0), R(j,1), R(j,2));
                    PyList_SetItem(symMatrix,j,row);
                    }
                PyList_SetItem(symMatrices,i+1,symMatrix);
            }
            return symMatrices;
        }
        catch (XmippError &xe)
        {
            PyErr_SetString(PyXmippError, xe.what());
        }
    }
    return nullptr;
}

SymList_new()

PyObject* SymList_new	(	PyTypeObject *	type,
		PyObject *	args,
		PyObject *	kwargs
	)

Definition at line 37 of file python_symmetry.cpp.

{
    auto *self = (SymListObject*)type->tp_alloc(type, 0);
    if (self != nullptr)
    {
        self->symlist = new SymList();
        //self->symlist->readSymmetryFile("i3");
    }
    return (PyObject *)self;
}

SymList_readSymmetryFile()

PyObject* SymList_readSymmetryFile	(	PyObject *	obj,
		PyObject *	args,
		PyObject *	kwargs
	)

Definition at line 57 of file python_symmetry.cpp.

{
    char * str = nullptr;

    if (PyArg_ParseTuple(args, "s", &str))
    {
        try
        {
            auto *self = (SymListObject*) obj;
            self->symlist->readSymmetryFile(str);
            Py_RETURN_NONE;
        }
        catch (XmippError &xe)
        {
            PyErr_SetString(PyXmippError, xe.what());
        }
    }
    return nullptr;
}

Variable Documentation

PyXmippError

PyObject* PyXmippError

Definition at line 47 of file xmippmodule.cpp.

SymList_methods

PyMethodDef SymList_methods[]

Definition at line 277 of file python_symmetry.cpp.

SymListType

PyTypeObject SymListType

Definition at line 295 of file python_symmetry.cpp.