Where people come together to learn, code, and play. Custom-built HTTP server, site generator, and website from scratch using no external libraries. Goal is to be as minimalistic and fun as possible. http://projectdivar.com
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
SigPlace/readers/LoveLiveReader.java

377 lines
16 KiB

package readers;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.regex.Pattern;
import javax.imageio.ImageIO;
import readers.fonts.Font;
import readers.fonts.Glyph;
public class LoveLiveReader extends Reader{
List<Box> extraRegions = new ArrayList<>();
static int lastJump=0;
public LoveLiveReader(){
addRegion(new Box(713,401,232,50),NUMBER); //score[0]
addRegion(new Box(867,710,286,156)); //rank[1]
extraRegions.add(new Box(65,604,250,53)); //perfect outline[0]
extraRegions.add(new Box(65,680,250,53)); //great outline[1]
extraRegions.add(new Box(65,760,250,53)); //good outline[2]
extraRegions.add(new Box(65,840,250,53)); //bad outline[3]
extraRegions.add(new Box(65,920,250,53)); //miss outline[4]
addRegion(new Box(509,604,190,54),NUMBER); //notes[2]
addRegion(new Box(509,680,190,54),NUMBER); //notes[3]
addRegion(new Box(509,760,190,54),NUMBER); //notes[4]
addRegion(new Box(509,840,190,54),NUMBER); //notes[5]
addRegion(new Box(509,920,190,54),NUMBER); //notes[6]
addRegion(new Box(26,374,265,36)); //difficulty[7]
addRegion(new Box(277,165,572,40)); //title[8]
addRegion(new Box(716,502,226,45),NUMBER); //pct[9]
addRegion(new Box(782,452,158,50),NUMBER); //maxcombo[10]
addRegion(new Box(100,470,84,42)); //difficultylv[11]
init();
}
void ColorFilter(int[] arr,int region,int width) {
switch (region) {
case 0:{
final ColorRange TARGETCOLOR = new ColorRange(240,255,130,150,0,10);
final ColorRange SEEKINGCOLOR = new ColorRange(140,255,110,255,0,200);
final Color FINALCOLOR = Color.MAGENTA;
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (TARGETCOLOR.colorInRange(col)) {
seek(arr,i,SEEKINGCOLOR,FINALCOLOR,width);
}
}
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (!col.equals(Color.MAGENTA)) {
arr[i]=TRANSPARENT;
}
}
}break;
case 2:
case 3:
case 4:
case 5:
case 6:
case 8:
case 11:{
final ColorRange TARGETCOLOR = new ColorRange(255,255,255,255,255,255);
final ColorRange SEEKINGCOLOR = new ColorRange(240,255,240,255,240,255);
final Color FINALCOLOR = Color.MAGENTA;
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (TARGETCOLOR.colorInRange(col)) {
seek(arr,i,SEEKINGCOLOR,FINALCOLOR,width);
}
}
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (!col.equals(Color.MAGENTA)) {
arr[i]=TRANSPARENT;
}
}
}break;
case 9:{
final ColorRange TARGETCOLOR = new ColorRange(0,80,135,190,75,145);
final ColorRange SEEKINGCOLOR = new ColorRange(0,200,135,240,75,240);
final Color FINALCOLOR = Color.RED;
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (TARGETCOLOR.colorInRange(col)) {
seek(arr,i,SEEKINGCOLOR,FINALCOLOR,width);
}
}
final ColorRange SEEKINGCOLOR2 = new ColorRange(255,255,0,0,0,0);
final Color FINALCOLOR2 = new Color(TRANSPARENT,true);
for (int x=0;x<width;x++) {
//30 pixels from top.
if (arr[30*width+x]==Color.RED.getRGB()) {
System.out.println("Start Jump: "+x);
x=seek(arr,30*width+x,SEEKINGCOLOR2,FINALCOLOR2,width,x);
System.out.println("End Jump: "+x);
}
}
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (!col.equals(Color.RED)) {
arr[i]=TRANSPARENT;
} else {
arr[i]=Color.MAGENTA.getRGB();
}
}
}break;
case 10:{
final ColorRange TARGETCOLOR = new ColorRange(240,255,30,50,0,5);
final ColorRange SEEKINGCOLOR = new ColorRange(180,255,10,120,0,120);
final Color FINALCOLOR = Color.MAGENTA;
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (TARGETCOLOR.colorInRange(col)) {
seek(arr,i,SEEKINGCOLOR,FINALCOLOR,width);
}
}
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (!col.equals(Color.MAGENTA)) {
arr[i]=TRANSPARENT;
}
}
}break;
case 402:{
final ColorRange TARGETCOLOR = new ColorRange(80,120,100,130,100,140);
final ColorRange SEEKINGCOLOR = new ColorRange(220,255,220,255,220,255);
final Color FINALCOLOR = Color.MAGENTA;
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (TARGETCOLOR.colorInRange(col)) {
seek(arr,i-width,SEEKINGCOLOR,FINALCOLOR,width);
}
}
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (!col.equals(Color.MAGENTA)) {
arr[i]=TRANSPARENT;
}
}
}break;
case 400:
case 401:
case 403:
case 404:{
final ColorRange TARGETCOLOR = new ColorRange(255,255,255,255,255,255);
final ColorRange SEEKINGCOLOR = new ColorRange(220,255,220,255,220,255);
final Color FINALCOLOR = Color.RED;
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (TARGETCOLOR.colorInRange(col)) {
seek(arr,i,SEEKINGCOLOR,FINALCOLOR,width);
}
}
final ColorRange SEEKINGCOLOR2 = new ColorRange(255,255,0,0,0,0);
final Color FINALCOLOR2 = new Color(TRANSPARENT,true);
for (int x=0;x<width;x++) {
//30 pixels from top.
if (arr[30*width+x]==Color.RED.getRGB()) {
System.out.println("Start Jump: "+x);
x=seek(arr,30*width+x,SEEKINGCOLOR2,FINALCOLOR2,width,x);
System.out.println("End Jump: "+x);
}
}
for (int i=0;i<arr.length;i++) {
Color col = new Color(arr[i],true);
if (!col.equals(Color.RED)) {
arr[i]=TRANSPARENT;
} else {
arr[i]=Color.MAGENTA.getRGB();
}
}
}break;
}
}
boolean checkRank(ColorRange range,int[] arr) {
return checkRank(range,arr,4096);
}
boolean checkRank(ColorRange range,int[] arr,int amt) {
int colorCount=0;
for (int j=0;j<arr.length;j++) {
Color col = new Color(arr[j],true);
if (range.colorInRange(col)) {
colorCount++;
}
}
System.out.println("Found "+colorCount+" pixels.");
return colorCount>amt;
}
public void interpretBoxes(Path img){
/*String dataString = readAllBoxes(img);
String[] data = dataString.split(Pattern.quote("\n"));
String[] ja_data = data[0].split(Pattern.quote(")"));
String[] en_data = data[2].split(Pattern.quote(")"));
trimAllData(ja_data);
trimAllData(en_data);
System.out.println(Arrays.toString(ja_data));
System.out.println(Arrays.toString(en_data));*/
int regionHeights = 0;
int maxWidth = 0;
int counter = 0;
for (int i=0;i<readRegions.size();i++) {
regionHeights+=readRegions.get(i).h+REGION_PADDING;
if (readRegions.get(i).w>maxWidth) {
maxWidth=readRegions.get(i).w;
}
}
try {
BufferedImage originalImg = ImageIO.read(img.toFile());
BufferedImage cutImg = new BufferedImage(maxWidth,regionHeights,BufferedImage.TYPE_INT_ARGB);
Graphics2D g = cutImg.createGraphics();
int currentHeight=0;
for (int i=0;i<readRegions.size();i++) {
BufferedImage subRegion = new BufferedImage(readRegions.get(i).w, readRegions.get(i).h,BufferedImage.TYPE_INT_ARGB);
subRegion.setRGB(0,0,readRegions.get(i).w,readRegions.get(i).h,originalImg.getRGB(readRegions.get(i).x, readRegions.get(i).y, readRegions.get(i).w, readRegions.get(i).h, null, 0, readRegions.get(i).w),0,readRegions.get(i).w);
int[] arr = subRegion.getRGB(0, 0, readRegions.get(i).w, readRegions.get(i).h, null, 0, readRegions.get(i).w);
//System.out.println(Arrays.toString(arr));
//System.out.println(i);
//ImageIO.write(originalImg.getSubimage(readRegions.get(i).x, readRegions.get(i).y, readRegions.get(i).w, readRegions.get(i).h),"png",new File("cut.png"));
ColorFilter(arr,i,readRegions.get(i).w);
subRegion.setRGB(0,0,readRegions.get(i).w,readRegions.get(i).h,arr,0,readRegions.get(i).w);
ImageIO.write(subRegion,"png",new File("sub.png"));
cutImg.setRGB(0,currentHeight,readRegions.get(i).w,readRegions.get(i).h,arr,0,readRegions.get(i).w);
if (i==1) {
final ColorRange dRange = new ColorRange(130,160,130,190,200,250);
final ColorRange cRange = new ColorRange(50,200,180,230,10,115);
final ColorRange bRange = new ColorRange(240,252,0,140,150,200);
final ColorRange aRange = new ColorRange(180,230,40,200,230,250);
final ColorRange sRange = new ColorRange(245,255,150,255,0,100);
Color col = new Color(subRegion.getRGB(269, 19),true);
System.out.println(col);
final ColorRange sssRange = new ColorRange(245,255,60,190,0,150);
Color col2 = new Color(subRegion.getRGB(228, 19),true);
final ColorRange ssRange = new ColorRange(245,255,60,190,0,150);
System.out.println(col2);
if (checkRank(dRange,arr)) {
rank=6;/*D*/
} else
if (checkRank(cRange,arr)) {
rank=5;/*C*/
} else
if (checkRank(bRange,arr)) {
rank=4;/*B*/
} else
if (checkRank(aRange,arr)) {
rank=3;/*A*/
} else
if (sssRange.colorInRange(col)) {
rank=0;/*SSS*/
} else
if (ssRange.colorInRange(col2)) {
rank=1;/*SS*/
} else
if (checkRank(sRange,arr)) {
rank=2;/*S*/
}
}
String val = interpretImage(subRegion,i);
sig_data[counter++]=val;
//System.out.println(i+": "+currentHeight+"-"+(currentHeight+readRegions.get(i).h+REGION_PADDING));
currentHeight+=readRegions.get(i).h+REGION_PADDING;
}
Path output = Paths.get("result.png");
ImageIO.write(cutImg,"png",output.toFile());
String dataString = readAllBoxes(output);
String[] data = dataString.split(Pattern.quote("\n"));
String[] ja_data = parseOutCommas(data[0]).split(Pattern.quote(","));
String[] en_data = parseOutCommas(data[2]).split(Pattern.quote(","));
trimAllData(ja_data);
trimAllData(en_data);
System.out.println(Arrays.toString(ja_data));
System.out.println(Arrays.toString(en_data));
System.out.println(Arrays.toString(sig_data));
interpretOutput(ja_data,en_data,sig_data);
g.dispose();
} catch (IOException e) {
e.printStackTrace();
}
//System.out.println(data[0]);
//System.out.println(data[2]);
}
private String interpretImage(BufferedImage cutImg, int i) {
switch (i) {
case 0:
case 10:{
Font f = Font.FONT_LOVELIVE_SCORE;
List<Glyph> glyphs = Glyph.split(cutImg);
return f.convertGlyphs(glyphs);
}
case 2:
case 3:
case 4:
case 5:
case 6:{
Font f = Font.FONT_LOVELIVE_NOTECOUNT;
List<Glyph> glyphs = Glyph.split(cutImg);
return f.convertGlyphs(glyphs);
}
case 9:{
Font f = Font.FONT_LOVELIVE_PCT;
List<Glyph> glyphs = Glyph.split(cutImg);
glyphs.remove(glyphs.size()-3);
glyphs.remove(glyphs.size()-1);
String res = f.convertGlyphs(glyphs,true);
return res.substring(0,glyphs.size()-1)+"."+res.substring(glyphs.size()-1,glyphs.size());
}
}
return "";
}
String interpretResults(String[] finalData) {
for (int i=0;i<finalData.length;i++) {
String[] splitter = finalData[i].split(Pattern.quote("\n"));
switch (i) {
case 0:{
score=convertToInt(splitter);
}break;
case 1:{}break; //We will handle the grade with another system.
case 2:{
notes[0]=convertToInt("PERFECT",splitter);
}break;
case 3:{
notes[1]=convertToInt("GREAT",splitter);
}break;
case 4:{
notes[2]=convertToInt("GOOD",splitter);
}break;
case 5:{
notes[3]=convertToInt("BAD",splitter);
}break;
case 6:{
notes[4]=convertToInt("MISS",splitter);
}break;
case 7:{
//difficulty=getDifficulty(convertToString(splitter).toLowerCase()); //I think we'll use the number instead. Assume CHALLENGE.
}break;
case 8:{
title=convertToString(splitter);
}break;
case 9:{
//pct=convertToPct(splitter); //No longer used.
pct=convertToDouble(splitter);
}break;
case 10:{
maxcombo=convertToInt(splitter);
}break;
case 11:{
difficulty=convertToInt(splitter);
}break;
}
}
return toString();
}
int getDifficulty(String str) {
final String[] diffs={"easy","normal","hard","master","challenge"};
for (int i=0;i<diffs.length;i++) {
String diff = diffs[i].toLowerCase();
if (str.contains(diff)) {
return i;
}
}
return -1;
}
}