lars/geant4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
dab42d20185975fb0910dadfc6aa9998d9d9e06d
geant4/source/analysis/g4tools/include/tools/wps
T
Gabriele Cosmo dab42d2018 Import Geant4 10.7.0 source tree
2020-12-04 12:30:43 +01:00

511 lines
15 KiB
Plaintext
Raw Blame History

// Copyright (C) 2010, Guy Barrand. All rights reserved.
// See the file tools.license for terms.
#ifndef tools_wps
#define tools_wps
#include <ctime>
#include <string>
#include <ostream>
#include <cstring>
#include "sprintf"
namespace tools {
class wps {
typedef unsigned char Uchar;
typedef unsigned int Uint;
public:
typedef float VCol;
typedef bool(*rgb_func)(void*,unsigned int,unsigned int,VCol&,VCol&,VCol&);
public:
wps(std::ostream& a_out):m_out(a_out){
m_deviceWidth = (8.5f-1.f)*72*METAFILE_SCALE(); // 540 * METAFILE_SCALE
m_deviceHeight = 11*72*METAFILE_SCALE(); // 792 * METAFILE_SCALE
m_pageNumber = 0;
//m_pagePos = 0;
//m_markSize = 2;
m_file = 0;
m_fname.clear();
m_string.clear();
m_gsave = 0;
m_buffer = new Uchar[METAFILE_RECORD_LENGTH()+1];
m_number = 0;
//m_param.shade = shade_color;
}
virtual ~wps(){
if(m_file) close_file();
m_string.clear();
if(m_gsave) {
m_out << "tools::wps::~wps :"
<< " bad gsave/grestore balance : " << m_gsave
<< std::endl;
}
m_gsave = 0;
delete [] m_buffer;
}
//protected:
private:
wps(const wps& a_from):m_out(a_from.m_out){}
wps& operator=(const wps&){return *this;}
public:
//const std::string& file_name() const {return m_fname;}
//FILE* get_FILE() const {return m_file;}
bool open_file(const std::string& a_name,bool a_anonymous = false){
if(m_file) return false;
m_file = ::fopen(a_name.c_str(),"wb");
if(!m_file) return false;
m_fname = a_name;
m_number = 0;
m_buffer[METAFILE_RECORD_LENGTH()]= '\0';
m_pageNumber = 0;
// Header :
PrintFLN("%%!PS-Adobe-2.0");
if(a_anonymous) { //usefull for integration tests.
} else {
PrintFLN("%%%%Creator: tools::wps.");
PrintFLN("%%%%CreationDate: %s",get_date());
PrintFLN("%%%%Title: %s",m_fname.c_str());
}
PrintFLN("%%%%Pages: (atend)");
PrintFLN("%%%%BoundingBox: 0 0 %d %d",(int)m_deviceWidth,(int)m_deviceHeight);
PrintFLN("%%%%DocumentFonts: Courier-Bold");
PrintFLN("%%%%DocumentPaperSizes: a4");
PrintFLN("%%%%EndComments");
// postscript :
PS_SAVE();
/*
// General :
in_buffer("/n {newpath} def ");
in_buffer("/cl {closepath} def ");
in_buffer("/s {stroke} def ");
in_buffer("/f {fill} def ");
// Move :
in_buffer("/m {moveto} def ");
in_buffer("/rm {rmoveto} def ");
in_buffer("/rl {rlineto} def ");
// Line :
in_buffer("/lc {setlinecap} def ");
in_buffer("/lw {setlinewidth} def ");
in_buffer("/rgb {setrgbcolor} def ");
in_buffer("/ss {[] 0 setdash} def "); // style solid.
in_buffer("/sd {[12 6] 0 setdash} def "); // style dashed
in_buffer("/so {[6 12] 0 setdash} def "); // style dotted
in_buffer("/sdo {[18 12 6 12] 0 setdash} def "); // dash dotted
// Mark :
m_markSize = 2.;
in_buffer("/ms 2. def /msi .5 def "); // mark size
in_buffer("/cross {ms ms scale -1. -1. rm ");
in_buffer ("2. 2. rl 0. -2. rm -2. 2. rl msi msi scale} def ");
in_buffer("/plus {ms ms scale -1. 0. rm 2. 0. rl ");
in_buffer("-1. 1. rm 0. -2. rl msi msi scale} def ");
in_buffer("/asterisk {ms ms scale -1. 0. rm 2. 0. rl -1. 1. rm ");
in_buffer("0. -2. rl 0. 1. rm -0.707 -0.707 rm 1.414 1.414 rl ");
in_buffer("0. -1.414 rm -1.414 1.414 rl msi msi scale} def ");
in_buffer("/star {ms ms scale 0. 1. rm -0.6 -1.5 rl ");
in_buffer("1.2 0. rl -0.6 1.5 rl msi msi scale} def ");
// Text :
in_buffer("/sh {show} def ");
in_buffer("/df {/Courier-Bold findfont} def ");
in_buffer("/mf {makefont setfont} def ");
*/
PrintFLN("%%%%EndProlog");
return true;
}
bool close_file(){
if(!m_file) return false;
PS_RESTORE();
PrintFLN("%%%%Trailer");
PrintFLN("%%%%Pages: %d",m_pageNumber);
PrintFLN("%%%%EOF");
::fclose(m_file);
m_file = 0;
m_fname.clear();
return true;
}
void PS_PAGE_SCALE(float a_width,float a_height,bool a_portrait = true){
//NOTE : no check done on [a_width,a_height]<=0
PS_SCALE(1/METAFILE_SCALE(),1/METAFILE_SCALE());
PS_TRANSLATE(m_deviceWidth/20,m_deviceHeight/30);
float scale;
if(m_deviceWidth<=m_deviceHeight) {
scale = (a_height<=a_width?m_deviceWidth/a_width:m_deviceWidth/a_height);
} else {
scale = (a_height<=a_width?m_deviceHeight/a_width:m_deviceHeight/a_height);
}
{float xtra,ytra;
if(a_portrait) {
xtra = (m_deviceWidth - scale * a_width)/2;
ytra = (m_deviceHeight - scale * a_height)/2;
} else {
PS_TRANSLATE(m_deviceWidth,0);
PS_ROTATE(90);
xtra = (m_deviceHeight - scale * a_width)/2;
ytra = (m_deviceWidth - scale * a_height)/2;
}
PS_TRANSLATE(xtra,ytra);}
PS_SCALE(scale,scale);
}
void PS_SCALE(float a_x,float a_y) {
in_buffer("%.2f %.2f scale ",a_x,a_y);
}
void PS_TRANSLATE(float a_x,float a_y){
in_buffer("%.2f %.2f translate ",a_x,a_y);
}
void PS_ROTATE(float a_x){in_buffer("%.2f rotate ",a_x);}
void PS_SAVE(){in_buffer("gsave ");m_gsave++;}
void PS_RESTORE(){in_buffer("grestore ");m_gsave--;}
void PS_BEGIN_PAGE(){
m_pageNumber++;
PrintFLN("%%%%Page: %d %d",m_pageNumber,m_pageNumber);
PS_SAVE();
}
void PS_END_PAGE(){
in_buffer("showpage ");
PS_RESTORE();
}
enum rgb_nbit {
rgb_bw = 0,
rgb_2 = 2,
rgb_4 = 4,
rgb_8 = 8
};
void PS_IMAGE(Uint a_width,Uint a_height,rgb_nbit a_nbit,rgb_func a_proc,void* a_tag){
//NOTE : no check done on [a_width,a_height] being zero.
Uint row,col;
VCol dr,dg,db;
Uchar red,green,blue,b;
PS_SAVE();
in_buffer("%d %d scale ", a_width, a_height );
bool status = true;
if(a_nbit==rgb_bw) { //grey or black_white
in_buffer ("/picstr %d string def ",a_width);
in_buffer ("%d %d %d ",a_width,a_height,8);
in_buffer ("[ %d 0 0 -%d 0 %d ] ",a_width,a_height,a_height);
in_buffer ("{ currentfile picstr readhexstring pop } " );
PrintFLN ("image " );
for ( row = 0; row < a_height; row++ ) {
for ( col = 0; col < a_width; col++) {
status = a_proc(a_tag,col,row,dr,dg,db)?status:false;
VCol fgrey = rgb2grey(dr,dg,db);
Uchar grey = (Uchar) ( 255.0F * fgrey);
WriteByte(grey);
}
}
int nbhex = a_width * a_height * 2;
PrintFLN ("%%%% nbhex digit :%d ",nbhex);
PrintFLN ("%%%% nbhex/record_length :%d ",
int(nbhex/METAFILE_RECORD_LENGTH()));
PrintFLN ("%%%% nbhex%%record_length :%d ",
int(nbhex%METAFILE_RECORD_LENGTH()));
} else if(a_nbit==rgb_2) {
int nbyte2 = (a_width * 3)/4;
nbyte2 /=3;
nbyte2 *=3;
Uint col_max = (nbyte2 * 4)/3;
// 2 bit for r and g and b.
// rgbs following each other.
in_buffer ("/rgbstr %d string def ",nbyte2);
in_buffer ("%d %d %d ",col_max,a_height,2);
in_buffer ("[ %d 0 0 -%d 0 %d ] ",col_max,a_height,a_height);
in_buffer ("{ currentfile rgbstr readhexstring pop } " );
in_buffer ("false 3 " );
PrintFLN ("colorimage " );
for ( row = 0; row < a_height; row++ ) {
for ( col = 0; col < col_max; col+=4) {
status = a_proc(a_tag,col,row,dr,dg,db)?status:false;
red = (Uchar) ( 3.0F * dr);
green = (Uchar) ( 3.0F * dg);
blue = (Uchar) ( 3.0F * db);
b = red;
b = (b<<2)+green;
b = (b<<2)+blue;
status = a_proc(a_tag,col+1,row,dr,dg,db)?status:false;
red = (Uchar) ( 3.0F * dr);
green = (Uchar) ( 3.0F * dg);
blue = (Uchar) ( 3.0F * db);
b = (b<<2)+red;
WriteByte(b);
b = green;
b = (b<<2)+blue;
status = a_proc(a_tag,col+2,row,dr,dg,db)?status:false;
red = (Uchar) ( 3.0F * dr);
green = (Uchar) ( 3.0F * dg);
blue = (Uchar) ( 3.0F * db);
b = (b<<2)+red;
b = (b<<2)+green;
WriteByte(b);
b = blue;
status = a_proc(a_tag,col+3,row,dr,dg,db)?status:false;
red = (Uchar) ( 3.0F * dr);
green = (Uchar) ( 3.0F * dg);
blue = (Uchar) ( 3.0F * db);
b = (b<<2)+red;
b = (b<<2)+green;
b = (b<<2)+blue;
WriteByte(b);
}
}
} else if(a_nbit==rgb_4) {
int nbyte4 = (a_width * 3)/2;
nbyte4 /=3;
nbyte4 *=3;
Uint col_max = (nbyte4 * 2)/3;
// 4 bit for r and g and b.
// rgbs following each other.
in_buffer ("/rgbstr %d string def ",nbyte4);
in_buffer ("%d %d %d ",col_max,a_height,4);
in_buffer ("[ %d 0 0 -%d 0 %d ] ",col_max,a_height,a_height);
in_buffer ("{ currentfile rgbstr readhexstring pop } " );
in_buffer ("false 3 " );
PrintFLN ("colorimage " );
for ( row = 0; row < a_height; row++ ) {
for ( col = 0; col < col_max; col+=2) {
status = a_proc(a_tag,col,row,dr,dg,db)?status:false;
red = (Uchar) ( 15.0F * dr);
green = (Uchar) ( 15.0F * dg);
in_buffer ("%x%x",red,green);
blue = (Uchar) ( 15.0F * db);
status = a_proc(a_tag,col+1,row,dr,dg,db)?status:false;
red = (Uchar) ( 15.0F * dr);
in_buffer ("%x%x",blue,red);
green = (Uchar) ( 15.0F * dg);
blue = (Uchar) ( 15.0F * db);
in_buffer ("%x%x",green,blue);
}
}
} else if(a_nbit==rgb_8) {
int nbyte8 = a_width * 3;
// 8 bit for r and g and b.
in_buffer ("/rgbstr %d string def ",nbyte8);
in_buffer ("%d %d %d ",a_width,a_height,8);
in_buffer ("[ %d 0 0 -%d 0 %d ] ",a_width,a_height,a_height);
in_buffer ("{ currentfile rgbstr readhexstring pop } " );
in_buffer ("false 3 " );
PrintFLN ("colorimage " );
for ( row = 0; row < a_height; row++ ) {
for ( col = 0; col < a_width; col++) {
status = a_proc(a_tag,col,row,dr,dg,db)?status:false;
red = (Uchar) ( 255.0F * dr);
WriteByte (red);
green = (Uchar) ( 255.0F * dg);
WriteByte (green);
blue = (Uchar) ( 255.0F * db);
WriteByte (blue);
}
}
} else {
m_out << "PS_IMAGE :"
<< " unknown rgb nbit " << a_nbit
<< std::endl;
}
if(!status) {
m_out << "PS_IMAGE :"
<< " problem to retrieve some pixel rgb."
<< std::endl;
}
PS_RESTORE();
}
protected:
static size_t METAFILE_RECORD_LENGTH() {return 80;}
static float METAFILE_SCALE() {return 1.0f;}
static char* get_date(){
char* string;
time_t d;
::time(&d);
string = ::ctime(&d);
string[24] = '\0';
return string;
}
static VCol rgb2grey(VCol a_red,VCol a_green,VCol a_blue){
return (0.30f * a_red + 0.59f * a_green + 0.11f * a_blue);
}
bool in_buffer(const char* a_format,...){
va_list args;
va_start(args,a_format);
bool status = vsprintf(m_string,2048,a_format,args);
va_end(args);
if(!status) {
m_out << "tools::wps::in_buffer : overflow." << std::endl;
return false;
}
size_t length = m_string.size();
if(length>METAFILE_RECORD_LENGTH()) {
m_out << "tools::wps::in_buffer : overflow." << std::endl;
return false;
}
size_t nlength = m_number + length;
if(nlength>METAFILE_RECORD_LENGTH()) {
m_buffer[m_number] = '\0';
if(::fprintf(m_file,"%s\n",(char*)m_buffer)<0) {
m_out << "tools::wps::in_buffer : fprintf failed." << std::endl;
}
m_number = 0;
nlength = length;
}
Uchar* pointer = m_buffer + m_number;
::strcpy((char*)pointer,m_string.c_str());
m_number = nlength;
return true;
}
bool PrintFLN(const char* a_format,...){
va_list args;
va_start(args,a_format);
bool status = vsprintf(m_string,2048,a_format,args);
va_end(args);
if(!status) {
m_out << "tools::wps::PrintFLN : overflow." << std::endl;
return false;
}
// put buffer in file :
if(m_number>0) {
m_buffer[m_number] = '\0';
if(::fprintf(m_file,"%s\n",(char*)m_buffer)<0) {
m_out << "tools::wps::PrintFLN : fprintf failed." << std::endl;
}
m_number = 0;
}
if(::fprintf(m_file,"%s\n",m_string.c_str())<0) {
m_out << "tools::wps::PrintFLN : fprintf failed." << std::endl;
}
return true;
}
void WriteByte(Uchar a_byte){
Uchar h = a_byte / 16;
Uchar l = a_byte % 16;
in_buffer("%x%x",h,l);
}
protected:
std::ostream& m_out;
float m_deviceWidth;
float m_deviceHeight;
unsigned int m_pageNumber;
//double m_markSize;
FILE* m_file;
std::string m_fname;
std::string m_string;
int m_gsave;
Uchar* m_buffer;
size_t m_number;
//struct {
// int shade;
// int nbit;
//} m_param;
};
}
#endif
/*
enum {
shade_color = 0,
shade_grey = 1,
shade_black_white = 2
};
void PS_BACKGROUND(double a_r,double a_g,double a_b,
double a_width,double a_height,
bool a_do_back = true){
PS_NEWPATH();
PS_MOVE (0.,0.);
PS_RLINETO (a_width,0.);
PS_RLINETO (0.,a_height);
PS_RLINETO (-a_width,0.);
PS_RLINETO (0.,-a_height);
PS_CLOSEPATH();
if(a_do_back) {
PS_SAVE();
PS_RGB(a_r,a_g,a_b);
PS_FILL();
PS_RESTORE();
}
in_buffer("clip ");
}
void PS_RGB(double a_r,double a_g,double a_b){
if(m_param.shade==shade_color)
in_buffer("%.2f %.2f %.2f rgb ",a_r,a_g,a_b);
else if(m_param.shade==shade_grey)
in_buffer("%.2f setgray ",rgb2grey(a_r,a_g,a_b));
else if(m_param.shade==shade_black_white)
in_buffer("0. setgray ",rgb2grey(a_r,a_g,a_b));
}
void PS_LINE_WIDTH(int a_width){in_buffer("%.1f lw ",float(a_width));}
void PS_NEWPATH() {in_buffer("n ");}
void PS_STROKE() {in_buffer("s ");}
void PS_FILL() {in_buffer("f ");}
void PS_CLOSEPATH() {in_buffer("cl ");}
void PS_CAP(double a_x) {in_buffer ("%1d lc ",a_x);}
void PS_RLINETO(double a_x,double a_y) {
in_buffer ("%.2f %.2f rl ",a_x,a_y);
}
void PS_MOVE(double a_x,double a_y) {
in_buffer ("%.2f %.2f m ",a_x,a_y);
}
void PS_FRAME(double a_r,double a_g,double a_b,
double a_width,double a_height){
PS_NEWPATH ();
PS_MOVE (0.,0.);
PS_RLINETO (a_width,0.);
PS_RLINETO (0.,a_height);
PS_RLINETO (-a_width,0.);
PS_RLINETO (0.,-a_height);
PS_CLOSEPATH ();
PS_RGB (a_r,a_g,a_b);
PS_LINE_WIDTH(1);
PS_CAP (1);
in_buffer ("ss ");
PS_STROKE();
}
*/
Reference in New Issue View Git Blame Copy Permalink