| 1 | // |
|---|
| 2 | // ******************************************************************** |
|---|
| 3 | // * License and Disclaimer * |
|---|
| 4 | // * * |
|---|
| 5 | // * The Geant4 software is copyright of the Copyright Holders of * |
|---|
| 6 | // * the Geant4 Collaboration. It is provided under the terms and * |
|---|
| 7 | // * conditions of the Geant4 Software License, included in the file * |
|---|
| 8 | // * LICENSE and available at http://cern.ch/geant4/license . These * |
|---|
| 9 | // * include a list of copyright holders. * |
|---|
| 10 | // * * |
|---|
| 11 | // * Neither the authors of this software system, nor their employing * |
|---|
| 12 | // * institutes,nor the agencies providing financial support for this * |
|---|
| 13 | // * work make any representation or warranty, express or implied, * |
|---|
| 14 | // * regarding this software system or assume any liability for its * |
|---|
| 15 | // * use. Please see the license in the file LICENSE and URL above * |
|---|
| 16 | // * for the full disclaimer and the limitation of liability. * |
|---|
| 17 | // * * |
|---|
| 18 | // * This code implementation is the result of the scientific and * |
|---|
| 19 | // * technical work of the GEANT4 collaboration. * |
|---|
| 20 | // * By using, copying, modifying or distributing the software (or * |
|---|
| 21 | // * any work based on the software) you agree to acknowledge its * |
|---|
| 22 | // * use in resulting scientific publications, and indicate your * |
|---|
| 23 | // * acceptance of all terms of the Geant4 Software license. * |
|---|
| 24 | // ******************************************************************** |
|---|
| 25 | // |
|---|
| 26 | #ifdef G4VIS_BUILD_OI_DRIVER |
|---|
| 27 | |
|---|
| 28 | /*----------------------------HEPVis----------------------------------------*/ |
|---|
| 29 | /* */ |
|---|
| 30 | /* Node: SoImageWriter */ |
|---|
| 31 | /* Author: Guy Barrand */ |
|---|
| 32 | /* */ |
|---|
| 33 | /*--------------------------------------------------------------------------*/ |
|---|
| 34 | |
|---|
| 35 | // this : |
|---|
| 36 | #include <HEPVis/nodes/SoImageWriter.h> |
|---|
| 37 | |
|---|
| 38 | #include <Inventor/errors/SoDebugError.h> |
|---|
| 39 | #include <Inventor/elements/SoViewportRegionElement.h> |
|---|
| 40 | #include <Inventor/actions/SoGLRenderAction.h> |
|---|
| 41 | |
|---|
| 42 | #include <HEPVis/SbGL.h> |
|---|
| 43 | #include <HEPVis/SbPainterPS.h> |
|---|
| 44 | //#include <HEPVis/SbGIF.h> |
|---|
| 45 | |
|---|
| 46 | #include <stdlib.h> |
|---|
| 47 | |
|---|
| 48 | typedef struct { |
|---|
| 49 | unsigned char red; |
|---|
| 50 | unsigned char green; |
|---|
| 51 | unsigned char blue; |
|---|
| 52 | } Pixel; |
|---|
| 53 | typedef unsigned char Uchar; |
|---|
| 54 | |
|---|
| 55 | //static void getImagePixels(int,int,float*,int&, |
|---|
| 56 | // Uchar*&,Uchar*&,Uchar*&,Uchar*&); |
|---|
| 57 | |
|---|
| 58 | static int sWidth = 0; |
|---|
| 59 | static int sHeight = 0; |
|---|
| 60 | static float* sImage = 0; |
|---|
| 61 | static int getRGB(unsigned int,unsigned int,double&,double&,double&); |
|---|
| 62 | |
|---|
| 63 | SO_NODE_SOURCE(SoImageWriter) |
|---|
| 64 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 65 | void SoImageWriter::initClass ( |
|---|
| 66 | ) |
|---|
| 67 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 68 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 69 | { |
|---|
| 70 | SO_NODE_INIT_CLASS(SoImageWriter,SoNode,"Node"); |
|---|
| 71 | } |
|---|
| 72 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 73 | SoImageWriter::SoImageWriter( |
|---|
| 74 | ) |
|---|
| 75 | :fEnabled(FALSE) |
|---|
| 76 | ,fStatus(FALSE) |
|---|
| 77 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 78 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 79 | { |
|---|
| 80 | SO_NODE_CONSTRUCTOR(SoImageWriter); |
|---|
| 81 | //SO_NODE_ADD_FIELD(format,(POST_SCRIPT)); |
|---|
| 82 | SO_NODE_ADD_FIELD(fileName,("out.ps")); |
|---|
| 83 | |
|---|
| 84 | //SO_NODE_DEFINE_ENUM_VALUE(Format,POST_SCRIPT); |
|---|
| 85 | //SO_NODE_DEFINE_ENUM_VALUE(Format,GIF); |
|---|
| 86 | |
|---|
| 87 | //SO_NODE_SET_SF_ENUM_TYPE(format,Format); |
|---|
| 88 | } |
|---|
| 89 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 90 | SoImageWriter::~SoImageWriter ( |
|---|
| 91 | ) |
|---|
| 92 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 93 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 94 | { |
|---|
| 95 | } |
|---|
| 96 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 97 | void SoImageWriter::enable( |
|---|
| 98 | ) |
|---|
| 99 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 100 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 101 | { |
|---|
| 102 | fEnabled = TRUE; |
|---|
| 103 | } |
|---|
| 104 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 105 | void SoImageWriter::disable( |
|---|
| 106 | ) |
|---|
| 107 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 108 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 109 | { |
|---|
| 110 | fEnabled = FALSE; |
|---|
| 111 | } |
|---|
| 112 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 113 | SbBool SoImageWriter::getStatus( |
|---|
| 114 | ) const |
|---|
| 115 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 116 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 117 | { |
|---|
| 118 | return fStatus; |
|---|
| 119 | } |
|---|
| 120 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 121 | void SoImageWriter::GLRender( |
|---|
| 122 | SoGLRenderAction* aAction |
|---|
| 123 | ) |
|---|
| 124 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 125 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 126 | { |
|---|
| 127 | fStatus = FALSE; |
|---|
| 128 | //printf("debug : SoImageWriter::GLRender : enabled : %d\n",fEnabled); |
|---|
| 129 | if(!fEnabled) return; |
|---|
| 130 | SbViewportRegion vpr = SoViewportRegionElement::get(aAction->getState()); |
|---|
| 131 | const SbVec2s& win = vpr.getWindowSize(); |
|---|
| 132 | int w = win[0]; |
|---|
| 133 | int h = win[1]; |
|---|
| 134 | if((w*h)<=0) { |
|---|
| 135 | SoDebugError::postInfo("SoImageWriter::GLRender","null area window !"); |
|---|
| 136 | return; |
|---|
| 137 | } |
|---|
| 138 | |
|---|
| 139 | int x = 0; |
|---|
| 140 | int y = 0; |
|---|
| 141 | int s = 3 * w * h; |
|---|
| 142 | float* image = new float[s]; |
|---|
| 143 | if(!image) return; |
|---|
| 144 | |
|---|
| 145 | //printf("debug : SoImageWriter::GLRender : %d %d %d %d\n",x,y,w,h); |
|---|
| 146 | |
|---|
| 147 | //glReadPixels(x,y,w,h,GL_RGB,GL_UNSIGNED_BYTE,image); Don't work ! |
|---|
| 148 | glReadPixels(x,y,w,h,GL_RGB,GL_FLOAT,image); |
|---|
| 149 | |
|---|
| 150 | //Format fm = (Format)format.getValue(); |
|---|
| 151 | //if(fm==GIF) { |
|---|
| 152 | /* |
|---|
| 153 | FILE* file = fopen(fileName.getValue().getString(),"wb"); |
|---|
| 154 | if(!file) { |
|---|
| 155 | SoDebugError::postInfo("SoImageWriter::GLRender", |
|---|
| 156 | "can't open file \"%s\".",fileName.getValue().getString()); |
|---|
| 157 | } else { |
|---|
| 158 | int coln; |
|---|
| 159 | Uchar* rs; |
|---|
| 160 | Uchar* gs; |
|---|
| 161 | Uchar* bs; |
|---|
| 162 | Uchar* data; |
|---|
| 163 | getImagePixels(w,h,image,coln,rs,gs,bs,data); |
|---|
| 164 | |
|---|
| 165 | SbGIF::putBytesInStream(file,data,w,h,coln,rs,gs,bs); |
|---|
| 166 | |
|---|
| 167 | delete [] data; |
|---|
| 168 | |
|---|
| 169 | if(rs) free(rs); |
|---|
| 170 | if(gs) free(gs); |
|---|
| 171 | if(bs) free(bs); |
|---|
| 172 | |
|---|
| 173 | fclose(file); |
|---|
| 174 | |
|---|
| 175 | fStatus = TRUE; |
|---|
| 176 | } |
|---|
| 177 | } else { |
|---|
| 178 | */ |
|---|
| 179 | |
|---|
| 180 | SbPainterPS painterPS; |
|---|
| 181 | painterPS.openFileForWriting(fileName.getValue().getString()); |
|---|
| 182 | if(!painterPS.getStream()) { |
|---|
| 183 | SoDebugError::postInfo("SoImageWriter::GLRender", |
|---|
| 184 | "can't open file \"%s\".",fileName.getValue().getString()); |
|---|
| 185 | } else { |
|---|
| 186 | painterPS.setWindowSize(w,h); |
|---|
| 187 | //painterPS.setBitsPerPixel(8); |
|---|
| 188 | painterPS.setBitsPerPixel(4); |
|---|
| 189 | painterPS.beginTraversal(); |
|---|
| 190 | painterPS.clearColorBuffer(1.,1.,1.); |
|---|
| 191 | |
|---|
| 192 | sWidth = w; |
|---|
| 193 | sHeight = h; |
|---|
| 194 | sImage = image; |
|---|
| 195 | painterPS.putImageInStream((unsigned int)w,(unsigned int)h,getRGB); |
|---|
| 196 | |
|---|
| 197 | painterPS.endTraversal(); |
|---|
| 198 | |
|---|
| 199 | painterPS.closeStream(); |
|---|
| 200 | |
|---|
| 201 | fStatus = TRUE; |
|---|
| 202 | } |
|---|
| 203 | //} |
|---|
| 204 | delete [] image; |
|---|
| 205 | |
|---|
| 206 | } |
|---|
| 207 | /* |
|---|
| 208 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 209 | void getImagePixels ( |
|---|
| 210 | int aWidth |
|---|
| 211 | ,int aHeight |
|---|
| 212 | ,float* aImage |
|---|
| 213 | ,int& aColorn |
|---|
| 214 | ,Uchar*& aReds |
|---|
| 215 | ,Uchar*& aGreens |
|---|
| 216 | ,Uchar*& aBlues |
|---|
| 217 | ,Uchar*& aData |
|---|
| 218 | ) |
|---|
| 219 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 220 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 221 | { |
|---|
| 222 | aColorn = 0; |
|---|
| 223 | aReds = 0; |
|---|
| 224 | aGreens = 0; |
|---|
| 225 | aBlues = 0; |
|---|
| 226 | aData = 0; |
|---|
| 227 | if( (aWidth * aHeight) <=0) return; |
|---|
| 228 | int size = 256; |
|---|
| 229 | Uchar* rs = (Uchar*)malloc(size * sizeof(Uchar)); |
|---|
| 230 | Uchar* gs = (Uchar*)malloc(size * sizeof(Uchar)); |
|---|
| 231 | Uchar* bs = (Uchar*)malloc(size * sizeof(Uchar)); |
|---|
| 232 | Uchar* data = new Uchar[aWidth * aHeight]; |
|---|
| 233 | if( !rs || !gs || !bs || !data ) { |
|---|
| 234 | if(rs) free(rs); |
|---|
| 235 | if(gs) free(gs); |
|---|
| 236 | if(bs) free(bs); |
|---|
| 237 | delete [] data; |
|---|
| 238 | return; |
|---|
| 239 | } |
|---|
| 240 | int pixeln = 0; |
|---|
| 241 | int row,col; |
|---|
| 242 | Uchar red,green,blue; |
|---|
| 243 | Uchar ored = 0,ogreen = 0,oblue = 0; |
|---|
| 244 | float* pimag = aImage; |
|---|
| 245 | Uchar* pdata = 0; |
|---|
| 246 | Uchar index = 0; |
|---|
| 247 | int status = 0; |
|---|
| 248 | for(row=0;row<aHeight;row++) { |
|---|
| 249 | pdata = data + (aHeight - 1 - row) * aWidth; |
|---|
| 250 | for(col=0;col<aWidth;col++){ |
|---|
| 251 | red = (Uchar)(255 * (*pimag));pimag++; |
|---|
| 252 | green = (Uchar)(255 * (*pimag));pimag++; |
|---|
| 253 | blue = (Uchar)(255 * (*pimag));pimag++; |
|---|
| 254 | //printf("debug : %d %d : %d %d %d\n",row,col,red,green,blue); |
|---|
| 255 | if( (pixeln==0) || (red!=ored) || (green!=ogreen) || (blue!=oblue) ){ |
|---|
| 256 | // Search exact color : |
|---|
| 257 | int found = 0; |
|---|
| 258 | for(int count=0;count<pixeln;count++){ |
|---|
| 259 | if( (red==rs[count]) && (green==gs[count]) && (blue==bs[count]) ){ |
|---|
| 260 | found = 1; |
|---|
| 261 | index = count; |
|---|
| 262 | break; |
|---|
| 263 | } |
|---|
| 264 | } |
|---|
| 265 | if(found==0){ |
|---|
| 266 | if(pixeln>=256) { |
|---|
| 267 | // We can't store more than 256 on an Uchar. |
|---|
| 268 | // Search closest color : |
|---|
| 269 | int dr,dg,db; |
|---|
| 270 | int PRECISION = 20; |
|---|
| 271 | int closest = 0; |
|---|
| 272 | for(int count=0;count<pixeln;count++){ |
|---|
| 273 | dr = red - rs[count];dr = dr<0 ? -dr : dr; |
|---|
| 274 | dg = green - gs[count];dg = dg<0 ? -dg : dg; |
|---|
| 275 | db = blue - bs[count];db = db<0 ? -db : db; |
|---|
| 276 | if( (dr<=PRECISION) && (dg<=PRECISION) && (db<=PRECISION) ){ |
|---|
| 277 | closest = 1; |
|---|
| 278 | index = count; |
|---|
| 279 | break; |
|---|
| 280 | } |
|---|
| 281 | } |
|---|
| 282 | if(closest==0) { |
|---|
| 283 | index = 0; |
|---|
| 284 | status = 1; |
|---|
| 285 | } |
|---|
| 286 | } else { |
|---|
| 287 | if(pixeln>=size){ |
|---|
| 288 | size += 256; |
|---|
| 289 | rs = (Uchar*)realloc(rs,size * sizeof(Uchar)); |
|---|
| 290 | gs = (Uchar*)realloc(gs,size * sizeof(Uchar)); |
|---|
| 291 | bs = (Uchar*)realloc(bs,size * sizeof(Uchar)); |
|---|
| 292 | if( !rs || !gs || !bs ) { |
|---|
| 293 | if(rs) free(rs); |
|---|
| 294 | if(gs) free(gs); |
|---|
| 295 | if(bs) free(bs); |
|---|
| 296 | delete [] data; |
|---|
| 297 | return; |
|---|
| 298 | } |
|---|
| 299 | } |
|---|
| 300 | //printf("debug : SoImageWriter pixeln %d : %d %d %d\n", |
|---|
| 301 | // pixeln,red,green,blue); |
|---|
| 302 | rs[pixeln] = red; |
|---|
| 303 | gs[pixeln] = green; |
|---|
| 304 | bs[pixeln] = blue; |
|---|
| 305 | index = pixeln; |
|---|
| 306 | pixeln++; |
|---|
| 307 | } |
|---|
| 308 | } |
|---|
| 309 | } |
|---|
| 310 | *pdata = index; |
|---|
| 311 | pdata++; |
|---|
| 312 | ored = red; |
|---|
| 313 | ogreen = green; |
|---|
| 314 | oblue = blue; |
|---|
| 315 | } |
|---|
| 316 | } |
|---|
| 317 | if(status==1) |
|---|
| 318 | printf("SoImageWriter : more than 256 colors in picture ; some colors approximated.\n"); |
|---|
| 319 | aColorn = pixeln; |
|---|
| 320 | aReds = rs; |
|---|
| 321 | aGreens = gs; |
|---|
| 322 | aBlues = bs; |
|---|
| 323 | aData = data; |
|---|
| 324 | } |
|---|
| 325 | */ |
|---|
| 326 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 327 | int getRGB( |
|---|
| 328 | unsigned int aX |
|---|
| 329 | ,unsigned int aY |
|---|
| 330 | ,double& aRed |
|---|
| 331 | ,double& aGreen |
|---|
| 332 | ,double& aBlue |
|---|
| 333 | ) |
|---|
| 334 | ////////////////////////////////////////////////////////////////////////////// |
|---|
| 335 | // OpenGL image is from down to up. |
|---|
| 336 | // PS image is up to down. |
|---|
| 337 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
|---|
| 338 | { |
|---|
| 339 | float* pimag = sImage + 3 * (sWidth * (sHeight - 1 - aY) + aX); |
|---|
| 340 | aRed = *pimag;pimag++; |
|---|
| 341 | aGreen = *pimag;pimag++; |
|---|
| 342 | aBlue = *pimag;pimag++; |
|---|
| 343 | return 1; |
|---|
| 344 | } |
|---|
| 345 | |
|---|
| 346 | #endif |
|---|
