| [0b990d] | 1 | //
 | 
|---|
 | 2 | // localrect.cc
 | 
|---|
 | 3 | //
 | 
|---|
 | 4 | // Copyright (C) 1996 Limit Point Systems, Inc.
 | 
|---|
 | 5 | //
 | 
|---|
 | 6 | // Author: Curtis Janssen <cljanss@limitpt.com>
 | 
|---|
 | 7 | // Maintainer: LPS
 | 
|---|
 | 8 | //
 | 
|---|
 | 9 | // This file is part of the SC Toolkit.
 | 
|---|
 | 10 | //
 | 
|---|
 | 11 | // The SC Toolkit is free software; you can redistribute it and/or modify
 | 
|---|
 | 12 | // it under the terms of the GNU Library General Public License as published by
 | 
|---|
 | 13 | // the Free Software Foundation; either version 2, or (at your option)
 | 
|---|
 | 14 | // any later version.
 | 
|---|
 | 15 | //
 | 
|---|
 | 16 | // The SC Toolkit is distributed in the hope that it will be useful,
 | 
|---|
 | 17 | // but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
|---|
 | 18 | // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
|---|
 | 19 | // GNU Library General Public License for more details.
 | 
|---|
 | 20 | //
 | 
|---|
 | 21 | // You should have received a copy of the GNU Library General Public License
 | 
|---|
 | 22 | // along with the SC Toolkit; see the file COPYING.LIB.  If not, write to
 | 
|---|
 | 23 | // the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
 | 
|---|
 | 24 | //
 | 
|---|
 | 25 | // The U.S. Government is granted a limited license as per AL 91-7.
 | 
|---|
 | 26 | //
 | 
|---|
 | 27 | 
 | 
|---|
 | 28 | #include <math.h>
 | 
|---|
 | 29 | 
 | 
|---|
 | 30 | #include <util/misc/formio.h>
 | 
|---|
 | 31 | #include <util/keyval/keyval.h>
 | 
|---|
 | 32 | #include <math/scmat/local.h>
 | 
|---|
 | 33 | #include <math/scmat/cmatrix.h>
 | 
|---|
 | 34 | #include <math/scmat/elemop.h>
 | 
|---|
 | 35 | 
 | 
|---|
 | 36 | using namespace std;
 | 
|---|
 | 37 | using namespace sc;
 | 
|---|
 | 38 | 
 | 
|---|
 | 39 | extern "C" {
 | 
|---|
 | 40 |     int sing_(double *q, int *lq, int *iq, double *s, double *p,
 | 
|---|
 | 41 |               int *lp, int *ip, double *a, int *la, int *m, int *n,
 | 
|---|
 | 42 |               double *w);
 | 
|---|
 | 43 | };
 | 
|---|
 | 44 | 
 | 
|---|
 | 45 | /////////////////////////////////////////////////////////////////////////////
 | 
|---|
 | 46 | // LocalSCMatrix member functions
 | 
|---|
 | 47 | 
 | 
|---|
 | 48 | static ClassDesc LocalSCMatrix_cd(
 | 
|---|
 | 49 |   typeid(LocalSCMatrix),"LocalSCMatrix",1,"public SCMatrix",
 | 
|---|
 | 50 |   0, 0, 0);
 | 
|---|
 | 51 | 
 | 
|---|
 | 52 | static double **
 | 
|---|
 | 53 | init_rect_rows(double *data, int ni,int nj)
 | 
|---|
 | 54 | {
 | 
|---|
 | 55 |   double** r = new double*[ni];
 | 
|---|
 | 56 |   int i;
 | 
|---|
 | 57 |   for (i=0; i<ni; i++) r[i] = &data[i*nj];
 | 
|---|
 | 58 |   return r;
 | 
|---|
 | 59 | }
 | 
|---|
 | 60 | 
 | 
|---|
 | 61 | LocalSCMatrix::LocalSCMatrix(const RefSCDimension&a,const RefSCDimension&b,
 | 
|---|
 | 62 |                              LocalSCMatrixKit*kit):
 | 
|---|
 | 63 |   SCMatrix(a,b,kit),
 | 
|---|
 | 64 |   rows(0)
 | 
|---|
 | 65 | {
 | 
|---|
 | 66 |   resize(a->n(),b->n());
 | 
|---|
 | 67 | }
 | 
|---|
 | 68 | 
 | 
|---|
 | 69 | LocalSCMatrix::~LocalSCMatrix()
 | 
|---|
 | 70 | {
 | 
|---|
 | 71 |   if (rows) delete[] rows;
 | 
|---|
 | 72 | }
 | 
|---|
 | 73 | 
 | 
|---|
 | 74 | int
 | 
|---|
 | 75 | LocalSCMatrix::compute_offset(int i,int j) const
 | 
|---|
 | 76 | {
 | 
|---|
 | 77 |   if (i<0 || j<0 || i>=d1->n() || j>=d2->n()) {
 | 
|---|
 | 78 |       ExEnv::errn() << indent << "LocalSCMatrix: index out of bounds\n";
 | 
|---|
 | 79 |       abort();
 | 
|---|
 | 80 |     }
 | 
|---|
 | 81 |   return i*(d2->n()) + j;
 | 
|---|
 | 82 | }
 | 
|---|
 | 83 | 
 | 
|---|
 | 84 | void
 | 
|---|
 | 85 | LocalSCMatrix::resize(int nr, int nc)
 | 
|---|
 | 86 | {
 | 
|---|
 | 87 |   block = new SCMatrixRectBlock(0,nr,0,nc);
 | 
|---|
 | 88 |   if (rows) delete[] rows;
 | 
|---|
 | 89 |   rows = init_rect_rows(block->data,nr,nc);
 | 
|---|
 | 90 | }
 | 
|---|
 | 91 | 
 | 
|---|
 | 92 | double *
 | 
|---|
 | 93 | LocalSCMatrix::get_data()
 | 
|---|
 | 94 | {
 | 
|---|
 | 95 |   return block->data;
 | 
|---|
 | 96 | }
 | 
|---|
 | 97 | 
 | 
|---|
 | 98 | double **
 | 
|---|
 | 99 | LocalSCMatrix::get_rows()
 | 
|---|
 | 100 | {
 | 
|---|
 | 101 |   return rows;
 | 
|---|
 | 102 | }
 | 
|---|
 | 103 | 
 | 
|---|
 | 104 | double
 | 
|---|
 | 105 | LocalSCMatrix::get_element(int i,int j) const
 | 
|---|
 | 106 | {
 | 
|---|
 | 107 |   int off = compute_offset(i,j);
 | 
|---|
 | 108 |   return block->data[off];
 | 
|---|
 | 109 | }
 | 
|---|
 | 110 | 
 | 
|---|
 | 111 | void
 | 
|---|
 | 112 | LocalSCMatrix::set_element(int i,int j,double a)
 | 
|---|
 | 113 | {
 | 
|---|
 | 114 |   int off = compute_offset(i,j);
 | 
|---|
 | 115 |   block->data[off] = a;
 | 
|---|
 | 116 | }
 | 
|---|
 | 117 | 
 | 
|---|
 | 118 | void
 | 
|---|
 | 119 | LocalSCMatrix::accumulate_element(int i,int j,double a)
 | 
|---|
 | 120 | {
 | 
|---|
 | 121 |   int off = compute_offset(i,j);
 | 
|---|
 | 122 |   block->data[off] += a;
 | 
|---|
 | 123 | }
 | 
|---|
 | 124 | 
 | 
|---|
 | 125 | SCMatrix *
 | 
|---|
 | 126 | LocalSCMatrix::get_subblock(int br, int er, int bc, int ec)
 | 
|---|
 | 127 | {
 | 
|---|
 | 128 |   int nsrow = er-br+1;
 | 
|---|
 | 129 |   int nscol = ec-bc+1;
 | 
|---|
 | 130 | 
 | 
|---|
 | 131 |   if (nsrow > nrow() || nscol > ncol()) {
 | 
|---|
 | 132 |       ExEnv::errn() << indent <<
 | 
|---|
 | 133 |           "LocalSCMatrix::get_subblock: trying to get too big a subblock (" <<
 | 
|---|
 | 134 |           nsrow << "," << nscol << ") from (" <<
 | 
|---|
 | 135 |           nrow() << "," << ncol() << ")\n";
 | 
|---|
 | 136 |       abort();
 | 
|---|
 | 137 |     }
 | 
|---|
 | 138 |   
 | 
|---|
 | 139 |   RefSCDimension dnrow;
 | 
|---|
 | 140 |   if (nsrow==nrow()) dnrow = rowdim();
 | 
|---|
 | 141 |   else dnrow = new SCDimension(nsrow);
 | 
|---|
 | 142 | 
 | 
|---|
 | 143 |   RefSCDimension dncol;
 | 
|---|
 | 144 |   if (nscol==ncol()) dncol = coldim();
 | 
|---|
 | 145 |   else dncol = new SCDimension(nscol);
 | 
|---|
 | 146 | 
 | 
|---|
 | 147 |   SCMatrix * sb = kit()->matrix(dnrow,dncol);
 | 
|---|
 | 148 |   sb->assign(0.0);
 | 
|---|
 | 149 | 
 | 
|---|
 | 150 |   LocalSCMatrix *lsb =
 | 
|---|
 | 151 |     require_dynamic_cast<LocalSCMatrix*>(sb, "LocalSCMatrix::get_subblock");
 | 
|---|
 | 152 | 
 | 
|---|
 | 153 |   for (int i=0; i < nsrow; i++)
 | 
|---|
 | 154 |     for (int j=0; j < nscol; j++)
 | 
|---|
 | 155 |       lsb->rows[i][j] = rows[i+br][j+bc];
 | 
|---|
 | 156 |       
 | 
|---|
 | 157 |   return sb;
 | 
|---|
 | 158 | }
 | 
|---|
 | 159 | 
 | 
|---|
 | 160 | void
 | 
|---|
 | 161 | LocalSCMatrix::assign_subblock(SCMatrix*sb, int br, int er, int bc, int ec,
 | 
|---|
 | 162 |                                int source_br, int source_bc)
 | 
|---|
 | 163 | {
 | 
|---|
 | 164 |   LocalSCMatrix *lsb =
 | 
|---|
 | 165 |     require_dynamic_cast<LocalSCMatrix*>(sb, "LocalSCMatrix::assign_subblock");
 | 
|---|
 | 166 | 
 | 
|---|
 | 167 |   int nsrow = er-br+1;
 | 
|---|
 | 168 |   int nscol = ec-bc+1;
 | 
|---|
 | 169 | 
 | 
|---|
 | 170 |   if (nsrow > nrow() || nscol > ncol()) {
 | 
|---|
 | 171 |       ExEnv::errn() << indent <<
 | 
|---|
 | 172 |           "LocalSCMatrix::assign_subblock: trying to assign too big a " <<
 | 
|---|
 | 173 |           "subblock (" << nsrow << "," << nscol << " to (" <<
 | 
|---|
 | 174 |           nrow() << "," << ncol() << ")\n";
 | 
|---|
 | 175 |       abort();
 | 
|---|
 | 176 |     }
 | 
|---|
 | 177 |   
 | 
|---|
 | 178 |   for (int i=0; i < nsrow; i++)
 | 
|---|
 | 179 |     for (int j=0; j < nscol; j++)
 | 
|---|
 | 180 |       rows[i+br][j+bc] = lsb->rows[source_br + i][source_bc + j];
 | 
|---|
 | 181 | }
 | 
|---|
 | 182 | 
 | 
|---|
 | 183 | void
 | 
|---|
 | 184 | LocalSCMatrix::accumulate_subblock(SCMatrix*sb, int br, int er, int bc, int ec,
 | 
|---|
 | 185 |                                    int source_br, int source_bc)
 | 
|---|
 | 186 | {
 | 
|---|
 | 187 |   LocalSCMatrix *lsb =
 | 
|---|
 | 188 |     require_dynamic_cast<LocalSCMatrix*>(sb, "LocalSCMatrix::accumulate_subblock");
 | 
|---|
 | 189 | 
 | 
|---|
 | 190 |   int nsrow = er-br+1;
 | 
|---|
 | 191 |   int nscol = ec-bc+1;
 | 
|---|
 | 192 | 
 | 
|---|
 | 193 |   if (nsrow > nrow() || nscol > ncol()) {
 | 
|---|
 | 194 |       ExEnv::errn() << indent << "LocalSCMatrix::accumulate_subblock: " <<
 | 
|---|
 | 195 |           "trying to accumulate too big a subblock (" <<
 | 
|---|
 | 196 |           nsrow << "," << nscol << " to (" <<
 | 
|---|
 | 197 |           nrow() << "," << ncol() << ")\n";
 | 
|---|
 | 198 |       abort();
 | 
|---|
 | 199 |     }
 | 
|---|
 | 200 |   
 | 
|---|
 | 201 |   for (int i=0; i < nsrow; i++)
 | 
|---|
 | 202 |     for (int j=0; j < nscol; j++)
 | 
|---|
 | 203 |       rows[i+br][j+bc] += lsb->rows[source_br + i][source_bc + j]; 
 | 
|---|
 | 204 | }
 | 
|---|
 | 205 | 
 | 
|---|
 | 206 | SCVector *
 | 
|---|
 | 207 | LocalSCMatrix::get_row(int i)
 | 
|---|
 | 208 | {
 | 
|---|
 | 209 |   if (i >= nrow()) {
 | 
|---|
 | 210 |       ExEnv::errn() << indent << "LocalSCMatrix::get_row: trying to get invalid row " <<
 | 
|---|
 | 211 |           i << " max " << nrow() << endl;
 | 
|---|
 | 212 |       abort();
 | 
|---|
 | 213 |     }
 | 
|---|
 | 214 |   
 | 
|---|
 | 215 |   SCVector * v = kit()->vector(coldim());
 | 
|---|
 | 216 | 
 | 
|---|
 | 217 |   LocalSCVector *lv =
 | 
|---|
 | 218 |     require_dynamic_cast<LocalSCVector*>(v, "LocalSCMatrix::get_row");
 | 
|---|
 | 219 | 
 | 
|---|
 | 220 |   for (int j=0; j < ncol(); j++)
 | 
|---|
 | 221 |     lv->set_element(j,rows[i][j]);
 | 
|---|
 | 222 |       
 | 
|---|
 | 223 |   return v;
 | 
|---|
 | 224 | }
 | 
|---|
 | 225 | 
 | 
|---|
 | 226 | void
 | 
|---|
 | 227 | LocalSCMatrix::assign_row(SCVector *v, int i)
 | 
|---|
 | 228 | {
 | 
|---|
 | 229 |   if (i >= nrow()) {
 | 
|---|
 | 230 |       ExEnv::errn() << indent <<
 | 
|---|
 | 231 |           "LocalSCMatrix::assign_row: trying to assign invalid row " <<
 | 
|---|
 | 232 |           i << " max " << nrow() << endl;
 | 
|---|
 | 233 |       abort();
 | 
|---|
 | 234 |     }
 | 
|---|
 | 235 |   
 | 
|---|
 | 236 |   if (v->n() != ncol()) {
 | 
|---|
 | 237 |       ExEnv::errn() << indent << "LocalSCMatrix::assign_row: vector is wrong size " <<
 | 
|---|
 | 238 |           " is " << v->n() << ", should be " << ncol() << endl;
 | 
|---|
 | 239 |       abort();
 | 
|---|
 | 240 |     }
 | 
|---|
 | 241 |   
 | 
|---|
 | 242 |   LocalSCVector *lv =
 | 
|---|
 | 243 |     require_dynamic_cast<LocalSCVector*>(v, "LocalSCMatrix::assign_row");
 | 
|---|
 | 244 | 
 | 
|---|
 | 245 |   for (int j=0; j < ncol(); j++)
 | 
|---|
 | 246 |     rows[i][j] = lv->get_element(j);
 | 
|---|
 | 247 | }
 | 
|---|
 | 248 | 
 | 
|---|
 | 249 | void
 | 
|---|
 | 250 | LocalSCMatrix::accumulate_row(SCVector *v, int i)
 | 
|---|
 | 251 | {
 | 
|---|
 | 252 |   if (i >= nrow()) {
 | 
|---|
 | 253 |       ExEnv::errn() << indent <<
 | 
|---|
 | 254 |           "LocalSCMatrix::accumulate_row: trying to assign invalid row " <<
 | 
|---|
 | 255 |           i << " max " << nrow() << endl;
 | 
|---|
 | 256 |       abort();
 | 
|---|
 | 257 |     }
 | 
|---|
 | 258 |   
 | 
|---|
 | 259 |   if (v->n() != ncol()) {
 | 
|---|
 | 260 |       ExEnv::errn() << indent <<
 | 
|---|
 | 261 |           "LocalSCMatrix::accumulate_row: vector is wrong size " <<
 | 
|---|
 | 262 |           "is " << v->n() << ", should be " << ncol() << endl;
 | 
|---|
 | 263 |       abort();
 | 
|---|
 | 264 |     }
 | 
|---|
 | 265 |   
 | 
|---|
 | 266 |   LocalSCVector *lv =
 | 
|---|
 | 267 |     require_dynamic_cast<LocalSCVector*>(v, "LocalSCMatrix::accumulate_row");
 | 
|---|
 | 268 | 
 | 
|---|
 | 269 |   for (int j=0; j < ncol(); j++)
 | 
|---|
 | 270 |     rows[i][j] += lv->get_element(j);
 | 
|---|
 | 271 | }
 | 
|---|
 | 272 | 
 | 
|---|
 | 273 | SCVector *
 | 
|---|
 | 274 | LocalSCMatrix::get_column(int i)
 | 
|---|
 | 275 | {
 | 
|---|
 | 276 |   if (i >= ncol()) {
 | 
|---|
 | 277 |       ExEnv::errn() << indent <<
 | 
|---|
 | 278 |           "LocalSCMatrix::get_column: trying to get invalid column " <<
 | 
|---|
 | 279 |           i << " max " << ncol() << endl;
 | 
|---|
 | 280 |       abort();
 | 
|---|
 | 281 |     }
 | 
|---|
 | 282 |   
 | 
|---|
 | 283 |   SCVector * v = kit()->vector(rowdim());
 | 
|---|
 | 284 | 
 | 
|---|
 | 285 |   LocalSCVector *lv =
 | 
|---|
 | 286 |     require_dynamic_cast<LocalSCVector*>(v, "LocalSCMatrix::get_column");
 | 
|---|
 | 287 | 
 | 
|---|
 | 288 |   for (int j=0; j < nrow(); j++)
 | 
|---|
 | 289 |     lv->set_element(j,rows[j][i]);
 | 
|---|
 | 290 |       
 | 
|---|
 | 291 |   return v;
 | 
|---|
 | 292 | }
 | 
|---|
 | 293 | 
 | 
|---|
 | 294 | void
 | 
|---|
 | 295 | LocalSCMatrix::assign_column(SCVector *v, int i)
 | 
|---|
 | 296 | {
 | 
|---|
 | 297 |   if (i >= ncol()) {
 | 
|---|
 | 298 |       ExEnv::errn() << indent <<
 | 
|---|
 | 299 |           "LocalSCMatrix::assign_column: trying to assign invalid column " <<
 | 
|---|
 | 300 |           i << " max " << ncol() << endl;
 | 
|---|
 | 301 |       abort();
 | 
|---|
 | 302 |     }
 | 
|---|
 | 303 |   
 | 
|---|
 | 304 |   if (v->n() != nrow()) {
 | 
|---|
 | 305 |       ExEnv::errn() << indent <<
 | 
|---|
 | 306 |           "LocalSCMatrix::assign_column: vector is wrong size " <<
 | 
|---|
 | 307 |           "is " << v->n() << ", should be " << nrow() << endl;
 | 
|---|
 | 308 |       abort();
 | 
|---|
 | 309 |     }
 | 
|---|
 | 310 |   
 | 
|---|
 | 311 |   LocalSCVector *lv =
 | 
|---|
 | 312 |     require_dynamic_cast<LocalSCVector*>(v, "LocalSCMatrix::assign_column");
 | 
|---|
 | 313 | 
 | 
|---|
 | 314 |   for (int j=0; j < nrow(); j++)
 | 
|---|
 | 315 |     rows[j][i] = lv->get_element(j);
 | 
|---|
 | 316 | }
 | 
|---|
 | 317 | 
 | 
|---|
 | 318 | void
 | 
|---|
 | 319 | LocalSCMatrix::accumulate_column(SCVector *v, int i)
 | 
|---|
 | 320 | {
 | 
|---|
 | 321 |   if (i >= ncol()) {
 | 
|---|
 | 322 |       ExEnv::errn() << indent <<
 | 
|---|
 | 323 |           "LocalSCMatrix::accumulate_column: trying to assign invalid column "
 | 
|---|
 | 324 |            << i << " max " << ncol() << endl;
 | 
|---|
 | 325 |       abort();
 | 
|---|
 | 326 |     }
 | 
|---|
 | 327 |   
 | 
|---|
 | 328 |   if (v->n() != nrow()) {
 | 
|---|
 | 329 |       ExEnv::errn() << indent <<
 | 
|---|
 | 330 |           "LocalSCMatrix::accumulate_column: vector is wrong size " <<
 | 
|---|
 | 331 |           "is " << v->n() << ", should be " << nrow() << endl;
 | 
|---|
 | 332 |       abort();
 | 
|---|
 | 333 |     }
 | 
|---|
 | 334 |   
 | 
|---|
 | 335 |   LocalSCVector *lv =
 | 
|---|
 | 336 |     require_dynamic_cast<LocalSCVector*>(v, "LocalSCMatrix::accumulate_column");
 | 
|---|
 | 337 | 
 | 
|---|
 | 338 |   for (int j=0; j < nrow(); j++)
 | 
|---|
 | 339 |     rows[j][i] += lv->get_element(j);
 | 
|---|
 | 340 | }
 | 
|---|
 | 341 | 
 | 
|---|
 | 342 | void
 | 
|---|
 | 343 | LocalSCMatrix::assign_val(double a)
 | 
|---|
 | 344 | {
 | 
|---|
 | 345 |   int n = d1->n() * d2->n();
 | 
|---|
 | 346 |   double *data = block->data;
 | 
|---|
 | 347 |   for (int i=0; i<n; i++) data[i] = a;
 | 
|---|
 | 348 | }
 | 
|---|
 | 349 | 
 | 
|---|
 | 350 | void
 | 
|---|
 | 351 | LocalSCMatrix::accumulate_product_rr(SCMatrix*a,SCMatrix*b)
 | 
|---|
 | 352 | {
 | 
|---|
 | 353 |   const char* name = "LocalSCMatrix::accumulate_product";
 | 
|---|
 | 354 |   // make sure that the arguments are of the correct type
 | 
|---|
 | 355 |   LocalSCMatrix* la = require_dynamic_cast<LocalSCMatrix*>(a,name);
 | 
|---|
 | 356 |   LocalSCMatrix* lb = require_dynamic_cast<LocalSCMatrix*>(b,name);
 | 
|---|
 | 357 | 
 | 
|---|
 | 358 |   // make sure that the dimensions match
 | 
|---|
 | 359 |   if (!rowdim()->equiv(la->rowdim()) || !coldim()->equiv(lb->coldim()) ||
 | 
|---|
 | 360 |       !la->coldim()->equiv(lb->rowdim())) {
 | 
|---|
 | 361 |       ExEnv::errn() << indent << "LocalSCMatrix::accumulate_product: bad dim" << endl;
 | 
|---|
 | 362 |       ExEnv::errn() << indent << "this row and col dimension:" << endl;
 | 
|---|
 | 363 |       rowdim()->print(ExEnv::errn());
 | 
|---|
 | 364 |       coldim()->print(ExEnv::errn());
 | 
|---|
 | 365 |       ExEnv::errn() << indent << "a row and col dimension:" << endl;
 | 
|---|
 | 366 |       a->rowdim()->print(ExEnv::errn());
 | 
|---|
 | 367 |       a->coldim()->print(ExEnv::errn());
 | 
|---|
 | 368 |       ExEnv::errn() << indent << "b row and col dimension:" << endl;
 | 
|---|
 | 369 |       b->rowdim()->print(ExEnv::errn());
 | 
|---|
 | 370 |       b->coldim()->print(ExEnv::errn());
 | 
|---|
 | 371 |       abort();
 | 
|---|
 | 372 |     }
 | 
|---|
 | 373 | 
 | 
|---|
 | 374 |   cmat_mxm(la->rows, 0,
 | 
|---|
 | 375 |            lb->rows, 0,
 | 
|---|
 | 376 |            rows, 0,
 | 
|---|
 | 377 |            nrow(), la->ncol(), this->ncol(),
 | 
|---|
 | 378 |            1);
 | 
|---|
 | 379 | }
 | 
|---|
 | 380 | 
 | 
|---|
 | 381 | // does the outer product a x b.  this must have rowdim() == a->dim() and
 | 
|---|
 | 382 | // coldim() == b->dim()
 | 
|---|
 | 383 | void
 | 
|---|
 | 384 | LocalSCMatrix::accumulate_outer_product(SCVector*a,SCVector*b)
 | 
|---|
 | 385 | {
 | 
|---|
 | 386 |   const char* name = "LocalSCMatrix::accumulate_outer_product";
 | 
|---|
 | 387 |   // make sure that the arguments are of the correct type
 | 
|---|
 | 388 |   LocalSCVector* la = require_dynamic_cast<LocalSCVector*>(a,name);
 | 
|---|
 | 389 |   LocalSCVector* lb = require_dynamic_cast<LocalSCVector*>(b,name);
 | 
|---|
 | 390 | 
 | 
|---|
 | 391 |   // make sure that the dimensions match
 | 
|---|
 | 392 |   if (!rowdim()->equiv(la->dim()) || !coldim()->equiv(lb->dim())) {
 | 
|---|
 | 393 |       ExEnv::errn() << indent <<
 | 
|---|
 | 394 |           "LocalSCMatrix::accumulate_outer_product(SCVector*a,SCVector*b): " <<
 | 
|---|
 | 395 |           "dimensions don't match" << endl;
 | 
|---|
 | 396 |       abort();
 | 
|---|
 | 397 |     }
 | 
|---|
 | 398 | 
 | 
|---|
 | 399 |   int nr = a->n();
 | 
|---|
 | 400 |   int nc = b->n();
 | 
|---|
 | 401 |   int i, j;
 | 
|---|
 | 402 |   double* adat = la->block->data;
 | 
|---|
 | 403 |   double* bdat = lb->block->data;
 | 
|---|
 | 404 |   double** thisdat = rows;
 | 
|---|
 | 405 |   for (i=0; i<nr; i++) {
 | 
|---|
 | 406 |       for (j=0; j<nc; j++) {
 | 
|---|
 | 407 |           thisdat[i][j] += adat[i] * bdat[j];
 | 
|---|
 | 408 |         }
 | 
|---|
 | 409 |     }
 | 
|---|
 | 410 | }
 | 
|---|
 | 411 | 
 | 
|---|
 | 412 | void
 | 
|---|
 | 413 | LocalSCMatrix::accumulate_product_rs(SCMatrix*a,SymmSCMatrix*b)
 | 
|---|
 | 414 | {
 | 
|---|
 | 415 |   const char* name = "LocalSCMatrix::accumulate_product";
 | 
|---|
 | 416 |   // make sure that the arguments are of the correct type
 | 
|---|
 | 417 |   LocalSCMatrix* la = require_dynamic_cast<LocalSCMatrix*>(a,name);
 | 
|---|
 | 418 |   LocalSymmSCMatrix* lb = require_dynamic_cast<LocalSymmSCMatrix*>(b,name);
 | 
|---|
 | 419 | 
 | 
|---|
 | 420 |   // make sure that the dimensions match
 | 
|---|
 | 421 |   if (!rowdim()->equiv(la->rowdim()) || !coldim()->equiv(lb->dim()) ||
 | 
|---|
 | 422 |       !la->coldim()->equiv(lb->dim())) {
 | 
|---|
 | 423 |       ExEnv::errn() << indent <<
 | 
|---|
 | 424 |           "LocalSCMatrix::accumulate_product_rs(SCMatrix*a,SymmSCMatrix*b): " <<
 | 
|---|
 | 425 |           "dimensions don't match" << endl;
 | 
|---|
 | 426 |       abort();
 | 
|---|
 | 427 |     }
 | 
|---|
 | 428 | 
 | 
|---|
 | 429 |   double **cd = rows;
 | 
|---|
 | 430 |   double **ad = la->rows;
 | 
|---|
 | 431 |   double **bd = lb->rows;
 | 
|---|
 | 432 |   int ni = a->rowdim().n();
 | 
|---|
 | 433 |   int njk = b->dim().n();
 | 
|---|
 | 434 |   int i, j, k;
 | 
|---|
 | 435 |   for (i=0; i<ni; i++) {
 | 
|---|
 | 436 |       for (j=0; j<njk; j++) {
 | 
|---|
 | 437 |           for (k=0; k<=j; k++) {
 | 
|---|
 | 438 |               cd[i][k] += ad[i][j]*bd[j][k];
 | 
|---|
 | 439 |             }
 | 
|---|
 | 440 |           for (; k<njk; k++) {
 | 
|---|
 | 441 |               cd[i][k] += ad[i][j]*bd[k][j];
 | 
|---|
 | 442 |             }
 | 
|---|
 | 443 |         }
 | 
|---|
 | 444 |     }
 | 
|---|
 | 445 | }
 | 
|---|
 | 446 | 
 | 
|---|
 | 447 | void
 | 
|---|
 | 448 | LocalSCMatrix::accumulate_product_rd(SCMatrix*a,DiagSCMatrix*b)
 | 
|---|
 | 449 | {
 | 
|---|
 | 450 |   const char* name = "LocalSCMatrix::accumulate_product_rd";
 | 
|---|
 | 451 |   // make sure that the arguments are of the correct type
 | 
|---|
 | 452 |   LocalSCMatrix* la = require_dynamic_cast<LocalSCMatrix*>(a,name);
 | 
|---|
 | 453 |   LocalDiagSCMatrix* lb = require_dynamic_cast<LocalDiagSCMatrix*>(b,name);
 | 
|---|
 | 454 | 
 | 
|---|
 | 455 |   // make sure that the dimensions match
 | 
|---|
 | 456 |   if (!rowdim()->equiv(la->rowdim()) || !coldim()->equiv(lb->dim()) ||
 | 
|---|
 | 457 |       !la->coldim()->equiv(lb->dim())) {
 | 
|---|
 | 458 |       ExEnv::errn() << indent <<
 | 
|---|
 | 459 |           "LocalSCMatrix::accumulate_product_rd(SCMatrix*a,DiagSCMatrix*b): " <<
 | 
|---|
 | 460 |           "dimensions don't match" << endl;
 | 
|---|
 | 461 |       abort();
 | 
|---|
 | 462 |     }
 | 
|---|
 | 463 | 
 | 
|---|
 | 464 |   double **cd = rows;
 | 
|---|
 | 465 |   double **ad = la->rows;
 | 
|---|
 | 466 |   double *bd = lb->block->data;
 | 
|---|
 | 467 |   int ni = a->rowdim().n();
 | 
|---|
 | 468 |   int nj = b->dim().n();
 | 
|---|
 | 469 |   int i, j;
 | 
|---|
 | 470 |   for (i=0; i<ni; i++) {
 | 
|---|
 | 471 |       for (j=0; j<nj; j++) {
 | 
|---|
 | 472 |           cd[i][j] += ad[i][j]*bd[j];
 | 
|---|
 | 473 |         }
 | 
|---|
 | 474 |     }
 | 
|---|
 | 475 | }
 | 
|---|
 | 476 | 
 | 
|---|
 | 477 | void
 | 
|---|
 | 478 | LocalSCMatrix::accumulate(const SCMatrix*a)
 | 
|---|
 | 479 | {
 | 
|---|
 | 480 |   // make sure that the arguments is of the correct type
 | 
|---|
 | 481 |   const LocalSCMatrix* la
 | 
|---|
 | 482 |     = require_dynamic_cast<const LocalSCMatrix*>(a,"LocalSCMatrix::accumulate");
 | 
|---|
 | 483 | 
 | 
|---|
 | 484 |   // make sure that the dimensions match
 | 
|---|
 | 485 |   if (!rowdim()->equiv(la->rowdim()) || !coldim()->equiv(la->coldim())) {
 | 
|---|
 | 486 |       ExEnv::errn() << indent << "LocalSCMatrix::accumulate(SCMatrix*a): " <<
 | 
|---|
 | 487 |           "dimensions don't match" << endl;
 | 
|---|
 | 488 |       abort();
 | 
|---|
 | 489 |     }
 | 
|---|
 | 490 | 
 | 
|---|
 | 491 |   int nelem = this->ncol() * this->nrow();
 | 
|---|
 | 492 |   int i;
 | 
|---|
 | 493 |   for (i=0; i<nelem; i++) block->data[i] += la->block->data[i];
 | 
|---|
 | 494 | }
 | 
|---|
 | 495 | 
 | 
|---|
 | 496 | void
 | 
|---|
 | 497 | LocalSCMatrix::accumulate(const SymmSCMatrix*a)
 | 
|---|
 | 498 | {
 | 
|---|
 | 499 |   // make sure that the arguments is of the correct type
 | 
|---|
 | 500 |   const LocalSymmSCMatrix* la
 | 
|---|
 | 501 |     = require_dynamic_cast<const LocalSymmSCMatrix*>(a,"LocalSCMatrix::accumulate");
 | 
|---|
 | 502 | 
 | 
|---|
 | 503 |   // make sure that the dimensions match
 | 
|---|
 | 504 |   if (!rowdim()->equiv(la->dim()) || !coldim()->equiv(la->dim())) {
 | 
|---|
 | 505 |       ExEnv::errn() << indent << "LocalSCMatrix::accumulate(SymmSCMatrix*a): " <<
 | 
|---|
 | 506 |           "dimensions don't match" << endl;
 | 
|---|
 | 507 |       abort();
 | 
|---|
 | 508 |     }
 | 
|---|
 | 509 | 
 | 
|---|
 | 510 |   int n = this->ncol();
 | 
|---|
 | 511 |   double *dat = la->block->data;
 | 
|---|
 | 512 |   int i, j;
 | 
|---|
 | 513 |   for (i=0; i<n; i++) {
 | 
|---|
 | 514 |       for (j=0; j<i; j++) {
 | 
|---|
 | 515 |           double tmp = *dat;
 | 
|---|
 | 516 |           block->data[i*n+j] += tmp;
 | 
|---|
 | 517 |           block->data[j*n+i] += tmp;
 | 
|---|
 | 518 |           dat++;
 | 
|---|
 | 519 |         }
 | 
|---|
 | 520 |       block->data[i*n+i] += *dat++;
 | 
|---|
 | 521 |     }
 | 
|---|
 | 522 | }
 | 
|---|
 | 523 | 
 | 
|---|
 | 524 | void
 | 
|---|
 | 525 | LocalSCMatrix::accumulate(const DiagSCMatrix*a)
 | 
|---|
 | 526 | {
 | 
|---|
 | 527 |   // make sure that the arguments is of the correct type
 | 
|---|
 | 528 |   const LocalDiagSCMatrix* la
 | 
|---|
 | 529 |     = require_dynamic_cast<const LocalDiagSCMatrix*>(a,"LocalSCMatrix::accumulate");
 | 
|---|
 | 530 | 
 | 
|---|
 | 531 |   // make sure that the dimensions match
 | 
|---|
 | 532 |   if (!rowdim()->equiv(la->dim()) || !coldim()->equiv(la->dim())) {
 | 
|---|
 | 533 |       ExEnv::errn() << indent << "LocalSCMatrix::accumulate(DiagSCMatrix*a): " <<
 | 
|---|
 | 534 |           "dimensions don't match\n";
 | 
|---|
 | 535 |       abort();
 | 
|---|
 | 536 |     }
 | 
|---|
 | 537 | 
 | 
|---|
 | 538 |   int n = this->ncol();
 | 
|---|
 | 539 |   double *dat = la->block->data;
 | 
|---|
 | 540 |   int i;
 | 
|---|
 | 541 |   for (i=0; i<n; i++) {
 | 
|---|
 | 542 |       block->data[i*n+i] += *dat++;
 | 
|---|
 | 543 |     }
 | 
|---|
 | 544 | }
 | 
|---|
 | 545 | 
 | 
|---|
 | 546 | void
 | 
|---|
 | 547 | LocalSCMatrix::accumulate(const SCVector*a)
 | 
|---|
 | 548 | {
 | 
|---|
 | 549 |   // make sure that the arguments is of the correct type
 | 
|---|
 | 550 |   const LocalSCVector* la
 | 
|---|
 | 551 |     = require_dynamic_cast<const LocalSCVector*>(a,"LocalSCVector::accumulate");
 | 
|---|
 | 552 | 
 | 
|---|
 | 553 |   // make sure that the dimensions match
 | 
|---|
 | 554 |   if (!((rowdim()->equiv(la->dim()) && coldim()->n() == 1)
 | 
|---|
 | 555 |         || (coldim()->equiv(la->dim()) && rowdim()->n() == 1))) {
 | 
|---|
 | 556 |       ExEnv::errn() << indent << "LocalSCMatrix::accumulate(SCVector*a): " <<
 | 
|---|
 | 557 |           "dimensions don't match" << endl;
 | 
|---|
 | 558 |       abort();
 | 
|---|
 | 559 |     }
 | 
|---|
 | 560 | 
 | 
|---|
 | 561 |   int n = this->ncol();
 | 
|---|
 | 562 |   double *dat = la->block->data;
 | 
|---|
 | 563 |   int i;
 | 
|---|
 | 564 |   for (i=0; i<n; i++) {
 | 
|---|
 | 565 |       block->data[i*n+i] += *dat++;
 | 
|---|
 | 566 |     }
 | 
|---|
 | 567 | }
 | 
|---|
 | 568 | 
 | 
|---|
 | 569 | void
 | 
|---|
 | 570 | LocalSCMatrix::transpose_this()
 | 
|---|
 | 571 | {
 | 
|---|
 | 572 |   cmat_transpose_matrix(rows,nrow(),ncol());
 | 
|---|
 | 573 |   delete[] rows;
 | 
|---|
 | 574 |   rows = new double*[ncol()];
 | 
|---|
 | 575 |   cmat_matrix_pointers(rows,block->data,ncol(),nrow());
 | 
|---|
 | 576 |   RefSCDimension tmp = d1;
 | 
|---|
 | 577 |   d1 = d2;
 | 
|---|
 | 578 |   d2 = tmp;
 | 
|---|
 | 579 | 
 | 
|---|
 | 580 |   int itmp = block->istart;
 | 
|---|
 | 581 |   block->istart = block->jstart;
 | 
|---|
 | 582 |   block->jstart = itmp;
 | 
|---|
 | 583 | 
 | 
|---|
 | 584 |   itmp = block->iend;
 | 
|---|
 | 585 |   block->iend = block->jend;
 | 
|---|
 | 586 |   block->jend = itmp;
 | 
|---|
 | 587 | }
 | 
|---|
 | 588 | 
 | 
|---|
 | 589 | double
 | 
|---|
 | 590 | LocalSCMatrix::invert_this()
 | 
|---|
 | 591 | {
 | 
|---|
 | 592 |   if (nrow() != ncol()) {
 | 
|---|
 | 593 |       ExEnv::errn() << indent << "LocalSCMatrix::invert_this: matrix is not square\n";
 | 
|---|
 | 594 |       abort();
 | 
|---|
 | 595 |     }
 | 
|---|
 | 596 |   return cmat_invert(rows,0,nrow());
 | 
|---|
 | 597 | }
 | 
|---|
 | 598 | 
 | 
|---|
 | 599 | double
 | 
|---|
 | 600 | LocalSCMatrix::determ_this()
 | 
|---|
 | 601 | {
 | 
|---|
 | 602 |   if (nrow() != ncol()) {
 | 
|---|
 | 603 |       ExEnv::errn() << indent << "LocalSCMatrix::determ_this: matrix is not square\n";
 | 
|---|
 | 604 |       abort();
 | 
|---|
 | 605 |     }
 | 
|---|
 | 606 |   return cmat_determ(rows,0,nrow());
 | 
|---|
 | 607 | }
 | 
|---|
 | 608 | 
 | 
|---|
 | 609 | double
 | 
|---|
 | 610 | LocalSCMatrix::trace()
 | 
|---|
 | 611 | {
 | 
|---|
 | 612 |   if (nrow() != ncol()) {
 | 
|---|
 | 613 |       ExEnv::errn() << indent << "LocalSCMatrix::trace: matrix is not square\n";
 | 
|---|
 | 614 |       abort();
 | 
|---|
 | 615 |     }
 | 
|---|
 | 616 |   double ret=0;
 | 
|---|
 | 617 |   int i;
 | 
|---|
 | 618 |   for (i=0; i < nrow(); i++)
 | 
|---|
 | 619 |     ret += rows[i][i];
 | 
|---|
 | 620 |   return ret;
 | 
|---|
 | 621 | }
 | 
|---|
 | 622 | 
 | 
|---|
 | 623 | void
 | 
|---|
 | 624 | LocalSCMatrix::svd_this(SCMatrix *U, DiagSCMatrix *sigma, SCMatrix *V)
 | 
|---|
 | 625 | {
 | 
|---|
 | 626 |   LocalSCMatrix* lU =
 | 
|---|
 | 627 |     require_dynamic_cast<LocalSCMatrix*>(U,"LocalSCMatrix::svd_this");
 | 
|---|
 | 628 |   LocalSCMatrix* lV =
 | 
|---|
 | 629 |     require_dynamic_cast<LocalSCMatrix*>(V,"LocalSCMatrix::svd_this");
 | 
|---|
 | 630 |   LocalDiagSCMatrix* lsigma =
 | 
|---|
 | 631 |     require_dynamic_cast<LocalDiagSCMatrix*>(sigma,"LocalSCMatrix::svd_this");
 | 
|---|
 | 632 | 
 | 
|---|
 | 633 |   RefSCDimension mdim = rowdim();
 | 
|---|
 | 634 |   RefSCDimension ndim = coldim();
 | 
|---|
 | 635 |   int m = mdim.n();
 | 
|---|
 | 636 |   int n = ndim.n();
 | 
|---|
 | 637 | 
 | 
|---|
 | 638 |   RefSCDimension pdim;
 | 
|---|
 | 639 |   if (m == n && m == sigma->dim().n())
 | 
|---|
 | 640 |     pdim = sigma->dim();
 | 
|---|
 | 641 |   else if (m<n)
 | 
|---|
 | 642 |     pdim = mdim;
 | 
|---|
 | 643 |   else
 | 
|---|
 | 644 |     pdim = ndim;
 | 
|---|
 | 645 | 
 | 
|---|
 | 646 |   int p = pdim.n();
 | 
|---|
 | 647 | 
 | 
|---|
 | 648 |   if (!mdim->equiv(lU->rowdim()) ||
 | 
|---|
 | 649 |       !mdim->equiv(lU->coldim()) ||
 | 
|---|
 | 650 |       !ndim->equiv(lV->rowdim()) ||
 | 
|---|
 | 651 |       !ndim->equiv(lV->coldim()) ||
 | 
|---|
 | 652 |       !pdim->equiv(sigma->dim())) {
 | 
|---|
 | 653 |       ExEnv::errn() << indent << "LocalSCMatrix: svd_this: dimension mismatch\n";
 | 
|---|
 | 654 |       abort();
 | 
|---|
 | 655 |     }
 | 
|---|
 | 656 | 
 | 
|---|
 | 657 |   // form a fortran style matrix for the SVD routines
 | 
|---|
 | 658 |   double *dA = new double[m*n];
 | 
|---|
 | 659 |   double *dU = new double[m*m];
 | 
|---|
 | 660 |   double *dV = new double[n*n];
 | 
|---|
 | 661 |   double *dsigma = new double[n];
 | 
|---|
 | 662 |   double *w = new double[(3*p-1>m)?(3*p-1):m];
 | 
|---|
 | 663 | 
 | 
|---|
 | 664 |   int i,j;
 | 
|---|
 | 665 |   for (i=0; i<m; i++) {
 | 
|---|
 | 666 |       for (j=0; j<n; j++) {
 | 
|---|
 | 667 |           dA[i + j*m] = this->block->data[i*n + j];
 | 
|---|
 | 668 |         }
 | 
|---|
 | 669 |     }
 | 
|---|
 | 670 | 
 | 
|---|
 | 671 |   int three = 3;
 | 
|---|
 | 672 | 
 | 
|---|
 | 673 |   sing_(dU, &m, &three, dsigma, dV, &n, &three, dA, &m, &m, &n, w);
 | 
|---|
 | 674 | 
 | 
|---|
 | 675 |   for (i=0; i<m; i++) {
 | 
|---|
 | 676 |       for (j=0; j<m; j++) {
 | 
|---|
 | 677 |           lU->block->data[i*m + j] = dU[i + j*m];
 | 
|---|
 | 678 |         }
 | 
|---|
 | 679 |     }
 | 
|---|
 | 680 | 
 | 
|---|
 | 681 |   for (i=0; i<n; i++) {
 | 
|---|
 | 682 |       for (j=0; j<n; j++) {
 | 
|---|
 | 683 |           lV->block->data[i*n + j] = dV[i + j*n];
 | 
|---|
 | 684 |         }
 | 
|---|
 | 685 |     }
 | 
|---|
 | 686 | 
 | 
|---|
 | 687 |   for (i=0; i<p; i++) {
 | 
|---|
 | 688 |       lsigma->block->data[i] = dsigma[i];
 | 
|---|
 | 689 |     }
 | 
|---|
 | 690 | 
 | 
|---|
 | 691 |   delete[] dA;
 | 
|---|
 | 692 |   delete[] dU;
 | 
|---|
 | 693 |   delete[] dV;
 | 
|---|
 | 694 |   delete[] dsigma;
 | 
|---|
 | 695 |   delete[] w;
 | 
|---|
 | 696 | }
 | 
|---|
 | 697 | 
 | 
|---|
 | 698 | double
 | 
|---|
 | 699 | LocalSCMatrix::solve_this(SCVector*v)
 | 
|---|
 | 700 | {
 | 
|---|
 | 701 |   LocalSCVector* lv =
 | 
|---|
 | 702 |     require_dynamic_cast<LocalSCVector*>(v,"LocalSCMatrix::solve_this");
 | 
|---|
 | 703 |   
 | 
|---|
 | 704 |   // make sure that the dimensions match
 | 
|---|
 | 705 |   if (!rowdim()->equiv(lv->dim())) {
 | 
|---|
 | 706 |       ExEnv::errn() << indent << "LocalSCMatrix::solve_this(SCVector*v): " <<
 | 
|---|
 | 707 |           "dimensions don't match" << endl;
 | 
|---|
 | 708 |       abort();
 | 
|---|
 | 709 |     }
 | 
|---|
 | 710 | 
 | 
|---|
 | 711 |   return cmat_solve_lin(rows,0,lv->block->data,nrow());
 | 
|---|
 | 712 | }
 | 
|---|
 | 713 | 
 | 
|---|
 | 714 | void
 | 
|---|
 | 715 | LocalSCMatrix::schmidt_orthog(SymmSCMatrix *S, int nc)
 | 
|---|
 | 716 | {
 | 
|---|
 | 717 |   LocalSymmSCMatrix* lS =
 | 
|---|
 | 718 |     require_dynamic_cast<LocalSymmSCMatrix*>(S,"LocalSCMatrix::schmidt_orthog");
 | 
|---|
 | 719 |   
 | 
|---|
 | 720 |   // make sure that the dimensions match
 | 
|---|
 | 721 |   if (!rowdim()->equiv(lS->dim())) {
 | 
|---|
 | 722 |       ExEnv::errn() << indent << "LocalSCMatrix::schmidt_orthog(): " <<
 | 
|---|
 | 723 |           "dimensions don't match\n";
 | 
|---|
 | 724 |       abort();
 | 
|---|
 | 725 |     }
 | 
|---|
 | 726 | 
 | 
|---|
 | 727 |   cmat_schmidt(rows,lS->block->data,nrow(),nc);
 | 
|---|
 | 728 | }
 | 
|---|
 | 729 | 
 | 
|---|
 | 730 | int
 | 
|---|
 | 731 | LocalSCMatrix::schmidt_orthog_tol(SymmSCMatrix *S, double tol, double *res)
 | 
|---|
 | 732 | {
 | 
|---|
 | 733 |   LocalSymmSCMatrix* lS =
 | 
|---|
 | 734 |     require_dynamic_cast<LocalSymmSCMatrix*>(S,"LocalSCMatrix::schmidt_orthog");
 | 
|---|
 | 735 |   
 | 
|---|
 | 736 |   // make sure that the dimensions match
 | 
|---|
 | 737 |   if (!rowdim()->equiv(lS->dim())) {
 | 
|---|
 | 738 |       ExEnv::errn() << indent << "LocalSCMatrix::schmidt_orthog(): " <<
 | 
|---|
 | 739 |           "dimensions don't match\n";
 | 
|---|
 | 740 |       abort();
 | 
|---|
 | 741 |     }
 | 
|---|
 | 742 | 
 | 
|---|
 | 743 |   return cmat_schmidt_tol(rows,lS->block->data,nrow(),ncol(),tol,res);
 | 
|---|
 | 744 | }
 | 
|---|
 | 745 | 
 | 
|---|
 | 746 | void
 | 
|---|
 | 747 | LocalSCMatrix::element_op(const Ref<SCElementOp>& op)
 | 
|---|
 | 748 | {
 | 
|---|
 | 749 |   op->process_spec_rect(block.pointer());
 | 
|---|
 | 750 | }
 | 
|---|
 | 751 | 
 | 
|---|
 | 752 | void
 | 
|---|
 | 753 | LocalSCMatrix::element_op(const Ref<SCElementOp2>& op,
 | 
|---|
 | 754 |                           SCMatrix* m)
 | 
|---|
 | 755 | {
 | 
|---|
 | 756 |   LocalSCMatrix *lm
 | 
|---|
 | 757 |       = require_dynamic_cast<LocalSCMatrix*>(m,"LocalSCMatrix::element_op");
 | 
|---|
 | 758 | 
 | 
|---|
 | 759 |   if (!rowdim()->equiv(lm->rowdim()) || !coldim()->equiv(lm->coldim())) {
 | 
|---|
 | 760 |       ExEnv::errn() << indent << "LocalSCMatrix: bad element_op\n";
 | 
|---|
 | 761 |       abort();
 | 
|---|
 | 762 |     }
 | 
|---|
 | 763 |   op->process_spec_rect(block.pointer(), lm->block.pointer());
 | 
|---|
 | 764 | }
 | 
|---|
 | 765 | 
 | 
|---|
 | 766 | void
 | 
|---|
 | 767 | LocalSCMatrix::element_op(const Ref<SCElementOp3>& op,
 | 
|---|
 | 768 |                           SCMatrix* m,SCMatrix* n)
 | 
|---|
 | 769 | {
 | 
|---|
 | 770 |   LocalSCMatrix *lm
 | 
|---|
 | 771 |       = require_dynamic_cast<LocalSCMatrix*>(m,"LocalSCMatrix::element_op");
 | 
|---|
 | 772 |   LocalSCMatrix *ln
 | 
|---|
 | 773 |       = require_dynamic_cast<LocalSCMatrix*>(n,"LocalSCMatrix::element_op");
 | 
|---|
 | 774 | 
 | 
|---|
 | 775 |   if (!rowdim()->equiv(lm->rowdim()) || !coldim()->equiv(lm->coldim()) ||
 | 
|---|
 | 776 |       !rowdim()->equiv(ln->rowdim()) || !coldim()->equiv(ln->coldim())) {
 | 
|---|
 | 777 |       ExEnv::errn() << indent << "LocalSCMatrix: bad element_op\n";
 | 
|---|
 | 778 |       abort();
 | 
|---|
 | 779 |     }
 | 
|---|
 | 780 |   op->process_spec_rect(block.pointer(),
 | 
|---|
 | 781 |                         lm->block.pointer(), ln->block.pointer());
 | 
|---|
 | 782 | }
 | 
|---|
 | 783 | 
 | 
|---|
 | 784 | // from Ed Seidl at the NIH
 | 
|---|
 | 785 | void
 | 
|---|
 | 786 | LocalSCMatrix::vprint(const char *title, ostream& os, int prec) const
 | 
|---|
 | 787 | {
 | 
|---|
 | 788 |   int ii,jj,kk,nn;
 | 
|---|
 | 789 |   int i,j;
 | 
|---|
 | 790 |   int lwidth,width;
 | 
|---|
 | 791 |   double max=this->maxabs();
 | 
|---|
 | 792 | 
 | 
|---|
 | 793 |   max = (max==0.0) ? 1.0 : log10(max);
 | 
|---|
 | 794 |   if (max < 0.0) max=1.0;
 | 
|---|
 | 795 | 
 | 
|---|
 | 796 |   lwidth = prec + 5 + (int) max;
 | 
|---|
 | 797 |   width = 75/(lwidth+SCFormIO::getindent(os));
 | 
|---|
 | 798 | 
 | 
|---|
 | 799 |   if (title)
 | 
|---|
 | 800 |     os << endl << indent << title << endl;
 | 
|---|
 | 801 |   else
 | 
|---|
 | 802 |     os << endl;
 | 
|---|
 | 803 | 
 | 
|---|
 | 804 |   if (nrow()==0 || ncol()==0) {
 | 
|---|
 | 805 |     os << indent << "empty matrix\n";
 | 
|---|
 | 806 |     return;
 | 
|---|
 | 807 |   }
 | 
|---|
 | 808 | 
 | 
|---|
 | 809 |   for (ii=jj=0;;) {
 | 
|---|
 | 810 |     ii++; jj++;
 | 
|---|
 | 811 |     kk=width*jj;
 | 
|---|
 | 812 |     nn = (ncol()>kk) ? kk : ncol();
 | 
|---|
 | 813 | 
 | 
|---|
 | 814 |     // print column indices
 | 
|---|
 | 815 |     os << indent;
 | 
|---|
 | 816 |     for (i=ii; i <= nn; i++)
 | 
|---|
 | 817 |       os << scprintf("%*d",lwidth,i);
 | 
|---|
 | 818 |     os << endl;
 | 
|---|
 | 819 | 
 | 
|---|
 | 820 |     // print the rows
 | 
|---|
 | 821 |     for (i=0; i < nrow() ; i++) {
 | 
|---|
 | 822 |       os << indent << scprintf("%5d",i+1);
 | 
|---|
 | 823 |       for (j=ii-1; j < nn; j++)
 | 
|---|
 | 824 |         os << scprintf("%*.*f",lwidth,prec,rows[i][j]);
 | 
|---|
 | 825 |       os << endl;
 | 
|---|
 | 826 |     }
 | 
|---|
 | 827 |     os << endl;
 | 
|---|
 | 828 | 
 | 
|---|
 | 829 |     if (ncol() <= kk) {
 | 
|---|
 | 830 |       os.flush();
 | 
|---|
 | 831 |       return;
 | 
|---|
 | 832 |     }
 | 
|---|
 | 833 | 
 | 
|---|
 | 834 |     ii=kk;
 | 
|---|
 | 835 |   }
 | 
|---|
 | 836 | }
 | 
|---|
 | 837 | 
 | 
|---|
 | 838 | Ref<SCMatrixSubblockIter>
 | 
|---|
 | 839 | LocalSCMatrix::local_blocks(SCMatrixSubblockIter::Access access)
 | 
|---|
 | 840 | {
 | 
|---|
 | 841 |   if (messagegrp()->n() > 1) {
 | 
|---|
 | 842 |       ExEnv::errn() << indent
 | 
|---|
 | 843 |            << "LocalSCMatrix::local_blocks: not valid for local matrices"
 | 
|---|
 | 844 |            << endl;
 | 
|---|
 | 845 |       abort();
 | 
|---|
 | 846 |     }
 | 
|---|
 | 847 |   Ref<SCMatrixSubblockIter> iter
 | 
|---|
 | 848 |       = new SCMatrixSimpleSubblockIter(access, block.pointer());
 | 
|---|
 | 849 |   return iter;
 | 
|---|
 | 850 | }
 | 
|---|
 | 851 | 
 | 
|---|
 | 852 | Ref<SCMatrixSubblockIter>
 | 
|---|
 | 853 | LocalSCMatrix::all_blocks(SCMatrixSubblockIter::Access access)
 | 
|---|
 | 854 | {
 | 
|---|
 | 855 |   if (access == SCMatrixSubblockIter::Write) {
 | 
|---|
 | 856 |       ExEnv::errn() << indent << "LocalSCMatrix::all_blocks: "
 | 
|---|
 | 857 |            << "Write access permitted for local blocks only"
 | 
|---|
 | 858 |            << endl;
 | 
|---|
 | 859 |       abort();
 | 
|---|
 | 860 |     }
 | 
|---|
 | 861 |   return local_blocks(access);
 | 
|---|
 | 862 | }
 | 
|---|
 | 863 | 
 | 
|---|
 | 864 | /////////////////////////////////////////////////////////////////////////////
 | 
|---|
 | 865 | 
 | 
|---|
 | 866 | // Local Variables:
 | 
|---|
 | 867 | // mode: c++
 | 
|---|
 | 868 | // c-file-style: "CLJ"
 | 
|---|
 | 869 | // End:
 | 
|---|