diff options
author | DeDiamondPro <67508414+DeDiamondPro@users.noreply.github.com> | 2023-05-26 14:21:04 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-05-26 22:21:04 +1000 |
commit | 009c6cca93f0b8ee3dbc81a2929867c673d96f4f (patch) | |
tree | e771cfba365924eda677cd91b854e70b40f0df59 /src/main/kotlin | |
parent | fbf651d95e0a5b1b138511e7cd899bc4d663a83a (diff) | |
download | NotEnoughUpdates-009c6cca93f0b8ee3dbc81a2929867c673d96f4f.tar.gz NotEnoughUpdates-009c6cca93f0b8ee3dbc81a2929867c673d96f4f.tar.bz2 NotEnoughUpdates-009c6cca93f0b8ee3dbc81a2929867c673d96f4f.zip |
Graph v2 (#680)
Diffstat (limited to 'src/main/kotlin')
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 +} |