00001 #ifndef _DiffSTP_ 00002 #define _DiffSTP_ 00003 #include "transportbase.h" 00004 00005 00006 class DiffusiveStep 00007 { 00008 TransportBase *trans; 00009 DynamicBase *_dyn; 00010 public: 00011 DiffusiveStep(){trans = NULL;_dyn=NULL;} 00012 virtual ~DiffusiveStep(){} 00013 00014 void setTransport(TransportBase &atrans){trans=&atrans;} 00015 TransportBase& getTransport(){assert(trans);return *trans;} 00016 00017 void setDynamic(DynamicBase &dyn){_dyn=&dyn;} 00018 DynamicBase& getDynamic(){assert(_dyn);return *_dyn;} 00019 00020 00021 00022 virtual void iterate(double dt)=0; 00023 virtual void printOutput()=0; 00024 00025 00026 }; 00027 00028 #endif