forked from google/grumpy
/
complex.go
72 lines (62 loc) · 1.93 KB
/
complex.go
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
// Copyright 2016 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package grumpy
import (
"fmt"
"reflect"
"strconv"
)
// ComplexType is the object representing the Python 'complex' type.
var ComplexType = newBasisType("complex", reflect.TypeOf(Complex{}), toComplexUnsafe, ObjectType)
// Complex represents Python 'complex' objects.
type Complex struct {
Object
value complex128
}
// NewComplex returns a new Complex holding the given complex value.
func NewComplex(value complex128) *Complex {
return &Complex{Object{typ: ComplexType}, value}
}
func toComplexUnsafe(o *Object) *Complex {
return (*Complex)(o.toPointer())
}
// ToObject upcasts c to an Object.
func (c *Complex) ToObject() *Object {
return &c.Object
}
// Value returns the underlying complex value held by c.
func (c *Complex) Value() complex128 {
return c.value
}
func complexRepr(f *Frame, o *Object) (*Object, *BaseException) {
c := toComplexUnsafe(o).Value()
rs, is := "", ""
pre, post := "", ""
sign := ""
if real(c) == 0.0 {
is = strconv.FormatFloat(imag(c), 'g', -1, 64)
} else {
pre = "("
rs = strconv.FormatFloat(real(c), 'g', -1, 64)
is = strconv.FormatFloat(imag(c), 'g', -1, 64)
if imag(c) >= 0.0 {
sign = "+"
}
post = ")"
}
return NewStr(fmt.Sprintf("%s%s%s%sj%s", pre, rs, sign, is, post)).ToObject(), nil
}
func initComplexType(dict map[string]*Object) {
ComplexType.slots.Repr = &unaryOpSlot{complexRepr}
}