Documented the sorter

master
Michael Dvorkin 11 years ago
parent ebde4db119
commit 3c9f3534a6
  1. 48
      sorter.go

@ -1,16 +1,22 @@
// Copyright (c) 2013 by Michael Dvorkin. All Rights Reserved. // Copyright (c) 2013 by Michael Dvorkin. All Rights Reserved.
// Use of this source code is governed by a MIT-style // Use of this source code is governed by a MIT-style license that can
// license that can be found in the LICENSE file. // be found in the LICENSE file.
//-----------------------------------------------------------------------------
package mop package mop
import ( import (
`sort` `sort`
`strings`
`strconv` `strconv`
`strings`
) )
// Sorter gets called to sort stock quotes by one of the columns. The
// setup is rather lengthy; there should probably be more concise way
// that uses reflection and avoids hardcoding the column names.
type Sorter struct {
profile *Profile // Pointer to where we store sort column and order.
}
type sortable []Stock type sortable []Stock
func (list sortable) Len() int { return len(list) } func (list sortable) Len() int { return len(list) }
func (list sortable) Swap(i, j int) { list[i], list[j] = list[j], list[i] } func (list sortable) Swap(i, j int) { list[i], list[j] = list[j], list[i] }
@ -79,20 +85,19 @@ func (list byDividendDesc) Less(i, j int) bool { return list.sortable[j].Divi
func (list byYieldDesc) Less(i, j int) bool { return list.sortable[j].Yield < list.sortable[i].Yield } func (list byYieldDesc) Less(i, j int) bool { return list.sortable[j].Yield < list.sortable[i].Yield }
func (list byMarketCapDesc) Less(i, j int) bool { return m(list.sortable[j].MarketCap) < m(list.sortable[i].MarketCap) } func (list byMarketCapDesc) Less(i, j int) bool { return m(list.sortable[j].MarketCap) < m(list.sortable[i].MarketCap) }
type Sorter struct { // Initialize simply saves the pointer to Profile for later use.
profile *Profile func (sorter *Sorter) Initialize(profile *Profile) *Sorter {
} sorter.profile = profile
func (self *Sorter) Initialize(profile *Profile) *Sorter {
self.profile = profile
return self return sorter
} }
func (self *Sorter) SortByCurrentColumn(stocks []Stock) *Sorter { // SortByCurrentColumn builds a list of sort interface based on current sort
// order, then calls sort.Sort to do the actual job.
func (sorter *Sorter) SortByCurrentColumn(stocks []Stock) *Sorter {
var interfaces []sort.Interface var interfaces []sort.Interface
if self.profile.Ascending { if sorter.profile.Ascending {
interfaces = []sort.Interface{ interfaces = []sort.Interface{
byTickerAsc { stocks }, byTickerAsc { stocks },
byLastTradeAsc { stocks }, byLastTradeAsc { stocks },
@ -130,22 +135,25 @@ func (self *Sorter) SortByCurrentColumn(stocks []Stock) *Sorter {
} }
} }
sort.Sort(interfaces[self.profile.SortColumn]) sort.Sort(interfaces[sorter.profile.SortColumn])
return self return sorter
} }
// The same exact method is used to sort by Change and Change%. In both cases // The same exact method is used to sort by $Change and Change%. In both cases
// we sort by the value of Change% so that $0.00 change gets sorted proferly. // we sort by the value of Change% so that multiple $0.00s get sorted proferly.
func c(str string) float32 { func c(str string) float32 {
trimmed := strings.Replace(strings.Trim(str, ` %`), `$`, ``, 1) trimmed := strings.Replace(strings.Trim(str, ` %`), `$`, ``, 1)
value, _ := strconv.ParseFloat(trimmed, 32) value, _ := strconv.ParseFloat(trimmed, 32)
return float32(value) return float32(value)
} }
// When sorting by the market value we must first convert 42B etc. notations
// to proper numeric values.
func m(str string) float32 { func m(str string) float32 {
multiplier := 1.0 multiplier := 1.0
switch str[len(str)-1:len(str)] {
switch str[len(str)-1:len(str)] { // Check the last character.
case `B`: case `B`:
multiplier = 1000000000.0 multiplier = 1000000000.0
case `M`: case `M`:
@ -153,7 +161,9 @@ func m(str string) float32 {
case `K`: case `K`:
multiplier = 1000.0 multiplier = 1000.0
} }
trimmed := strings.Trim(str, ` $BMK`)
trimmed := strings.Trim(str, ` $BMK`) // Get rid of non-numeric characters.
value, _ := strconv.ParseFloat(trimmed, 32) value, _ := strconv.ParseFloat(trimmed, 32)
return float32(value * multiplier) return float32(value * multiplier)
} }

Loading…
Cancel
Save