Added LinearRankSelection.

[genetic.git] / src / generation.h

diff --git a/src/generation.h b/src/generation.h
index c1adc73cb4fcb6299a2fcffb371839e858e1bb4e..88942ec0d9fd2a0cac8e19b3762bee8cef227da8 100644 (file)
--- a/src/generation.h
+++ b/src/generation.h
@@ -9,18 +9,65 @@
 using namespace std;
 
 namespace genetic {
+    /**
+     * Class representing Generation of given Chromosomes.
+     */
     template < typename _Chromosome >
     class Generation {
     protected:
+        /**
+         * Chromosomes in the given Generation
+         */
         vector<_Chromosome> chromosomes;
 
+        template<typename> friend class Generation;
     public:
+        /**
+         * Default constructor
+         */
+        Generation() {}
+
+        /**
+         * Class constructor. Initializes generation with the given chromosomes
+         *
+         * @param chromosomes vector containing Chromosomes to use in Generation
+         */
         Generation(vector<_Chromosome> chromosomes) {
             this->chromosomes = chromosomes;
         }
 
-        vector<_Chromosome> get() {
-            return this->chromosomes;
+        /** Copy constructor */
+        Generation(const Generation& generation)
+            : chromosomes(generation.chromosomes) {
+        }
+
+        /**
+         * Copy operator.
+         *
+         * @param generation Generation from which the Chromosomes should be copied.
+         * @return Generation instance containing copied Chromosomes
+         */
+        Generation& operator=(const Generation& generation){
+            this->chromosomes = generation.chromosomes;
+            return *this;
+        }
+
+        /**
+         * Returns number of Chromosomes in current Generation
+         *
+         * @return number of Chromosomes in current Generation
+         */
+        unsigned int size() const {
+            return this->chromosomes.size();
+        }
+
+        /**
+         * Returns i-th Chromosome in the current Generation
+         *
+         * @return i-th Chromosome in the current Generation
+         */
+        _Chromosome& operator[](unsigned int i) {
+            return this->chromosomes[i];
         }
     };
 }