python-flint/src/flint/types/qqbar.pyx at edb2242ebff90aa823ef62dfa12c83753d35c31e · flintlib/python-flint · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
from flint.flintlib.functions.qqbar cimport (
    qqbar_init,
    qqbar_clear,
    qqbar_set,
    qqbar_set_si,
    qqbar_printn,
    qqbar_set_fmpz,
    qqbar_set_fmpq,
    qqbar_add,
    qqbar_sub,
    qqbar_mul,
    qqbar_div
)
from flint.types.fmpz cimport fmpz
from flint.types.fmpq cimport fmpq

cdef class qqbar(flint_scalar):
    """
    The qqbar class represents algebraic numbers.
    """

    def __cinit__(self):
        qqbar_init(self.val)

    def __dealloc__(self):
        qqbar_clear(self.val)

    def __init__(self, val=None):
        if val is not None:
            if isinstance(val, qqbar):
                qqbar_set(self.val, (<qqbar>val).val)
            elif isinstance(val, int):
                qqbar_set_si(self.val, val)
            elif isinstance(val, fmpz):
                qqbar_set_fmpz(self.val, (<fmpz>val).val)
            elif isinstance(val, fmpq):
                qqbar_set_fmpq(self.val, (<fmpq>val).val)
            else:
                raise TypeError("Cannot initialize qqbar type from input")

    def str(self, Py_ssize_t digits=15):
        """
        Converts the algebraic number to a decimal string representation by capturing C-level stdout.
        """
        import os
        import sys

        cdef int stdout_fd = 1
        cdef int saved_stdout = os.dup(stdout_fd)

        pipe_read, pipe_write = os.pipe()
        os.dup2(pipe_write, stdout_fd)

        try:
            qqbar_printn(self.val, digits)
            sys.stdout.flush()
        finally:
            os.close(pipe_write)
            os.dup2(saved_stdout, stdout_fd)
            os.close(saved_stdout)

        cdef bytes captured = os.read(pipe_read, 4096)
        os.close(pipe_read)

        return captured.decode('utf-8').strip()

    def __str__(self):
        return self.str()

    def __repr__(self):
        return f"qqbar({self.str()})"

    def __add__(x, y):
        cdef qqbar res = qqbar()
        cdef qqbar c_x = x if isinstance(x, qqbar) else qqbar(x)
        cdef qqbar c_y = y if isinstance(y, qqbar) else qqbar(y)
        qqbar_add(res.val, c_x.val, c_y.val)
        return res

    def __sub__(x, y):
        cdef qqbar res = qqbar()
        cdef qqbar c_x = x if isinstance(x, qqbar) else qqbar(x)
        cdef qqbar c_y = y if isinstance(y, qqbar) else qqbar(y)
        qqbar_sub(res.val, c_x.val, c_y.val)
        return res

    def __mul__(x, y):
        cdef qqbar res = qqbar()
        cdef qqbar c_x = x if isinstance(x, qqbar) else qqbar(x)
        cdef qqbar c_y = y if isinstance(y, qqbar) else qqbar(y)
        qqbar_mul(res.val, c_x.val, c_y.val)
        return res

    def __truediv__(x, y):
        cdef qqbar res = qqbar()
        cdef qqbar c_x = x if isinstance(x, qqbar) else qqbar(x)
        cdef qqbar c_y = y if isinstance(y, qqbar) else qqbar(y)
        qqbar_div(res.val, c_x.val, c_y.val)
        return res