X-Git-Url: https://git.dlugolecki.net.pl/?a=blobdiff_plain;f=src%2Fchromosome.h;h=08d24c92c5b422290c0c3198ebf0be700b418963;hb=c9db29f895be287471f280bbed6cfb5271de3803;hp=a20c793db43d1852f1a75e79d3757594746888dd;hpb=46357591de1336612d4c26170cd59c13543b5169;p=genetic.git

diff --git a/src/chromosome.h b/src/chromosome.h
index a20c793..08d24c9 100644
--- a/src/chromosome.h
+++ b/src/chromosome.h
@@ -3,36 +3,77 @@
 
 #include <vector>
 
-#include <gene.h>
+#include "gene.h"
 
 using namespace std;
 
 namespace genetic {
-    template < typename _Chromosome >
+    /**
+     * Chromosome of given Genes.
+     */
+    template < typename _Gene >
     class Chromosome {
+    public:
+        /**
+         * Type representing Chromosome Gene
+         */
+        typedef _Gene GeneType;
     protected:
-        vector<_Chromosome> genes;
+        /**
+         * Genes of the chromosome
+         */
+        vector<_Gene> genes;
 
+        template<typename> friend class Chromosome;
     public:
+        /**
+         * Default constructor
+         */
         Chromosome() {}
 
-        Chromosome(vector<_Chromosome> genes) {
+        /**
+         * Class constructor. Initializes Chromosome with the given Gene's
+         *
+         * @param genes vector containing Genes to use in Chromosome
+         */
+        Chromosome(vector<_Gene> genes) {
             this->genes = genes;
         }
 
         /** Copy constructor */
         Chromosome(const Chromosome& chromosome)
-            : genes(chromosome.get()) {
+            : genes(chromosome.genes) {
         }
 
-        Chromosome& operator=(const Chromosome&){
+        /**
+         * Copy operator.
+         *
+         * @param chromosome Chromosome from which the Genes should be copied.
+         * @return Chromosome instance containing copied Genes
+         */
+        Chromosome& operator=(const Chromosome& chromosome) {
+            this->genes = chromosome.genes;
             return *this;
         }
 
-        vector<_Chromosome> get() {
-            return this->genes;
+        /**
+         * Returns number of Genes in current Chromosome
+         *
+         * @return number of Genes in current Chromosome
+         */
+        unsigned int size() const {
+            return this->genes.size();
+        }
+
+        /**
+         * Returns i-th Gene in the current Chromosome
+         *
+         * @return i-th Gene in the current Chromosome
+         */
+        _Gene& operator[](unsigned int i) {
+            return this->genes[i];
         }
     };
 }
 
-#endif /* __CHROMOSOME_H */
\ No newline at end of file
+#endif /* __CHROMOSOME_H */