-
Notifications
You must be signed in to change notification settings - Fork 47
/
PosProvider.kt
132 lines (109 loc) · 4.96 KB
/
PosProvider.kt
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
/*
* Copyright (c) 2019. JetBrains s.r.o.
* Use of this source code is governed by the MIT license that can be found in the LICENSE file.
*/
package org.jetbrains.letsPlot.core.plot.builder.assemble
import org.jetbrains.letsPlot.core.plot.base.PositionAdjustment
import org.jetbrains.letsPlot.core.plot.base.pos.PositionAdjustments
import org.jetbrains.letsPlot.core.plot.base.pos.StackablePos
import org.jetbrains.letsPlot.core.plot.base.pos.StackingMode
import org.jetbrains.letsPlot.core.plot.builder.PosProviderContext
import kotlin.jvm.JvmOverloads
abstract class PosProvider {
abstract fun createPos(ctx: PosProviderContext): PositionAdjustment
abstract fun handlesGroups(): Boolean
companion object {
fun wrap(pos: PositionAdjustment): PosProvider {
return object : PosProvider() {
override fun createPos(ctx: PosProviderContext): PositionAdjustment {
return pos
}
override fun handlesGroups(): Boolean {
return pos.handlesGroups()
}
}
}
fun barStack(vjust: Double? = null, stackingMode: StackingMode = StackablePos.DEF_STACKING_MODE): PosProvider {
return object : PosProvider() {
override fun createPos(ctx: PosProviderContext): PositionAdjustment {
return PositionAdjustments.stack(ctx.aesthetics, vjust, stackingMode)
}
override fun handlesGroups(): Boolean {
return PositionAdjustments.Meta.STACK.handlesGroups()
}
}
}
@JvmOverloads
fun dodge(width: Double? = null): PosProvider {
return object : PosProvider() {
override fun createPos(ctx: PosProviderContext): PositionAdjustment {
val aesthetics = ctx.aesthetics
val groupCount = ctx.groupCount
return PositionAdjustments.dodge(aesthetics, groupCount, width)
}
override fun handlesGroups(): Boolean {
return PositionAdjustments.Meta.DODGE.handlesGroups()
}
}
}
fun dodgev(height: Double? = null): PosProvider {
return object : PosProvider() {
override fun createPos(ctx: PosProviderContext): PositionAdjustment {
val aesthetics = ctx.aesthetics
val groupCount = ctx.groupCount
return PositionAdjustments.dodgev(aesthetics, groupCount, height)
}
override fun handlesGroups(): Boolean {
return PositionAdjustments.Meta.DODGE.handlesGroups()
}
}
}
fun fill(vjust: Double? = null, stackingMode: StackingMode = StackablePos.DEF_STACKING_MODE): PosProvider {
return object : PosProvider() {
override fun createPos(ctx: PosProviderContext): PositionAdjustment {
return PositionAdjustments.fill(ctx.aesthetics, vjust, stackingMode)
}
override fun handlesGroups(): Boolean {
return PositionAdjustments.Meta.FILL.handlesGroups()
}
}
}
fun jitter(width: Double?, height: Double?, seed: Long?): PosProvider {
return object : PosProvider() {
override fun createPos(ctx: PosProviderContext): PositionAdjustment {
return PositionAdjustments.jitter(width, height, seed)
}
override fun handlesGroups(): Boolean {
return PositionAdjustments.Meta.JITTER.handlesGroups()
}
}
}
fun nudge(width: Double?, height: Double?): PosProvider {
return object : PosProvider() {
override fun createPos(ctx: PosProviderContext): PositionAdjustment {
return PositionAdjustments.nudge(width, height)
}
override fun handlesGroups(): Boolean {
return PositionAdjustments.Meta.NUDGE.handlesGroups()
}
}
}
/*
public static PosProvider jitterDodge(Double jitterWidth, Double jitterHeight) {
return jitterDodge(null, jitterWidth, jitterHeight);
}
*/
fun jitterDodge(width: Double?, jitterWidth: Double?, jitterHeight: Double?): PosProvider {
return object : PosProvider() {
override fun createPos(ctx: PosProviderContext): PositionAdjustment {
val aesthetics = ctx.aesthetics
val groupCount = ctx.groupCount
return PositionAdjustments.jitterDodge(aesthetics, groupCount, width, jitterWidth, jitterHeight)
}
override fun handlesGroups(): Boolean {
return PositionAdjustments.Meta.JITTER_DODGE.handlesGroups()
}
}
}
}
}