2 # Copyright (C) 2008-2009 EDF R&D
4 # This library is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU General Public
6 # License as published by the Free Software Foundation; either
7 # version 2.1 of the License.
9 # This library is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 # Lesser General Public License for more details.
14 # You should have received a copy of the GNU Lesser General Public
15 # License along with this library; if not, write to the Free Software
16 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
19 # Author : André RIBES (EDF R&D)
28 logging.basicConfig(level=logging.DEBUG)
30 #----------- Templates Part ---------------#
31 begin_catalog_file = """
32 # -*- coding: utf-8 -*-
34 # --------------------------------------------------------
35 # generated by AdaoCatalogGenerator at ${date}
36 # --------------------------------------------------------
41 JdC = JDC_CATA (code = 'ADAO',
43 regles = ( AU_MOINS_UN ('ASSIMILATION_STUDY'), AU_PLUS_UN ('ASSIMILATION_STUDY')),
48 def F_${data_name}(statut) : return FACT(statut = statut,
49 FROM = SIMP(statut = "o", typ = "TXM", into=(${data_into}), defaut=${data_default}),
50 SCRIPT_DATA = BLOC ( condition = " FROM in ( 'Script', ) ",
52 SCRIPT_FILE = SIMP(statut = "o", typ = "Fichier"),
54 STRING_DATA = BLOC ( condition = " FROM in ( 'String', ) ",
56 STRING = SIMP(statut = "o", typ = "TXM"),
58 FUNCTIONDICT_DATA = BLOC ( condition = " FROM in ( 'FunctionDict', ) ",
60 FUNCTIONDICT_FILE = SIMP(statut = "o", typ = "Fichier"),
66 def F_InitChoice() : return ("Background",
70 "ObservationOperator",
71 "AlgorithmParameters",
75 def F_Init(statut) : return FACT(statut = statut,
76 INIT_FILE = SIMP(statut = "o", typ = "Fichier"),
77 TARGET_LIST = SIMP(statut = "o", typ = "TXM", min=1, max="**", into=F_InitChoice(), validators=(VerifExiste(2))),
81 assim_data_method = """
82 def F_${assim_name}(statut) : return FACT(statut=statut,
83 INPUT_TYPE = SIMP(statut="o", typ = "TXM", into=(${choices}), defaut=${default_choice}),
88 assim_data_choice = """
89 ${choice_name} = BLOC ( condition = " INPUT_TYPE in ( '${choice_name}', ) ",
90 data = F_${choice_name}("o"),
95 ASSIMILATION_STUDY = PROC(nom="ASSIMILATION_STUDY",
98 Study_name = SIMP(statut="o", typ = "TXM"),
99 Study_repertory = SIMP(statut="f", typ = "TXM"),
100 Debug = SIMP(statut="o", typ = "I", into=(0, 1), defaut=0),
101 Algorithm = SIMP(statut="o", typ = "TXM", into=(${algos_names})),
102 Background = F_Background("o"),
103 BackgroundError = F_BackgroundError("o"),
104 Observation = F_Observation("o"),
105 ObservationError = F_ObservationError("o"),
106 ObservationOperator = F_ObservationOperator("o"),
107 AlgorithmParameters = F_AlgorithmParameters("f"),
108 UserDataInit = F_Init("f"),
109 UserPostAnalysis = F_Analysis("f")
113 begin_catalog_file = string.Template(begin_catalog_file)
114 data_method = string.Template(data_method)
115 assim_data_method = string.Template(assim_data_method)
116 assim_data_choice = string.Template(assim_data_choice)
117 assim_study = string.Template(assim_study)
119 #----------- End of Templates Part ---------------#
123 #----------- Begin generation script -----------#
124 print "-- Starting AdaoCalatogGenerator.py --"
128 import daYacsSchemaCreator
129 import daCore.AssimilationStudy
130 import daYacsSchemaCreator.infos_daComposant as infos
132 logging.fatal("Import of ADAO python modules failed !" +
133 "\n add ADAO python installation directory in your PYTHONPATH")
134 traceback.print_exc()
137 def check_args(args):
138 logging.debug("Arguments are :" + str(args))
140 logging.fatal("Bad number of arguments: you have to provide two arguments (%d given)" % (len(args)))
144 from optparse import OptionParser
145 usage = "usage: %prog [options] catalog_path catalog_name"
147 my_parser = OptionParser(usage=usage, version=version)
148 (options, args) = my_parser.parse_args()
151 catalog_path = args[0]
152 catalog_name = args[1]
154 # Generates into a string
155 mem_file = StringIO.StringIO()
158 from time import strftime
159 mem_file.write(begin_catalog_file.substitute(date=strftime("%Y-%m-%d %H:%M:%S")))
161 # Step 1: A partir des infos, on crée les fonctions qui vont permettre
162 # d'entrer les données utilisateur
163 for data_input_name in infos.DataTypeDict.keys():
164 logging.debug('A data input Type is found: ' + data_input_name)
165 data_name = data_input_name
169 # On récupère les différentes façon d'entrer les données
170 for basic_type in infos.DataTypeDict[data_input_name]:
171 data_into += "\"" + basic_type + "\", "
173 # On choisit le défault
174 data_default = "\"" + infos.DataTypeDefaultDict[data_input_name] + "\""
176 mem_file.write(data_method.substitute(data_name = data_name,
177 data_into = data_into,
178 data_default = data_default))
180 # Step 2: On crée les fonctions qui permettent de rentrer les données des algorithmes
181 for assim_data_input_name in infos.AssimDataDict.keys():
182 logging.debug("An assimilation algorithm data input is found: " + assim_data_input_name)
183 assim_name = assim_data_input_name
188 for choice in infos.AssimDataDict[assim_data_input_name]:
189 choices += "\"" + choice + "\", "
190 decl_choices += assim_data_choice.substitute(choice_name = choice)
191 default_choice = "\"" + infos.AssimDataDefaultDict[assim_data_input_name] + "\""
193 mem_file.write(assim_data_method.substitute(assim_name = assim_name,
195 decl_choices = decl_choices,
196 default_choice=default_choice))
198 # Step 3: On ajoute les fonctions représentant les options possibles
200 for opt_name in infos.OptDict.keys():
201 logging.debug("An optional node is found: " + opt_name)
206 for choice in infos.OptDict[opt_name]:
207 data_into += "\"" + choice + "\", "
208 data_default = "\"" + infos.OptDefaultDict[opt_name] + "\""
210 mem_file.write(data_method.substitute(data_name = data_name,
211 data_into = data_into,
212 data_default = data_default))
214 opt_names.append(opt_name)
216 # Step 4: On ajoute la méthode optionnelle init
217 # TODO uniformiser avec le step 3
218 mem_file.write(init_method)
220 # Final step: Add algorithm and assim_study
224 assim_study_object = daCore.AssimilationStudy.AssimilationStudy()
225 algos_list = assim_study_object.get_available_algorithms()
226 for algo_name in algos_list:
227 logging.debug("An assimilation algorithm is found: " + algo_name)
228 algos_names += "\"" + algo_name + "\", "
230 mem_file.write(assim_study.substitute(algos_names=algos_names,
231 decl_algos=decl_algos))
233 final_file = open(catalog_path + "/" + catalog_name, "wr")
234 final_file.write(mem_file.getvalue())