Better unsafe funcs for UnivPoly and a few others (#1838)

src/Matrix.jl

@@ -239,14 +239,6 @@ zero(x::MatrixElem{T}, R::NCRing=base_ring(x)) where T <: NCRingElement = zero(x
 zero(x::MatrixElem{T}, R::NCRing, r::Int, c::Int) where T <: NCRingElement = zero!(similar(x, R, r, c))
 zero(x::MatrixElem{T}, r::Int, c::Int) where T <: NCRingElement = zero(x, base_ring(x), r, c)
 
-function zero!(x::MatrixElem{T}) where T <: NCRingElement
-   R = base_ring(x)
-   for i = 1:nrows(x), j = 1:ncols(x)
-      x[i, j] = zero(R)
-   end
-   x
-end
-
 @doc raw"""
     one(a::MatSpace)
 
@@ -867,6 +859,20 @@ function *(x::MatElem{T}, y::MatElem{T}) where {T <: NCRingElement}
    return A
 end
 
+###############################################################################
+#
+#   Unsafe functions
+#
+###############################################################################
+
+function zero!(x::MatrixElem{T}) where T <: NCRingElement
+   R = base_ring(x)
+   for i = 1:nrows(x), j = 1:ncols(x)
+      x[i, j] = zero(R)
+   end
+   x
+end
+
 function add!(c::MatrixElem{T}, a::MatrixElem{T}, b::MatrixElem{T}) where T <: NCRingElement
    check_parent(a, b)
    check_parent(a, c)

src/generic/AbsMSeries.jl

@@ -615,7 +615,7 @@ end
 
 ###############################################################################
 #
-#   Unsafe operators
+#   Unsafe functions
 #
 ###############################################################################
 

src/generic/FunctionField.jl

@@ -1103,7 +1103,7 @@ end
 
 ###############################################################################
 #
-#   Unsafe operators
+#   Unsafe functions
 #
 ###############################################################################
 

src/generic/MatRing.jl

@@ -90,7 +90,7 @@ end
 
 ###############################################################################
 #
-#   Unsafe operators
+#   Unsafe functions
 #
 ###############################################################################
 

src/generic/PuiseuxSeries.jl

@@ -732,7 +732,7 @@ rand(S::PuiseuxSeriesRingOrField, val_range, scale_range, v...) =
 
 ###############################################################################
 #
-#   Unsafe operations
+#   Unsafe functions
 #
 ###############################################################################
 

src/generic/RationalFunctionField.jl

@@ -468,7 +468,7 @@ end
 
 ###############################################################################
 #
-#   Unsafe operators and functions
+#   Unsafe functions
 #
 ###############################################################################
 

src/generic/TotalFraction.jl

@@ -391,18 +391,27 @@ end
 
 ###############################################################################
 #
-#   Unsafe operators and functions
+#   Unsafe functions
 #
 ###############################################################################
 
 function zero!(c::TotFrac)
    c.num = zero!(c.num)
-   if !isone(c.den)
-      c.den = one(base_ring(c))
-   end
+   c.den = one!(c.den)
+   return c
+end
+
+function one!(c::TotFrac)
+   c.num = one!(c.num)
+   c.den = one!(c.den)
    return c
 end
 
+function neg!(z::TotFrac, c::TotFrac)
+   z.num = neg!(z.num, c.num)
+   return z
+end
+
 function mul!(c::TotFrac{T}, a::TotFrac{T}, b::TotFrac{T}) where {T <: RingElem}
    d1 = denominator(a, false)
    d2 = denominator(b, false)

src/generic/UnivPoly.jl

@@ -1004,6 +1004,21 @@ end
 #
 ###############################################################################
 
+function zero!(a::UnivPoly{T}) where {T <: RingElement}
+   a.p = zero!(a.p)
+   return a
+end
+
+function one!(a::UnivPoly{T}) where {T <: RingElement}
+   a.p = one!(a.p)
+   return a
+end
+
+function neg!(z::UnivPoly{T}, a::UnivPoly{T}) where {T <: RingElement}
+   z.p = neg!(a.p)
+   return z
+end
+
 function fit!(a::UnivPoly, n::Int)
    fit!(data(a), n)
 end
@@ -1013,21 +1028,40 @@ function add!(a::UnivPoly{T}, b::UnivPoly{T}, c::UnivPoly{T}) where {T <: RingEl
    return a
 end
 
+function add!(a::UnivPoly{T}, b::UnivPoly{T}, c::RingElement) where {T <: RingElement}
+   a.p = add!(a.p, b.p, c)
+   return a
+end
+
+add!(a::UnivPoly{T}, b::RingElement, c::UnivPoly{T}) where {T <: RingElement} = add!(a, c, b)
+
 function sub!(a::UnivPoly{T}, b::UnivPoly{T}, c::UnivPoly{T}) where {T <: RingElement}
    a.p = sub!(a.p, b.p, c.p)
    return a
 end
 
+function sub!(a::UnivPoly{T}, b::UnivPoly{T}, c::RingElement) where {T <: RingElement}
+   a.p = sub!(a.p, b.p, c)
+   return a
+end
+
+function sub!(a::UnivPoly{T}, b::RingElement, c::UnivPoly{T}) where {T <: RingElement}
+   a.p = sub!(a.p, b, c.p)
+   return a
+end
+
 function mul!(a::UnivPoly{T}, b::UnivPoly{T}, c::UnivPoly{T}) where {T <: RingElement}
    a.p = mul!(a.p, b.p, c.p)
    return a
 end
 
-function zero!(a::UnivPoly{T}) where {T <: RingElement}
-   a.p = zero!(a.p)
+function mul!(a::UnivPoly{T}, b::UnivPoly{T}, c::RingElement) where {T <: RingElement}
+   a.p = mul!(a.p, b.p, c)
    return a
 end
 
+mul!(a::UnivPoly{T}, b::RingElement, c::UnivPoly{T}) where {T <: RingElement} = mul!(a, c, b)
+
 ###############################################################################
 #
 #   Promotion rules

src/generic/imports.jl

@@ -178,6 +178,7 @@ import ..AbstractAlgebra: number_of_generators
 import ..AbstractAlgebra: number_of_rows
 import ..AbstractAlgebra: number_of_variables
 import ..AbstractAlgebra: numerator
+import ..AbstractAlgebra: one!
 import ..AbstractAlgebra: order
 import ..AbstractAlgebra: parent_type
 import ..AbstractAlgebra: pol_length
@@ -198,6 +199,8 @@ import ..AbstractAlgebra: shift_left
 import ..AbstractAlgebra: shift_right
 import ..AbstractAlgebra: snf
 import ..AbstractAlgebra: sqrt
+import ..AbstractAlgebra: sub!
+import ..AbstractAlgebra: submul!
 import ..AbstractAlgebra: symbols
 import ..AbstractAlgebra: tail
 import ..AbstractAlgebra: term_degree