---

netloader.h

/***************************************************************************
                          netloader.h  -  description
                             -------------------
    begin                : Mon Apr 29 2002
    copyright            : (C) 2002 by Rüdiger Koch
    email                : rkoch@rkoch.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.                                   *
 *                                                                         *
 ***************************************************************************/

#ifndef NETLOADER_H
#define NETLOADER_H

using namespace std;
#include <amygdala/types.h>
#if GCC_VERSION >= 30000
    #include <ext/hash_map>
#else
    #include <hash_map>
#endif
#include <string>
#include <strings.h>
#include <stdio.h>


#include <amygdala/neuron.h>

class Layer;

class NeuronFactoryBase {
public:
    NeuronFactoryBase() {}
    virtual Neuron * NewNeuron(AmIdInt id) = 0;
    virtual string GetType() = 0;
};

template<class neuron>
class NeuronFactory : public NeuronFactoryBase {
public:
    NeuronFactory(){
    }
    virtual ~NeuronFactory(){}

    virtual Neuron * NewNeuron(AmIdInt id){
        neuron *n = new neuron(id);
        return n;
    }
    virtual string GetType() {
        // We instantiate a dummy neuron here so we can call the virtual
        // function.
        neuron n(0);
        return n.ClassId();
    }
};


class NetLoader {
public:
    NetLoader(Network *_net);
    ~NetLoader();
    void RegisterNeuronClass(NeuronFactoryBase *factory);
    void LoadXML(const char *filename);
    void SaveXML(string filename);
    Layer* BuildLayer(int size, int startNrnId, LayerConstants layerConst, string neuronType);

    void SAXStartElement1(const char *name, const char **attrs);
    void SAXEndElement1(const char *name);

    void SAXStartElement2(const char *name, const char **attrs);
    void SAXEndElement2(const char *name);
    void SAXError();

protected: // Protected attributes
    struct eqstr {
        bool operator()(const char* s1, const char* s2) const {
            return strcmp(s1, s2) == 0;
        }
    };
    hash_map <const char*, NeuronFactoryBase*, hash<const char*>, eqstr > registry;

    Network* net;
    Layer *currLayer;
    AmIdInt currNeuron;
    unsigned int saxErrors;

private: // Private methods
    NetLoader();


protected: // Protected methods
    void WriteSynapses(FILE *netFile, Neuron* preNrn);
    void SaveLayers(FILE *netFile);
    string LayerTypeStr(LayerType lType);
    void SAXPass(const char* filename, int pass);


    void ParseLayerConstants(LayerConstants *layerConst, char **name, const char **attrs);
};

#endif

---