aimsGraphW_d.h

Go to the documentation of this file.

/* This software and supporting documentation are distributed by
 *     Institut Federatif de Recherche 49
 *     CEA/NeuroSpin, Batiment 145,
 *     91191 Gif-sur-Yvette cedex
 *     France
 *
 * This software is governed by the CeCILL-B license under
 * French law and abiding by the rules of distribution of free software.
 * You can  use, modify and/or redistribute the software under the
 * terms of the CeCILL-B license as circulated by CEA, CNRS
 * and INRIA at the following URL "http://www.cecill.info".
 *
 * As a counterpart to the access to the source code and  rights to copy,
 * modify and redistribute granted by the license, users are provided only
 * with a limited warranty  and the software's author,  the holder of the
 * economic rights,  and the successive licensors  have only  limited
 * liability.
 *
 * In this respect, the user's attention is drawn to the risks associated
 * with loading,  using,  modifying and/or developing or reproducing the
 * software by the user in light of its specific status of free software,
 * that may mean  that it is complicated to manipulate,  and  that  also
 * therefore means  that it is reserved for developers  and  experienced
 * professionals having in-depth computer knowledge. Users are therefore
 * encouraged to load and test the software's suitability as regards their
 * requirements in conditions enabling the security of their systems and/or
 * data to be ensured and,  more generally, to use and operate it in the
 * same conditions as regards security.
 *
 * The fact that you are presently reading this means that you have had
 * knowledge of the CeCILL-B license and that you accept its terms.
 */

#ifndef AIMS_IO_AIMSGRAPHW_D_H
#define AIMS_IO_AIMSGRAPHW_D_H

#include <aims/io/aimsGraphW.h>
#include <aims/graph/graphmanip.h>
#include <aims/io/writer.h>
#include <aims/io/datatypecode.h>
#include <aims/io/finder.h>
#include <aims/data/data.h>
#include <graph/graph/graph.h>
#include <cartobase/smart/rcptr.h>
#include <cartobase/stream/fileutil.h>
#include <cartobase/stream/directory.h>
#include <list>
#include <vector>

namespace aims
{

  struct AimsGraphWriter_Private
  {
    AimsGraphWriter_Private();
    ~AimsGraphWriter_Private();
    std::string                                 filename;
    std::string                                 directory;
    std::string                                 elemfname;
    int                                         elemindex;
    GraphElementCode                            elemcode;
    AimsGraphWriter::SavingMode                 mode;
    AimsGraphWriter::SavingMode                 oldmode;
    AimsGraphWriter::SavingMode                 newmode;
    AimsGraphWriter::SavingMode                 prevmode;
    std::map<std::string,AimsGraphWriter::Wrapper *>    globalobjects;
    std::map<std::string, bool>                 globalmodified;
    int                                         genindexcount;
    bool                                        delayindex;
    bool                                        modified;
    bool                                        saveall;
    carto::rc_ptr<std::map<std::string,std::vector<int> > >     colors;
  };


  AimsGraphWriter_Private::AimsGraphWriter_Private()
    : colors( new std::map<std::string,std::vector<int> > )
  {
  }


  AimsGraphWriter_Private::~AimsGraphWriter_Private()
  {
    std::map<std::string,AimsGraphWriter::Wrapper *>::iterator  i, e 
      = globalobjects.end();
    for ( i=globalobjects.begin(); i!=e; ++i )
      {
        // i->second->destroy(); ? no, object can be kept in graph
        delete i->second;
      }
  }


  template<typename T> bool 
  AimsGraphWriter::ObjectWrapper<T>::write( const AimsGraphWriter & wp ) const
  {
    return wp.write( *data );
  }


  template<typename T> int 
  AimsGraphWriter::ObjectWrapper<T>::freeindex() const
  {
    typename T::const_iterator i = data->begin(), e = data->end();
    int n = 1;
    if( i != e && i->first == 0 )
      n = 0;
    for ( ; i!=e && i->first==n; ++i, ++n ) {}
    return n;
  }


  template<typename T> std::string 
  AimsGraphWriter::ObjectWrapper<T>::objectType() const
  {
    return ::carto::DataTypeCode<T>().objectType();
  }


  template<typename T> std::string 
  AimsGraphWriter::ObjectWrapper<T>::dataType() const
  {
    return carto::DataTypeCode<T>().dataType();
  }


  template<class T> 
  bool AimsGraphWriter::write( const T & obj ) const
  {
    // write in file or in global object
    if ( d->mode == Local )
      {
        if ( d->saveall || ( d->modified && !obj.empty() ) )
          {
            std::string dir = carto::FileUtil::dirname( d->elemfname );
            if( carto::FileUtil::fileStat( dir ).find( '+' ) 
                == std::string::npos )
              {
                carto::Directory        cd( dir );
                cd.makedirs();
              }
            Writer<T>   w( d->elemfname );
            return w.write( obj );
          }
        return true;
      }
    else
      {
        Wrapper *& glob = d->globalobjects[ d->elemcode.global_filename ];
        if ( !glob )
          {
            //cout << "create " << d->elemcode.global_filename << endl;
            carto::rc_ptr<T>    gobject;
            if ( !d->elemcode.global_attribute.empty() 
                 && info.graph->getProperty( d->elemcode.global_attribute, 
                                             gobject ) )
              glob = new ObjectWrapper<T>( gobject.get() );
            else
              {
                glob = new ObjectWrapper<T>( new T );
                T* data = ((ObjectWrapper<T> *) glob)->data;
                DataManip::adjustVoxelSize( *data, obj );
                DataManip::setHeaderInfo( *data, *info.graph );
              }
          }
        if( !d->delayindex && d->mode == Global )
          {
            /*cout << "insert index " << d->elemindex << " in " 
              << d->elemcode.global_filename << endl;*/
            DataManip::insertElement( *((ObjectWrapper<T> *) glob)->data, 
                                      d->elemindex, obj );
          }
      }

    return true;
  }


  template<class T> 
  bool AimsGraphWriter::defaultTakeObject( Process & p, const std::string &, 
                                           Finder & )
  {
    AimsGraphWriter     & wp = (AimsGraphWriter &) p;
    ElementInfo         & info = wp.info;
    carto::rc_ptr<T>    obj;
    
    try
      {
        if ( !info.element->getProperty( info.attribute, obj ) )
          return false;
        info.object = new ObjectWrapper<T>( obj.get() );
        return true;
      }
    catch( std::exception & e )
      {
        std::cerr << e.what() << std::endl;
        return false;
      }
  }

}

#endif