source: cpp/frams/_demos/geometry/geometrytestutils.cpp @ 662

Last change on this file since 662 was 662, checked in by Maciej Komosinski, 3 years ago
  • Property svn:eol-style set to native
File size: 8.4 KB
Line 
1// This file is a part of Framsticks SDK.  http://www.framsticks.com/
2// Copyright (C) 1999-2017  Maciej Komosinski and Szymon Ulatowski.
3// See LICENSE.txt for details.
4
5#include "geometrytestutils.h"
6
7#include "../genotypeloader.h"
8#include "frams/genetics/preconfigured.h"
9#include <common/virtfile/stdiofile.h>
10#include <common/loggers/loggertostdout.h>
11#include <math.h>
12#include <stdio.h>
13#include <stdlib.h>
14#include <time.h>
15
16int printGenotypesList(const char *file)
17{
18        long count = 0;
19        long totalSize = 0;
20        MiniGenotypeLoader loader(file);
21        MiniGenotype *genotype;
22       
23        while (genotype = loader.loadNextGenotype())
24        {
25                count++;
26                totalSize += genotype->genotype.len();
27               
28                fprintf(stderr, "%d. (%6d chars) %s\n", count, genotype->genotype.len(),
29                        genotype->name.c_str());
30        }
31       
32        if (loader.getStatus() == MiniGenotypeLoader::OnError)
33        {
34                fprintf(stderr, "Error: %s\n", loader.getError().c_str());
35                return 2;
36        }
37        else
38        {
39                fprintf(stderr, "\ntotal: %d items, %d chars\n", count, totalSize);
40                return 0;
41        }
42}
43
44class TestInvoker
45{
46        public:
47                virtual void operator()(Model &model) = 0;
48};
49
50int executeTestUsingLoadedModel(const char *file, const char *genoId, TestInvoker &test)
51{
52        const char* genoName = genoId;
53        const int genoIndex = isdigit(genoId[0]) ? atol(genoId) : 0;
54        long count = 0;
55        MiniGenotypeLoader loader(file);
56        MiniGenotype *genotype;
57       
58        while (genotype = loader.loadNextGenotype())
59        {
60                count++;
61               
62                if ((genoIndex == count) || (strcmp(genotype->name.c_str(), genoName) == 0))
63                {
64                        Model model(genotype->genotype);
65                       
66                        if (!model.isValid())
67                        {
68                                fprintf(stderr, "Cannot build a valid Model from this genotype!\n");
69                                return 4;
70                        }
71                        SolidsShapeTypeModel sst_model(model);
72                        test(sst_model);
73                        return 0;
74                }
75        }
76       
77        if (loader.getStatus() == MiniGenotypeLoader::OnError)
78        {
79                fprintf(stderr, "Error: %s\n", loader.getError().c_str());
80                return 2;
81        }
82        else
83        {
84                fprintf(stderr, "Genotype %s not found in %s\n", genoId, file);
85                return 3;
86        }
87}
88
89int executeTestUsingRandomModel(int shape, TestInvoker &test)
90{
91        Model model;
92        model.open();
93
94        if ((shape < 1) || (shape > 3))
95        {
96                shape = (rand()%3) + 1;
97        }
98
99        Part *part = model.addNewPart(Part::Shape(shape));
100        GeometryTestUtils::randomizePositionScaleAndOrient(part);
101
102        model.close();
103        test(model);
104        GeometryTestUtils::describePart(part, stdout);
105        return 0;
106}
107
108class ModelBasedTestInvoker: public TestInvoker
109{
110        private:
111                void (*test)(Model &);
112        public:
113                ModelBasedTestInvoker(void (*_test)(Model &)):
114                        test(_test)
115                {}
116                void operator()(Model &model)
117                {
118                        test(model);
119                }
120};
121
122int GeometryTestUtils::execute(const SString header, int argc, char *argv[], void (*test)(Model &))
123{
124        LoggerToStdout messages_to_stdout(LoggerBase::Enable); //comment this object out to mute error/warning messages
125        StdioFileSystem_autoselect stdiofilesys;
126        PreconfiguredGenetics genetics;
127       
128        srand(time(NULL));
129
130        if ((argc == 3) && (strcmp("-l", argv[1]) == 0))
131        {
132                return printGenotypesList(argv[2]);
133        }
134       
135        if ((argc == 4) && (strcmp("-l", argv[1]) == 0))
136        {
137                ModelBasedTestInvoker invoker(test);
138                return executeTestUsingLoadedModel(argv[2], argv[3], invoker);
139        }
140       
141        if ((argc == 2) && (strcmp("-c", argv[1]) == 0))
142        {
143                ModelBasedTestInvoker invoker(test);
144                return executeTestUsingRandomModel(-1, invoker);
145        }
146       
147        if ((argc == 3) && (strcmp("-c", argv[1]) == 0) && isdigit(argv[2][0]))
148        {
149                int shape = atol(argv[2]);
150                ModelBasedTestInvoker invoker(test);
151                return executeTestUsingRandomModel(shape, invoker);
152        }
153       
154        fprintf(stderr,
155                "%s\n\n"
156                "argument lists:\n"
157                "-l FILENAME            - to print list of models in file\n"
158                "-l FILENAME GENO_ID    - to load model from file and run test\n"
159                "-c [SHAPE]             - to create simple random model and run test\n\n"
160                "FILENAME - name of file containing named f0 genotypes\n"
161                "GENO_ID - either genotype name or index (1-based)\n"
162                "SHAPE - 1=ellipsoid, 2=cuboid, 3=cylinder, others or none=random\n",
163                header.c_str());
164        return 1;
165}
166
167class ModelAndDensityBasedTestInvoker: public TestInvoker
168{
169        private:
170                void (*test)(Model &, const double);
171                double density;
172        public:
173                ModelAndDensityBasedTestInvoker(void (*_test)(Model &, const double), double _density):
174                        test(_test),
175                        density(_density)
176                {}
177               
178                void operator()(Model &model)
179                {
180                        test(model, density);
181                }
182};
183
184int GeometryTestUtils::execute(const SString header, int argc, char *argv[],
185        void (*test)(Model &, const double))
186{
187        LoggerToStdout messages_to_stdout(LoggerBase::Enable); //comment this object out to mute error/warning messages
188        StdioFileSystem_autoselect stdiofilesys;
189        PreconfiguredGenetics genetics;
190       
191        srand(time(NULL));
192
193        if ((argc == 3) && (strcmp("-l", argv[1]) == 0))
194        {
195                return printGenotypesList(argv[2]);
196        }
197
198        if ((argc == 5) && (strcmp("-l", argv[1]) == 0) && isdigit(argv[4][0]))
199        {
200                double density = atol(argv[4]);
201                ModelAndDensityBasedTestInvoker invoker(test, density);
202                return executeTestUsingLoadedModel(argv[2], argv[3], invoker);
203        }
204       
205        if ((argc == 3) && (strcmp("-c", argv[1]) == 0) && isdigit(argv[2][0]))
206        {
207                double density = atol(argv[2]);
208                ModelAndDensityBasedTestInvoker invoker(test, density);
209                return executeTestUsingRandomModel(-1, invoker);
210        }
211       
212        if ((argc == 4) && (strcmp("-c", argv[1]) == 0) && isdigit(argv[2][0]) && isdigit(argv[3][0]))
213        {
214                double density = atol(argv[2]);
215                int shape = atol(argv[3]);
216                ModelAndDensityBasedTestInvoker invoker(test, density);
217                return executeTestUsingRandomModel(shape, invoker);
218        }
219       
220        fprintf(stderr,
221                "%s\n\n"
222                "argument list:\n"
223                "-l FILENAME                    - to print the list of models in file\n"
224                "-l FILENAME GENO_ID DENSITY    - to load the model from the file and run test\n"
225                "-c DENSITY [SHAPE]             - to create a simple random model and run test\n\n"
226                "FILENAME - name of the file containing named f0 genotypes\n"
227                "GENO_ID - either genotype name or index (1-based)\n"
228                "DENSITY - minimal number of samples per unit\n"
229                "SHAPE - 1=ellipsoid, 2=cuboid, 3=cylinder, others or none=random\n",
230                header.c_str());
231        return 1;
232}
233
234void GeometryTestUtils::addAnchorToModel(Model &model)
235{
236        Part *part = model.addNewPart(Part::SHAPE_ELLIPSOID);
237       
238        part->p = Pt3D(0);
239        part->scale = Pt3D(0.1);
240        part->vcolor = Pt3D(1.0, 0.0, 1.0);
241       
242        addAxesToModel(Pt3D(0.5), Orient(Orient_1), Pt3D(0.0), model);
243}
244
245void GeometryTestUtils::addPointToModel(const Pt3D &markerLocation, Model &model)
246{
247        Part *anchor = model.getPart(0);
248        Part *part = model.addNewPart(Part::SHAPE_ELLIPSOID);
249       
250        part->p = Pt3D(markerLocation);
251        part->scale = Pt3D(0.05);
252        part->vcolor = Pt3D(1.0, 1.0, 0.0);
253       
254        model.addNewJoint(anchor, part, Joint::SHAPE_FIXED);
255}
256
257void GeometryTestUtils::addAxesToModel(const Pt3D &sizes, const Orient &axes, const Pt3D &center,
258        Model &model)
259{
260        Part *anchor = model.getPart(0);
261        Part *part;
262       
263        part = model.addNewPart(Part::SHAPE_CUBOID);
264        part->scale = Pt3D(sizes.x, 0.05, 0.05);
265        part->setOrient(axes);
266        part->p = center;
267        part->vcolor = Pt3D(1.0, 0.0, 0.0);
268        model.addNewJoint(anchor, part, Joint::SHAPE_FIXED);
269       
270        part = model.addNewPart(Part::SHAPE_CUBOID);
271        part->scale = Pt3D(0.05, sizes.y, 0.05);
272        part->setOrient(axes);
273        part->p = center;
274        part->vcolor = Pt3D(0.0, 1.0, 0.0);
275        model.addNewJoint(anchor, part, Joint::SHAPE_FIXED);
276       
277        part = model.addNewPart(Part::SHAPE_CUBOID);
278        part->scale = Pt3D(0.05, 0.05, sizes.z);
279        part->setOrient(axes);
280        part->p = center;
281        part->vcolor = Pt3D(0.0, 0.0, 1.0);
282        model.addNewJoint(anchor, part, Joint::SHAPE_FIXED);
283}
284
285void GeometryTestUtils::mergeModels(Model &target, Model &source)
286{
287        Part *targetAnchor = target.getPart(0);
288        Part *sourceAnchor = source.getPart(0);
289       
290        target.moveElementsFrom(source);
291       
292        target.addNewJoint(targetAnchor, sourceAnchor, Joint::SHAPE_FIXED);
293}
294
295double frand(double from, double width)
296{
297        return from + width * ((rand()%10000) / 10000.0);
298}
299
300void GeometryTestUtils::randomizePositionScaleAndOrient(Part *part)
301{
302        part->p = Pt3D(frand(1.5, 1.0), frand(1.5, 1.0), frand(1.5, 1.0));
303        part->scale = Pt3D(frand(0.1, 0.9), frand(0.1, 0.9), frand(0.1, 0.9));
304        part->setRot(Pt3D(frand(0.0, M_PI), frand(0.0, M_PI), frand(0.0, M_PI)));
305}
306
307void GeometryTestUtils::describePart(const Part *part, FILE *output)
308{
309        fprintf(output, "# shape=%d\n", part->shape);
310        fprintf(output, "# x=%f\n", part->p.x);
311        fprintf(output, "# y=%f\n", part->p.y);
312        fprintf(output, "# z=%f\n", part->p.z);
313        fprintf(output, "# sx=%f\n", part->scale.x);
314        fprintf(output, "# sy=%f\n", part->scale.y);
315        fprintf(output, "# sz=%f\n", part->scale.z);
316        fprintf(output, "# rx=%f\n", part->rot.x);
317        fprintf(output, "# ry=%f\n", part->rot.y);
318        fprintf(output, "# rz=%f\n", part->rot.z);
319}
Note: See TracBrowser for help on using the repository browser.