#include "renderer_light.hpp"

#include <functional>
#include <string>
#include <math.h>

#include "Types.h"

#include "modules/Gui.h"
#include "modules/Screen.h"
#include "modules/Maps.h"
#include "modules/Units.h"

#include "df/graphic.h"
#include "df/viewscreen_dwarfmodest.h"
#include "df/flow_info.h"
#include "df/world.h"
#include "df/building.h"
#include "df/building_doorst.h"
#include "df/plant.h"
#include "df/plant_raw.h"

using df::global::gps;
using namespace DFHack;
using df::coord2d;

const float levelDim=0.2f;

rect2d getMapViewport()
{
    const int AREA_MAP_WIDTH = 23;
    const int MENU_WIDTH = 30;
    if(!gps || !df::viewscreen_dwarfmodest::_identity.is_instance(DFHack::Gui::getCurViewscreen()))
        return mkrect_wh(0,0,0,0);
    int w=gps->dimx;
    int h=gps->dimy;
    int view_height=h-2;
    int area_x2 = w-AREA_MAP_WIDTH-2;
    int menu_x2=w-MENU_WIDTH-2;
    int menu_x1=area_x2-MENU_WIDTH-1;
    int view_rb=w-1;
    
    int area_pos=*df::global::ui_area_map_width;
    int menu_pos=*df::global::ui_menu_width;
    if(area_pos<3)
    {
        view_rb=area_x2;
    }
    if (menu_pos<area_pos || df::global::ui->main.mode!=0)
    {
        if (menu_pos >= area_pos) 
            menu_pos = area_pos-1;
        int menu_x = menu_x2;
        if(menu_pos < 2) menu_x = menu_x1;
        view_rb = menu_x;
    }
    return mkrect_wh(1,1,view_rb,view_height+1);
}
lightingEngineViewscreen::lightingEngineViewscreen(renderer_light* target):lightingEngine(target)
{
    reinit();
    initRawSpecific();
}

void lightingEngineViewscreen::reinit()
{
    if(!gps)
        return;
    w=gps->dimx;
    h=gps->dimy;
    size_t size=w*h;
    lightMap.resize(size,lightCell(1,1,1));
    ocupancy.resize(size);
    lights.resize(size);
}
void plotCircle(int xm, int ym, int r,std::function<void(int,int)> setPixel)
{
    int x = -r, y = 0, err = 2-2*r; /* II. Quadrant */ 
    do {
        setPixel(xm-x, ym+y); /*   I. Quadrant */
        setPixel(xm-y, ym-x); /*  II. Quadrant */
        setPixel(xm+x, ym-y); /* III. Quadrant */
        setPixel(xm+y, ym+x); /*  IV. Quadrant */
        r = err;
        if (r <= y) err += ++y*2+1;           /* e_xy+e_y < 0 */
        if (r > x || err > y) err += ++x*2+1; /* e_xy+e_x > 0 or no 2nd y-step */
    } while (x < 0);
}
void plotLine(int x0, int y0, int x1, int y1,std::function<bool(int,int,int,int)> setPixel)
{
    int dx =  abs(x1-x0), sx = x0<x1 ? 1 : -1;
    int dy = -abs(y1-y0), sy = y0<y1 ? 1 : -1; 
    int err = dx+dy, e2; /* error value e_xy */
    int rdx=0;
    int rdy=0;
    for(;;){  /* loop */
        if(!setPixel(rdx,rdy,x0,y0))
            return;
        if (x0==x1 && y0==y1) break;
        e2 = 2*err;
        rdx=rdy=0;
        if (e2 >= dy) { err += dy; x0 += sx; rdx=sx;} /* e_xy+e_x > 0 */
        if (e2 <= dx) { err += dx; y0 += sy; rdy=sy;} /* e_xy+e_y < 0 */
    }
}
lightCell blend(lightCell a,lightCell b)
{
    return lightCell(std::max(a.r,b.r),std::max(a.g,b.g),std::max(a.b,b.b));
}
bool lightingEngineViewscreen::lightUpCell(lightCell& power,int dx,int dy,int tx,int ty)
{
    
    if(tx>=mapPort.first.x && ty>=mapPort.first.y && tx<=mapPort.second.x && ty<=mapPort.second.y)
    {
        size_t tile=getIndex(tx,ty);
        float dsq=dx*dx+dy*dy;
        float dt=sqrt(dsq);
        lightCell& v=ocupancy[tile];
        lightSource& ls=lights[tile];
        bool wallhack=false;
        if(v.r+v.g+v.b==0)
            wallhack=true;
        
        if (dsq>0 && !wallhack)
        {
            power.r=power.r*(pow(v.r,dt));
            power.g=power.g*(pow(v.g,dt));
            power.b=power.b*(pow(v.b,dt));
        }
        if(ls.radius>0 && dsq>0)
        {
            if(power<ls.power)
                return false;
        }
        //float dt=sqrt(dsq);
        lightCell oldCol=lightMap[tile];
        lightCell ncol=blend(power,oldCol);
        lightMap[tile]=ncol;
        
        if(wallhack)
            return false;
        float pwsq=power.r*power.r+power.g*power.g+power.b*power.b;
        return pwsq>levelDim*levelDim;
    }
    else
        return false;
}
void lightingEngineViewscreen::doRay(lightCell power,int cx,int cy,int tx,int ty)
{
    using namespace std::placeholders;
    lightCell curPower=power;
    plotLine(cx,cy,tx,ty,std::bind(&lightingEngineViewscreen::lightUpCell,this,std::ref(curPower),_1,_2,_3,_4));
}
void lightingEngineViewscreen::doFovs()
{
    mapPort=getMapViewport();
    using namespace std::placeholders;

    for(int i=mapPort.first.x;i<mapPort.second.x;i++)
        for(int j=mapPort.first.y;j<mapPort.second.y;j++)
        {
            lightSource& csource=lights[getIndex(i,j)];
            if(csource.radius>0)
            {
                lightCell power=csource.power;
                int radius =csource.radius;
                if(csource.flicker)
                {
                    float flicker=(rand()/(float)RAND_MAX)/2.0f+0.5f;
                    radius*=flicker;
                    power=power*flicker;
                }
                int surrounds = 0;
                lightCell curPower;

                if(lightUpCell(curPower = power, 0, 0,i+0, j+0))
                {
                    surrounds += lightUpCell(curPower = power, 0, 1,i+0, j+1);
                    surrounds += lightUpCell(curPower = power, 1, 1,i+1, j+1);
                    surrounds += lightUpCell(curPower = power, 1, 0,i+1, j+0);
                    surrounds += lightUpCell(curPower = power, 1,-1,i+1, j-1);
                    surrounds += lightUpCell(curPower = power, 0,-1,i+0, j-1);
                    surrounds += lightUpCell(curPower = power,-1,-1,i-1, j-1);
                    surrounds += lightUpCell(curPower = power,-1, 0,i-1, j+0);
                    surrounds += lightUpCell(curPower = power,-1, 1,i-1, j+1);
                }
                if(surrounds)
                    plotCircle(i,j,radius,
                    std::bind(&lightingEngineViewscreen::doRay,this,power,i,j,_1,_2));
            }
        }
}
void lightingEngineViewscreen::calculate()
{
    rect2d vp=getMapViewport();
    const lightCell dim(levelDim,levelDim,levelDim);
    lightMap.assign(lightMap.size(),lightCell(1,1,1));
    lights.assign(lights.size(),lightSource());
    for(int i=vp.first.x;i<vp.second.x;i++)
    for(int j=vp.first.y;j<vp.second.y;j++)
    {
        lightMap[getIndex(i,j)]=dim;
    }
    doOcupancyAndLights();
    doFovs();
    //for each lightsource in viewscreen+x do light
}
void lightingEngineViewscreen::updateWindow()
{
    tthread::lock_guard<tthread::fast_mutex> guard(myRenderer->dataMutex);
    if(lightMap.size()!=myRenderer->lightGrid.size())
    {
        reinit();
        myRenderer->invalidate();
        return;
    }
    std::swap(lightMap,myRenderer->lightGrid);
    rect2d vp=getMapViewport();
    
    //myRenderer->invalidateRect(vp.first.x,vp.first.y,vp.second.x-vp.first.x,vp.second.y-vp.first.y);
    myRenderer->invalidate();
    //std::copy(lightMap.begin(),lightMap.end(),myRenderer->lightGrid.begin());
}
void lightSource::combine(const lightSource& other)
{
    power=blend(power,other.power);
    radius=std::max(other.radius,radius);//hack... but who cares
}
bool lightingEngineViewscreen::addLight(int tileId,const lightSource& light)
{
    bool wasLight=lights[tileId].radius>0;
    lights[tileId].combine(light);
    if(light.flicker)
        lights[tileId].flicker=true;
    return wasLight;
}
lightCell getStandartColor(int colorId)
{
    return lightCell(df::global::enabler->ccolor[colorId][0]/255.0f,
        df::global::enabler->ccolor[colorId][1]/255.0f,
        df::global::enabler->ccolor[colorId][2]/255.0f);
}
int getPlantNumber(const std::string& id)
{
    std::vector<df::plant_raw*>& vec=df::plant_raw::get_vector();
    for(int i=0;i<vec.size();i++)
    {
        if(vec[i]->id==id)
            return i;
    }
    return -1;
}
void addPlant(const std::string& id,std::map<int,lightSource>& map,const lightSource& v)
{
    int nId=getPlantNumber(id);
    if(nId>0)
    {
        map[nId]=v;
    }
}
void lightingEngineViewscreen::initRawSpecific()
{   
    addPlant("TOWER_CAP",glowPlants,lightSource(lightCell(0.65,0.65,0.65),6));
    addPlant("MUSHROOM_CUP_DIMPLE",glowPlants,lightSource(lightCell(0.03,0.03,0.5),3));
    addPlant("CAVE MOSS",glowPlants,lightSource(lightCell(0.1,0.1,0.4),2));
    addPlant("MUSHROOM_HELMET_PLUMP",glowPlants,lightSource(lightCell(0.2,0.1,0.6),2));
}
static size_t max_list_size = 100000; // Avoid iterating over huge lists
void lightingEngineViewscreen::doOcupancyAndLights()
{
    // TODO better curve (+red dawn ?)
    float daycol = abs((*df::global::cur_year_tick % 1200) - 600.0) / 600.0;
    lightCell sky_col(daycol, daycol, daycol);
    lightSource sky(sky_col, 15);

    lightSource lava(lightCell(0.8f,0.2f,0.2f),5);
    lightSource candle(lightCell(0.96f,0.84f,0.03f),5);
    lightSource torch(lightCell(0.9f,0.75f,0.3f),8);
    rect2d vp=getMapViewport();

    int window_x=*df::global::window_x;
    int window_y=*df::global::window_y;
    int window_z=*df::global::window_z;
    int vpW=vp.second.x-vp.first.x;
    int vpH=vp.second.y-vp.first.y;
    int endBlockx = (window_x+vpW)/16;
    int endBlocky = (window_y+vpH)/16;
    if(endBlockx >= df::global::world->map.x_count_block) endBlockx = df::global::world->map.x_count_block-1;
    if(endBlocky >= df::global::world->map.y_count_block) endBlocky = df::global::world->map.y_count_block-1;
    for(int blockx=window_x/16;blockx<=endBlockx;blockx++)
    for(int blocky=window_y/16;blocky<=endBlocky;blocky++)
    {

        lightCell cellArray[16][16];
        for(int block_x = 0; block_x < 16; block_x++)
        for(int block_y = 0; block_y < 16; block_y++)
        {
            cellArray[block_x][block_y] = sky_col;
        }
        int totalBlank = 0;
        int topLevel = df::global::world->map.z_count-1;
        for(int ZZ = topLevel; (ZZ >= window_z) && totalBlank < 256; ZZ--)
        {
            df::map_block* block=Maps::getBlock(blockx,blocky,ZZ);
            totalBlank = 0;
            if(block)
            for(int block_x = 0; block_x < 16; block_x++)
            for(int block_y = 0; block_y < 16; block_y++)
            {
                df::tiletype type = block->tiletype[block_x][block_y];
                df::tile_designation d = block->designation[block_x][block_y];
                df::tile_occupancy o = block->occupancy[block_x][block_y];
                df::tiletype_shape shape = ENUM_ATTR(tiletype,shape,type);
                df::tiletype_shape_basic basic_shape = ENUM_ATTR(tiletype_shape, basic_shape, shape);

                if(basic_shape==df::tiletype_shape_basic::Wall)
                {
                    cellArray[block_x][block_y]=lightCell(0,0,0);
                }
                else if(basic_shape==df::tiletype_shape_basic::Floor || basic_shape==df::tiletype_shape_basic::Ramp || shape==df::tiletype_shape::STAIR_UP)
                {
                    if(ZZ!=window_z)
                    {
                        cellArray[block_x][block_y]=lightCell(0,0,0);
                    }
                }
                else if(shape==df::tiletype_shape::STAIR_DOWN || shape==df::tiletype_shape::STAIR_UPDOWN)
                {
                    cellArray[block_x][block_y]*=lightCell(0.9,0.9,0.9);
                }
                if(d.bits.liquid_type == df::enums::tile_liquid::Water && d.bits.flow_size)
                {
                    cellArray[block_x][block_y] *= (lightCell(1,1,1) - (lightCell(1,1,1) - lightCell(0.63f,0.63f,0.75f))*(d.bits.flow_size/7));
                }
                else if(d.bits.liquid_type == df::enums::tile_liquid::Magma && d.bits.flow_size > 3)
                {
                    cellArray[block_x][block_y]=lightCell(0,0,0);
                }
                if(cellArray[block_x][block_y].r < 0.003f && cellArray[block_x][block_y].g < 0.003f && cellArray[block_x][block_y].b < 0.003f)
                    totalBlank++;
            }
        }
        for(int block_x = 0; block_x < 16; block_x++)
        for(int block_y = 0; block_y < 16; block_y++)
        {
            df::coord2d pos;
            pos.x = blockx*16+block_x;
            pos.y = blocky*16+block_y;
            int wx=pos.x-window_x+vp.first.x;
            int wy=pos.y-window_y+vp.first.y;
            if(wx>=vp.first.x && wy>=vp.first.y && wx<=vp.second.x && wy<=vp.second.y)
            if(cellArray[block_x][block_y].r >= 0.003f && cellArray[block_x][block_y].g >= 0.003f && cellArray[block_x][block_y].b >= 0.003f)
            {
                lightSource sun=lightSource(cellArray[block_x][block_y],15);
                addLight(getIndex(wx,wy),sun);
            }
        }

    }
    for(int x=window_x;x<window_x+vpW;x++)
    for(int y=window_y;y<window_y+vpH;y++)
    {
        int wx=x-window_x+vp.first.x;
        int wy=y-window_y+vp.first.y;
        int tile=getIndex(wx,wy);
        lightCell& curCell=ocupancy[tile];
        curCell=lightCell(0.85f,0.85f,0.85f);
        df::tiletype* type = Maps::getTileType(x,y,window_z);
        if(!type)
        {
            //unallocated, do sky
            addLight(tile,sky);
            continue;
        }
        df::tiletype_shape shape = ENUM_ATTR(tiletype,shape,*type);
        df::tile_designation* d=Maps::getTileDesignation(x,y,window_z);
        df::tile_designation* d2=Maps::getTileDesignation(x,y,window_z-1);
        df::tile_occupancy* o=Maps::getTileOccupancy(x,y,window_z);
        df::tiletype_material m=ENUM_ATTR(tiletype,material,*type);
        if(!o || !d )
            continue;
        if(shape==df::tiletype_shape::BROOK_BED || shape==df::tiletype_shape::WALL || shape==df::tiletype_shape::TREE || d->bits.hidden )
        {
            if(shape==df::tiletype_shape::WALL && m==df::tiletype_material::FROZEN_LIQUID)
                curCell*=lightCell(0.7,0.7,0.9);
            else
                curCell=lightCell(0,0,0);
        }
        else if(o->bits.building)
        {
            // Fixme: don't iterate the list every frame
            size_t count = df::global::world->buildings.all.size();
            if (count <= max_list_size)
            {
                for(size_t i = 0; i < count; i++)
                {
                    df::building *bld = df::global::world->buildings.all[i];

                    if (window_z == bld->z && 
                        x >= bld->x1 && x <= bld->x2 &&
                        y >= bld->y1 && y <= bld->y2)
                    {
                        df::building_type type = bld->getType();

                        if (type == df::enums::building_type::WindowGlass)
                        {
                            if(bld->mat_type == 3)//green glass
                            {
                                curCell*=lightCell(0.1f,0.9f,0.5f);
                            }
                            else if(bld->mat_type == 4)//clear glass
                            {
                                curCell*=lightCell(0.5f,0.95f,0.9f);
                            }
                            else if(bld->mat_type == 5)//crystal glass
                            {
                                curCell*=lightCell(0.75f,0.95f,0.95f);
                            }
                        }
                        if (type == df::enums::building_type::Table)
                        {
                            addLight(tile,candle);
                        }
                        if (type==df::enums::building_type::Statue)
                        {
                            addLight(tile,torch);
                        }
                        if (type==df::enums::building_type::WindowGem)
                        {
                            DFHack::MaterialInfo mat(bld->mat_index,bld->mat_type);
                            if(mat.isInorganic())
                            {
                                int color=mat.inorganic->material.basic_color[0]+8*mat.inorganic->material.basic_color[2];
                                curCell*=getStandartColor(color);
                            }
                        }
                        if(type==df::enums::building_type::Door)
                        {
                            df::building_doorst* door=static_cast<df::building_doorst*>(bld);
                            if(door->door_flags.bits.closed)
                                curCell*=lightCell(0,0,0);
                        }
                    }
                }
            }
        }
        else if(!d->bits.liquid_type && d->bits.flow_size>3 )
        {
            curCell*=lightCell(0.7f,0.7f,0.8f);
        }
        //lights
        if((d->bits.liquid_type && d->bits.flow_size>0)|| 
            (
            (shape==df::tiletype_shape::EMPTY || shape==df::tiletype_shape::RAMP_TOP || shape==df::tiletype_shape::STAIR_DOWN || shape==df::tiletype_shape::STAIR_UPDOWN )
            && d2 && d2->bits.liquid_type && d2->bits.flow_size>0)
            )
        {
            
            addLight(tile,lava);
        }
        if(d->bits.outside && d->bits.flow_size==0)
        {
            addLight(tile,sky);
        }
        
    }
    for(int blockx=window_x/16;blockx<=endBlockx;blockx++)
    for(int blocky=window_y/16;blocky<=endBlocky;blocky++)
    {
        df::map_block* block=Maps::getBlock(blockx,blocky,window_z);
        
        if(!block)
            continue;
        for(int i=0;i<block->flows.size();i++)
        {
            df::flow_info* f=block->flows[i];
            if(f && f->density>0 && f->type==df::flow_type::Dragonfire || f->type==df::flow_type::Fire)
            {
                df::coord2d pos=f->pos;
                int wx=pos.x-window_x+vp.first.x;
                int wy=pos.y-window_y+vp.first.y;
                int tile=getIndex(wx,wy);
                if(wx>=vp.first.x && wy>=vp.first.y && wx<=vp.second.x && wy<=vp.second.y)
                {
                    lightCell fireColor;
                    if(f->density>60)
                    {
                        fireColor=lightCell(0.98f,0.91f,0.30f);
                    }
                    else if(f->density>30)
                    {
                        fireColor=lightCell(0.93f,0.16f,0.16f);
                    }
                    else
                    {
                        fireColor=lightCell(0.64f,0.0f,0.0f);
                    }
                    lightSource fire(fireColor,f->density/5);
                    addLight(tile,fire);
                }
            }
        }
        for(int i=0;i<block->plants.size();i++)
        {
            df::plant* cPlant=block->plants[i];
            
            df::coord2d pos=cPlant->pos;
            int wx=pos.x-window_x+vp.first.x;
            int wy=pos.y-window_y+vp.first.y;
            int tile=getIndex(wx,wy);
            if(wx>=vp.first.x && wy>=vp.first.y && wx<=vp.second.x && wy<=vp.second.y)
            {
                auto it=glowPlants.find(cPlant->material);
                if(it!=glowPlants.end())
                {
                    addLight(tile,it->second);
                }
            }
        }
    }
    if(df::global::cursor->x>-30000)
    {
        lightSource cursor(lightCell(0.96f,0.84f,0.03f),11);
        cursor.flicker=true;
        int wx=df::global::cursor->x-window_x+vp.first.x;
        int wy=df::global::cursor->y-window_y+vp.first.y;
        int tile=getIndex(wx,wy);
        addLight(tile,cursor);
    }
    lightSource citizen(lightCell(0.80f,0.80f,0.90f),6);
    for (int i=0;i<df::global::world->units.active.size();++i)
    {
        df::unit *u = df::global::world->units.active[i];
        if (u->pos.z != window_z ||
                (u->pos.x < window_x || u->pos.x >= window_x+vpW) ||
                (u->pos.y < window_y || u->pos.y >= window_y+vpH))
            continue;
        if (DFHack::Units::isCitizen(u))
            addLight(getIndex(u->pos.x-window_x+1, u->pos.y-window_y+1),citizen);
    }
}