/* 1:1 function mappings for natmath.h in libparted */
PyObject *py_ped_alignment_duplicate(PyObject *s, PyObject *args) {
PedAlignment *alignment = NULL, *align = NULL;
_ped_Alignment *ret = NULL;
alignment = _ped_Alignment2PedAlignment(s);
if (alignment == NULL) {
return NULL;
}
align = ped_alignment_duplicate(alignment);
ped_alignment_destroy(alignment);
if (align) {
ret = PedAlignment2_ped_Alignment(align);
}
else {
PyErr_SetString(CreateException, "Could not duplicate alignment");
return NULL;
}
ped_alignment_destroy(align);
return (PyObject *) ret;
}
/**
* Release the memory allocated for a PedConstraint constructed with
* ped_constraint_init().
*/
void
ped_constraint_done (PedConstraint* constraint)
{
PED_ASSERT (constraint != NULL);
ped_alignment_destroy (constraint->start_align);
ped_alignment_destroy (constraint->end_align);
ped_geometry_destroy (constraint->start_range);
ped_geometry_destroy (constraint->end_range);
}
/**
* Return a constraint that requires a region to satisfy both \p a and \p b.
*
* Moreover, any region satisfying \p a and \p b will also satisfy the returned
* constraint.
*
* \return \c NULL if no solution could be found (note that \c NULL is a valid
* PedConstraint).
*/
PedConstraint*
ped_constraint_intersect (const PedConstraint* a, const PedConstraint* b)
{
PedAlignment* start_align;
PedAlignment* end_align;
PedGeometry* start_range;
PedGeometry* end_range;
PedSector min_size;
PedSector max_size;
PedConstraint* constraint;
if (!a || !b)
return NULL;
start_align = ped_alignment_intersect (a->start_align, b->start_align);
if (!start_align)
goto empty;
end_align = ped_alignment_intersect (a->end_align, b->end_align);
if (!end_align)
goto empty_destroy_start_align;
start_range = ped_geometry_intersect (a->start_range, b->start_range);
if (!start_range)
goto empty_destroy_end_align;
end_range = ped_geometry_intersect (a->end_range, b->end_range);
if (!end_range)
goto empty_destroy_start_range;
min_size = PED_MAX (a->min_size, b->min_size);
max_size = PED_MIN (a->max_size, b->max_size);
constraint = ped_constraint_new (
start_align, end_align, start_range, end_range,
min_size, max_size);
if (!constraint)
goto empty_destroy_end_range;
ped_alignment_destroy (start_align);
ped_alignment_destroy (end_align);
ped_geometry_destroy (start_range);
ped_geometry_destroy (end_range);
return constraint;
empty_destroy_end_range:
ped_geometry_destroy (end_range);
empty_destroy_start_range:
ped_geometry_destroy (start_range);
empty_destroy_end_align:
ped_alignment_destroy (end_align);
empty_destroy_start_align:
ped_alignment_destroy (start_align);
empty:
return NULL;
}
PyObject *py_ped_alignment_is_aligned(PyObject *s, PyObject *args) {
int ret = -1;
PyObject *in_geom = NULL;
PedAlignment *align = NULL;
PedGeometry *out_geom = NULL;
PedSector sector;
if (!PyArg_ParseTuple(args, "O!L", &_ped_Geometry_Type_obj, &in_geom, §or)) {
return NULL;
}
align = _ped_Alignment2PedAlignment(s);
if (align == NULL) {
return NULL;
}
out_geom = _ped_Geometry2PedGeometry(in_geom);
if (out_geom == NULL) {
return NULL;
}
ret = ped_alignment_is_aligned(align, out_geom, sector);
ped_alignment_destroy(align);
if (ret) {
Py_RETURN_TRUE;
} else {
Py_RETURN_FALSE;
}
}
PyObject *py_ped_alignment_align_nearest(PyObject *s, PyObject *args) {
PyObject *in_geom = NULL;
PedAlignment *align = NULL;
PedGeometry *out_geom = NULL;
PedSector sector, ret;
if (!PyArg_ParseTuple(args, "O!L", &_ped_Geometry_Type_obj, &in_geom, §or)) {
return NULL;
}
align = _ped_Alignment2PedAlignment(s);
if (align == NULL) {
return NULL;
}
out_geom = _ped_Geometry2PedGeometry(in_geom);
if (out_geom == NULL) {
return NULL;
}
ret = ped_alignment_align_nearest(align, out_geom, sector);
ped_alignment_destroy(align);
if (ret == -1) {
PyErr_SetString(PyExc_ArithmeticError, "Could not align to closest sector");
return NULL;
}
return PyLong_FromLong(ret);
}
PyObject *py_ped_alignment_intersect(PyObject *s, PyObject *args) {
PyObject *in_b = NULL;
PedAlignment *out_a = NULL, *out_b = NULL, *align = NULL;
_ped_Alignment *ret = NULL;
if (!PyArg_ParseTuple(args, "O!", &_ped_Alignment_Type_obj, &in_b)) {
return NULL;
}
out_a = _ped_Alignment2PedAlignment(s);
if (out_a == NULL) {
return NULL;
}
out_b = _ped_Alignment2PedAlignment(in_b);
if (out_b == NULL) {
return NULL;
}
align = ped_alignment_intersect(out_a, out_b);
ped_alignment_destroy(out_a);
ped_alignment_destroy(out_b);
if (align) {
ret = PedAlignment2_ped_Alignment(align);
}
else {
PyErr_SetString(PyExc_ArithmeticError, "Could not find alignment intersection");
return NULL;
}
ped_alignment_destroy(align);
return (PyObject *) ret;
}
/* _ped_Constraint -> PedConstraint functions */
PedConstraint *_ped_Constraint2PedConstraint(PyObject *s) {
PedConstraint *ret = NULL;
PedAlignment *start_align = NULL, *end_align = NULL;
PedGeometry *start_range = NULL, *end_range = NULL;
_ped_Constraint *constraint = (_ped_Constraint *) s;
if (constraint == NULL) {
PyErr_SetString(PyExc_TypeError, "Empty _ped.Constraint()");
return NULL;
}
start_align = _ped_Alignment2PedAlignment(constraint->start_align);
if (start_align == NULL) {
return NULL;
}
end_align = _ped_Alignment2PedAlignment(constraint->end_align);
if (end_align == NULL) {
ped_alignment_destroy(start_align);
return NULL;
}
start_range = _ped_Geometry2PedGeometry(constraint->start_range);
if (start_range == NULL) {
ped_alignment_destroy(start_align);
ped_alignment_destroy(end_align);
return NULL;
}
end_range = _ped_Geometry2PedGeometry(constraint->end_range);
if (end_range == NULL) {
ped_alignment_destroy(start_align);
ped_alignment_destroy(end_align);
return NULL;
}
ret = ped_constraint_new(start_align, end_align, start_range, end_range,
constraint->min_size, constraint->max_size);
if (ret == NULL) {
/* Fall through to clean up memory, but set the error condition now. */
PyErr_NoMemory();
}
ped_alignment_destroy(start_align);
ped_alignment_destroy(end_align);
return ret;
}
int _ped_Alignment_init(_ped_Alignment *self, PyObject *args, PyObject *kwds) {
static char *kwlist[] = {"offset", "grain_size", NULL};
PedAlignment *alignment = NULL;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "LL", kwlist,
&self->offset, &self->grain_size)) {
return -1;
} else {
alignment = ped_alignment_new(self->offset, self->grain_size);
if (!alignment) {
PyErr_SetString(CreateException, "Could not create new alignment");
return -1;
}
self->offset = alignment->offset;
self->grain_size = alignment->grain_size;
ped_alignment_destroy(alignment);
return 0;
}
}
PyObject *py_ped_device_get_optimum_alignment(PyObject *s, PyObject *args) {
PedDevice *device = NULL;
PedAlignment *alignment = NULL;
_ped_Alignment *ret = NULL;
device = _ped_Device2PedDevice(s);
if (device == NULL) {
return NULL;
}
alignment = ped_device_get_optimum_alignment(device);
if (!alignment) {
PyErr_SetString(CreateException, "Could not get alignment for device");
return NULL;
}
ret = PedAlignment2_ped_Alignment(alignment);
ped_alignment_destroy(alignment);
return (PyObject *) ret;
}
int _ped_Constraint_init(_ped_Constraint *self, PyObject *args,
PyObject *kwds) {
static char *kwlist[] = {"start_align", "end_align", "start_range",
"end_range", "min_size", "max_size", NULL};
PedConstraint *constraint = NULL;
PedAlignment *start_align = NULL, *end_align = NULL;
PedGeometry *start_range = NULL, *end_range = NULL;
if (kwds == NULL) {
if (!PyArg_ParseTuple(args, "O!O!O!O!LL",
&_ped_Alignment_Type_obj, &self->start_align,
&_ped_Alignment_Type_obj, &self->end_align,
&_ped_Geometry_Type_obj, &self->start_range,
&_ped_Geometry_Type_obj, &self->end_range,
&self->min_size, &self->max_size)) {
self->start_align = self->end_align = NULL;
self->start_range = self->end_range = NULL;
return -1;
}
} else {
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O!O!O!O!LL", kwlist,
&_ped_Alignment_Type_obj,
&self->start_align,
&_ped_Alignment_Type_obj,
&self->end_align,
&_ped_Geometry_Type_obj,
&self->start_range,
&_ped_Geometry_Type_obj,
&self->end_range,
&self->min_size,
&self->max_size)) {
self->start_align = self->end_align = NULL;
self->start_range = self->end_range = NULL;
return -2;
}
}
/*
* try to call libparted with provided information,
* on failure, raise an exception
*/
start_align = _ped_Alignment2PedAlignment(self->start_align);
end_align = _ped_Alignment2PedAlignment(self->end_align);
start_range = _ped_Geometry2PedGeometry(self->start_range);
end_range = _ped_Geometry2PedGeometry(self->end_range);
constraint = ped_constraint_new(start_align, end_align,
start_range, end_range,
self->min_size, self->max_size);
if (constraint == NULL) {
PyErr_SetString(CreateException, "Could not create new constraint");
ped_alignment_destroy(start_align);
ped_alignment_destroy(end_align);
self->start_align = NULL;
self->end_align = NULL;
self->start_range = NULL;
self->end_range = NULL;
return -3;
}
/* increment reference count for PyObjects read by PyArg_ParseTuple */
Py_INCREF(self->start_align);
Py_INCREF(self->end_align);
Py_INCREF(self->start_range);
Py_INCREF(self->end_range);
/* clean up libparted objects we created */
ped_alignment_destroy(start_align);
ped_alignment_destroy(end_align);
ped_constraint_destroy(constraint);
return 0;
}
