704 lines
15 KiB
C++
704 lines
15 KiB
C++
#include<hgl/type/StringList.h>
|
||
#include<hgl/type/LoadStringList.h>
|
||
#include<hgl/type/Gradient.h>
|
||
#include<hgl/math/Vector.h>
|
||
#include<hgl/util/imgfmt/tga.h>
|
||
#include<hgl/io/FileInputStream.h>
|
||
#include<hgl/io/FileOutputStream.h>
|
||
#include<hgl/filesystem/Filename.h>
|
||
#include<iostream>
|
||
#include"BitmapFont.h"
|
||
|
||
using namespace hgl;
|
||
|
||
OSString csv_filename;
|
||
|
||
struct Bitmap
|
||
{
|
||
uint width,height;
|
||
uint channels;
|
||
|
||
uint8 *data;
|
||
|
||
public:
|
||
|
||
Bitmap()
|
||
{
|
||
data=nullptr;
|
||
}
|
||
|
||
~Bitmap()
|
||
{
|
||
delete[] data;
|
||
}
|
||
};
|
||
|
||
Bitmap *BackgroundBitmap=nullptr;
|
||
|
||
bool LoadBackgroundBitmap()
|
||
{
|
||
io::OpenFileInputStream fis(OS_TEXT("mini_map.tga"));
|
||
|
||
if(!fis)
|
||
return(false);
|
||
|
||
util::TGAHeader tga_header;
|
||
util::TGAImageDesc tga_desc;
|
||
|
||
fis->Read(&tga_header,sizeof(util::TGAHeader));
|
||
|
||
if(tga_header.image_type!=util::TGA_IMAGE_TYPE_TRUE_COLOR)
|
||
return(false);
|
||
|
||
if(tga_header.bit!=24)
|
||
return(false);
|
||
|
||
tga_desc.image_desc=tga_header.image_desc;
|
||
|
||
BackgroundBitmap=new Bitmap;
|
||
|
||
BackgroundBitmap->width=tga_header.width;
|
||
BackgroundBitmap->height=tga_header.height;
|
||
BackgroundBitmap->channels=tga_header.bit/8;
|
||
|
||
uint total_bytes=BackgroundBitmap->width*BackgroundBitmap->height*BackgroundBitmap->channels;
|
||
|
||
BackgroundBitmap->data=new uint8[total_bytes];
|
||
|
||
if(fis->Read(BackgroundBitmap->data,total_bytes)!=total_bytes)
|
||
{
|
||
delete BackgroundBitmap;
|
||
BackgroundBitmap=nullptr;
|
||
return(false);
|
||
}
|
||
|
||
if(tga_desc.direction==util::TGA_DIRECTION_LOWER_LEFT)
|
||
{
|
||
uint line_bytes=tga_header.width*3;
|
||
|
||
uint8 *temp=new uint8[line_bytes];
|
||
uint8 *tp=BackgroundBitmap->data;
|
||
uint8 *bp=tp+line_bytes*(tga_header.height-1);
|
||
|
||
while(tp<bp)
|
||
{
|
||
memcpy(temp,tp,line_bytes);
|
||
memcpy(tp,bp,line_bytes);
|
||
memcpy(bp,temp,line_bytes);
|
||
|
||
tp+=line_bytes;
|
||
bp-=line_bytes;
|
||
}
|
||
}
|
||
|
||
return(true);
|
||
}
|
||
|
||
uint CHAR_BITMAP_WIDTH=0;
|
||
uint CHAR_BITMAP_HEIGHT=0;
|
||
|
||
bool InitBitmapFont()
|
||
{
|
||
if(!LoadBitmapFont())
|
||
return(false);
|
||
|
||
CHAR_BITMAP_WIDTH=GetCharWidth();
|
||
CHAR_BITMAP_HEIGHT=GetCharHeight();
|
||
|
||
return(true);
|
||
}
|
||
|
||
constexpr const float LOW_GAP=0.2f;
|
||
constexpr const Vector3u8 black_color={0,0,0};
|
||
constexpr const Vector3u8 white_color={255,255,255};
|
||
|
||
constexpr const Vector3u8 stop_color[]=
|
||
{
|
||
{255,0,0},
|
||
{255,255,0},
|
||
{0,255,0},
|
||
{0,255,255},
|
||
{0,0,255}
|
||
};
|
||
|
||
constexpr const uint STOP_COUNT=sizeof(stop_color)/sizeof(Vector3u8);
|
||
|
||
HGL_GRADIENT_DEFINE(GradientColor3u8,uint,Vector3u8)
|
||
{
|
||
result.r=start.r+float(end.r-start.r)*pos;
|
||
result.g=start.g+float(end.g-start.g)*pos;
|
||
result.b=start.b+float(end.b-start.b)*pos;
|
||
}
|
||
|
||
GradientColor3u8 ColorGradient;
|
||
|
||
void InitGradient(uint max_count)
|
||
{
|
||
for(uint i=0;i<STOP_COUNT;i++)
|
||
ColorGradient.Add(max_count*(1.0-float(i)/float(STOP_COUNT-1)),stop_color[i]);
|
||
}
|
||
|
||
bool ParseLine(Vector2i *result,const UTF8String &str)
|
||
{
|
||
if(!result)return(false);
|
||
|
||
if(str.Length()<=0)return(false);
|
||
|
||
const char *sp=str.c_str();
|
||
const char *cp;
|
||
|
||
if(*sp!='X')return(false);
|
||
|
||
sp+=2;
|
||
cp=hgl::strchr(sp,'Y');
|
||
|
||
if(!cp)return(false);
|
||
|
||
if(!hgl::stoi(sp,result->x))
|
||
return(false);
|
||
|
||
sp=cp+2;
|
||
|
||
cp=hgl::strchr(sp,'Z');
|
||
|
||
if(!cp)return(false);
|
||
|
||
if(!hgl::stoi(sp,result->y))
|
||
return(false);
|
||
|
||
return(true);
|
||
}
|
||
|
||
struct PositionStat
|
||
{
|
||
uint count;
|
||
|
||
Vector2i minp,maxp;
|
||
Vector2i gap;
|
||
|
||
Vector2i *data;
|
||
|
||
public:
|
||
|
||
PositionStat(const uint c)
|
||
{
|
||
count=c;
|
||
|
||
hgl_zero(minp);
|
||
hgl_zero(maxp);
|
||
hgl_zero(gap);
|
||
|
||
data=new Vector2i[count];
|
||
|
||
hgl_zero(data,count);
|
||
}
|
||
|
||
~PositionStat()
|
||
{
|
||
delete[] data;
|
||
}
|
||
};
|
||
|
||
PositionStat *ToVector2i(const UTF8StringList &sl)
|
||
{
|
||
const uint count=sl.GetCount();
|
||
|
||
PositionStat *ps=new PositionStat(count);
|
||
|
||
Vector2i *p=ps->data;
|
||
|
||
UTF8String str;
|
||
uint result=0;
|
||
|
||
for(int i=0;i<count;i++)
|
||
{
|
||
str=sl[i];
|
||
|
||
if(str.Length()<=0)
|
||
continue;
|
||
|
||
if(!ParseLine(p,str))
|
||
continue;
|
||
|
||
(*p)/=100; //Unreal单位为cm,把单位缩到米
|
||
(*p)/=4;
|
||
|
||
if(p->x>=BackgroundBitmap->width
|
||
||p->y>=BackgroundBitmap->height)
|
||
continue;
|
||
|
||
//std::cout<<"X="<<p->x<<",Y="<<p->y<<std::endl;
|
||
|
||
// if(p->x<ps->minp.x)ps->minp.x=p->x;
|
||
// if(p->x>ps->maxp.x)ps->maxp.x=p->x;
|
||
// if(p->y<ps->minp.y)ps->minp.y=p->y;
|
||
// if(p->y>ps->maxp.y)ps->maxp.y=p->y;
|
||
|
||
++result;
|
||
++p;
|
||
}
|
||
|
||
//std::cout<<"minp: "<<ps->minp.x<<","<<ps->minp.y<<std::endl;
|
||
//std::cout<<"maxp: "<<ps->maxp.x<<","<<ps->maxp.y<<std::endl;
|
||
|
||
ps->maxp.x=1023;
|
||
ps->maxp.y=1023;
|
||
|
||
ps->count=result;
|
||
|
||
return ps;
|
||
}
|
||
|
||
struct Chart
|
||
{
|
||
uint width,height;
|
||
|
||
uint max_count;
|
||
|
||
uint32 *count_data;
|
||
uint32 *circle_data;
|
||
uint8 *chart_data;
|
||
|
||
public:
|
||
|
||
Chart(const uint w,const uint h)
|
||
{
|
||
width=w;
|
||
height=h;
|
||
|
||
count_data=new uint32[width*height];
|
||
circle_data=new uint32[width*height];
|
||
chart_data=new uint8[width*height*4];
|
||
|
||
hgl_zero(count_data,width*height);
|
||
hgl_zero(circle_data,width*height);
|
||
hgl_zero(chart_data,width*height*4);
|
||
|
||
max_count=0;
|
||
}
|
||
|
||
~Chart()
|
||
{
|
||
delete[] count_data;
|
||
delete[] circle_data;
|
||
delete[] chart_data;
|
||
}
|
||
|
||
void DrawCircle(uint x,uint y,uint radius)
|
||
{
|
||
uint r2=radius*radius;
|
||
uint length;
|
||
|
||
for(int col=x-radius;col<=x+radius;col++)
|
||
{
|
||
if(col<0||col>=width)continue;
|
||
|
||
for(int row=y-radius;row<=y+radius;row++)
|
||
{
|
||
if(row<0||row>=height)continue;
|
||
|
||
length=(col-x)*(col-x)+(row-y)*(row-y);
|
||
|
||
if(length<=r2)
|
||
{
|
||
++circle_data[col+row*width];
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
void DrawBar(const uint x,const uint y,const uint size,const Vector3u8 &stop_color,const uint8 alpha)
|
||
{
|
||
uint8 *tp=chart_data+(x+y*width)*4;
|
||
uint line_bytes=(width-size)*4;
|
||
|
||
for(uint row=0;row<size;row++)
|
||
{
|
||
for(uint col=0;col<size;col++)
|
||
{
|
||
tp[0]=stop_color.b;
|
||
tp[1]=stop_color.g;
|
||
tp[2]=stop_color.r;
|
||
tp[3]=alpha;
|
||
|
||
tp+=4;
|
||
}
|
||
|
||
tp+=line_bytes;
|
||
}
|
||
}
|
||
|
||
void DrawChar(const char ch,const uint x,const uint y,const Vector3u8 &stop_color,const uint8 alpha)
|
||
{
|
||
const uint8 *sp=GetBitmapChar(ch);
|
||
uint8 bit;
|
||
uint8 *tp=chart_data+(x+y*width)*4;
|
||
|
||
uint line_wrap_bytes=(width-CHAR_BITMAP_WIDTH)*4;
|
||
|
||
for(uint row=0;row<CHAR_BITMAP_HEIGHT;row++)
|
||
{
|
||
bit=1<<7;
|
||
|
||
for(uint col=0;col<CHAR_BITMAP_WIDTH;col++)
|
||
{
|
||
if(*sp&bit)
|
||
{
|
||
tp[0]=stop_color.b;
|
||
tp[1]=stop_color.g;
|
||
tp[2]=stop_color.r;
|
||
tp[3]=alpha;
|
||
}
|
||
|
||
tp+=4;
|
||
|
||
bit>>=1;
|
||
}
|
||
|
||
tp+=line_wrap_bytes;
|
||
++sp;
|
||
}
|
||
}
|
||
|
||
void DrawString(const AnsiString &str,const uint x,const uint y,const Vector3u8 &stop_color,const uint8 alpha)
|
||
{
|
||
const char *sp=str.c_str();
|
||
const uint len=str.Length();
|
||
|
||
uint pos=x;
|
||
|
||
for(uint i=0;i<len;i++)
|
||
{
|
||
if(*sp!=' ')
|
||
DrawChar(*sp,pos,y,stop_color,alpha);
|
||
|
||
pos+=CHAR_BITMAP_WIDTH;
|
||
++sp;
|
||
}
|
||
}
|
||
|
||
void DrawGradient(const uint left,const uint top,const uint w,const uint h)
|
||
{
|
||
uint8 *tp=chart_data+(left+top*width)*4;
|
||
Vector3u8 color;
|
||
|
||
uint low,high,gap;
|
||
|
||
ColorGradient.GetLowest(low);
|
||
ColorGradient.GetHighest(high);
|
||
|
||
gap=high-low;
|
||
|
||
for(uint i=0;i<h;i++)
|
||
{
|
||
ColorGradient.Get(color,(1.0f-float(i)/float(h))*float(gap)+float(low));
|
||
|
||
for(uint j=0;j<w;j++)
|
||
{
|
||
tp[0]=color.b;
|
||
tp[1]=color.g;
|
||
tp[2]=color.r;
|
||
tp[3]=255;
|
||
|
||
tp+=4;
|
||
}
|
||
|
||
tp+=(width-w)*4;
|
||
}
|
||
}
|
||
};
|
||
|
||
Chart *ToChart32(const PositionStat *ps)
|
||
{
|
||
uint width=BackgroundBitmap->width;
|
||
uint height=BackgroundBitmap->height;
|
||
|
||
std::cout<<"width: "<<width<<",height: "<<height<<std::endl;
|
||
|
||
Chart *chart=new Chart(width,height);
|
||
uint max_count=0;
|
||
uint step_count[STOP_COUNT];
|
||
|
||
//统计每个格子数据数量
|
||
{
|
||
uint x,y;
|
||
uint32 *cp32=chart->count_data;
|
||
|
||
const Vector2i *p=ps->data;
|
||
|
||
for(uint i=0;i<ps->count;i++)
|
||
{
|
||
x=p->x-ps->minp.x;
|
||
y=p->y-ps->minp.y;
|
||
|
||
cp32=chart->count_data+(x+y*width);
|
||
|
||
++(*cp32);
|
||
|
||
if(*cp32>max_count)max_count=*cp32;
|
||
|
||
++p;
|
||
}
|
||
}
|
||
|
||
//统计占比
|
||
{
|
||
uint32 *cp32=chart->count_data;
|
||
|
||
hgl_zero(step_count);
|
||
|
||
for(uint y=0;y<height;y++)
|
||
{
|
||
for(uint x=0;x<width;x++)
|
||
{
|
||
if(*cp32>0)
|
||
for(uint i=0;i<STOP_COUNT;i++)
|
||
if(*cp32>max_count*(STOP_COUNT-1-i)/STOP_COUNT)
|
||
{
|
||
step_count[i]+=*cp32;
|
||
break;
|
||
}
|
||
|
||
++cp32;
|
||
}
|
||
}
|
||
}
|
||
|
||
//画圆
|
||
{
|
||
uint32 *cp32=chart->count_data;
|
||
|
||
for(uint y=0;y<height;y++)
|
||
{
|
||
for(uint x=0;x<width;x++)
|
||
{
|
||
if(*cp32>0)
|
||
chart->DrawCircle(x,y,(*cp32));
|
||
|
||
++cp32;
|
||
}
|
||
}
|
||
}
|
||
|
||
//统计最大值
|
||
{
|
||
uint32 *cp32=chart->circle_data;
|
||
|
||
for(uint i=0;i<width*height;i++)
|
||
{
|
||
if(*cp32>chart->max_count)chart->max_count=*cp32;
|
||
|
||
++cp32;
|
||
}
|
||
|
||
std::cout<<"max_count: "<<chart->max_count<<std::endl;
|
||
}
|
||
|
||
InitGradient(chart->max_count);
|
||
|
||
//生成权重图
|
||
{
|
||
uint32 *cp32=chart->circle_data;
|
||
uint8 *cp8=chart->chart_data;
|
||
|
||
float alpha;
|
||
Vector3u8 final_color;
|
||
|
||
for(uint i=0;i<width*height;i++)
|
||
{
|
||
alpha=float(*cp32)/float(chart->max_count);
|
||
|
||
ColorGradient.Get(final_color,*cp32);
|
||
|
||
if(*cp32>0) //为了避免最后什么都看不见，所以把没数据的挑出来，剩下的透明度全部加0.25
|
||
{
|
||
alpha+=LOW_GAP;
|
||
|
||
if(alpha>1)
|
||
alpha=1;
|
||
}
|
||
|
||
cp8[0]=final_color.b;
|
||
cp8[1]=final_color.g;
|
||
cp8[2]=final_color.r;
|
||
cp8[3]=alpha*255.0f;
|
||
|
||
++cp32;
|
||
cp8+=4;
|
||
}
|
||
}
|
||
|
||
if(CHAR_BITMAP_HEIGHT==0)
|
||
return chart;
|
||
|
||
//写入数值
|
||
{
|
||
uint col=10;
|
||
uint row=10;
|
||
uint stop_str_width=0;
|
||
|
||
AnsiString str;
|
||
AnsiString num_str;
|
||
const AnsiString str_total=AnsiString::numberOf(ps->count);
|
||
|
||
AnsiString step_str[STOP_COUNT];
|
||
const uint dradient_bar_height=CHAR_BITMAP_HEIGHT*STOP_COUNT;
|
||
|
||
char space[32];
|
||
|
||
memset(space,' ',32);
|
||
|
||
for(uint i=0;i<STOP_COUNT;i++)
|
||
{
|
||
step_str[i]=AnsiString::numberOf(uint(max_count*(1.0-float(i)/float(STOP_COUNT))));
|
||
|
||
if(stop_str_width<step_str[i].Length())
|
||
stop_str_width=step_str[i].Length();
|
||
}
|
||
|
||
str="Source: "+ToAnsiString(csv_filename);
|
||
|
||
chart->DrawString(str,col,row,black_color,255);
|
||
|
||
row+=CHAR_BITMAP_HEIGHT*2;
|
||
|
||
str=AnsiString("Total: ")+str_total;
|
||
|
||
chart->DrawString(str,col,row,white_color,255);
|
||
row+=CHAR_BITMAP_HEIGHT;
|
||
|
||
chart->DrawGradient(col,row,CHAR_BITMAP_WIDTH,dradient_bar_height);
|
||
|
||
col+=CHAR_BITMAP_WIDTH*2;
|
||
|
||
chart->DrawGradient(col+(stop_str_width+1)*CHAR_BITMAP_WIDTH,
|
||
row,CHAR_BITMAP_WIDTH,dradient_bar_height);
|
||
|
||
chart->DrawGradient(col+(str_total.Length()+stop_str_width+4)*CHAR_BITMAP_WIDTH,
|
||
row,CHAR_BITMAP_WIDTH,dradient_bar_height);
|
||
|
||
for(uint i=0;i<STOP_COUNT;i++)
|
||
{
|
||
str.Strcpy(space,stop_str_width-step_str[i].Length());
|
||
|
||
str+=step_str[i];
|
||
|
||
num_str=AnsiString::numberOf(step_count[i]);
|
||
|
||
str.Strcat(space,str_total.Length()-num_str.Length()+3);
|
||
|
||
str+=num_str;
|
||
|
||
if(step_count[i]>0)
|
||
{
|
||
num_str=AnsiString::floatOf(float(step_count[i])*100.0f/float(ps->count),4);
|
||
|
||
str.Strcat(space,3+(8-num_str.Length()));
|
||
str+=num_str;
|
||
str+="%";
|
||
}
|
||
|
||
chart->DrawString(str,col,row,stop_color[i],255);
|
||
row+=CHAR_BITMAP_HEIGHT;
|
||
}
|
||
}
|
||
|
||
//混合底图
|
||
if(BackgroundBitmap)
|
||
{
|
||
uint8 *p=chart->chart_data;
|
||
uint8 *bp=BackgroundBitmap->data;
|
||
uint8 alpha;
|
||
|
||
for(uint row=0;row<height;row++)
|
||
{
|
||
for(uint col=0;col<width;col++)
|
||
{
|
||
alpha=p[3];
|
||
|
||
p[0]=(p[0]*alpha+bp[0]*(255-alpha))/255;
|
||
p[1]=(p[1]*alpha+bp[1]*(255-alpha))/255;
|
||
p[2]=(p[2]*alpha+bp[2]*(255-alpha))/255;
|
||
p[3]=255;
|
||
|
||
p+=4;
|
||
bp+=3;
|
||
}
|
||
}
|
||
}
|
||
|
||
return chart;
|
||
}
|
||
|
||
int os_main(int argc,os_char **argv)
|
||
{
|
||
std::cout<<"Distribution Chart 2D"<<std::endl<<std::endl;
|
||
|
||
if(argc<2)
|
||
{
|
||
std::cout<<"example: DistributionChart2D data.csv"<<std::endl<<std::endl;
|
||
return 0;
|
||
}
|
||
|
||
if(!LoadBackgroundBitmap())
|
||
{
|
||
std::cerr<<"can't load background mini_map.tga !"<<std::endl;
|
||
return 1;
|
||
}
|
||
|
||
if(!InitBitmapFont())
|
||
{
|
||
std::cerr<<"can't load font file VGA8.F16 !"<<std::endl;
|
||
return 2;
|
||
}
|
||
|
||
UTF8StringList sl;
|
||
|
||
csv_filename=argv[1];
|
||
|
||
int line_count=LoadStringListFromTextFile(sl,csv_filename);
|
||
|
||
if(line_count<=1)
|
||
{
|
||
os_out<<OS_TEXT("Load file ")<<csv_filename.c_str()<<OS_TEXT(" failed!")<<std::endl;
|
||
return(3);
|
||
}
|
||
|
||
os_out<<OS_TEXT("Load file ")<<csv_filename.c_str()<<OS_TEXT(" OK!")<<std::endl;
|
||
|
||
std::cout<<"file total line: "<<line_count<<std::endl;
|
||
|
||
AutoDelete<PositionStat> ps=ToVector2i(sl);
|
||
|
||
AutoDelete<Chart> chart=ToChart32(ps);
|
||
|
||
OSString tga_filename;
|
||
{
|
||
tga_filename=filesystem::ReplaceExtName(csv_filename,OSString(OS_TEXT(".tga")));
|
||
|
||
os_out<<OS_TEXT("output: ")<<tga_filename.c_str()<<std::endl;
|
||
}
|
||
|
||
{
|
||
util::TGAHeader tga_header;
|
||
|
||
util::FillTGAHeader(&tga_header,chart->width,chart->height,4);
|
||
|
||
io::OpenFileOutputStream fos(tga_filename.c_str(),io::FileOpenMode::CreateTrunc);
|
||
|
||
if(fos)
|
||
{
|
||
fos->Write(&tga_header,util::TGAHeaderSize);
|
||
fos->Write(chart->chart_data,chart->width*chart->height*4);
|
||
fos->Close();
|
||
}
|
||
else
|
||
std::cerr<<"Create chart.tga failed!"<<std::endl;
|
||
}
|
||
|
||
delete BackgroundBitmap;
|
||
ClearBitmapFont();
|
||
|
||
return 0;
|
||
}
|