aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin
diff options
context:
space:
mode:
authorDeDiamondPro <67508414+DeDiamondPro@users.noreply.github.com>2023-05-26 14:21:04 +0200
committerGitHub <noreply@github.com>2023-05-26 22:21:04 +1000
commit009c6cca93f0b8ee3dbc81a2929867c673d96f4f (patch)
treee771cfba365924eda677cd91b854e70b40f0df59 /src/main/kotlin
parentfbf651d95e0a5b1b138511e7cd899bc4d663a83a (diff)
downloadNotEnoughUpdates-009c6cca93f0b8ee3dbc81a2929867c673d96f4f.tar.gz
NotEnoughUpdates-009c6cca93f0b8ee3dbc81a2929867c673d96f4f.tar.bz2
NotEnoughUpdates-009c6cca93f0b8ee3dbc81a2929867c673d96f4f.zip
Graph v2 (#680)
Diffstat (limited to 'src/main/kotlin')
-rw-r--r--src/main/kotlin/io/github/moulberry/notenoughupdates/commands/dev/DevTestCommand.kt5
-rw-r--r--src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/GraphDataProvider.kt27
-rw-r--r--src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/GuiPriceGraph.kt596
-rw-r--r--src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/LocalGraphDataProvider.kt192
-rw-r--r--src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/PriceObject.kt24
-rw-r--r--src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/ServerGraphDataProvider.kt42
-rw-r--r--src/main/kotlin/io/github/moulberry/notenoughupdates/util/KotlinNumberUtils.kt28
7 files changed, 912 insertions, 2 deletions
diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/commands/dev/DevTestCommand.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/commands/dev/DevTestCommand.kt
index a5ff48d2..93c4ab90 100644
--- a/src/main/kotlin/io/github/moulberry/notenoughupdates/commands/dev/DevTestCommand.kt
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/commands/dev/DevTestCommand.kt
@@ -29,8 +29,8 @@ import io.github.moulberry.notenoughupdates.events.RegisterBrigadierCommandEvent
import io.github.moulberry.notenoughupdates.miscfeatures.FishingHelper
import io.github.moulberry.notenoughupdates.miscfeatures.customblockzones.CustomBiomes
import io.github.moulberry.notenoughupdates.miscfeatures.customblockzones.LocationChangeEvent
-import io.github.moulberry.notenoughupdates.miscgui.GuiPriceGraph
import io.github.moulberry.notenoughupdates.miscgui.minionhelper.MinionHelperManager
+import io.github.moulberry.notenoughupdates.miscgui.pricegraph.GuiPriceGraph
import io.github.moulberry.notenoughupdates.util.PronounDB
import io.github.moulberry.notenoughupdates.util.SBInfo
import io.github.moulberry.notenoughupdates.util.TabListUtils
@@ -131,7 +131,8 @@ class DevTestCommand {
thenLiteral("pricetest") {
thenArgument("item", StringArgumentType.string()) { item ->
thenExecute {
- NotEnoughUpdates.INSTANCE.openGui = GuiPriceGraph(this[item])
+ NotEnoughUpdates.INSTANCE.openGui =
+ GuiPriceGraph(this[item])
}
}.withHelp("Display the price graph for an item by id")
thenExecute {
diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/GraphDataProvider.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/GraphDataProvider.kt
new file mode 100644
index 00000000..575c34b9
--- /dev/null
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/GraphDataProvider.kt
@@ -0,0 +1,27 @@
+/*
+ * Copyright (C) 2023 NotEnoughUpdates contributors
+ *
+ * This file is part of NotEnoughUpdates.
+ *
+ * NotEnoughUpdates is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation, either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * NotEnoughUpdates is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package io.github.moulberry.notenoughupdates.miscgui.pricegraph
+
+import java.time.Instant
+import java.util.concurrent.CompletableFuture
+
+interface GraphDataProvider {
+ fun loadData(itemId: String,): CompletableFuture<Map<Instant, PriceObject>?>
+}
diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/GuiPriceGraph.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/GuiPriceGraph.kt
new file mode 100644
index 00000000..0fe6d843
--- /dev/null
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/GuiPriceGraph.kt
@@ -0,0 +1,596 @@
+/*
+ * Copyright (C) 2023 NotEnoughUpdates contributors
+ *
+ * This file is part of NotEnoughUpdates.
+ *
+ * NotEnoughUpdates is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation, either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * NotEnoughUpdates is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package io.github.moulberry.notenoughupdates.miscgui.pricegraph
+
+import io.github.moulberry.notenoughupdates.NotEnoughUpdates
+import io.github.moulberry.notenoughupdates.util.SpecialColour
+import io.github.moulberry.notenoughupdates.util.Utils
+import io.github.moulberry.notenoughupdates.util.roundToDecimals
+import net.minecraft.client.Minecraft
+import net.minecraft.client.gui.GuiScreen
+import net.minecraft.client.renderer.GlStateManager
+import net.minecraft.item.ItemStack
+import net.minecraft.util.EnumChatFormatting
+import net.minecraft.util.ResourceLocation
+import org.lwjgl.opengl.GL11
+import java.text.DecimalFormat
+import java.text.NumberFormat
+import java.text.SimpleDateFormat
+import java.time.Duration
+import java.time.Instant
+import java.util.*
+import java.util.concurrent.CompletableFuture
+import kotlin.math.abs
+
+private const val X_SIZE = 364
+private const val Y_SIZE = 215
+private val dateFormat = SimpleDateFormat("'§b'd MMMMM yyyy '§eat§b' HH:mm")
+private val numberFormat = NumberFormat.getInstance()
+private val config = NotEnoughUpdates.INSTANCE.config
+
+class GuiPriceGraph(itemId: String) : GuiScreen() {
+ private val TEXTURE: ResourceLocation = when (config.ahGraph.graphStyle) {
+ 1 -> ResourceLocation("notenoughupdates:price_graph_gui/price_information_gui_dark.png")
+ 2 -> ResourceLocation("notenoughupdates:price_graph_gui/price_information_gui_phqdark.png")
+ 3 -> ResourceLocation("notenoughupdates:price_graph_gui/price_information_gui_fsr.png")
+ else -> ResourceLocation("notenoughupdates:price_graph_gui/price_information_gui.png")
+ }
+ private val dataProvider: GraphDataProvider = when (config.ahGraph.dataSource) {
+ 0 -> ServerGraphDataProvider
+ else -> LocalGraphDataProvider
+ }
+ private val rawData: CompletableFuture<Map<Instant, PriceObject>?> = dataProvider.loadData(itemId)
+ private var data: Map<Instant, PriceObject> = mapOf()
+ private var processedData = false
+ private var hasSellData = false
+ private var firstTime: Instant = Instant.now()
+ private var lastTime: Instant = Instant.now()
+ private var lowestPrice: Double = 0.0
+ private var highestPrice: Double = 1.0
+ private var guiLeft = 0
+ private var guiTop = 0
+
+ /**
+ * 0 = hour
+ * 1 = day
+ * 2 = week
+ * 3 = all
+ * 4 = custom
+ */
+ private var mode = config.ahGraph.defaultMode
+
+ private var itemName: String? = null
+ private var itemStack: ItemStack? = null
+
+ private var customSelecting = false
+ private var customSelectionStart = 0
+ private var customSelectionEnd = 0
+
+ private val buyPoints = mutableMapOf<Double, Pair<Double, Double>>()
+ private val sellPoints = mutableMapOf<Double, Pair<Double, Double>>()
+
+ private val buyMovingAverage = mutableMapOf<Double, Double>()
+ private val buyMovingAveragePoints = mutableMapOf<Double, Pair<Double, Double>>()
+ private val sellMovingAverage = mutableMapOf<Double, Double>()
+ private val sellMovingAveragePoints = mutableMapOf<Double, Pair<Double, Double>>()
+
+ init {
+ if (NotEnoughUpdates.INSTANCE.manager.itemInformation.containsKey(itemId)) {
+ val itemInfo = NotEnoughUpdates.INSTANCE.manager.itemInformation[itemId]
+ itemName = itemInfo!!["displayname"].asString
+ itemStack = NotEnoughUpdates.INSTANCE.manager.jsonToStack(itemInfo)
+ }
+ }
+
+ override fun drawScreen(mouseX: Int, mouseY: Int, partialTicks: Float) {
+ drawDefaultBackground()
+
+ guiLeft = (width - X_SIZE) / 2
+ guiTop = (height - Y_SIZE) / 2
+
+ if (customSelecting) customSelectionEnd = // Update custom selecting box
+ if (mouseX < guiLeft + 17) guiLeft + 17 else mouseX.coerceAtMost(guiLeft + 315)
+
+ Minecraft.getMinecraft().textureManager.bindTexture(TEXTURE)
+ GlStateManager.color(1f, 1f, 1f, 1f)
+ Utils.drawTexturedRect( // Draw main background
+ guiLeft.toFloat(), guiTop.toFloat(), X_SIZE.toFloat(), Y_SIZE.toFloat(),
+ 0f, X_SIZE / 512f, 0f, Y_SIZE / 512f, GL11.GL_NEAREST
+ )
+ for (i in 0..3) Utils.drawTexturedRect( // Draw buttons
+ (guiLeft + 245 + 18 * i).toFloat(), (guiTop + 17).toFloat(), 16f, 16f,
+ (0f + 16f * i) / 512f, (16 + 16f * i) / 512f,
+ (if (mode == i) 215 else 231) / 512f,
+ (if (mode == i) 231 else 247) / 512f, GL11.GL_NEAREST
+ )
+
+ if (itemName != null && itemStack != null) { // Draw item name and icon
+ Utils.drawItemStack(itemStack, guiLeft + 16, guiTop + 11)
+ Utils.drawStringScaledMax(
+ itemName, (guiLeft + 35).toFloat(), (guiTop + 13).toFloat(), false,
+ 0xffffff, 1.77f, 208
+ )
+ }
+
+ if (!rawData.isDone) {
+ Utils.drawStringCentered( // Loading text
+ "Loading...",
+ (guiLeft + 166).toFloat(), (guiTop + 116).toFloat(),
+ false, -0x100
+ )
+ } else if (rawData.get() == null || rawData.get()!!.size < 2 || data.isEmpty() && processedData) {
+ Utils.drawStringCentered( // Error text
+ "No data found.",
+ (guiLeft + 166).toFloat(),
+ (guiTop + 116).toFloat(),
+ false,
+ -0x10000
+ )
+ } else if (data.isEmpty()) {
+ processData() // Process the data if needed, done here so no race conditions of any kind can occur
+ } else {
+ val buyColor = SpecialColour.specialToChromaRGB(config.ahGraph.graphColor)
+ val sellColor = SpecialColour.specialToChromaRGB(config.ahGraph.graphColor2)
+ val buyMovingAverageColor = SpecialColour.specialToChromaRGB(config.ahGraph.movingAverageColor)
+ val sellMovingAverageColor = SpecialColour.specialToChromaRGB(config.ahGraph.movingAverageColor2)
+ var prevX: Double? = null
+ var prevY: Double? = null
+
+ if (hasSellData) { // Draw sell gradient
+ drawGradient(sellColor)
+ for (point in data) {
+ if (point.value.sellPrice == null) continue
+ val x = getX(point.key)
+ val y = getY(point.value.sellPrice!!)
+
+ if (prevX != null && prevY != null) drawCoveringQuad(x, y, prevX, prevY)
+
+ prevX = x
+ prevY = y
+ }
+ }
+
+ prevX = null
+ prevY = null
+ var closestPoint = data.entries.first()
+ var closestDistance = Double.MAX_VALUE
+
+ drawGradient(buyColor) // Draw buy gradient
+ for (point in data) {
+ val x = getX(point.key)
+ val y = getY(point.value.buyPrice)
+
+ if (prevX != null && prevY != null) drawCoveringQuad(x, y, prevX, prevY)
+
+ val distance = abs(mouseX - x) // Find the closest point to show tooltip
+ if (closestDistance > distance) {
+ closestPoint = point
+ closestDistance = distance
+ }
+
+ prevX = x
+ prevY = y
+ }
+ prevX = null
+
+ // Draw lines last to make sure nothing cuts into it
+ for (point in data) {
+ val x = getX(point.key)
+ if (prevX != null) {
+ if (config.ahGraph.movingAverages) drawLine(
+ x, prevX,
+ buyMovingAveragePoints[x], sellMovingAveragePoints[x],
+ buyMovingAverageColor, sellMovingAverageColor
+ )
+ drawLine(x, prevX, buyPoints[x], sellPoints[x], buyColor, sellColor)
+ }
+ prevX = x
+ }
+
+ // Draw axis
+ for (i in 0..6) { // Y-axis with price
+ val price = map(i.toDouble(), 0.0, 6.0, highestPrice, lowestPrice).toLong()
+ Utils.drawStringF(
+ formatPrice(price), guiLeft + 320f,
+ map(i.toDouble(), 0.0, 6.0, guiTop + 35.0, guiTop + 198.0).toFloat()
+ - Minecraft.getMinecraft().fontRendererObj.FONT_HEIGHT / 2f, false, 0x8b8b8b
+ )
+ }
+ // X-axis with hour or date
+ val showDays = lastTime.epochSecond - firstTime.epochSecond > 86400
+ val amountOfTime = (lastTime.epochSecond - firstTime.epochSecond) / (if (showDays) 86400.0 else 3600.0)
+ val pixelsPerTime = 298.0 / amountOfTime
+ var time = firstTime.plusSeconds(
+ if (showDays) (24 - Date.from(firstTime).hours) * 3600L
+ else (60 - Date.from(firstTime).minutes) * 60L
+ )
+ var xPos = getX(time)
+ var lastX = -100.0
+ while (xPos < guiLeft + 315) {
+ if (abs(xPos - lastX) > 30) {
+ Utils.drawStringCentered(
+ Date.from(time).let { if (showDays) it.date else it.hours }.toString(),
+ xPos.toFloat(), (guiTop + 206).toFloat(),
+ false, 0x8b8b8b
+ )
+ lastX = xPos
+ }
+ time = time.plusSeconds(if (showDays) 86400L else 3600L)
+ xPos += pixelsPerTime
+ }
+
+ if (
+ mouseX >= guiLeft + 17 && mouseX <= guiLeft + 315 &&
+ mouseY >= guiTop + 35 && mouseY <= guiTop + 198 && !customSelecting
+ ) {
+ // Draw tooltip with price info
+ val text = ArrayList<String>()
+ val x = getX(closestPoint.key)
+ val y = getY(closestPoint.value.buyPrice)
+ text.add(dateFormat.format(Date.from(closestPoint.key)))
+ if (closestPoint.value.sellPrice == null) {
+ text.add(
+ "${EnumChatFormatting.YELLOW}${EnumChatFormatting.BOLD}Lowest BIN: ${EnumChatFormatting.GOLD}" +
+ "${EnumChatFormatting.BOLD}${numberFormat.format(closestPoint.value.buyPrice)}"
+ )
+ if (config.ahGraph.movingAverages && buyMovingAverage[x] != null) text.add(
+ "${EnumChatFormatting.YELLOW}${EnumChatFormatting.BOLD}Lowest BIN Moving Average: ${EnumChatFormatting.GOLD}" +
+ "${EnumChatFormatting.BOLD}${numberFormat.format(buyMovingAverage[x])}"
+ )
+ } else {
+ text.add(
+ "${EnumChatFormatting.YELLOW}${EnumChatFormatting.BOLD}Bazaar Insta-Buy: ${EnumChatFormatting.GOLD}" +
+ "${EnumChatFormatting.BOLD}${numberFormat.format(closestPoint.value.buyPrice)}"
+ )
+ text.add(
+ "${EnumChatFormatting.YELLOW}${EnumChatFormatting.BOLD}Bazaar Insta-Sell: ${EnumChatFormatting.GOLD}" +
+ "${EnumChatFormatting.BOLD}${numberFormat.format(closestPoint.value.sellPrice)}"
+ )
+ if (config.ahGraph.movingAverages) {
+ if (buyMovingAverage[x] != null) text.add(
+ "${EnumChatFormatting.YELLOW}${EnumChatFormatting.BOLD}Bazaar Insta-Buy Moving Average: ${EnumChatFormatting.GOLD}${EnumChatFormatting.BOLD}" +
+ numberFormat.format(buyMovingAverage[x])
+ )
+ if (sellMovingAverage[x] != null) text.add(
+ "${EnumChatFormatting.YELLOW}${EnumChatFormatting.BOLD}Bazaar Insta-Sell Moving Average: ${EnumChatFormatting.GOLD}${EnumChatFormatting.BOLD}" +
+ numberFormat.format(sellMovingAverage[x])
+ )
+ }
+ }
+ Utils.drawLine(
+ x.toFloat(), (guiTop + 35).toFloat(),
+ x.toFloat(), (guiTop + 198).toFloat(),
+ 2, 0x4D8b8b8b
+ )
+ Minecraft.getMinecraft().textureManager.bindTexture(TEXTURE)
+ GlStateManager.color(1f, 1f, 1f, 1f)
+ Utils.drawTexturedRect(
+ x.toFloat() - 2.5f, y.toFloat() - 2.5f, 5f, 5f,
+ 0f, 5 / 512f, 247 / 512f, 252 / 512f, GL11.GL_NEAREST
+ )
+ if (closestPoint.value.sellPrice != null) Utils.drawTexturedRect(
+ x.toFloat() - 2.5f, getY(closestPoint.value.sellPrice!!).toFloat() - 2.5f, 5f, 5f,
+ 0f, 5 / 512f, 247 / 512f, 252 / 512f, GL11.GL_NEAREST
+ )
+ if (config.ahGraph.movingAverages) {
+ val buyAverageY = buyMovingAveragePoints[x]?.second
+ if (buyAverageY != null) Utils.drawTexturedRect(
+ x.toFloat() - 2.5f, buyAverageY.toFloat() - 2.5f, 5f, 5f,
+ 0f, 5 / 512f, 247 / 512f, 252 / 512f, GL11.GL_NEAREST
+ )
+ val sellAverageY = sellMovingAveragePoints[x]?.second
+ if (sellAverageY != null) Utils.drawTexturedRect(
+ x.toFloat() - 2.5f, sellAverageY.toFloat() - 2.5f, 5f, 5f,
+ 0f, 5 / 512f, 247 / 512f, 252 / 512f, GL11.GL_NEAREST
+ )
+ }
+
+ drawHoveringText(text, x.toInt(), y.toInt())
+ }
+
+ if (customSelecting) { // Draw selecting box
+ Utils.drawDottedLine(
+ customSelectionStart.toFloat(), guiTop + 36f,
+ customSelectionStart.toFloat(), guiTop + 197f,
+ 2, 10, -0x39393a
+ )
+ Utils.drawDottedLine(
+ customSelectionEnd.toFloat(), guiTop + 36f,
+ customSelectionEnd.toFloat(), guiTop + 197f,
+ 2, 10, -0x39393a
+ )
+ Utils.drawDottedLine(
+ customSelectionStart.toFloat(), guiTop + 36f,
+ customSelectionEnd.toFloat(), guiTop + 36f,
+ 2, 10, -0x39393a
+ )
+ Utils.drawDottedLine(
+ customSelectionStart.toFloat(), guiTop + 197f,
+ customSelectionEnd.toFloat(), guiTop + 197f,
+ 2, 10, -0x39393a
+ )
+ }
+ }
+
+ // Draw item tooltips
+ if (mouseY >= guiTop + 17 && mouseY <= guiTop + 35 && mouseX >= guiLeft + 244 && mouseX <= guiLeft + 316) {
+ val index = (mouseX - guiLeft - 245) / 18
+ drawRect(
+ guiLeft + 245 + 18 * index,
+ guiTop + 17, guiLeft + 261 + 18 * index,
+ guiTop + 33, -0x7f000001
+ )
+ drawHoveringText(
+ listOf(
+ when (index) {
+ 0 -> "Show 1 Hour"
+ 1 -> "Show 1 Day"
+ 2 -> "Show 1 Week"
+ else -> if (dataProvider is LocalGraphDataProvider) "Show All" else "Show 1 Month"
+ }
+ ), mouseX, mouseY
+ )
+ }
+ }
+
+ private fun getX(time: Instant) = map(
+ time.epochSecond.toDouble(),
+ firstTime.epochSecond.toDouble(),
+ lastTime.epochSecond.toDouble(),
+ guiLeft + 17.0,
+ guiLeft + 315.0
+ )
+
+ private fun getY(price: Double) = map(
+ price,
+ highestPrice + 1,
+ lowestPrice - 1,
+ guiTop + 45.0,
+ guiTop + 188.0
+ )
+
+ override fun mouseClicked(mouseX: Int, mouseY: Int, mouseButton: Int) {
+ super.mouseClicked(mouseX, mouseY, mouseButton)
+ if (mouseY >= guiTop + 17 && mouseY <= guiTop + 35 && mouseX >= guiLeft + 244 && mouseX <= guiLeft + 316) {
+ selectMode((mouseX - guiLeft - 245) / 18)
+ Utils.playPressSound()
+ } else if (mouseY >= guiTop + 35 && mouseY <= guiTop + 198 && mouseX >= guiLeft + 17 && mouseX <= guiLeft + 315) {
+ customSelecting = true
+ customSelectionStart = mouseX
+ customSelectionEnd = mouseX
+ }
+ }
+
+ override fun mouseReleased(mouseX: Int, mouseY: Int, state: Int) {
+ super.mouseReleased(mouseX, mouseY, state)
+ if (customSelecting) {
+ customSelecting = false
+ customSelectionEnd =
+ if (mouseX < guiLeft + 17) guiLeft + 17 else mouseX.coerceAtMost(guiLeft + 315)
+ if (customSelectionStart > customSelectionEnd) {
+ val temp = customSelectionStart
+ customSelectionStart = customSelectionEnd
+ customSelectionEnd = temp
+ }
+ if (customSelectionStart - customSelectionEnd == 0) return
+ selectMode(4)
+ }
+ }
+
+ private fun processData() {
+ processedData = true
+ // Filter based on time
+ val now = Instant.now()
+ val startTime = when (mode) {
+ 0 -> now.minus(Duration.ofHours(1))
+ 1 -> now.minus(Duration.ofDays(1))
+ 2 -> now.minus(Duration.ofDays(7))
+ // The server only deletes old data every hour, so you could get 30 days and 1 hour, this is just for consistency
+ 3 -> if (dataProvider is ServerGraphDataProvider) now.minus(Duration.ofDays(30)) else null
+ 4 -> Instant.ofEpochSecond(
+ map(
+ customSelectionStart.toDouble(),
+ guiLeft + 17.0,
+ guiLeft + 315.0,
+ firstTime.epochSecond.toDouble(),
+ lastTime.epochSecond.toDouble()
+ ).toLong()
+ )
+
+ else -> error("$mode is not a valid mode!")
+ }
+ val endTime = when (mode) {
+ 4 -> Instant.ofEpochSecond(
+ map(
+ customSelectionEnd.toDouble(),
+ guiLeft + 17.0,
+ guiLeft + 315.0,
+ firstTime.epochSecond.toDouble(),
+ lastTime.epochSecond.toDouble()
+ ).toLong()
+ )
+
+ else -> null
+ }
+ val cutData = rawData.get()!!.filter {
+ (startTime == null || it.key >= startTime) && (endTime == null || it.key <= endTime)
+ }
+ if (cutData.isEmpty()) return
+
+ // Smooth data
+ val zones = config.ahGraph.graphZones
+ val first = cutData.minOf { it.key }
+ val last = cutData.maxOf { it.key }
+ val trimmedData = mutableMapOf<Instant, PriceObject>()
+ for (i in 0..zones) {
+ val zoneStart = Instant.ofEpochSecond(
+ map(
+ i.toDouble(), 0.0, zones.toDouble(),
+ first.epochSecond.toDouble(), last.epochSecond.toDouble()
+ ).toLong()
+ )
+ val zoneEnd = Instant.ofEpochSecond(
+ map(
+ i + 1.0, 0.0, zones.toDouble(),
+ first.epochSecond.toDouble(), last.epochSecond.toDouble()
+ ).toLong()
+ )
+ val dataInZone = cutData.filter { it.key >= zoneStart && it.key < zoneEnd }
+ if (dataInZone.isEmpty()) {
+ continue
+ } else {
+ val averageTime = Instant.ofEpochSecond(dataInZone.keys.sumOf { it.epochSecond } / dataInZone.size)
+ val averageBuyPrice = (dataInZone.values.sumOf { it.buyPrice } / dataInZone.size).roundToDecimals(1)
+ val sellPoints = dataInZone.values.filter { it.sellPrice != null }
+ val averageSellPrice = if (sellPoints.isEmpty()) null
+ else (sellPoints.sumOf { it.sellPrice ?: 0.0 } / sellPoints.size).roundToDecimals(1)
+ trimmedData[averageTime] = PriceObject(averageBuyPrice, averageSellPrice)
+ }
+ }
+ data = trimmedData
+ if (data.isEmpty()) return
+
+ // Populate variables required for graphs
+ firstTime = data.minOf { it.key }
+ lastTime = data.maxOf { it.key }
+ lowestPrice = data.minOf {
+ if (it.value.sellPrice != null) it.value.buyPrice.coerceAtMost(it.value.sellPrice!!)
+ else it.value.buyPrice
+ }
+ highestPrice = data.maxOf {
+ if (it.value.sellPrice != null) it.value.buyPrice.coerceAtLeast(it.value.sellPrice!!)
+ else it.value.buyPrice
+ }
+
+ // Populate line variables
+ buyPoints.clear()
+ sellPoints.clear()
+ buyMovingAveragePoints.clear()
+ sellMovingAveragePoints.clear()
+ val movingAveragePeriod = (lastTime.epochSecond - firstTime.epochSecond) * config.ahGraph.movingAveragePercent
+ var prevBuyY: Double? = null
+ var prevSellY: Double? = null
+ var prevBuyAverageY: Double? = null
+ var prevSellAverageY: Double? = null
+ for (point in data) {
+ val x = getX(point.key)
+ val buyY = getY(point.value.buyPrice)
+ if (prevBuyY != null) buyPoints[x] = prevBuyY to buyY
+ prevBuyY = buyY
+ if (point.value.sellPrice != null) {
+ val sellY = getY(point.value.sellPrice!!)
+ if (prevSellY != null) sellPoints[x] = prevSellY to sellY
+ prevSellY = sellY
+ }
+ // Moving average stuff
+ if (!config.ahGraph.movingAverages) continue
+ val dataInPeriod = rawData.get()?.filterKeys {
+ it >= point.key.minusSeconds(movingAveragePeriod.toLong()) && it <= point.key
+ }
+ if (dataInPeriod.isNullOrEmpty()) continue
+ val buyAverage = dataInPeriod.values.sumOf { it.buyPrice } / dataInPeriod.size
+ buyMovingAverage[x] = buyAverage.roundToDecimals(1)
+ val buyAverageY = getY((buyAverage).coerceAtLeast(lowestPrice).coerceAtMost(highestPrice))
+ if (prevBuyAverageY != null) buyMovingAveragePoints[x] = prevBuyAverageY to buyAverageY
+ prevBuyAverageY = buyAverageY
+ val sellData = dataInPeriod.filterValues { it.sellPrice != null }
+ if (sellData.isEmpty()) continue
+ val sellAverage = sellData.values.sumOf { it.sellPrice!! } / sellData.size
+ sellMovingAverage[x] = sellAverage.roundToDecimals(1)
+ val sellAverageY = getY((sellAverage).coerceAtLeast(lowestPrice).coerceAtMost(highestPrice))
+ if (prevSellAverageY != null) sellMovingAveragePoints[x] = prevSellAverageY to sellAverageY
+ prevSellAverageY = sellAverageY
+ }
+ hasSellData = sellPoints.isNotEmpty()
+ }
+
+ private fun selectMode(mode: Int) {
+ this.mode = mode
+ data = mapOf()
+ processedData = false
+ }
+
+ private fun drawGradient(color: Int) {
+ Utils.drawGradientRect(
+ 0,
+ guiLeft + 17,
+ guiTop + 35,
+ guiLeft + 315,
+ guiTop + 198,
+ changeAlpha(color, 120),
+ changeAlpha(color, 10)
+ )
+ }
+
+ private fun drawCoveringQuad(x: Double, y: Double, prevX: Double, prevY: Double) {
+ Minecraft.getMinecraft().textureManager.bindTexture(TEXTURE)
+ GlStateManager.color(1f, 1f, 1f, 1f)
+ Utils.drawTexturedQuad(
+ prevX.toFloat(), prevY.toFloat(),
+ x.toFloat(), y.toFloat(),
+ x.toFloat(), guiTop + 35f,
+ prevX.toFloat(), guiTop + 35f,
+ 18 / 512f, 19 / 512f,
+ 36 / 512f, 37 / 512f,
+ GL11.GL_NEAREST
+ )
+ }
+
+ private fun drawLine(
+ x: Double,
+ prevX: Double,
+ buyLine: Pair<Double, Double>?,
+ sellLine: Pair<Double, Double>?,
+ buyColor: Int,
+ sellColor: Int
+ ) {
+ if (buyLine != null) Utils.drawLine(
+ prevX.toFloat(), buyLine.first.toFloat() + 0.5f,
+ x.toFloat(), buyLine.second.toFloat() + 0.5f,
+ 2, buyColor
+ )
+ if (sellLine != null) Utils.drawLine(
+ prevX.toFloat(), sellLine.first.toFloat() + 0.5f,
+ x.toFloat(), sellLine.second.toFloat() + 0.5f,
+ 2, sellColor
+ )
+ }
+
+ private fun formatPrice(price: Long): String {
+ val df = DecimalFormat("#.00")
+ if (price >= 1000000000) {
+ return df.format((price / 1000000000f).toDouble()) + "B"
+ } else if (price >= 1000000) {
+ return df.format((price / 1000000f).toDouble()) + "M"
+ } else if (price >= 1000) {
+ return df.format((price / 1000f).toDouble()) + "K"
+ }
+ return price.toString()
+ }
+
+ private fun changeAlpha(origColor: Int, alpha: Int): Int {
+ val color = origColor and 0x00ffffff //drop the previous alpha value
+ return alpha shl 24 or color //add the one the user inputted
+ }
+
+ private fun map(x: Double, inMin: Double, inMax: Double, outMin: Double, outMax: Double): Double {
+ return (x - inMin) * (outMax - outMin) / (inMax - inMin) + outMin
+ }
+}
diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/LocalGraphDataProvider.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/LocalGraphDataProvider.kt
new file mode 100644
index 00000000..be93a8d6
--- /dev/null
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/LocalGraphDataProvider.kt
@@ -0,0 +1,192 @@
+/*
+ * Copyright (C) 2023 NotEnoughUpdates contributors
+ *
+ * This file is part of NotEnoughUpdates.
+ *
+ * NotEnoughUpdates is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation, either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * NotEnoughUpdates is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package io.github.moulberry.notenoughupdates.miscgui.pricegraph
+
+import com.google.common.reflect.TypeToken
+import com.google.gson.GsonBuilder
+import com.google.gson.JsonElement
+import com.google.gson.JsonObject
+import io.github.moulberry.notenoughupdates.NotEnoughUpdates
+import java.io.*
+import java.nio.charset.StandardCharsets
+import java.text.SimpleDateFormat
+import java.time.Instant
+import java.util.*
+import java.util.concurrent.CompletableFuture
+import java.util.concurrent.atomic.AtomicBoolean
+import java.util.zip.GZIPInputStream
+import java.util.zip.GZIPOutputStream
+
+object LocalGraphDataProvider : GraphDataProvider {
+ private val priceDir = File("config/notenoughupdates/prices")
+ private val GSON = GsonBuilder().create()
+ private val format = SimpleDateFormat("dd-MM-yyyy")
+ private val config = NotEnoughUpdates.INSTANCE.config
+ private val fileLocked = AtomicBoolean(false)
+
+ override fun loadData(itemId: String): CompletableFuture<Map<Instant, PriceObject>?> {
+ return CompletableFuture.supplyAsync {
+ if (!priceDir.exists() || !priceDir.isDirectory) return@supplyAsync null
+ if (fileLocked.get()) while (fileLocked.get()) { // Wait for file to become unlocked
+ Thread.sleep(100)
+ }
+ val response = mutableMapOf<Instant, PriceObject>()
+ val futures = mutableListOf<CompletableFuture<Map<Instant, PriceObject>?>>()
+ for (file in priceDir.listFiles { file -> file.extension == "gz" }!!) {
+ futures.add(CompletableFuture.supplyAsync {
+ val data = load(file)?.get(itemId) ?: return@supplyAsync null
+ (if (data.isBz()) data.bz?.map {
+ Instant.ofEpochSecond(it.key) to PriceObject(it.value.b, it.value.s)
+ } else data.ah?.map {
+ Instant.ofEpochSecond(it.key) to PriceObject(it.value.toDouble(), null)
+ })?.toMap()
+ })
+ }
+
+ for (future in futures) {
+ val result = future.get()
+ if (result != null) response.putAll(result)
+ }
+
+ return@supplyAsync response
+ }
+ }
+
+ private fun load(file: File): MutableMap<String, ItemData>? {
+ val type = object : TypeToken<MutableMap<String, ItemData>?>() {}.type
+ if (file.exists()) {
+ try {
+ BufferedReader(
+ InputStreamReader(
+ GZIPInputStream(FileInputStream(file)),
+ StandardCharsets.UTF_8
+ )
+ ).use { reader ->
+ return GSON.fromJson(
+ reader,
+ type
+ )
+ }
+ } catch (e: Exception) {
+ println("Deleting " + file.name + " because it is probably corrupted.")
+ file.delete()
+ }
+ }
+ return null
+ }
+
+ fun savePrices(items: JsonObject, bazaar: Boolean) {
+ try {
+ if (!priceDir.exists() && !priceDir.mkdir()) return
+ val files = priceDir.listFiles()
+ val dataRetentionTime =
+ System.currentTimeMillis() - config.ahGraph.dataRetention * 86400000L
+ files?.filter { it.extension == "gz" && it.lastModified() < dataRetentionTime }?.forEach { it.delete() }
+
+ if (!config.ahGraph.graphEnabled || config.ahGraph.dataSource != 1) return
+ if (fileLocked.get()) while (fileLocked.get()) { // Wait for file to become unlocked
+ Thread.sleep(100)
+ }
+ fileLocked.set(true)
+
+ val date = Date()
+ val epochSecond = date.toInstant().epochSecond
+ val file = File(priceDir, "prices_" + format.format(date) + ".gz")
+ var prices = load(file) ?: mutableMapOf()
+ if (file.exists()) {
+ val tempPrices = load(file)
+ if (tempPrices != null) prices = tempPrices
+ }
+ for (item in items.entrySet()) {
+ addOrUpdateItemPriceInfo(item, prices, epochSecond, bazaar)
+ }
+ file.createNewFile()
+ BufferedWriter(
+ OutputStreamWriter(
+ GZIPOutputStream(FileOutputStream(file)),
+ StandardCharsets.UTF_8
+ )
+ ).use { writer -> writer.write(GSON.toJson(prices)) }
+ fileLocked.set(false)
+ } catch (e: java.lang.Exception) {
+ e.printStackTrace()
+ fileLocked.set(false)
+ }
+ }
+
+ private fun addOrUpdateItemPriceInfo(
+ item: Map.Entry<String, JsonElement>,
+ prices: MutableMap<String, ItemData>,
+ timestamp: Long,
+ bazaar: Boolean
+ ) {
+ val itemName = item.key
+ var existingItemData: ItemData? = null
+ if (prices.containsKey(itemName)) {
+ existingItemData = prices[itemName]
+ }
+
+ // Handle transitions from ah to bz (the other direction typically doesn't happen)
+ if (existingItemData != null) {
+ if (existingItemData.isBz() && !bazaar) {
+ return
+ }
+ if (!existingItemData.isBz() && bazaar) {
+ prices.remove(itemName)
+ existingItemData = null
+ }
+ }
+ if (bazaar) {
+ if (!item.value.asJsonObject.has("curr_buy") ||
+ !item.value.asJsonObject.has("curr_sell")
+ ) {
+ return
+ }
+ val bzData = BzData(
+ item.value.asJsonObject["curr_buy"].asDouble,
+ item.value.asJsonObject["curr_sell"].asDouble
+ )
+ if (existingItemData != null) {
+ existingItemData.bz?.set(timestamp, bzData)
+ } else {
+ val mapData = mutableMapOf(timestamp to bzData)
+ prices[item.key] = ItemData(bz = mapData)
+ }
+ } else {
+ if (existingItemData != null) {
+ prices[item.key]!!.ah?.set(timestamp, item.value.asBigDecimal.toLong())
+ } else {
+ val mapData = mutableMapOf(timestamp to item.value.asLong)
+ prices[item.key] = ItemData(ah = mapData)
+ }
+ }
+ }
+}
+
+
+private data class ItemData(val ah: MutableMap<Long, Long>? = null, val bz: MutableMap<Long, BzData>? = null) {
+
+ fun get(): MutableMap<Long, *>? = if (!isBz()) ah else bz
+
+ fun isBz() = !bz.isNullOrEmpty()
+}
+
+private class BzData(val b: Double, val s: Double)
+
diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/PriceObject.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/PriceObject.kt
new file mode 100644
index 00000000..41f1d02f
--- /dev/null
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/PriceObject.kt
@@ -0,0 +1,24 @@
+/*
+ * Copyright (C) 2023 NotEnoughUpdates contributors
+ *
+ * This file is part of NotEnoughUpdates.
+ *
+ * NotEnoughUpdates is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation, either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * NotEnoughUpdates is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package io.github.moulberry.notenoughupdates.miscgui.pricegraph
+
+import com.google.gson.annotations.SerializedName
+
+data class PriceObject(@SerializedName("b") val buyPrice: Double, @SerializedName("s") val sellPrice: Double? = null)
diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/ServerGraphDataProvider.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/ServerGraphDataProvider.kt
new file mode 100644
index 00000000..65c76bca
--- /dev/null
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/miscgui/pricegraph/ServerGraphDataProvider.kt
@@ -0,0 +1,42 @@
+/*
+ * Copyright (C) 2023 NotEnoughUpdates contributors
+ *
+ * This file is part of NotEnoughUpdates.
+ *
+ * NotEnoughUpdates is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation, either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * NotEnoughUpdates is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package io.github.moulberry.notenoughupdates.miscgui.pricegraph
+
+import com.google.gson.GsonBuilder
+import io.github.moulberry.notenoughupdates.NotEnoughUpdates
+import java.time.Instant
+import java.util.concurrent.CompletableFuture
+
+object ServerGraphDataProvider : GraphDataProvider {
+ private val gson = GsonBuilder().create()
+
+ override fun loadData(itemId: String): CompletableFuture<Map<Instant, PriceObject>?> {
+ return CompletableFuture.supplyAsync {
+ val request = NotEnoughUpdates.INSTANCE.manager.apiUtils.request()
+ .url("https://${NotEnoughUpdates.INSTANCE.config.ahGraph.serverUrl}")
+ .queryArgument("item", itemId).requestJson().get()?.asJsonObject ?: return@supplyAsync null
+
+ val response = mutableMapOf<Instant, PriceObject>()
+ for (element in request.entrySet()) response[Instant.parse(element.key)] =
+ gson.fromJson(element.value, PriceObject::class.java)
+ return@supplyAsync response
+ }
+ }
+}
diff --git a/src/main/kotlin/io/github/moulberry/notenoughupdates/util/KotlinNumberUtils.kt b/src/main/kotlin/io/github/moulberry/notenoughupdates/util/KotlinNumberUtils.kt
new file mode 100644
index 00000000..f20da4e4
--- /dev/null
+++ b/src/main/kotlin/io/github/moulberry/notenoughupdates/util/KotlinNumberUtils.kt
@@ -0,0 +1,28 @@
+/*
+ * Copyright (C) 2023 NotEnoughUpdates contributors
+ *
+ * This file is part of NotEnoughUpdates.
+ *
+ * NotEnoughUpdates is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation, either
+ * version 3 of the License, or (at your option) any later version.
+ *
+ * NotEnoughUpdates is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with NotEnoughUpdates. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package io.github.moulberry.notenoughupdates.util
+
+import kotlin.math.pow
+import kotlin.math.round
+
+fun Double.roundToDecimals(decimals: Int): Double {
+ val multiplier = 10.0.pow(decimals)
+ return round(this * multiplier) / multiplier
+}