Initial import

1 files changed, 402 insertions, 0 deletions

diff --git a/src/file.cpp b/src/file.cpp
new file mode 100644
index 0000000..8b6106e
--- /dev/null
+++ b/src/file.cpp
@@ -0,0 +1,402 @@
+/***************************************************************************
+*   Copyright (C) 2004-2009 by Thomas Fischer                             *
+*   [email protected]                                             *
+*                                                                         *
+*   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.                                   *
+*                                                                         *
+*   This program is distributed in the hope that it will be useful,       *
+*   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
+*   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
+*   GNU General Public License for more details.                          *
+*                                                                         *
+*   You should have received a copy of the GNU General Public License     *
+*   along with this program; if not, write to the                         *
+*   Free Software Foundation, Inc.,                                       *
+*   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
+***************************************************************************/
+#include <qfile.h>
+#include <qtextstream.h>
+#include <qiodevice.h>
+#include <qstringlist.h>
+#include <qregexp.h>
+
+#include <settings.h>
+#include <file.h>
+#include <entry.h>
+#include <element.h>
+#include <macro.h>
+#include <comment.h>
+#include <fileexporter.h>
+
+namespace BibTeX
+{
+
+    File::File( ) : QObject( ), fileName( QString::null )
+    {
+        // nothing
+    }
+
+    File::~File()
+    {
+        for ( ElementList::iterator it = elements.begin(); it != elements.end(); it++ )
+            delete * it;
+    }
+
+    unsigned int File::count() const
+    {
+        return elements.count();
+    }
+
+    Element* File::at( const unsigned int index )
+    {
+        return *( elements.at( index ) );
+    }
+
+    void File::append( const File *other, const Element *after )
+    {
+        for ( ElementList::ConstIterator it = other->constBegin(); it != other->constEnd(); it++ )
+            appendElement( cloneElement( *it ), after );
+    }
+
+    void File::appendElement( Element *element, const Element *after )
+    {
+        if ( after == NULL )
+            elements.append( element );
+        else
+        {
+            for ( ElementList::iterator it = elements.begin() ; it != elements.end(); it++ )
+                if (( *it ) == after )
+                {
+                    elements.insert( ++it, element );
+                    break;
+                }
+        }
+    }
+
+    void File::deleteElement( Element *element )
+    {
+        bool found = false;
+        for ( ElementList::iterator it = elements.begin(); it != elements.end(); it++ )
+            if ( ( found = ( *it == element ) ) )
+            {
+                elements.remove( it );
+                delete element;
+                break;
+            }
+
+        if ( !found )
+            qDebug( "BibTeX::File got told to delete an element which is not in this file." );
+    }
+
+    Element* File::cloneElement( Element *element )
+    {
+        Entry * entry = dynamic_cast<Entry*>( element );
+        if ( entry )
+            return new Entry( entry );
+        else
+        {
+            Macro *macro = dynamic_cast<Macro*>( element );
+            if ( macro )
+                return new Macro( macro );
+            else
+            {
+                Comment *comment = dynamic_cast<Comment*>( element );
+                if ( comment )
+                    return new Comment( comment );
+                else
+                    return NULL;
+            }
+        }
+    }
+
+    Element *File::containsKey( const QString &key )
+    {
+        for ( ElementList::iterator it = elements.begin(); it != elements.end(); it++ )
+        {
+            Entry* entry = dynamic_cast<Entry*>( *it );
+            if ( entry != NULL )
+            {
+                if ( entry->id() == key )
+                    return entry;
+            }
+            else
+            {
+                Macro* macro = dynamic_cast<Macro*>( *it );
+                if ( macro != NULL )
+                {
+                    if ( macro->key() == key )
+                        return macro;
+                }
+            }
+        }
+
+        return NULL;
+    }
+
+    const Element *File::containsKeyConst( const QString &key ) const
+    {
+        for ( ElementList::const_iterator it = elements.begin(); it != elements.end(); it++ )
+        {
+            Entry* entry = dynamic_cast<Entry*>( *it );
+            if ( entry != NULL )
+            {
+                if ( entry->id() == key )
+                    return entry;
+            }
+            else
+            {
+                Macro* macro = dynamic_cast<Macro*>( *it );
+                if ( macro != NULL )
+                {
+                    if ( macro->key() == key )
+                        return macro;
+                }
+            }
+        }
+
+        return NULL;
+    }
+
+    QStringList File::allKeys()
+    {
+        QStringList result;
+
+        for ( ElementList::iterator it = elements.begin(); it != elements.end(); it++ )
+        {
+            Entry* entry = dynamic_cast<Entry*>( *it );
+            if ( entry != NULL )
+                result.append( entry->id() );
+            else
+            {
+                Macro* macro = dynamic_cast<Macro*>( *it );
+                if ( macro != NULL )
+                    result.append( macro->key() );
+            }
+        }
+
+        return result;
+    }
+
+    QString File::text()
+    {
+        QString result;
+
+        for ( ElementList::iterator it = elements.begin(); it != elements.end(); it++ )
+        {
+            result.append(( *it )->text() );
+            result.append( "\n" );
+        }
+
+        return result;
+    }
+
+    File::ElementList::iterator File::begin()
+    {
+        return elements.begin();
+    }
+
+    File::ElementList::iterator File::end()
+    {
+        return elements.end();
+    }
+
+    File::ElementList::ConstIterator File::constBegin() const
+    {
+        return elements.constBegin();
+    }
+
+    File::ElementList::ConstIterator File::constEnd() const
+    {
+        return elements.constEnd();
+    }
+
+    QStringList File::getAllValuesAsStringList( const EntryField::FieldType fieldType ) const
+    {
+        QStringList result;
+        for ( ElementList::ConstIterator eit = elements.constBegin(); eit != elements.constEnd(); ++eit )
+        {
+            Entry* entry = dynamic_cast<Entry*>( *eit );
+            EntryField * field = NULL;
+            if ( entry != NULL && ( field = entry->getField( fieldType ) ) != NULL )
+            {
+                QValueList<ValueItem*> valueItems = field->value()->items;
+                for ( QValueList<ValueItem*>::ConstIterator vit = valueItems.begin(); vit != valueItems.end(); ++vit )
+                {
+                    switch ( fieldType )
+                    {
+                    case EntryField::ftKeywords :
+                        {
+                            KeywordContainer *container = dynamic_cast<KeywordContainer*>( *vit );
+                            if ( container != NULL )
+                                for ( QValueList<Keyword*>::ConstIterator kit = container->keywords.constBegin(); kit != container->keywords.constEnd(); ++kit )
+                                {
+                                    QString text = ( *kit )->text();
+                                    if ( !result.contains( text ) )
+                                        result.append( text );
+                                }
+                        }
+                        break;
+                    case EntryField::ftEditor :
+                    case EntryField::ftAuthor :
+                        {
+                            PersonContainer *container = dynamic_cast<PersonContainer*>( *vit );
+                            if ( container != NULL )
+                                for ( QValueList<Person*>::ConstIterator pit = container->persons.constBegin(); pit != container->persons.constEnd(); ++pit )
+                                {
+                                    QString text = ( *pit )->text();
+                                    if ( !result.contains( text ) )
+                                        result.append( text );
+                                }
+                        }
+                        break;
+                    default:
+                        {
+                            QString text = ( *vit )->text();
+                            if ( !result.contains( text ) )
+                                result.append( text );
+                        }
+                    }
+                }
+            }
+        }
+
+        result.sort();
+        return result;
+    }
+
+    QMap<QString, int> File::getAllValuesAsStringListWithCount( const EntryField::FieldType fieldType ) const
+    {
+        QMap<QString, int> result;
+        for ( ElementList::ConstIterator eit = elements.begin(); eit != elements.end(); ++eit )
+        {
+            Entry* entry = dynamic_cast<Entry*>( *eit );
+            EntryField * field = NULL;
+            if ( entry != NULL && ( field = entry->getField( fieldType ) ) != NULL )
+            {
+                QValueList<ValueItem*> valueItems = field->value()->items;
+                for ( QValueList<ValueItem*>::ConstIterator vit = valueItems.begin(); vit != valueItems.end(); ++vit )
+                {
+                    switch ( fieldType )
+                    {
+                    case EntryField::ftKeywords :
+                        {
+                            KeywordContainer *container = dynamic_cast<KeywordContainer*>( *vit );
+                            if ( container != NULL )
+                                for ( QValueList<Keyword*>::ConstIterator kit = container->keywords.constBegin(); kit != container->keywords.constEnd(); ++kit )
+                                {
+                                    QString text = ( *kit )->text();
+                                    if ( !result.contains( text ) )
+                                        result[text] = 1;
+                                    else
+                                        result[text] += 1;
+                                }
+                        }
+                        break;
+                    case EntryField::ftEditor :
+                    case EntryField::ftAuthor :
+                        {
+                            PersonContainer *container = dynamic_cast<PersonContainer*>( *vit );
+                            if ( container != NULL )
+                                for ( QValueList<Person*>::ConstIterator pit = container->persons.constBegin(); pit != container->persons.constEnd(); ++pit )
+                                {
+                                    QString text = ( *pit )->text();
+                                    if ( !result.contains( text ) )
+                                        result[text] = 1;
+                                    else
+                                        result[text] += 1;
+                                }
+                        }
+                        break;
+                    default:
+                        {
+                            QString text = ( *vit )->text();
+                            if ( !result.contains( text ) )
+                                result[text] = 1;
+                            else
+                                result[text] += 1;
+                        }
+                    }
+                }
+            }
+        }
+
+        return result;
+    }
+
+    void File::replaceValue( const QString& oldText, const QString& newText, const EntryField::FieldType fieldType )
+    {
+        qDebug( "Renaming all occurrences of '%s' to '%s' for fields of type '%s'", oldText.latin1(), newText.latin1(), EntryField::fieldTypeToString( fieldType ).latin1() );
+
+        for ( ElementList::ConstIterator it = elements.begin(); it != elements.end(); it++ )
+        {
+            Entry* entry = dynamic_cast<Entry*>( *it );
+            if ( entry != NULL )
+            {
+                if ( fieldType != EntryField::ftUnknown )
+                {
+                    EntryField * field = entry->getField( fieldType );
+                    if ( field != NULL )
+                        field->value() ->replace( oldText, newText );
+                }
+            }
+        }
+    }
+
+    BibTeX::Entry *File::completeReferencedFieldsConst( const BibTeX::Entry *entry ) const
+    {
+        BibTeX::Entry *myEntry = new BibTeX::Entry( entry );
+        completeReferencedFields( myEntry );
+        return myEntry;
+    }
+
+    void File::completeReferencedFields( BibTeX::Entry *entry ) const
+    {
+        BibTeX::EntryField *crossRefField = entry->getField( BibTeX::EntryField::ftCrossRef );
+        const BibTeX::Entry *parent = NULL;
+        if ( crossRefField != NULL && ( parent = dynamic_cast<const BibTeX::Entry*>( containsKeyConst( crossRefField->value()->text() ) ) ) != NULL )
+        {
+            for ( int ef = ( int )BibTeX::EntryField::ftAbstract; ef <= ( int )BibTeX::EntryField::ftYear; ++ef )
+            {
+                BibTeX::EntryField *entryField = entry->getField(( BibTeX::EntryField::FieldType ) ef );
+                if ( entryField == NULL )
+                {
+                    BibTeX::EntryField *parentEntryField = parent->getField(( BibTeX::EntryField::FieldType ) ef );
+                    if ( parentEntryField != NULL )
+                    {
+                        entryField = new BibTeX::EntryField(( BibTeX::EntryField::FieldType )ef );
+                        entryField->setValue( parentEntryField->value() );
+                        entry->addField( entryField );
+                    }
+                }
+            }
+
+            BibTeX::EntryField *entryField = entry->getField( EntryField::ftBookTitle );
+            BibTeX::EntryField *parentEntryField = parent->getField( EntryField::ftTitle );
+            if (( entry->entryType() == Entry::etInProceedings || entry->entryType() == Entry::etInBook ) && entryField == NULL && parentEntryField != NULL )
+            {
+                entryField = new BibTeX::EntryField( EntryField::ftBookTitle );
+                entryField->setValue( parentEntryField->value() );
+                entry->addField( entryField );
+            }
+        }
+
+        for ( int ef = ( int )BibTeX::EntryField::ftAbstract; ef <= ( int )BibTeX::EntryField::ftYear; ++ef )
+        {
+            BibTeX::EntryField *entryField = entry->getField(( BibTeX::EntryField::FieldType ) ef );
+            if ( entryField != NULL && entryField->value() != NULL && !entryField->value()->items.isEmpty() )
+            {
+                MacroKey *macroKey = dynamic_cast<MacroKey*>( entryField->value()->items.first() );
+                const Macro *macro = NULL;
+                if ( macroKey != NULL && ( macro = dynamic_cast<const Macro*>( containsKeyConst( macroKey->text() ) ) ) != NULL )
+                    entryField->setValue( macro->value() );
+            }
+        }
+    }
+
+}
+
+#include "file.moc"
+