QtFileDialog.cxx

Go to the documentation of this file.

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "QtFileDialog.h"

#include "CanvasWindow.h"
#include "ListDialog.h"

#include "datasrcs/DataSourceException.h"
#include "datasrcs/NTupleController.h"

#ifdef HAVE_CFITSIO
#include "fits/FitsController.h"
#endif

#ifdef HAVE_ROOT
#include "root/RootController.h"
#endif

#include <qmessagebox.h>

using std::exception;
using std::string;
using std::vector;

/* Use `=' syntax so that Doxygen picks it up. */
const string QtFileDialog::s_doc_suffix = ".hpo";
const string QtFileDialog::s_data_suffix = ".tnt";

QtFileDialog::
QtFileDialog ( const QString & dirName,
               QWidget * parent,
               const char * name,
               bool modal )
  : QFileDialog ( dirName, QString::null, parent, name, modal )
{
}

void
QtFileDialog::
createOpenFilter ()
{
  const string doc ( s_doc_suffix );
  const string xml ( ".xml" );
  const string tnt ( ".tnt" );

#if __GNUC__ < 3
#define STRING_CLEAR_DEFECT
#endif

#ifdef STRING_CLEAR_DEFECT
  m_open_filter.erase();
#else
  m_open_filter.clear();
#endif

  m_open_filter = "Documents (*";
  m_open_filter += doc;
  m_open_filter += " *" + xml;

  m_open_filter += ");;Text ntuples (*";
  m_open_filter += tnt;
  m_open_filter += ")";

#ifdef HAVE_ROOT
  const string root ( ".root" );
  m_open_filter += ";;ROOT ntuples(*";
  m_open_filter += root;
  m_open_filter += ")";
#endif

#ifdef HAVE_CFITSIO
  const string fits ( ".fits" );
  m_open_filter += ";;FITS ntuples(*";
  m_open_filter += fits;
  m_open_filter += ")";
#endif

  m_open_filter += ";;All files (";
  m_open_filter += "*.*";
  m_open_filter += ")";
}
void
QtFileDialog::
createBrowseFilter ()
{
  const string doc ( s_doc_suffix );
  const string xml ( ".xml" );
  const string tnt ( ".tnt" );

#if __GNUC__ < 3
#define STRING_CLEAR_DEFECT
#endif

#ifdef STRING_CLEAR_DEFECT
  m_open_filter.erase();
#else
  m_open_filter.clear();
#endif
  m_open_filter = "Documents (*";
  m_open_filter += doc;
  m_open_filter += " *" + xml;
  m_open_filter += ")";

  m_open_filter += ";;All files (";
  m_open_filter += "*.*";
  m_open_filter += ")";
}

const std::string &
QtFileDialog::
getDocSuffix ()
{
  return s_doc_suffix;
}

const std::string &
QtFileDialog::
getTextSuffix ()
{
  return s_data_suffix;
}

bool
QtFileDialog::
isDocSuffix ( const std::string & suffix )
{
  return suffix == s_doc_suffix || suffix == ".xml";
}

bool
QtFileDialog::
isTextSuffix ( const std::string & suffix )
{
  return suffix == s_data_suffix;
}

bool
QtFileDialog::
isRootSuffix ( const std::string & suffix ) const
{
  return suffix == ".root";
}

bool
QtFileDialog::
isFitsSuffix ( const std::string & suffix ) const
{
  return suffix == ".fits";
}

std::string
QtFileDialog::
getSaveDocFilename ()
{
  QString filter ( "Documents (*" );
#if defined(QT_NO_STL) || QT_VERSION < 0x030100
  filter += s_doc_suffix.c_str();
#else
  filter += s_doc_suffix;
#endif

  filter += ")";

  QString filename;
  while ( true ) {
    setFilters ( filter );
    setCaption ( "Save document as ..." );
    setMode ( QFileDialog::AnyFile );

    string tmp;
    int retval = QFileDialog::exec ();
    if ( retval == QDialog::Rejected ) return tmp;

    filename = selectedFile ();
      if ( filename.isEmpty () ) return tmp;

#if defined(QT_NO_STL)  || QT_VERSION < 0x030100
    if ( filename.endsWith ( s_doc_suffix.c_str() ) == false ) {
      filename += s_doc_suffix.c_str();
    }
#else
    if ( filename.endsWith ( s_doc_suffix ) == false ) {
      filename += s_doc_suffix;
    }
#endif

    QFileInfo info ( filename );
    filename = info.absFilePath();

    bool yes = info.exists ();
    if ( yes == false ) break;

    string message ( "File exists. \n\n" );
    message += "Over write existing file?";
    int result = QMessageBox::warning ( this, // parent
                                        "Warning", // caption
                                        message.c_str (), //
                                        QMessageBox::Yes,
                                        QMessageBox::No );
    if ( result == QMessageBox::Yes ) break;
  }
  string name = filename.latin1();

  return name;
}

int
QtFileDialog::
exec ()
{
  if ( m_open_filter.empty () ) {
    createOpenFilter ();
  }
  setFilters ( QString ( m_open_filter.c_str() ) );
  setMode ( QFileDialog::ExistingFile );

  return QFileDialog::exec ();
}


void
QtFileDialog::
openTextTuple ( const std::string & filename )
{
  NTupleController * controller = NTupleController::instance ();
  try {
    //    DataSource * tuple =
    controller -> createNTuple ( filename );
  }
  catch ( const DataSourceException & e )
    {

      QString message( "Unable to open file\n" );
      message.append(e.what ());

      QMessageBox::critical ( 0, // parent
                              "HippoDraw",
                              message,
                              QMessageBox::Ok,
                              Qt::NoButton );
    }

}

bool
QtFileDialog::
isFitsFile ( const std::string & fn ) const
{
  bool yes = false;
#ifdef HAVE_CFITSIO
  FitsController * controller = FitsController::instance ();
  FitsFile * fits_file = 0;
  try {
    fits_file = controller ->openFile ( fn );
   if ( fits_file != 0 ) yes = true;
  }
  catch ( ... ) {
    // ignore, was not FITS file
  }
#endif

  return yes;
}

void
QtFileDialog::
openFitsTuple ( const std::string & filename )
{
#ifdef HAVE_CFITSIO
  unsigned int index = 0;

  FitsController * controller = FitsController::instance ();
  const vector < string > & names
    = controller -> getNTupleNames ( filename );

  if ( names.size () > 1 ) {
    if ( names.size () >= 2  ) {
      ListDialog * dialog = new ListDialog ( this ); // take defaults for rest
      dialog -> setNames ( names );
      int retval = dialog -> exec ();

      if ( retval == QDialog::Accepted ) {
        index = dialog -> selectedItem ();
      }
      delete dialog;
      if ( retval == QDialog::Rejected ) return;
    }
    else { // was your typical empty image HDU
      index = 1;
    }
  }

  try {
    // FitsNTuple * tuple =
    controller -> createNTuple ( filename, names [ index ], index );
  }
  catch ( const DataSourceException & e ) {
    QMessageBox::information ( this, // parent
                               "HippoDraw", // caption
                               e.what (),
                               QMessageBox::Ok,
                               Qt::NoButton );
    return;
  }

#endif
}

void
QtFileDialog::
openRootTuple ( const std::string & filename )
{
#ifdef HAVE_ROOT
  unsigned int index = 0;

  RootController * controller = RootController::instance ();
  const vector < string > & tree_names
    = controller -> getNTupleNames ( filename );
  if ( tree_names.size () > 1 ) {
    ListDialog * dialog = new ListDialog ( this ); // take defaults for rest
    dialog -> setNames ( tree_names );

    int retval = dialog -> exec ();

    if ( retval == QDialog::Rejected ) return;
    index = dialog -> selectedItem ();
    delete dialog;
  }
  if ( tree_names.empty () ) {
    string message ( "There were no ntuple objects found in this file." );
    message += "\nFile not opened.";

    QMessageBox::information ( this, // parent
                               "HippoDraw", // caption
                               message.c_str(),
                               QMessageBox::Ok,
                               Qt::NoButton );
    return;
  }

  try {
    //    RootNTuple * tuple =
    controller -> createNTuple ( filename, tree_names[index] );
  }
  catch ( const DataSourceException & e ) {
    QMessageBox::information ( this, // parent
                               "HippoDraw", // caption
                               e.what (),
                               QMessageBox::Ok,
                               Qt::NoButton );
    return;
  }

#endif
}

void
QtFileDialog::
openFileError ( const std::string & filename, exception & e )
{
  string message = e.what();
  QMessageBox::critical ( 0,
                          "File error",
                          message.c_str(),
                          QMessageBox::Ok,
                          0 );
}

---