#include <cstdlib>
#include <iostream>
#include <boost/filesystem.hpp>
#include <cmath>

using namespace std;
namespace fs=boost::filesystem;



class Element
{
	protected:
	fs::path sciezka;
	public:
		double rozmiar;
		Element(){}
		Element(fs::path sc):sciezka(sc){}
		fs::path Sciezka(){return sciezka;}
		virtual double Rozmiar() =0;
};

class Plik : public Element{
	private :
	double  rozmiar;
		public: 
		Plik():Element(), rozmiar(0.0){} 
		Plik(fs::path s):Element(s){}
		
		double Rozmiar(){
			
			rozmiar = fs::file_size(Sciezka());
			return rozmiar;
		}
		
	void wyswietl(){
		cout<<Sciezka().string()<<endl;
	}
				
};

class Katalog : public Element{
	private:
	double rozmiar;
	vector<Element*>elements;
	public:
	Katalog():Element(),rozmiar(0.0){}
	Katalog(fs::path s):Element(s),rozmiar(0.0){}
	
	double Rozmiar(){
	//	rozmiar=fs::file_size(Sciezka());
		
	for(fs::directory_iterator iter(sciezka); iter!= fs::directory_iterator(); ++iter)
	{
		Element* element;
		if(is_directory(*iter))
			element = new Katalog(*iter);
		else
			element = new Plik(*iter);
			elements.push_back(element);
	}
	
	for(vector<Element*>::iterator iter = elements.begin();iter!= elements.end(); ++ iter)
		rozmiar += (*iter)->Rozmiar();
		
			return rozmiar;
	}
		
};


int main(){
	
//	vector<Element*>elementy;
	fs::path path ="./";
	Katalog *file = new Katalog(path);
	cout<<file->Rozmiar()<<endl;
	
	
	
	
return 0;
}