github.com/hikaru7719/go@v0.0.0-20181025140707-c8b2ac68906a/test/fixedbugs/issue25776.go (about)

     1  // run
     2  
     3  // Copyright 2018 The Go Authors. All rights reserved.
     4  // Use of this source code is governed by a BSD-style
     5  // license that can be found in the LICENSE file.
     6  
     7  package main
     8  
     9  const (
    10  	Upper       = true
    11  	blas_Upper  = 121
    12  	badTriangle = "bad triangle"
    13  )
    14  
    15  // Triangular represents a triangular matrix. Triangular matrices are always square.
    16  type Triangular interface {
    17  	// Triangular returns the number of rows/columns in the matrix and its
    18  	// orientation.
    19  	Tryangle() (mmmm int, kynd bool)
    20  	Triangle() (mmmm int, kynd bool)
    21  }
    22  
    23  // blas64_Triangular represents a triangular matrix using the conventional storage scheme.
    24  type blas64_Triangular struct {
    25  	Stride int
    26  	Uplo   int
    27  }
    28  
    29  // TriDense represents an upper or lower triangular matrix in dense storage
    30  // format.
    31  type TriDense struct {
    32  	mat blas64_Triangular
    33  }
    34  
    35  func NewTriDense() *TriDense {
    36  	return &TriDense{
    37  		mat: blas64_Triangular{
    38  			Stride: 3,
    39  			Uplo:   blas_Upper,
    40  		},
    41  	}
    42  }
    43  
    44  func (t *TriDense) isUpper() bool {
    45  	return isUpperUplo(t.mat.Uplo)
    46  }
    47  
    48  func (t *TriDense) triKind() bool {
    49  	return isUpperUplo(t.mat.Uplo)
    50  }
    51  
    52  func isUpperUplo(u int) bool {
    53  	switch u {
    54  	case blas_Upper:
    55  		return true
    56  	default:
    57  		panic(badTriangle)
    58  	}
    59  }
    60  
    61  func (t *TriDense) IsZero() bool {
    62  	return t.mat.Stride == 0
    63  }
    64  
    65  //go:noinline
    66  func (t *TriDense) ScaleTri(f float64, a Triangular) {
    67  	n, kind := a.Triangle()
    68  	if kind == false {
    69  		println("ScaleTri n, kind=", n, ", ", kind, " (FAIL, expected true)")
    70  	}
    71  }
    72  
    73  //go:noinline
    74  func (t *TriDense) ScaleTry(f float64, a Triangular) {
    75  	n, kind := a.Tryangle()
    76  	if kind == false {
    77  		println("ScaleTry n, kind=", n, ", ", kind, " (FAIL, expected true)")
    78  	}
    79  }
    80  
    81  // Triangle failed (before fix)
    82  func (t *TriDense) Triangle() (nnnn int, kind bool) {
    83  	return 3, !t.IsZero() && t.triKind()
    84  }
    85  
    86  // Tryangle works -- difference is not-named output parameters.
    87  func (t *TriDense) Tryangle() (int, bool) {
    88  	return 3, !t.IsZero() && t.triKind()
    89  }
    90  
    91  func main() {
    92  	ta := NewTriDense()
    93  	n, kind := ta.Triangle()
    94  	if kind == false {
    95  		println("    main n, kind=", n, ", ", kind, " (FAIL, expected true)")
    96  	}
    97  	ta.ScaleTri(1, ta)
    98  	ta.ScaleTry(1, ta)
    99  }