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

347 lines
14 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(613,290,65,36)); //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;
}
}
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);
String val = interpretImage(subRegion,i);
if (!val.isEmpty()) {
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 = 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));
System.out.println(Arrays.toString(sig_data));
//interpretOutput(ja_data,en_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 "";
}
void 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());
}break;
case 8:{
title=convertToString(splitter);
}break;
case 9:{
pct=convertToPct(splitter);
}break;
case 10:{
maxcombo=convertToInt(splitter);
}break;
}
}
System.out.println(this);
}
double convertToPct(String[] str) {
int front=convertToInt(new String[]{str[1]});
int decimal=convertToInt(new String[]{str[2]});
double numb=0;
int placeValue=0;
int decimalPlaces=0;
while (decimal!=0) {
numb+=(decimal%10)*(Math.pow(10,placeValue++));
decimal/=10;
decimalPlaces++;
}
while (front!=0) {
numb+=(front%10)*(Math.pow(10,placeValue++));
front/=10;
}
for (int i=0;i<decimalPlaces;i++) {
numb/=10;
}
return numb;
}
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;
}
}