#include "artclass.h"
#include <bits/stdc++.h>
using namespace std;
int avg(int a[][500], int x1=0, int y1=0, int x2=500, int y2=500){
int tot = 0;
for(int i = x1; i < x2; i++)
for(int j = y1; j < y2; j++)
tot+=a[i][j];
return tot/((x2-x1+1)*(y2-y1+1));
}
int cnt(int a[][500], int b[][500], int c[][500], int thl=0, int thr=255, int x1=0, int y1=0, int x2=500, int y2=500){
int tot = 0, x;
for(int i = x1; i < x2; i++)
for(int j = y1; j < y2; j++)
x = (a[i][j]+b[i][j]+c[i][j])/3, tot+=(x>=thl and x<=thr);
return tot;
}
int style(int H, int W, int R[500][500], int G[500][500], int B[500][500]) {
int r = avg(R), g = avg(G), b = avg(B);
int rt = avg(R,0,0,500,250), gt = avg(G,0,0,500,250), bt = avg(B,0,0,500,250);
int rb = avg(R,0,250,500,500), gb = avg(G,0,250,500,500), bb = avg(B,0,250,500,500);
int th = 125, nth = 100000;
if(avg(G,0,150,500,500)>=th) return 2; // average green from a third of the picture is high?
if(cnt(R,G,B,230,255) >= nth) return 1; // lots of white pixels
if(cnt(R,G,B,0,80) >= nth) return 1; // lots of dark pixels
if(rb-rt<=th and gb-gt<=th and bb-bt<=th) return 4; //top half of painting is near color of bottom half
if(r>=th and g>=th and b>=th) return 4; // lighter in general
if(r>=th and b>=th) return 3; //average color is close to brown? idk
if((r+g+b)/3>=th){ // lightish?
if(rand()%2) return 1;
return 4;
}
if(rand()%2) return 2;
return 3;
}
