You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdegraphics/kpovmodeler/pmpovrayrenderwidget.cpp

438 lines
12 KiB

/*
**************************************************************************
description
--------------------
copyright : (C) 2001-2003 by Andreas Zehender
email : zehender@kde.org
**************************************************************************
**************************************************************************
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
**************************************************************************/
#include "pmpovrayrenderwidget.h"
#include "pmdefaults.h"
#include "pmdebug.h"
#include "pmdragwidget.h"
#include <kprocess.h>
#include <kmessagebox.h>
#include <klocale.h>
#include <kconfig.h>
#include <kurl.h>
#include <ktempfile.h>
#include <tqcolor.h>
#include <tqpainter.h>
#include <tqtextstream.h>
#include <tqdragobject.h>
#ifdef KDE_NO_COMPAT
#undef KDE_NO_COMPAT
#endif
#include <kapplication.h>
TQString PMPovrayRenderWidget::s_povrayCommand = c_defaultPovrayCommand;
TQStringList PMPovrayRenderWidget::s_libraryPaths;
PMPovrayRenderWidget::PMPovrayRenderWidget( TQWidget* parent, const char* name )
: PMDragWidget( parent, name )
{
m_pProcess = 0;
m_bSuspended = false;
m_rcvHeader = false;
m_skipBytes = 0;
m_bPixmapUpToDate = false;
m_pTempFile = 0;
setBackgroundColor( TQColor( 0, 0, 0 ) );
}
PMPovrayRenderWidget::~PMPovrayRenderWidget( )
{
cleanup( );
}
bool PMPovrayRenderWidget::render( const TQByteArray& scene,
const PMRenderMode& m,
const KURL& documentURL )
{
cleanup( );
m_povrayOutput = "";
m_renderMode = m;
if( !scene.data( ) )
{
KMessageBox::sorry( this, i18n( "Can't render an empty scene.\n" ) );
return false;
}
// output to tmp file
m_pTempFile = new KTempFile( TQString(), ".pov" );
TQDataStream* dstr = m_pTempFile->dataStream( );
if( ( m_pTempFile->status( ) != 0 ) || !dstr )
{
KMessageBox::sorry( this, i18n( "Couldn't write the scene to a temp file.\n" ) );
return false;
}
dstr->writeRawBytes( scene.data( ), scene.size( ) );
m_pTempFile->close( );
m_pProcess = new KProcess( );
connect( m_pProcess, TQT_SIGNAL( receivedStdout( KProcess*, char*, int ) ),
TQT_SLOT( slotPovrayImage( KProcess*, char*, int ) ) );
connect( m_pProcess, TQT_SIGNAL( receivedStderr( KProcess*, char*, int ) ),
TQT_SLOT( slotPovrayMessage( KProcess*, char*, int ) ) );
connect( m_pProcess, TQT_SIGNAL( processExited( KProcess* ) ),
TQT_SLOT( slotRenderingFinished( KProcess* ) ) );
*m_pProcess << s_povrayCommand;
TQStringList::ConstIterator it;
TQStringList args = m_renderMode.commandLineSwitches( );
for( it = args.begin( ); it != args.end( ); ++it )
*m_pProcess << *it;
for( it = s_libraryPaths.begin( ); it != s_libraryPaths.end( ); ++it )
{
TQString path = *it;
if( path != TQString( "/" ) )
if( path.right( 1 ) == TQString( "/" ) )
path.truncate( path.length( ) - 1 );
*m_pProcess << ( TQString( "+L" ) + path );
}
*m_pProcess << TQString( "+I" ) + m_pTempFile->name( ) << "+O-" << "+FT"
<< "+K0.0" << "+KFI1" << "+KFF1" << "+KI0.0" << "+KF0.0"
<< "+SF1" << "+EF1" << "-KC" << "-D";
#if ( ( TDE_VERSION_MAJOR == 2 ) && ( TDE_VERSION_MINOR >= 9 ) ) || ( TDE_VERSION_MAJOR == 3 )
if( !documentURL.isEmpty( ) && documentURL.isLocalFile( ) )
m_pProcess->setWorkingDirectory( documentURL.directory( ) );
#endif
m_rcvHeader = true;
m_rcvHeaderBytes = 0;
m_rcvPixels = 0;
m_progress = 0;
m_numRestBytes = 0;
m_line = 0;
m_column = 0;
m_skipBytes = 0;
int width = m_renderMode.width( );
int height = m_renderMode.height( );
m_image.create( width, height, 32 );
m_image.setAlphaBuffer( m_renderMode.alpha( ) );
m_image.fill( tqRgb( 0, 0, 0 ) );
m_bPixmapUpToDate = false;
repaint( );
if( !m_pProcess->start( KProcess::NotifyOnExit, KProcess::AllOutput ) )
{
KMessageBox::error( this, i18n( "Couldn't call povray.\n"
"Please check your installation "
"or set another povray command." ) );
delete m_pProcess;
m_pProcess = 0;
return false;
}
m_bSuspended = false;
return true;
}
void PMPovrayRenderWidget::killRendering( )
{
if( m_pProcess )
{
if( m_bSuspended )
m_pProcess->kill( SIGCONT );
m_bSuspended = false;
m_pProcess->kill( );
}
}
void PMPovrayRenderWidget::suspendRendering( )
{
if( m_pProcess )
{
m_bSuspended = true;
m_pProcess->kill( SIGSTOP );
}
}
void PMPovrayRenderWidget::resumeRendering( )
{
if( m_pProcess )
{
m_pProcess->kill( SIGCONT );
m_bSuspended = false;
}
}
void PMPovrayRenderWidget::slotPovrayMessage( KProcess*,
char* buffer, int buflen )
{
TQString str;
str.setLatin1( buffer, buflen );
m_povrayOutput += str;
emit povrayMessage( str );
}
void PMPovrayRenderWidget::slotPovrayImage( KProcess*, char* buffer, int buflen )
{
int index = 0;
int i;
int oldLine = m_line;
if( m_rcvHeader )
{
// receive targa header
while( ( m_rcvHeaderBytes < 18 ) && ( index < buflen ) )
{
m_header[m_rcvHeaderBytes] = ( unsigned char ) buffer[index];
m_rcvHeaderBytes++;
index++;
}
if( m_rcvHeaderBytes == 18 )
{
// complete targa header received
m_rcvHeader = false;
m_skipBytes = m_header[0]; // id length
m_bytespp = m_header[16] / 8;
}
}
if( m_skipBytes > 0 )
{
int skip = buflen - index;
if( skip > m_skipBytes )
skip = m_skipBytes;
m_skipBytes -= skip;
index += skip;
}
if( ( m_numRestBytes > 0 ) && ( index < buflen ) )
{
while( ( m_numRestBytes < m_bytespp ) && ( index < buflen ) )
{
m_restBytes[m_numRestBytes] = ( unsigned char ) buffer[index];
index++;
m_numRestBytes++;
}
if( m_numRestBytes == m_bytespp )
{
m_numRestBytes = 0;
if( m_bytespp == 4 )
setPixel( m_column, m_line,
tqRgba( m_restBytes[2], m_restBytes[1],
m_restBytes[0], m_restBytes[3] ) );
else
setPixel( m_column, m_line,
tqRgb( m_restBytes[2], m_restBytes[1], m_restBytes[0] ) );
m_column++;
m_rcvPixels++;
if( m_column == m_renderMode.width( ) )
{
m_column = 0;
m_line++;
}
}
}
if( index < buflen )
{
int num = ( buflen - index ) / m_bytespp;
for( i = 0; i < num; i++ )
{
if( m_bytespp == 4 )
setPixel( m_column, m_line,
tqRgba( buffer[index+2], buffer[index+1],
buffer[index], buffer[index+3] ) );
else
setPixel( m_column, m_line,
tqRgb( buffer[index+2], buffer[index+1],
buffer[index] ) );
index += m_bytespp;
m_column++;
m_rcvPixels++;
if( m_column == m_renderMode.width( ) )
{
m_column = 0;
m_line++;
}
}
}
if( index < buflen )
{
m_numRestBytes = buflen - index;
for( i = 0; i < m_numRestBytes; i++ )
{
m_restBytes[i] = buffer[index];
index++;
}
}
if( m_line != oldLine )
{
TQPainter paint( this );
int offset = 0;
if( m_renderMode.subSection( ) )
{
double sr = m_renderMode.startRow( );
if( sr < 1 )
offset = ( int ) ( m_renderMode.height( ) * sr + 0.5 );
else
offset += ( int ) sr;
}
paint.drawImage( 0, offset + oldLine,
m_image.copy( 0, offset + oldLine, m_image.width( ), offset + m_line - oldLine ) );
emit lineFinished( m_line - 1 );
}
int oldProgress = m_progress;
int numPixels = 0;
if( m_renderMode.subSection( ) )
{
int sr = 0;
if( m_renderMode.startRow( ) < 1 )
sr = ( int ) ( m_renderMode.height( ) * m_renderMode.startRow( ) + 0.5 );
else
sr = ( int ) m_renderMode.startRow( );
int er = 0;
if( m_renderMode.endRow( ) < 1 )
er = ( int ) ( m_renderMode.height( ) * m_renderMode.endRow( ) + 0.5 );
else
er = ( int ) m_renderMode.endRow( );
numPixels = m_renderMode.width( ) * ( er - sr );
}
else
numPixels = m_renderMode.width( ) * m_renderMode.height( );
m_progress = m_rcvPixels * 100 / numPixels;
if( m_progress != oldProgress )
emit progress( m_progress );
m_bPixmapUpToDate = false;
}
void PMPovrayRenderWidget::setPixel( int x, int y, uint c )
{
if( m_renderMode.subSection( ) )
{
double sr = m_renderMode.startRow( );
if( sr < 1 )
y += ( int ) ( m_renderMode.height( ) * sr + 0.5 );
else
y += ( int ) sr;
}
if( x >= 0 && x < m_image.width( ) &&
y >= 0 && y < m_image.height( ) )
m_image.setPixel( x, y, c );
}
/**
void PMPovrayRenderWidget::slotWroteStdin( KProcess* )
{
if( m_pProcess )
m_pProcess->closeStdin( );
m_data.resize( 0 );
}
*/
void PMPovrayRenderWidget::slotRenderingFinished( KProcess* )
{
if( m_pProcess->normalExit( ) )
emit( finished( m_pProcess->exitStatus( ) ) );
else
emit( finished( -1000 ) );
cleanup( );
}
void PMPovrayRenderWidget::paintEvent( TQPaintEvent* ev )
{
if( !m_bPixmapUpToDate )
{
if( !m_image.isNull( ) )
m_pixmap.convertFromImage( m_image );
m_bPixmapUpToDate = true;
}
bitBlt( this, ev->rect( ).left( ), ev->rect( ).top( ),
&m_pixmap, ev->rect( ).left( ), ev->rect( ).top( ),
ev->rect( ).width( ), ev->rect( ).height( ), CopyROP );
}
void PMPovrayRenderWidget::cleanup( )
{
if( m_pProcess )
delete m_pProcess;
m_pProcess = 0;
if( m_pTempFile )
{
m_pTempFile->unlink( );
delete m_pTempFile;
}
m_pTempFile = 0;
}
TQSize PMPovrayRenderWidget::sizeHint( ) const
{
TQSize s;
if( m_image.isNull( ) )
s = TQSize( 200, 200 );
else
s = m_image.size( );
return s.expandedTo( minimumSize( ) );
}
void PMPovrayRenderWidget::saveConfig( KConfig* cfg )
{
cfg->setGroup( "Povray" );
#if ( ( TDE_VERSION_MAJOR == 3 ) && ( TDE_VERSION_MINOR <= 1 ) )
cfg->writeEntry( "PovrayCommand", s_povrayCommand );
cfg->writeEntry( "LibraryPaths", s_libraryPaths );
#else
cfg->writePathEntry( "PovrayCommand", s_povrayCommand );
cfg->writePathEntry( "LibraryPaths", s_libraryPaths );
#endif
}
void PMPovrayRenderWidget::restoreConfig( KConfig* cfg )
{
cfg->setGroup( "Povray" );
#if ( ( TDE_VERSION_MAJOR == 3 ) && ( TDE_VERSION_MINOR <= 1 ) )
s_povrayCommand = cfg->readEntry( "PovrayCommand", s_povrayCommand );
s_libraryPaths = cfg->readListEntry( "LibraryPaths" );
#else
s_povrayCommand = cfg->readPathEntry( "PovrayCommand", s_povrayCommand );
s_libraryPaths = cfg->readPathListEntry( "LibraryPaths" );
#endif
}
void PMPovrayRenderWidget::startDrag( )
{
TQImageDrag* d = new TQImageDrag( m_image, this );
d->dragCopy( );
}
#include "pmpovrayrenderwidget.moc"